From 46cb0cef9303940ade51ffe16831f55ee9cd8e1f Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Sat, 10 Aug 2013 18:28:36 +0200 Subject: o Start of loading referenced (singular) entities. --- .../io/trygvis/persistence/generators/EntityManagerGenerator.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java') diff --git a/container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java b/container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java index f621343..4a6df0f 100644 --- a/container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java +++ b/container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java @@ -60,14 +60,14 @@ public class EntityManagerGenerator { } public static MethodRef getSqlDao(SqlUnitModel unit, Imports imports, Map daoFields) { - TypeRef sqlDatoType = imports.add(new TypeRef(SqlDao.class)).args("Id", "T"); + TypeRef sqlDatoType = imports.add(new TypeRef(SqlDao.class)).args("Id", "T", unit.sessionType.className); Parameters p = new Parameters(); TypeRef klassType = new TypeRef(Class.class).args("T"); ParameterRef klass = p.addParameter(klassType, "klass"); List body = new ArrayList<>(); for (EntityMirror entity : unit.getEntities().values()) { body.add("if (klass == " + entity.type.plainName + ".class) {"); - body.add(" return (SqlDao) " + daoFields.get(entity).name + ";"); + body.add(" return (SqlDao) " + daoFields.get(entity).name + ";"); body.add("}"); } body.add("throw new RuntimeException(\"Type is not a part of this persistence unit: \" + klass);"); -- cgit v1.2.3