Skip to content

Commit 207d539

Browse files
DbEntity sync nhl#131
1 parent 5cb64d7 commit 207d539

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

link-move/src/main/java/com/nhl/link/move/runtime/task/createorupdatedb/CreateOrUpdateDbBuilder.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
import org.apache.cayenne.map.ObjAttribute;
2929
import org.apache.cayenne.map.ObjEntity;
3030

31+
import java.util.Collection;
3132
import java.util.Objects;
3233

3334
/**
@@ -68,7 +69,16 @@ public CreateOrUpdateDbBuilder(String dbEntityName,
6869
}
6970
this.dbEntity = entity;
7071

71-
ObjEntity objEntity = new ObjEntity(entity.getName() + "_temp");
72+
Collection<ObjEntity> objEntities = entity.mappedObjEntities();
73+
ObjEntity objEntity;
74+
if (objEntities.isEmpty()) {
75+
objEntity = new ObjEntity(entity.getName());
76+
} else if (objEntities.size() == 1) {
77+
objEntity = objEntities.iterator().next();
78+
} else {
79+
throw new LmRuntimeException("Db entity has several mapped object entities: " + dbEntityName);
80+
}
81+
7282
entity.getAttributes().forEach(a -> {
7383
if (!a.getName().equals("id")) {
7484
ObjAttribute objAttribute = new ObjAttribute(a.getName(), TypesMapping.getJavaBySqlType(a.getType()), objEntity);

0 commit comments

Comments
 (0)