初始提交

pom.xml

@@ -50,6 +50,11 @@
             <artifactId>springfox-swagger-ui</artifactId>
             <version>2.8.0</version>
         </dependency>
+        <dependency>
+            <groupId>com.baomidou</groupId>
+            <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
+            <version>3.4.1</version>
+        </dependency>
     </dependencies>
 
     <build>

src/main/java/com/tykj/datahouse/controller/TableInfoController.java

 package com.tykj.datahouse.controller;
 
+import com.tykj.datahouse.base.result.ResultObj;
+import com.tykj.datahouse.base.result.ResultUtil;
+import com.tykj.datahouse.entity.TableInfo;
 import com.tykj.datahouse.service.TableInfoService;
 import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.List;
+
 
 @Api(tags = "基础模型接口")
 @CrossOrigin
-@RequestMapping("/tableinfo")
+@RequestMapping("/table")
 @RestController
 public class TableInfoController {
 
     @Autowired
     private TableInfoService tableInfoService;
 
+
+    @PostMapping
+    @ApiOperation("创建一个新表")
+    public ResponseEntity<ResultObj<Object>> createTable(@RequestBody TableInfo tableInfo) {
+        tableInfoService.createTable(tableInfo);
+        return ResultUtil.success("操作成功");
+    }
+
+    @PutMapping
+    @ApiOperation("修改一个表")
+    public ResponseEntity<ResultObj<Object>> alterTable(@RequestBody TableInfo tableInfo) {
+        tableInfoService.alterTable(tableInfo);
+        return ResultUtil.success("操作成功");
+    }
+
+    @DeleteMapping("/{id}")
+    @ApiOperation("按指定id删除一个表")
+    public ResponseEntity<ResultObj<Object>> dropTable(@PathVariable Integer id) {
+        tableInfoService.deleteTable(id);
+        return ResultUtil.success("操作成功");
+    }
+
+    @GetMapping
+    @ApiOperation("查询当前所有表")
+    public ResponseEntity<ResultObj<List<TableInfo>>> findAll() {
+        List<TableInfo> all = tableInfoService.findAll();
+        return ResultUtil.success(all, "查询成功");
+    }
+
 }

src/main/java/com/tykj/datahouse/entity/ColumnInfo.java

@@ -7,9 +7,7 @@ import lombok.Data;
 import lombok.NoArgsConstructor;
 import lombok.experimental.Accessors;
 
-import javax.persistence.Entity;
-import javax.persistence.Id;
-import javax.persistence.Transient;
+import javax.persistence.*;
 import java.util.List;
 
 @Accessors(chain = true)
