提交 02f926b8 编写于 作者: ibiz4j's avatar ibiz4j

dameng批量

上级 61bf0450
...@@ -29,4 +29,23 @@ public interface DbEntityMapper extends BaseMapper<EntityObj>{ ...@@ -29,4 +29,23 @@ public interface DbEntityMapper extends BaseMapper<EntityObj>{
" ) T2 \n ${model.mergeSegment}"+ " ) T2 \n ${model.mergeSegment}"+
"</script>"}) "</script>"})
boolean save(@Param("model") EntityModel model, @Param("entity") EntityObj entity); boolean save(@Param("model") EntityModel model, @Param("entity") EntityObj entity);
@Insert({"<script> "+
"<foreach collection=\"list\" item=\"item\" index=\"index\" separator=\";\">\n" +
" MERGE INTO ${model.tableName} T1\n" +
" USING (\n" +
" SELECT ${model.mergeColumnSegment} FROM dual\n" +
" ) T2 \n ${model.mergeSegment}"+
"</foreach>\n" +
"</script>"})
boolean saveBatchDm(@Param("model") EntityModel model, @Param("list") List<EntityObj> list);
@Insert({"<script> "+" MERGE INTO ${model.tableName} T1\n" +
" USING (\n" +
" SELECT ${model.mergeColumnSegment} FROM dual\n" +
" ) T2 \n ${model.mergeSegment}"+
"</script>"})
boolean saveDm(@Param("model") EntityModel model, @Param("entity") EntityObj entity);
} }
\ No newline at end of file
...@@ -9,6 +9,8 @@ import cn.ibizlab.core.lite.extensions.domain.ModelObj; ...@@ -9,6 +9,8 @@ import cn.ibizlab.core.lite.extensions.domain.ModelObj;
import cn.ibizlab.core.lite.extensions.filter.DbEntitySearchContext; import cn.ibizlab.core.lite.extensions.filter.DbEntitySearchContext;
import cn.ibizlab.core.lite.extensions.model.DataModel; import cn.ibizlab.core.lite.extensions.model.DataModel;
import cn.ibizlab.util.filter.QueryFilter; import cn.ibizlab.util.filter.QueryFilter;
import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder; import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
...@@ -20,8 +22,10 @@ import org.springframework.beans.factory.annotation.Autowired; ...@@ -20,8 +22,10 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy; import org.springframework.context.annotation.Lazy;
import org.springframework.context.annotation.Primary; import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils; import org.springframework.util.StringUtils;
import javax.sql.DataSource;
import java.io.BufferedReader; import java.io.BufferedReader;
import java.io.Reader; import java.io.Reader;
import java.sql.Timestamp; import java.sql.Timestamp;
...@@ -211,6 +215,10 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl ...@@ -211,6 +215,10 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl
} }
} }
@Autowired
@Lazy
private DynamicRoutingDataSource dynamicRoutingDataSource;
@Override @Override
public boolean saveBatch(String dsName, EntityModel entityModel, List<EntityObj> list) { public boolean saveBatch(String dsName, EntityModel entityModel, List<EntityObj> list) {
try { try {
...@@ -219,6 +227,16 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl ...@@ -219,6 +227,16 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl
dstDataSourceService.initDataSource(dsName); dstDataSourceService.initDataSource(dsName);
DynamicDataSourceContextHolder.push(dsName); DynamicDataSourceContextHolder.push(dsName);
String dstype="";
Map<String, DataSource> dynamicDSMap = dynamicRoutingDataSource.getCurrentDataSources();
if (!ObjectUtils.isEmpty(dynamicDSMap) && dynamicDSMap.containsKey(dsName)) {
DataSource ds = dynamicDSMap.get(dsName);
if (ds instanceof DruidDataSource) {
DruidDataSource druidDataSource = (DruidDataSource) dynamicDSMap.get(dsName);
dstype=druidDataSource.getDbType();
}
}
list.forEach(entityObj -> { list.forEach(entityObj -> {
entityObj.keySet().forEach(key -> { entityObj.keySet().forEach(key -> {
if(!key.toUpperCase().equals(key)) if(!key.toUpperCase().equals(key))
...@@ -226,7 +244,14 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl ...@@ -226,7 +244,14 @@ public class DbEntityService extends ServiceImpl<DbEntityMapper, EntityObj> impl
}); });
}); });
if ("dameng".equals(dstype)||"dm".equals(dstype)) {
baseMapper.saveBatchDm(entityModel, list);
}
else
{
baseMapper.saveBatch(entityModel, list); baseMapper.saveBatch(entityModel, list);
}
return true; return true;
} catch (Exception ex) { } catch (Exception ex) {
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册