lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

实现DataBatch

Browse Source
This commit is contained in:
redkale
2023-04-01 14:43:41 +08:00
parent dba72cab46
commit 156131d4b5
4 changed files with 668 additions and 664 deletions
Download Patch File Download Diff File Expand all files Collapse all files

155
src/main/java/org/redkale/source/AbstractDataSource.java
View File

@@ -432,124 +432,6 @@ public abstract class AbstractDataSource extends AbstractService implements Data
return CompletableFuture.supplyAsync(supplier, getExecutor()); return CompletableFuture.supplyAsync(supplier, getExecutor());
} }
protected DefaultBatchInfo parseBatchInfo(DefaultDataBatch batch, Function<Class, EntityInfo> func) {
final List<BatchAction> actions = ((DefaultDataBatch) batch).actions;
final Map<Class, EntityInfo> infos = new HashMap<>();
final Map<String, EntityInfo> notInsertDisTables = new HashMap<>();
final Map<String, EntityInfo> notUpDelDisTables = new HashMap<>();
for (BatchAction action : actions) {
if (action instanceof InsertBatchAction1) {
InsertBatchAction1 act = (InsertBatchAction1) action;
Object entity = act.entity;
Class clazz = entity.getClass();
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(entity);
if (!info.containsDisTable(tableKey)) {
notInsertDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof DeleteBatchAction1) {
DeleteBatchAction1 act = (DeleteBatchAction1) action;
Object entity = act.entity;
Class clazz = entity.getClass();
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(entity);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof DeleteBatchAction2) {
DeleteBatchAction2 act = (DeleteBatchAction2) action;
Class clazz = act.clazz;
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(act.pk);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof DeleteBatchAction3) {
DeleteBatchAction3 act = (DeleteBatchAction3) action;
Class clazz = act.clazz;
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(act.node);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof UpdateBatchAction1) {
UpdateBatchAction1 act = (UpdateBatchAction1) action;
Object entity = act.entity;
Class clazz = entity.getClass();
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(entity);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof UpdateBatchAction2) {
UpdateBatchAction2 act = (UpdateBatchAction2) action;
Class clazz = act.clazz;
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(act.pk);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof UpdateBatchAction3) {
UpdateBatchAction3 act = (UpdateBatchAction3) action;
Class clazz = act.clazz;
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = info.getTable(act.node);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
} else if (action instanceof UpdateBatchAction4) {
UpdateBatchAction4 act = (UpdateBatchAction4) action;
Class clazz = act.entity.getClass();
if (!infos.containsKey(clazz)) {
EntityInfo info = func.apply(clazz);
infos.put(clazz, info);
if (info.getTableStrategy() != null) {
String tableKey = act.node == null ? info.getTable(act.entity) : info.getTable(act.node);
if (!info.containsDisTable(tableKey)) {
notUpDelDisTables.put(tableKey, info);
}
}
}
}
}
return new DefaultBatchInfo(infos, notInsertDisTables, notUpDelDisTables);
}
@Override @Override
public int batch(final DataBatch batch) { public int batch(final DataBatch batch) {
return batchAsync(batch).join(); return batchAsync(batch).join();
@@ -557,7 +439,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public CompletableFuture<Integer> batchAsync(final DataBatch batch) { public CompletableFuture<Integer> batchAsync(final DataBatch batch) {
throw new UnsupportedOperationException("Not supported yet."); return CompletableFuture.failedFuture(new UnsupportedOperationException("Not supported yet."));
} }
@Override @Override
@@ -1398,7 +1280,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
public <T> DataBatch insert(T... entitys) { public <T> DataBatch insert(T... entitys) {
for (T t : entitys) { for (T t : entitys) {
Objects.requireNonNull(t); Objects.requireNonNull(t);
if (t.getClass().getAnnotation(Entity.class) == null && t.getClass().getAnnotation(javax.persistence.Entity.class) == null) { if (t.getClass().getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity");
} }
this.actions.add(new InsertBatchAction1(t)); this.actions.add(new InsertBatchAction1(t));
@@ -1410,7 +1292,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
public <T> DataBatch delete(T... entitys) { public <T> DataBatch delete(T... entitys) {
for (T t : entitys) { for (T t : entitys) {
Objects.requireNonNull(t); Objects.requireNonNull(t);
if (t.getClass().getAnnotation(Entity.class) == null && t.getClass().getAnnotation(javax.persistence.Entity.class) == null) { if (t.getClass().getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity");
} }
this.actions.add(new DeleteBatchAction1(t)); this.actions.add(new DeleteBatchAction1(t));
@@ -1421,7 +1303,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public <T> DataBatch delete(Class<T> clazz, Serializable... pks) { public <T> DataBatch delete(Class<T> clazz, Serializable... pks) {
Objects.requireNonNull(clazz); Objects.requireNonNull(clazz);
if (clazz.getAnnotation(Entity.class) == null && clazz.getAnnotation(javax.persistence.Entity.class) == null) { if (clazz.getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity");
} }
if (pks.length < 1) { if (pks.length < 1) {
@@ -1442,7 +1324,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public <T> DataBatch delete(Class<T> clazz, FilterNode node, Flipper flipper) { public <T> DataBatch delete(Class<T> clazz, FilterNode node, Flipper flipper) {
Objects.requireNonNull(clazz); Objects.requireNonNull(clazz);
if (clazz.getAnnotation(Entity.class) == null && clazz.getAnnotation(javax.persistence.Entity.class) == null) { if (clazz.getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity");
} }
this.actions.add(new DeleteBatchAction3(clazz, node, flipper)); this.actions.add(new DeleteBatchAction3(clazz, node, flipper));
@@ -1453,7 +1335,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
public <T> DataBatch update(T... entitys) { public <T> DataBatch update(T... entitys) {
for (T t : entitys) { for (T t : entitys) {
Objects.requireNonNull(t); Objects.requireNonNull(t);
if (t.getClass().getAnnotation(Entity.class) == null && t.getClass().getAnnotation(javax.persistence.Entity.class) == null) { if (t.getClass().getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + t.getClass() + " must be on Annotation @Entity");
} }
this.actions.add(new UpdateBatchAction1(t)); this.actions.add(new UpdateBatchAction1(t));
@@ -1469,7 +1351,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public <T> DataBatch update(Class<T> clazz, Serializable pk, ColumnValue... values) { public <T> DataBatch update(Class<T> clazz, Serializable pk, ColumnValue... values) {
Objects.requireNonNull(clazz); Objects.requireNonNull(clazz);
if (clazz.getAnnotation(Entity.class) == null && clazz.getAnnotation(javax.persistence.Entity.class) == null) { if (clazz.getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity");
} }
Objects.requireNonNull(pk); Objects.requireNonNull(pk);
@@ -1496,7 +1378,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public <T> DataBatch update(Class<T> clazz, FilterNode node, Flipper flipper, ColumnValue... values) { public <T> DataBatch update(Class<T> clazz, FilterNode node, Flipper flipper, ColumnValue... values) {
Objects.requireNonNull(clazz); Objects.requireNonNull(clazz);
if (clazz.getAnnotation(Entity.class) == null && clazz.getAnnotation(javax.persistence.Entity.class) == null) { if (clazz.getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + clazz + " must be on Annotation @Entity");
} }
if (values.length < 1) { if (values.length < 1) {
@@ -1533,7 +1415,7 @@ public abstract class AbstractDataSource extends AbstractService implements Data
@Override @Override
public <T> DataBatch updateColumn(T entity, final FilterNode node, SelectColumn selects) { public <T> DataBatch updateColumn(T entity, final FilterNode node, SelectColumn selects) {
Objects.requireNonNull(entity); Objects.requireNonNull(entity);
if (entity.getClass().getAnnotation(Entity.class) == null && entity.getClass().getAnnotation(javax.persistence.Entity.class) == null) { if (entity.getClass().getAnnotation(Entity.class) == null) {
throw new SourceException("Entity Class " + entity.getClass() + " must be on Annotation @Entity"); throw new SourceException("Entity Class " + entity.getClass() + " must be on Annotation @Entity");
} }
Objects.requireNonNull(selects); Objects.requireNonNull(selects);
@@ -1543,25 +1425,6 @@ public abstract class AbstractDataSource extends AbstractService implements Data
} }
protected static class DefaultBatchInfo {
//EntityInfo对象
public Map<Class, EntityInfo> entityInfos;
//新增操作可能不存在的分表
public Map<String, EntityInfo> notInsertDisTables;
//删除修改操作可能不存在的分表
public Map<String, EntityInfo> notUpDelDisTables;
public DefaultBatchInfo(Map<Class, EntityInfo> entityInfos, Map<String, EntityInfo> notInsertDisTables, Map<String, EntityInfo> notUpDelDisTables) {
this.entityInfos = entityInfos;
this.notInsertDisTables = notInsertDisTables;
this.notUpDelDisTables = notUpDelDisTables;
}
}
protected abstract static class BatchAction { protected abstract static class BatchAction {
} }

3
src/main/java/org/redkale/source/AbstractDataSqlSource.java
View File

@@ -661,12 +661,15 @@ public abstract class AbstractDataSqlSource extends AbstractDataSource implement
} }
@Local @Local
@Override
public abstract int directExecute(String sql); public abstract int directExecute(String sql);
@Local @Local
@Override
public abstract int[] directExecute(String... sqls); public abstract int[] directExecute(String... sqls);
@Local @Local
@Override
public abstract <V> V directQuery(String sql, Function<DataResultSet, V> handler); public abstract <V> V directQuery(String sql, Function<DataResultSet, V> handler);
//是否异步 //是否异步

1163
src/main/java/org/redkale/source/DataJdbcSource.java
View File

File diff suppressed because it is too large Load Diff

11
src/main/java/org/redkale/source/EntityInfo.java
View File

@@ -1265,6 +1265,17 @@ public final class EntityInfo<T> {
return this.primary; return this.primary;
} }
/**
* 获取主键字段的值
*
* @param entity 实体对象
*
* @return 主键值
*/
public Serializable getPrimaryValue(T entity) {
return this.primary.get(entity);
}
/** /**
* 获取主键字段的Attribute单一元素数组 * 获取主键字段的Attribute单一元素数组
* *