@@ -20,6 +18,7 @@ import java.util.List;
 public class ColumnInfo {
 
     @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
     @ApiModelProperty("主键")
     @JsonIgnore
     private Integer id;
@@ -41,6 +40,9 @@ public class ColumnInfo {
     @ApiModelProperty(value = "长度", position = 5)
     private Integer length;
 
+    @ApiModelProperty(value = "长度", position = 5)
+    private Boolean notNull;
+
     @ApiModelProperty(value = "描述", position = 6)
     private String description;
 

src/main/java/com/tykj/datahouse/entity/TableInfo.java

@@ -19,10 +19,10 @@ import java.util.List;
 public class TableInfo {
 
     @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
     @ApiModelProperty("主键")
     private Integer id;
 
-
     @ApiModelProperty(value = "创建时间", position = 1)
     @Temporal(TemporalType.TIMESTAMP)
     @JsonFormat(pattern = "yyyy-MM-dd hh:mm")

src/main/java/com/tykj/datahouse/service/DataSourceManager.java

+package com.tykj.datahouse.service;
+
+import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
+import com.baomidou.dynamic.datasource.creator.BasicDataSourceCreator;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
+import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
+import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import javax.sql.DataSource;
+import java.util.Map;
+
+/**
+ * 数据源管理Service
+ * 可以动态增加和切换数据源
+ */
+@Service
+public class DataSourceManager {
+
+    @Autowired
+    private DataSource dataSource;
+
+    private BasicDataSourceCreator dataSourceCreator = new BasicDataSourceCreator(new DynamicDataSourceProperties());
+
+    public Map<String,DataSource> getDataSources(){
+        return ((DynamicRoutingDataSource) dataSource).getDataSources();
+    }
+
+    public void addDataSource(String name) {
+        DataSourceProperty dataSourceProperty = new DataSourceProperty();
+        DataSource basicDataSource = dataSourceCreator.createDataSource(dataSourceProperty);
+        ((DynamicRoutingDataSource) dataSource).addDataSource(name, basicDataSource);
+    }
+
+    public void switchToDataSource(String dataSourceName) {
+        DynamicDataSourceContextHolder.push(dataSourceName);
+    }
+
+    public void clear() {
+        DynamicDataSourceContextHolder.clear();
+    }
+
+}

src/main/java/com/tykj/datahouse/sql/SqlCreator.java

+package com.tykj.datahouse.sql;
+
+import com.tykj.datahouse.entity.ColumnInfo;
+import com.tykj.datahouse.entity.TableInfo;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * sql语句生成器
+ * 根据不同的操作对应不同的API
+ * 输入表信息或字段信息 返回对应的sql语句结果
+ */
+public interface SqlCreator {
+
+    /**
+     * 创建新表
+     * @param tableInfo 要创建的表
+     * @return 对应的sql语句结果
+     */
+    String createTableSql(TableInfo tableInfo);
+
+    /**
+     * 在表中添加字段
+     * @param tableInfo 被修改的表
+     * @param columnInfo 要添加的字段
+     * @return 对应的sql语句结果
+     */
+    String addColumnSql(TableInfo tableInfo, ColumnInfo columnInfo);
+
+    /**
+     * 修改表中的字段
+     * @param tableInfo 被修改的表
+     * @param oColumnInfo 原字段
+     * @param columnInfo 现字段
+     * @return 对应的sql语句结果
+     */
+    String alterColumnSql(TableInfo tableInfo, ColumnInfo oColumnInfo,ColumnInfo columnInfo);
+
+    /**
+     *
+     * @param tableInfo 被修改的表
+     * @param columnInfo 要删除的字段
+     * @return 对应的sql语句结果
+     */
+    String removeColumnSql(TableInfo tableInfo, ColumnInfo columnInfo);
+
+    /**
+     * 删除表
+     * @param tableInfo 被删除的表
+     * @return 对应的sql语句结果
+     */
+    String dropTableSql(TableInfo tableInfo);
+
+    /**
+     * 插入一条数据
+     * @param tableInfo 被插入数据的表
+     * @param row Map形式的数据对象
+     * @return 对应的sql语句结果
+     */
+    String insertSql(TableInfo tableInfo, Map<String, Object> row);
+
+    /**
+     * 更新一条数据
+     * @param tableInfo 被更新数据的表
+     * @param row Map形式的数据对象
+     * @return 对应的sql语句结果
+     */
+    String updateSql(TableInfo tableInfo, Map<String, Object> row);
+
+    /**
+     * 删除一条数据
+     * @param tableInfo 被删除数据的表
+     * @param row Map形式的数据对象
+     * @return 对应的sql语句结果
+     */
+    String deleteSql(TableInfo tableInfo, Map<String, Object> row);
+
+    String querySql(String tableName, List<String> columns);
+
+}

src/main/java/com/tykj/datahouse/sql/impl/MysqlSqlCreator.java

+package com.tykj.datahouse.sql.impl;
+
+import com.tykj.datahouse.entity.ColumnInfo;
+import com.tykj.datahouse.entity.TableInfo;
+import com.tykj.datahouse.sql.SqlCreator;
+
+import java.util.List;
+import java.util.Map;
+
+import static java.lang.String.format;
+import static java.util.Objects.nonNull;
+
+public class MysqlSqlCreator implements SqlCreator {
+
+    @Override
+    public String createTableSql(TableInfo tableInfo) {
+        List<ColumnInfo> columnInfos = tableInfo.getColumnInfos();
+        ColumnInfo primary = columnInfos.stream()
+                .filter(ColumnInfo::getIsPrimary)
+                .findAny()
+                .orElseThrow(() -> new RuntimeException("没有主键"));
+        StringBuilder result = new StringBuilder();
+        //起始
+        result.append(format("CREATE TABLE %s(\n", tableInfo.getName()));
+        //其他字段
+        for (ColumnInfo columnInfo : columnInfos) {
+            result.append(format("%s %s", columnInfo.getName(), columnInfo.getType()));
+            if (nonNull(columnInfo.getLength())) {
+                result.append(format("(%s)", columnInfo.getLength()));
+            }
+            if (nonNull(columnInfo.getNotNull())) {
+                result.append(" not null");
+            }
+            result.append(",\n");
+        }
+        //主键
+        result.append(format("PRIMARY KEY (%s)\n", primary.getName()));
+        //结尾
+        result.append(")");
+        return result.toString();
+    }
+
+    @Override
+    public String addColumnSql(TableInfo tableInfo, ColumnInfo columnInfo) {
+        StringBuilder result = new StringBuilder();
+        result.append(format("ALTER TABLE %s ADD COLUMN %s %s", tableInfo.getName(), columnInfo.getName(), columnInfo.getType()));
+        if (nonNull(columnInfo.getLength())) {
+            result.append(format("(%s)", columnInfo.getLength()));
+        }
+        if (nonNull(columnInfo.getNotNull())) {
+            result.append(" not null");
+        }
+        return result.toString();
+    }
+
+    @Override
+    public String alterColumnSql(TableInfo tableInfo, ColumnInfo oColumnInfo, ColumnInfo columnInfo) {
+        StringBuilder result = new StringBuilder();
+        result.append(format("ALTER TABLE %s CHANGE COLUMN %s %s %s", tableInfo.getName(), oColumnInfo.getName(), columnInfo.getName(), columnInfo.getType()));
+        if (nonNull(columnInfo.getLength())) {
+            result.append(format("(%s)", columnInfo.getLength()));
+        }
+        if (nonNull(columnInfo.getNotNull())) {
+            result.append(" not null");
+        }
+        return result.toString();
+    }
+
+    @Override
+    public String removeColumnSql(TableInfo tableInfo, ColumnInfo columnInfo) {
+        return format("ALTER TABLE %s DROP COLUMN %s", tableInfo.getName(), columnInfo.getName());
+    }
+
+    @Override
+    public String dropTableSql(TableInfo tableInfo) {
+        return format("DROP TABLE %s", tableInfo.getName());
+    }
+
+    @Override
+    public String insertSql(TableInfo tableInfo, Map<String, Object> row) {
+        List<ColumnInfo> columnInfos = tableInfo.getColumnInfos();
+        StringBuilder result = new StringBuilder();
+        StringBuilder columns = new StringBuilder();
+        StringBuilder values = new StringBuilder();
+        //起始
+        result.append(format("INSERT INTO %s ", tableInfo.getName()));
+        for (ColumnInfo columnInfo : columnInfos) {
+            columns.append(columnInfo.getName()).append(",");
+            values.append(format("\'%s\',", row.get(columnInfo.getName())));
+        }
+        int columnLastIndex = columns.lastIndexOf(",");
+        if (columnLastIndex > 0) {
+            columns.delete(columnLastIndex, columns.length());
+        }
+        int valueLastIndex = values.lastIndexOf(",");
+        if (valueLastIndex > 0) {
+            values.delete(valueLastIndex, values.length());
+        }
+        result.append(format("(%s) VALUES (%s)", columns.toString(), values.toString()));
+        return result.toString();
+    }
+
+    @Override
+    public String updateSql(TableInfo tableInfo, Map<String, Object> row) {
+        List<ColumnInfo> columnInfos = tableInfo.getColumnInfos();
+        ColumnInfo primary = columnInfos.stream()
+                .filter(ColumnInfo::getIsPrimary)
+                .findAny()
+                .orElseThrow(() -> new RuntimeException("没有主键"));
+        columnInfos.remove(primary);
+        StringBuilder result = new StringBuilder();
+        result.append(format("UPDATE %s SET ", tableInfo.getName()));
+        for (ColumnInfo columnInfo : columnInfos) {
+            result.append(format("%s = \'%s\',", columnInfo.getName(), row.get(columnInfo.getName())));
+        }
+        int lastIndex = result.lastIndexOf(",");
+        if (lastIndex > 0) {
+            result.delete(lastIndex, result.length());
+        }
+        result.append(format(" WHERE (%s = \'%s\')", primary.getName(), row.get(primary.getName())));
+        return result.toString();
+    }
+
+    @Override
+    public String deleteSql(TableInfo tableInfo, Map<String, Object> row) {
+        StringBuilder result = new StringBuilder();
+        ColumnInfo primary = tableInfo.getColumnInfos().stream()
+                .filter(ColumnInfo::getIsPrimary)
+                .findAny()
+                .orElseThrow(() -> new RuntimeException("没有主键"));
+        result.append(format("DELETE FROM %s WHERE (%s = \'%s\')", tableInfo.getName(), primary.getName(), row.get(primary.getName())));
+        return result.toString();
+    }
+
+    @Override
+    public String querySql(String tableName, List<String> columns) {
+        StringBuilder result = new StringBuilder();
+
+        return result.toString();
+    }
+
+}

src/main/resources/application-mysql.yml

@@ -3,21 +3,31 @@ spring:
     date-format: yyyy-MM-dd HH:mm:ss
     time-zone: GMT+8
   datasource:
-    username: root
-    password: root
-    url: jdbc:mysql://192.168.100.248:3306/datahouse?useSSL=true&verifyServerCertificate=false&useUnicode=true&autoReconnect=true&characterEncoding=utf-8&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true
-    driver-class-name: com.mysql.cj.jdbc.Driver
-    hikari:
-      read-only: false
-      connection-timeout: 60000
-      idle-timeout: 60000
-      validation-timeout: 3000
-      max-lifetime: 60000
-      login-timeout: 5
-      maximum-pool-size: 60
-      minimum-idle: 10
+    dynamic:
+      primary: master #设置默认的数据源或者数据源组,默认值即为master
+      strict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源
+      datasource:
+        master:
+          url: jdbc:mysql://192.168.100.248:3306/datahouse?useSSL=true&verifyServerCertificate=false&useUnicode=true&autoReconnect=true&characterEncoding=utf-8&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true
+          username: root
+          password: root
+          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0开始支持SPI可省略此配置
+#          hikari:
+#            read-only: false
+#            connection-timeout: 60000
+#            idle-timeout: 60000
+#            validation-timeout: 3000
+#            max-lifetime: 60000
+#            login-timeout: 5
+#            maximum-pool-size: 60
+#            minimum-idle: 10
+        target:
+          url: jdbc:mysql://192.168.100.248:3306/datahouse_data?useSSL=true&verifyServerCertificate=false&useUnicode=true&autoReconnect=true&characterEncoding=utf-8&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true
+          username: root
+          password: root
+          driver-class-name: com.mysql.cj.jdbc.Driver
   jpa:
-    show-sql: true
+    show-sql: false
     hibernate:
       naming:
         physical-strategy: com.tykj.datahouse.base.config.ToUpperCase