mysql数据库查询预览

@@ -74,4 +74,10 @@ public class TkDbConnectController extends BaseController {
     public ResponseEntity<TkDbConnectDTO> get(@RequestParam("id") String id) throws ThingsboardException {
     public ResponseEntity<TkDbConnectDTO> get(@RequestParam("id") String id) throws ThingsboardException {
         return ResponseEntity.ok(tkDbConnectService.get(id));
         return ResponseEntity.ok(tkDbConnectService.get(id));
     }
     }
+
+    @PostMapping("/dataView")
+    @PreAuthorize("hasAnyAuthority('TENANT_ADMIN','CUSTOMER_USER')")
+    public ResponseEntity<Object> dataView(@RequestBody TkDbConnectDTO tkDbConnectDTO) throws ThingsboardException {
+        return ResponseEntity.ok(tkDbConnectService.connectResult(tkDbConnectDTO));
+    }
 }
 }

@@ -59,4 +59,7 @@ public class TkDbConnectDTO extends TenantDTO {
 
 
     @ApiModelProperty("SSL通道配置信息")
     @ApiModelProperty("SSL通道配置信息")
     private TkDbConnectSslDTO ssl;
     private TkDbConnectSslDTO ssl;
+
+    @ApiModelProperty("预览sql")
+    private String sql;
 }
 }

@@ -305,7 +305,12 @@
             <artifactId>data-ext</artifactId>
             <artifactId>data-ext</artifactId>
             <scope>compile</scope>
             <scope>compile</scope>
         </dependency>
         </dependency>
-
+        <!-- MySQL JDBC 驱动 -->
+        <dependency>
+            <groupId>mysql</groupId>
+            <artifactId>mysql-connector-java</artifactId>
+            <version>8.0.28</version>
+        </dependency>
     </dependencies>
     </dependencies>
     <build>
     <build>
         <plugins>
         <plugins>

@@ -16,16 +16,12 @@ public class TkDbConnectSslEntity extends TenantBaseEntity {
 
 
     private String connectId;
     private String connectId;
 
 
-    private String ip;
+    private String caCert;
 
 
-    private String port;
+    private boolean verifyCaCert;
 
 
-    private String sshName;
+    private String clientKey;
 
 
-    private String sshPassword;
-
-    private String verifyMethod;
-
-    private String privateKey;
+    private String clientCert;
 
 
 }
 }

+package org.thingsboard.server.dao.yunteng.factory;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.thingsboard.server.dao.yunteng.service.BaseDbConnectService;
+
+import java.util.Map;
+
+@Service
+public class DbConnectServiceFactory {
+
+    @Autowired
+    private Map<String, BaseDbConnectService> serviceMap;
+
+    public BaseDbConnectService getService(String key) {
+        BaseDbConnectService service = serviceMap.get(key);
+        if (service == null) {
+            throw new IllegalArgumentException("未知数据库链接类型: " + key);
+        }
+
+        return service;
+    }
+}

