Commit 12245be469f41727f74f7bbc897a3564c0b6ff37

Authored by 杨鸣坤
1 parent 4bd451bc

接口调整

@@ -184,7 +184,7 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService { @@ -184,7 +184,7 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService {
184 List<Object> resultList = new ArrayList<>(); 184 List<Object> resultList = new ArrayList<>();
185 try { 185 try {
186 HikariConfig config = new HikariConfig(); 186 HikariConfig config = new HikariConfig();
187 - config.setJdbcUrl(replaceParam(tkDbConnectDTO.getUrl(), tkDbConnectDTO.getParamMap())); 187 + config.setJdbcUrl(tkDbConnectDTO.getUrl());
188 config.setUsername(tkDbConnectDTO.getUserName()); 188 config.setUsername(tkDbConnectDTO.getUserName());
189 config.setPassword(tkDbConnectDTO.getPassword()); 189 config.setPassword(tkDbConnectDTO.getPassword());
190 if (DbConnectTypeEnum.MySql.equals(tkDbConnectDTO.getType())) { 190 if (DbConnectTypeEnum.MySql.equals(tkDbConnectDTO.getType())) {
@@ -216,7 +216,7 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService { @@ -216,7 +216,7 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService {
216 216
217 dataSource = new HikariDataSource(config); 217 dataSource = new HikariDataSource(config);
218 connection = dataSource.getConnection(); 218 connection = dataSource.getConnection();
219 - statement = connection.prepareStatement(tkDbConnectDTO.getSql()); 219 + statement = connection.prepareStatement(replaceParam(tkDbConnectDTO.getSql(), tkDbConnectDTO.getParamMap()));
220 resultSet = statement.executeQuery(); 220 resultSet = statement.executeQuery();
221 ResultSetMetaData metaData = resultSet.getMetaData(); 221 ResultSetMetaData metaData = resultSet.getMetaData();
222 List<String> columnNames = new ArrayList<>(); 222 List<String> columnNames = new ArrayList<>();
@@ -257,9 +257,9 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService { @@ -257,9 +257,9 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService {
257 return resultList; 257 return resultList;
258 } 258 }
259 259
260 - private String replaceParam(String url, Map<String, Object> paramMap) { 260 + private String replaceParam(String sql, Map<String, Object> paramMap) {
261 if (MapUtils.isEmpty(paramMap)) { 261 if (MapUtils.isEmpty(paramMap)) {
262 - return url; 262 + return sql;
263 } 263 }
264 264
265 for (Map.Entry<String, Object> objectEntry : paramMap.entrySet()) { 265 for (Map.Entry<String, Object> objectEntry : paramMap.entrySet()) {
@@ -269,10 +269,10 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService { @@ -269,10 +269,10 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService {
269 continue; 269 continue;
270 } 270 }
271 271
272 - url = url.replaceAll("${" + key + "}", String.valueOf(value)); 272 + sql = sql.replaceAll("\\$\\{" + key + "\\}", String.valueOf(value));
273 } 273 }
274 274
275 - return url; 275 + return sql;
276 } 276 }
277 277
278 private Object getTypedValue(ResultSet rs, int index, int sqlType) throws SQLException { 278 private Object getTypedValue(ResultSet rs, int index, int sqlType) throws SQLException {
@@ -119,6 +119,7 @@ public class TkDbConnectServiceImpl extends AbstractBaseService<TkDbConnectMappe @@ -119,6 +119,7 @@ public class TkDbConnectServiceImpl extends AbstractBaseService<TkDbConnectMappe
119 } 119 }
120 120
121 dbDto.setSql(tkDbConnectDTO.getSql()); 121 dbDto.setSql(tkDbConnectDTO.getSql());
  122 + dbDto.setParamMap(tkDbConnectDTO.getParamMap());
122 return connectServiceFactory.getService("baseConnect").connect(dbDto); 123 return connectServiceFactory.getService("baseConnect").connect(dbDto);
123 } 124 }
124 } 125 }
@@ -41,6 +41,7 @@ public class TkDbDataSetServiceImpl extends AbstractBaseService<TkDbDataSetMappe @@ -41,6 +41,7 @@ public class TkDbDataSetServiceImpl extends AbstractBaseService<TkDbDataSetMappe
41 baseMapper.update(entity, filter); 41 baseMapper.update(entity, filter);
42 } 42 }
43 43
  44 + entity.copyToDTO(dto);
44 return dto; 45 return dto;
45 } 46 }
46 47
@@ -66,6 +67,7 @@ public class TkDbDataSetServiceImpl extends AbstractBaseService<TkDbDataSetMappe @@ -66,6 +67,7 @@ public class TkDbDataSetServiceImpl extends AbstractBaseService<TkDbDataSetMappe
66 dto = get(dto.getId()); 67 dto = get(dto.getId());
67 TkDbConnectDTO dbConnectDTO = dto.getTkDbConnectDTO(); 68 TkDbConnectDTO dbConnectDTO = dto.getTkDbConnectDTO();
68 dbConnectDTO.setParamMap(paramMap); 69 dbConnectDTO.setParamMap(paramMap);
  70 + dbConnectDTO.setSql(dto.getSql());
69 return connectServiceFactory.getService("baseConnect").connect(dbConnectDTO); 71 return connectServiceFactory.getService("baseConnect").connect(dbConnectDTO);
70 } 72 }
71 73