测试数据库是否可以链接

@@ -66,6 +66,12 @@ public class TkDbConnectController extends BaseController {
         return ResponseEntity.ok(dto);
     }
 
+    @PostMapping("/testDbConnect")
+    @PreAuthorize("hasAnyAuthority('TENANT_ADMIN','CUSTOMER_USER')")
+    public ResponseEntity<Boolean> testDbConnect(@RequestBody TkDbConnectDTO tkDbConnectDTO) throws ThingsboardException {
+        return ResponseEntity.ok(tkDbConnectService.checkConnect(tkDbConnectDTO));
+    }
+
     @GetMapping("/delete")
     @PreAuthorize("hasAnyAuthority('TENANT_ADMIN','CUSTOMER_USER')")
     public ResponseEntity<Boolean> delete(@RequestParam("id") String id) throws ThingsboardException {

@@ -34,9 +34,11 @@ import java.util.Properties;
 public class BaseDbConnectServiceImpl implements BaseDbConnectService {
 
     private static SSLContext sslContext = null;
+    private static Boolean checkConnect = null;
 
     @Override
-    public Object connect(TkDbConnectDTO tkDbConnectDTO) {
+    public Object connect(TkDbConnectDTO tkDbConnectDTO, boolean checkConnect) {
+        this.checkConnect = checkConnect;
         if (BooleanUtils.isNotTrue(tkDbConnectDTO.isOpenSsh()) && BooleanUtils.isNotTrue(tkDbConnectDTO.isOpenSsl())) {
             return commonConnect(tkDbConnectDTO);
         } else if (BooleanUtils.isTrue(tkDbConnectDTO.isOpenSsh()) && BooleanUtils.isNotTrue(tkDbConnectDTO.isOpenSsl())) {
@@ -216,6 +218,10 @@ public class BaseDbConnectServiceImpl implements BaseDbConnectService {
 
             dataSource = new HikariDataSource(config);
             connection = dataSource.getConnection();
+            if (BooleanUtils.isTrue(checkConnect)) {
+                return true;
+            }
+
             statement = connection.prepareStatement(replaceParam(tkDbConnectDTO.getSql(), tkDbConnectDTO.getParamMap()));
             resultSet = statement.executeQuery();
             ResultSetMetaData metaData = resultSet.getMetaData();

@@ -130,6 +130,16 @@ public class TkDbConnectServiceImpl extends AbstractBaseService<TkDbConnectMappe
 
         dbDto.setSql(tkDbConnectDTO.getSql());
         dbDto.setParamMap(tkDbConnectDTO.getParamMap());
-        return connectServiceFactory.getService("baseConnect").connect(dbDto);
+        return connectServiceFactory.getService("baseConnect").connect(dbDto, false);
+    }
+
+    @Override
+    public boolean checkConnect(TkDbConnectDTO tkDbConnectDTO) {
+        Object result = connectServiceFactory.getService("baseConnect").connect(tkDbConnectDTO, true);
+        if (result instanceof Boolean) {
+            return (Boolean) result;
+        }
+
+        return false;
     }
 }

@@ -68,7 +68,7 @@ public class TkDbDataSetServiceImpl extends AbstractBaseService<TkDbDataSetMappe
         TkDbConnectDTO dbConnectDTO = dto.getTkDbConnectDTO();
         dbConnectDTO.setParamMap(paramMap);
         dbConnectDTO.setSql(dto.getSql());
-        return connectServiceFactory.getService("baseConnect").connect(dbConnectDTO);
+        return connectServiceFactory.getService("baseConnect").connect(dbConnectDTO, false);
     }
 
     private void checkDto(TkDbDataSetDTO dto) {

@@ -10,5 +10,5 @@ public interface BaseDbConnectService {
      * @param tkDbConnectDTO
      * @return
      */
-    Object connect(TkDbConnectDTO tkDbConnectDTO);
+    Object connect(TkDbConnectDTO tkDbConnectDTO, boolean checkConnect);
 }

@@ -17,4 +17,6 @@ public interface TkDbConnectService extends BaseService<TkDbConnectEntity> {
     TkDbConnectDTO get(String id);
 
     Object connectResult(TkDbConnectDTO tkDbConnectDTO);
+
+    boolean checkConnect(TkDbConnectDTO tkDbConnectDTO);
 }