site.sorghum.anno._ddl.InitDataService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of anno-admin Show documentation
The newest version!
package site.sorghum.anno._ddl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.StopWatch;
import cn.hutool.core.util.StrUtil;
import jakarta.inject.Named;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.expression.operators.relational.ItemsList;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.insert.Insert;
import org.noear.wood.DbContext;
import org.noear.wood.annotation.Db;
import site.sorghum.anno._common.util.ScriptUtils;

import java.io.Reader;
import java.io.StringReader;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * 添加和升级预置数据
 *
 * @author songyinyin
 * @since 2023/7/9 11:49
 */
@Slf4j
@Named
public class InitDataService {

    @Db
    DbContext dbContext;

    public static Set systemFields = Set.of("create_time", "create_by", "update_time", "update_by", "del_flag");


    /**
     * 对比数据库中已有的数据，初始化数据或者升级历史数据
     *
     * @param sqlContent 预置数据文件
     */
    public void init(String sqlContent) throws Exception {
        // 初始化 init.sql
        initSql(new StringReader(sqlContent));
    }

    /**
     * 对比数据库中已有的数据，初始化数据或者升级历史数据
     *
     * @param sqlReader 预置数据 sql
     */
    private void initSql(Reader sqlReader) throws Exception {
        StopWatch stopWatch = new StopWatch("init sql");
        stopWatch.start("read sql");
        List statements = ScriptUtils.getStatements(sqlReader);
        stopWatch.stop();

        stopWatch.start("parse & execute sql");
        for (String statement : statements) {
            Statement parse = CCJSqlParserUtil.parse(statement);
            if (!(parse instanceof Insert)) {
                // 非 insert 语句，直接执行
                dbContext.exe(statement);
                continue;
            }

            // insert 语句，对比数据库中已有的数据，初始化数据或者升级历史数据
            List> list = new ArrayList<>();
            Map indexToColumn = new HashMap<>();
            for (int index = 0; index < ((Insert) parse).getColumns().size(); index++) {
                indexToColumn.put(index, ((Insert) parse).getColumns().get(index).getColumnName());
            }
            ItemsList itemsList = ((Insert) parse).getItemsList();
            itemsList.accept(new ExpressionListVisitor(list, indexToColumn));

            for (Map map : list) {
                String id = (String) map.get("id");
                if (StrUtil.isBlank(id)) {
                    continue;
                }
                String tableName = ((Insert) parse).getTable().getName();

                Set keys = map.keySet().stream().filter(k -> !systemFields.contains(k)).collect(Collectors.toSet());
                Map existMap = dbContext.table(tableName).whereEq("id", id).selectMap(String.join(",", keys));
                if (CollUtil.isEmpty(existMap)) {
                    dbContext.table(tableName).setMap(map).insert();
                } else {
                    systemFields.forEach(map::remove);
                    for (String key : keys) {
                        map.remove(key, existMap.get(key));
                    }
                    if (!map.isEmpty()) {
                        dbContext.table(tableName).setMap(map).whereEq("id", id).update();
                    }
                }
            }
        }
        stopWatch.stop();
        log.info("init sql(\n{}\n) finished, time: {}ms", statements, stopWatch.getTotal(TimeUnit.MILLISECONDS));
        if (log.isDebugEnabled()) {
            log.debug(stopWatch.prettyPrint(TimeUnit.MILLISECONDS));
        }
    }

}