+package org.thingsboard.server.dao.yunteng.impl;
+
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import org.apache.commons.lang3.BooleanUtils;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.stereotype.Service;
+import org.thingsboard.server.common.data.yunteng.dto.TkDbConnectDTO;
+import org.thingsboard.server.dao.yunteng.service.BaseDbConnectService;
+
+import java.sql.*;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * mySql类型数据库连接查询
+ */
+@Service("MySql")
+public class MySqlDbConnectServiceImpl implements BaseDbConnectService {
+    @Override
+    public Object connect(TkDbConnectDTO tkDbConnectDTO) {
+        if (BooleanUtils.isNotTrue(tkDbConnectDTO.isOpenSsh()) && BooleanUtils.isNotTrue(tkDbConnectDTO.isOpenSsl())) {
+            return commonConnect(tkDbConnectDTO);
+        }
+
+        return null;
+    }
+
+    private Object commonConnect(TkDbConnectDTO tkDbConnectDTO) {
+        PreparedStatement statement = null;
+        ResultSet resultSet = null;
+        Connection connection = null;
+        HikariDataSource dataSource = null;
+        List<Object> resultList = new ArrayList<>();
+        try {
+            HikariConfig config = new HikariConfig();
+            config.setJdbcUrl(tkDbConnectDTO.getUrl());
+            config.setUsername(tkDbConnectDTO.getUserName());
+            config.setPassword(tkDbConnectDTO.getPassword());
+            config.setDriverClassName("com.mysql.cj.jdbc.Driver");
+
+            // 连接池配置
+            config.setMaximumPoolSize(tkDbConnectDTO.getMaxPoolSize());          // 最大连接数
+            config.setMinimumIdle(tkDbConnectDTO.getMinIdle());               // 最小空闲连接数
+//            config.setIdleTimeout(600000);          // 空闲连接超时时间（毫秒）
+            config.setConnectionTimeout(tkDbConnectDTO.getConnectTimeout());     // 连接超时时间（毫秒）
+//            config.setMaxLifetime(1800000);         // 连接最大生命周期（毫秒）
+
+            // 连接校验配置
+            config.setConnectionTestQuery(StringUtils.isNotBlank(tkDbConnectDTO.getConnectTestQuery()) ?
+                    tkDbConnectDTO.getConnectTestQuery() : "SELECT 1");  // 连接校验语句
+//            config.setValidationTimeout(5000);         // 校验超时时间（毫秒）
+
+            dataSource = new HikariDataSource(config);
+            connection = dataSource.getConnection();
+            statement = connection.prepareStatement(tkDbConnectDTO.getSql());
+            resultSet = statement.executeQuery();
+            ResultSetMetaData metaData = resultSet.getMetaData();
+            List<String> columnNames = new ArrayList<>();
+            for (int index = 1; index <= metaData.getColumnCount(); index++) {
+                columnNames.add(metaData.getColumnLabel(index));
+            }
+
+            resultList.add(columnNames);
+            // 处理结果集
+            while (resultSet.next()) {
+                List<Object> result = new ArrayList<>(metaData.getColumnCount());
+                for (int index = 1; index <= metaData.getColumnCount(); index++) {
+                    int columnType = metaData.getColumnType(index);
+                    Object value = getTypedValue(resultSet, index, columnType);
+                    result.add(value);
+                }
+
+                resultList.add(result);
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+            throw new RuntimeException(e.getMessage());
+        } finally {
+            // 释放资源
+            try {
+                if (resultSet != null) resultSet.close();
+                if (statement != null) statement.close();
+                if (connection != null) connection.close();
+            } catch (SQLException e) {
+                e.printStackTrace();
+            }
+
+            if (dataSource != null && !dataSource.isClosed()) {
+                dataSource.close();
+            }
+        }
+
+        return resultList;
+    }
+
+    private Object getTypedValue(ResultSet rs, int index, int sqlType) throws SQLException {
+        Object value;
+
+        switch (sqlType) {
+            case Types.BIT:
+            case Types.BOOLEAN:
+                value = rs.getBoolean(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.TINYINT:
+            case Types.SMALLINT:
+            case Types.INTEGER:
+                value = rs.getInt(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.BIGINT:
+                value = rs.getLong(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.FLOAT:
+            case Types.REAL:
+                value = rs.getFloat(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.DOUBLE:
+                value = rs.getDouble(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.NUMERIC:
+            case Types.DECIMAL:
+                value = rs.getBigDecimal(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.CHAR:
+            case Types.VARCHAR:
+            case Types.LONGVARCHAR:
+            case Types.NCHAR:
+            case Types.NVARCHAR:
+            case Types.LONGNVARCHAR:
+                value = rs.getString(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.DATE:
+                Date date = rs.getDate(index);
+                return date != null ? date.toLocalDate() : null;
+
+            case Types.TIME:
+                Time time = rs.getTime(index);
+                return time != null ? time.toLocalTime() : null;
+
+            case Types.TIMESTAMP:
+                Timestamp timestamp = rs.getTimestamp(index);
+                return timestamp != null ? timestamp.toLocalDateTime() : null;
+
+            case Types.BINARY:
+            case Types.VARBINARY:
+            case Types.LONGVARBINARY:
+                value = rs.getBytes(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.BLOB:
+                value = rs.getBlob(index);
+                return rs.wasNull() ? null : value;
+
+            case Types.CLOB:
+                value = rs.getClob(index);
+                return rs.wasNull() ? null : value;
+
+            default:
+                value = rs.getObject(index);
+                return rs.wasNull() ? null : value;
+        }
+    }
+}

@@ -13,6 +13,7 @@ import org.thingsboard.server.common.data.yunteng.dto.TkDbConnectSshDTO;
 import org.thingsboard.server.common.data.yunteng.dto.TkDbConnectSslDTO;
 import org.thingsboard.server.common.data.yunteng.dto.TkDbConnectSslDTO;
 import org.thingsboard.server.common.data.yunteng.utils.tools.TkPageData;
 import org.thingsboard.server.common.data.yunteng.utils.tools.TkPageData;
 import org.thingsboard.server.dao.yunteng.entities.TkDbConnectEntity;
 import org.thingsboard.server.dao.yunteng.entities.TkDbConnectEntity;
+import org.thingsboard.server.dao.yunteng.factory.DbConnectServiceFactory;
 import org.thingsboard.server.dao.yunteng.mapper.TkDbConnectMapper;
 import org.thingsboard.server.dao.yunteng.mapper.TkDbConnectMapper;
 import org.thingsboard.server.dao.yunteng.service.AbstractBaseService;
 import org.thingsboard.server.dao.yunteng.service.AbstractBaseService;
 import org.thingsboard.server.dao.yunteng.service.TkDbConnectService;
 import org.thingsboard.server.dao.yunteng.service.TkDbConnectService;
@@ -28,6 +29,7 @@ public class TkDbConnectServiceImpl extends AbstractBaseService<TkDbConnectMappe
         implements TkDbConnectService {
         implements TkDbConnectService {
     private final TkDbConnectSshService tkDbConnectSshService;
     private final TkDbConnectSshService tkDbConnectSshService;
     private final TkDbConnectSslService tkDbConnectSslService;
     private final TkDbConnectSslService tkDbConnectSslService;
+    private final DbConnectServiceFactory connectServiceFactory;
 
 
     @Override
     @Override
     public TkPageData<TkDbConnectDTO> page(Map<String, Object> queryMap, String tenantId) {
     public TkPageData<TkDbConnectDTO> page(Map<String, Object> queryMap, String tenantId) {
@@ -104,4 +106,19 @@ public class TkDbConnectServiceImpl extends AbstractBaseService<TkDbConnectMappe
         tkDbConnectDTO.setSsl(ssl);
         tkDbConnectDTO.setSsl(ssl);
         return tkDbConnectDTO;
         return tkDbConnectDTO;
     }
     }
+
+    @Override
+    public Object connectResult(TkDbConnectDTO tkDbConnectDTO) {
+        if (tkDbConnectDTO == null || StringUtils.isBlank(tkDbConnectDTO.getSql()) || StringUtils.isBlank(tkDbConnectDTO.getId())) {
+            return new Object();
+        }
+
+        TkDbConnectDTO dbDto = get(tkDbConnectDTO.getId());
+        if (dbDto == null) {
+            return new Object();
+        }
+
+        dbDto.setSql(tkDbConnectDTO.getSql());
+        return connectServiceFactory.getService(dbDto.getType().toString()).connect(dbDto);
+    }
 }
 }

+package org.thingsboard.server.dao.yunteng.service;
+
+import org.thingsboard.server.common.data.yunteng.dto.TkDbConnectDTO;
+
+public interface BaseDbConnectService {
+
+    /**
+     * 获取数据库连接结果
+     *
+     * @param tkDbConnectDTO
+     * @return
+     */
+    Object connect(TkDbConnectDTO tkDbConnectDTO);
+}

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