游戏版本要回滚，还好我机智备份了数据库，代码直接拿走

package screw;
import cn.smallbun.screw.core.Configuration;
import cn.smallbun.screw.core.engine.EngineConfig;
import cn.smallbun.screw.core.engine.EngineFileType;
import cn.smallbun.screw.core.engine.EngineTemplateType;
import cn.smallbun.screw.core.execute.DocumentationExecute;
import cn.smallbun.screw.core.process.ProcessConfig;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import javax.sql.DataSource;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
public class App {
  public static void main(String[] ags) throws IOException, SQLException {
      String dbName = "fate";
      HikariConfig config = new HikariConfig();
      config.setDriverClassName("com.mysql.cj.jdbc.Driver");
      config.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/" + dbName +" ?serverTimezone=UTC");
      config.setUsername("root");
      config.setPassword("root");
      config.addDataSourceProperty("useInformationSchema", "true");
      config.setMinimumIdle(2);
      config.setMaximumPoolSize(5);
      DataSource ds = new HikariDataSource(config);
      String userDir = System.getProperty("user.dir") + "\\src\\test\\java\\com\\pdool\\";
      System.out.println(userDir);
      SimpleDateFormat dataFormat = new SimpleDateFormat("yyMMdd");
      String versionStr = dataFormat.format(new Date());
      List<String> ignoreTable = new ArrayList<>();
      List<String> ignorePrefix = new ArrayList<>();
      List<String> ignoreSuffix = new ArrayList<>();
      ignoreSuffix.add("_test");
      ignoreSuffix.add("test");
      for (int i = 0; i < 10; i++) {
          ignoreSuffix.add(String.valueOf(i));
      }
      createHtml(ds, userDir, versionStr, ignoreTable, ignorePrefix, ignoreSuffix);
      createSql(dbName, ds, userDir, versionStr, ignoreTable, ignorePrefix, ignoreSuffix);
  }
  public static void createHtml(DataSource dataSource, String userDir, String versionStr, List<String> ignoreTable, List<String> ignorePrefix, List<String> ignoreSuffix) {
      EngineConfig engineConfig = EngineConfig.builder()
              .fileOutputDir(userDir)
              .openOutputDir(false)
              .fileType(EngineFileType.HTML)
              .produceType(EngineTemplateType.freemarker)
              .build();
      ProcessConfig processConfig = ProcessConfig.builder()
              .ignoreTableName(ignoreTable)
              .ignoreTablePrefix(ignorePrefix)
              .ignoreTableSuffix(ignoreSuffix)
              .build();
      Configuration config = Configuration.builder()
              .version(versionStr)
              .description("数据库文档")
              .dataSource(dataSource)
              .engineConfig(engineConfig)
              .produceConfig(processConfig).build();
      new DocumentationExecute(config).execute();
  }
  public static void createSql(String dbName, DataSource dataSource, String userDir, String versionStr, List<String> ignoreTable, List<String> ignorePrefix, List<String> ignoreSuffix) throws IOException, SQLException {
      Statement tmt = null;
      PreparedStatement pstmt = null;
      List<String> createSqlList = new ArrayList<>();
      String sql = "select TABLE_NAME from INFORMATION_SCHEMA.TABLES where TABLE_SCHEMA = '"+dbName+"' and TABLE_TYPE = 'BASE TABLE'";
      tmt = dataSource.getConnection().createStatement();
      pstmt = dataSource.getConnection().prepareStatement(sql);
      ResultSet res = tmt.executeQuery(sql);
      while (res.next()) {
          String tableName = res.getString(1);
          if (tableName.contains("`")) {
              continue;
          }
          if (ignoreTable.contains(tableName)) {
              continue;
          }
          boolean isContinue = false;
          for (String prefix : ignorePrefix) {
              if (tableName.startsWith(prefix)) {
                  isContinue = true;
                  break;
              }
          }
          if (isContinue) {
              continue;
          }
          for (String suffix : ignoreSuffix) {
              if (tableName.startsWith(suffix)) {
                  isContinue = true;
                  break;
              }
          }
          if (isContinue) {
              continue;
          }
          ResultSet rs = pstmt.executeQuery("show create Table `" + tableName + "`");
          while (rs.next()) {
              createSqlList.add("DROP TABLE IF EXISTS '" + tableName + "'");
              createSqlList.add(rs.getString(2));
          }
      }
      String head = "-- 数据库建表语句 \r\n";
      head += "-- db:" + dbName + " version: " + versionStr + "\r\n";
      String collect = String.join(";\r\n", createSqlList);
      collect = head + collect + ";";
      string2file(collect, userDir + dbName + "_" + versionStr + ".sql");
  }
  public static void string2file(String collect, String dirStr) throws IOException {
      System.out.println("文件地址 "+ dirStr);
      OutputStreamWriter osw = null;
      try {
          osw = new OutputStreamWriter(new FileOutputStream(new File(dirStr)), StandardCharsets.UTF_8);
          osw.write(collect);
          osw.flush();
      } finally {
          if (osw != null) {
              osw.close();
          }
      }
  }
}
复制代码