diff options
author | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-10 18:28:36 +0200 |
---|---|---|
committer | Trygve Laugstøl <trygvis@inamo.no> | 2013-08-10 18:28:36 +0200 |
commit | 46cb0cef9303940ade51ffe16831f55ee9cd8e1f (patch) | |
tree | a89a3538ff0d3fd56b6d401cd296940b2336657d /container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java | |
parent | 5a1256a8ed931f7a5ba05c4328353411bae31f2b (diff) | |
download | container-playground-46cb0cef9303940ade51ffe16831f55ee9cd8e1f.tar.gz container-playground-46cb0cef9303940ade51ffe16831f55ee9cd8e1f.tar.bz2 container-playground-46cb0cef9303940ade51ffe16831f55ee9cd8e1f.tar.xz container-playground-46cb0cef9303940ade51ffe16831f55ee9cd8e1f.zip |
o Start of loading referenced (singular) entities.
Diffstat (limited to 'container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java')
-rw-r--r-- | container-compiler-plugin/src/main/java/io/trygvis/persistence/generators/EntityManagerGenerator.java | 4 |
1 files changed, 2 insertions, 2 deletions
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<EntityMirror, FieldRef> 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<String> body = new ArrayList<>(); for (EntityMirror entity : unit.getEntities().values()) { body.add("if (klass == " + entity.type.plainName + ".class) {"); - body.add(" return (SqlDao<Id, T>) " + daoFields.get(entity).name + ";"); + body.add(" return (SqlDao<Id, T, " + unit.sessionType.className + ">) " + daoFields.get(entity).name + ";"); body.add("}"); } body.add("throw new RuntimeException(\"Type is not a part of this persistence unit: \" + klass);"); |