From 86c475ecdfffaa1621d10bc80993726225633693 Mon Sep 17 00:00:00 2001 From: wangw <1594593906@qq.com> Date: Thu, 27 Mar 2025 16:03:56 +0800 Subject: [PATCH] =?UTF-8?q?sql=20=E8=AF=AD=E5=8F=A5=20=E5=88=9D=E5=A7=8B?= =?UTF-8?q?=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/czg/system/entity/SqlScript.java | 66 +++++++++ .../czg/system/service/SqlScriptService.java | 14 ++ .../system/mapper/SqlScriptMapper.java | 14 ++ .../service/impl/SqlScriptServiceImpl.java | 136 ++++++++++++++++++ .../main/resources/mapper/SqlScriptMapper.xml | 7 + 5 files changed, 237 insertions(+) create mode 100644 cash-common/cash-common-service/src/main/java/com/czg/system/entity/SqlScript.java create mode 100644 cash-common/cash-common-service/src/main/java/com/czg/system/service/SqlScriptService.java create mode 100644 cash-service/system-service/src/main/java/com/czg/service/system/mapper/SqlScriptMapper.java create mode 100644 cash-service/system-service/src/main/java/com/czg/service/system/service/impl/SqlScriptServiceImpl.java create mode 100644 cash-service/system-service/src/main/resources/mapper/SqlScriptMapper.xml diff --git a/cash-common/cash-common-service/src/main/java/com/czg/system/entity/SqlScript.java b/cash-common/cash-common-service/src/main/java/com/czg/system/entity/SqlScript.java new file mode 100644 index 00000000..c5c73b61 --- /dev/null +++ b/cash-common/cash-common-service/src/main/java/com/czg/system/entity/SqlScript.java @@ -0,0 +1,66 @@ +package com.czg.system.entity; + +import com.mybatisflex.annotation.Column; +import com.mybatisflex.annotation.Id; +import com.mybatisflex.annotation.Table; +import java.io.Serializable; +import java.time.LocalDateTime; + +import java.io.Serial; + +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +/** + * 实体类。 + * + * @author ww + * @since 2025-03-27 + */ +@Data +@Builder +@NoArgsConstructor +@AllArgsConstructor +@Table("sql_script") +public class SqlScript implements Serializable { + + @Serial + private static final long serialVersionUID = 1L; + + @Id + private Long id; + + /** + * 表名 + */ + private String tableName; + + /** + * 是否已执行 + */ + private Integer isUp; + + /** + * 该次执行描述 + */ + private String message; + + /** + * 执行的 sql语句 + */ + private String sqls; + + /** + * 操作日期 yyMMdd + */ + private String tradeDay; + + /** + * 时间 + */ + @Column(onInsertValue = "now()") + private LocalDateTime createTime; + +} diff --git a/cash-common/cash-common-service/src/main/java/com/czg/system/service/SqlScriptService.java b/cash-common/cash-common-service/src/main/java/com/czg/system/service/SqlScriptService.java new file mode 100644 index 00000000..bb58d4a1 --- /dev/null +++ b/cash-common/cash-common-service/src/main/java/com/czg/system/service/SqlScriptService.java @@ -0,0 +1,14 @@ +package com.czg.system.service; + +import com.mybatisflex.core.service.IService; +import com.czg.system.entity.SqlScript; + +/** + * 服务层。 + * + * @author ww + * @since 2025-03-27 + */ +public interface SqlScriptService extends IService { + +} diff --git a/cash-service/system-service/src/main/java/com/czg/service/system/mapper/SqlScriptMapper.java b/cash-service/system-service/src/main/java/com/czg/service/system/mapper/SqlScriptMapper.java new file mode 100644 index 00000000..fd965536 --- /dev/null +++ b/cash-service/system-service/src/main/java/com/czg/service/system/mapper/SqlScriptMapper.java @@ -0,0 +1,14 @@ +package com.czg.service.system.mapper; + +import com.mybatisflex.core.BaseMapper; +import com.czg.system.entity.SqlScript; + +/** + * 映射层。 + * + * @author ww + * @since 2025-03-27 + */ +public interface SqlScriptMapper extends BaseMapper { + +} diff --git a/cash-service/system-service/src/main/java/com/czg/service/system/service/impl/SqlScriptServiceImpl.java b/cash-service/system-service/src/main/java/com/czg/service/system/service/impl/SqlScriptServiceImpl.java new file mode 100644 index 00000000..c81d8a3d --- /dev/null +++ b/cash-service/system-service/src/main/java/com/czg/service/system/service/impl/SqlScriptServiceImpl.java @@ -0,0 +1,136 @@ +package com.czg.service.system.service.impl; + +import cn.hutool.core.date.DateUtil; +import cn.hutool.core.util.StrUtil; +import com.czg.service.system.mapper.SqlScriptMapper; +import com.czg.system.entity.SqlScript; +import com.czg.system.service.SqlScriptService; +import com.mybatisflex.spring.service.impl.ServiceImpl; +import jakarta.annotation.PostConstruct; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Profile; +import org.springframework.jdbc.core.JdbcTemplate; +import org.springframework.stereotype.Service; + +import java.io.BufferedReader; +import java.io.File; +import java.io.FileReader; +import java.io.IOException; +import java.sql.Connection; +import java.sql.Statement; +import java.util.Arrays; +import java.util.Date; +import java.util.List; + +/** + * 服务层实现。 + * + * @author ww + * @since 2025-03-27 + */ +@Slf4j +@Service +@Profile({"prod"}) +public class SqlScriptServiceImpl extends ServiceImpl implements SqlScriptService { + + + @Autowired + private JdbcTemplate jdbcTemplate; + + @PostConstruct + public void init() { + // 读取数据库表 sql_script 最后一条 日期 + String lastTradeDay = queryChain().select(SqlScript::getTradeDay).orderBy(SqlScript::getTradeDay).desc().oneAs(String.class); + if (StrUtil.isBlank(lastTradeDay)) { + lastTradeDay = "250301"; + } + // 读取目录 sqls/ 其中是以日期为目录的 多个目录 获取 日期 大于 最后一条 日期的所有目录 + File sqlDirectory = new File("sqls"); + if (sqlDirectory.exists() && sqlDirectory.isDirectory()) { + File[] dateDirectories = sqlDirectory.listFiles(); + if (dateDirectories != null) { + for (File dateDir : dateDirectories) { + if (dateDir.isDirectory()) { + String dirDate = dateDir.getName(); + if (isDateGreater(dirDate, lastTradeDay)) { + // 遍历执行 sql 文件 + executeSqlFilesInDirectory(dateDir); + } + } + } + } + } + log.info("初始化执行SQL初始化完成"); + } + + // 判断日期大小 + private boolean isDateGreater(String date1, String date2) { + return date1.compareTo(date2) > 0; + } + + // 执行指定目录下的 SQL 文件 + private void executeSqlFilesInDirectory(File directory) { + File[] sqlFiles = directory.listFiles((dir, name) -> name.endsWith(".sql")); + if (sqlFiles != null) { + for (File sqlFile : sqlFiles) { + try { + if (!isSqlExecuted(directory.getName(), sqlFile.getName())) { + String sql = readSqlStatementsFromFile(sqlFile); + if (StrUtil.isNotBlank(sql)) { + executeSql(sql); + } + markSqlAsExecuted(directory.getName(), sqlFile.getName(), sql); + } + } catch (IOException e) { + log.error("执行SQL文件异常:{}", sqlFile.getName()); + } + } + } + } + + // 从文件中读取 SQL 语句 + private String readSqlStatementsFromFile(File file) throws IOException { + try (BufferedReader reader = new BufferedReader(new FileReader(file))) { + StringBuilder statement = new StringBuilder(); + String line; + while ((line = reader.readLine()) != null) { + statement.append(line).append("\n"); + } + return statement.toString(); + } + } + + // 判断 SQL 是否已经执行过 + private boolean isSqlExecuted(String date, String sqlFileName) { + return queryChain().eq(SqlScript::getTradeDay, date).eq(SqlScript::getTableName, sqlFileName).eq(SqlScript::getIsUp, 1).exists(); + } + + // 执行 SQL 语句 + private void executeSql(String sqls) { + // 这里应该是实际的数据库执行逻辑，暂时打印 SQL 语句 + List sqlList = Arrays.asList(sqls.split("\\r?\\n")); + for (String sql : sqlList) { + if (!sql.trim().isEmpty()) { + jdbcTemplate.execute(sql); + } + } + log.info("初始化执行SQL:{}", sqls); + } + + // 标记 SQL 为已执行 + private void markSqlAsExecuted(String date, String sqlFileName, String sqls) { + // 这里应该是实际的数据库更新逻辑，暂时打印标记信息 + SqlScript sqlScript = queryChain().eq(SqlScript::getTradeDay, date).eq(SqlScript::getTableName, sqlFileName).eq(SqlScript::getIsUp, 1).one(); + if (sqlScript == null) { + sqlScript = new SqlScript(); + sqlScript.setTableName(sqlFileName); + sqlScript.setIsUp(1); + sqlScript.setSqls(sqls); + sqlScript.setTradeDay(DateUtil.format(new Date(), "yyMMdd")); + save(sqlScript); + } else { + updateChain().set(SqlScript::getSqls, sqls).set(SqlScript::getIsUp, 1).eq(SqlScript::getId, sqlScript.getId()).update(); + } + } +} diff --git a/cash-service/system-service/src/main/resources/mapper/SqlScriptMapper.xml b/cash-service/system-service/src/main/resources/mapper/SqlScriptMapper.xml new file mode 100644 index 00000000..fdce3c02 --- /dev/null +++ b/cash-service/system-service/src/main/resources/mapper/SqlScriptMapper.xml @@ -0,0 +1,7 @@ + + + + +