com.anji.plus.gaea.archiver.service.DBOperatorService Maven / Gradle / Ivy

Go to download
package com.anji.plus.gaea.archiver.service;

import com.anji.plus.gaea.archiver.config.ArchiverTable;
import com.anji.plus.gaea.archiver.consant.DBType;
import com.anji.plus.gaea.archiver.sqlbuilder.ISQLBuilder;
import com.anji.plus.gaea.archiver.sqlbuilder.SQLBuilderFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.support.rowset.SqlRowSet;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.sql.SQLException;
import java.util.*;
/**
 * 数据库操作实现类
 *
 * @author 木子李·De
 * @since 2021/2/3 14:16
 */
@Service
public class DBOperatorService implements IDBOperatorService {

    private static DBType dbType;

    @Autowired
    private JdbcTemplate jdbcTemplate;

    private DBType getDBType() throws SQLException {
        if(dbType != null){
            return dbType;
        }
        String databaseProductName=jdbcTemplate.getDataSource().getConnection().getMetaData().getDatabaseProductName();
        if(databaseProductName == null || databaseProductName.trim().length() == 0){
            throw new RuntimeException(String.format("%s database not supported", databaseProductName));
        }
        dbType=DBType.getDbType(databaseProductName);
        return Optional.ofNullable(dbType).orElseThrow(()->
            new RuntimeException(String.format("%s database not supported", databaseProductName))
        );
    }

    private ISQLBuilder getSQLBuilder(){
        try{
            return SQLBuilderFactory.getSQLBuilder(getDBType());
        }catch (Exception e){
            e.printStackTrace();
            throw new RuntimeException(e.getMessage());
        }
    }

    @Override
    public boolean existTable(String tableName) {
        try{
            String sql = getSQLBuilder().existTable(tableName);
            jdbcTemplate.queryForRowSet(sql);
            return true;
        }catch (Exception e){
            return false;
        }
    }

    @Override
    public boolean existTableField(String tableName, String fieldName) {
        try{
            String sql = getSQLBuilder().existTableField(tableName, fieldName);
            SqlRowSet sqlRowSet = jdbcTemplate.queryForRowSet(sql);

            boolean existField = false;
            while (sqlRowSet.next()){
                String fileName = sqlRowSet.getString("Field");
                if(fileName.toLowerCase().trim().equals(fieldName.trim())){
                    existField = true;
                    break;
                }
            }
            return existField;
        }catch (Exception e){
            return false;
        }
    }

    @Override
    public Date getNow() {
        String sql = getSQLBuilder().getNow();
        Map map = jdbcTemplate.queryForMap(sql);
        Date now = (Date)map.get("now");
        return null;
    }

    @Override
    public Long countRows(String sourceTableName) {
        return null;
    }

    @Override
    public List getCrossMonthList(String tablename, String timefield, Integer maxDaysBeforeArchive) {
        String sql = getSQLBuilder().getCrossMonthList(tablename, timefield, maxDaysBeforeArchive);
        List> list = jdbcTemplate.queryForList(sql);

        //yyyy-MM
        List monthList = new ArrayList();
        if(list == null ||list.isEmpty()){
            return monthList;
        }

        for(Map item: list){
            String monthStr = (String) item.get("month");
            monthList.add(monthStr);
        }
        return monthList;
    }

    @Override
    public String createArchiverTableIfNotExist(String sourceTableName, String monthStr){
        monthStr = monthStr.replace("-","");
        String archiveTableName = String.format("%s_%s", sourceTableName, monthStr);
        if(existTable(archiveTableName) == false){
            String sql = getSQLBuilder().createArchiverTableIfNotExist(archiveTableName, sourceTableName);
            jdbcTemplate.execute(sql);
        }
        return archiveTableName;
    }

    @Override
    public boolean doArchiver(String sourceTableName, Date startTime, Date endTime) {
        return false;
    }

    /**
     * 扫描以表名为前缀的所有归档表
     *
     * @param sourceTableName
     * @return list
     */
    @Override
    public List scanArchiver(String sourceTableName, String dbName) {
        String sql = getSQLBuilder().scanArchive(sourceTableName, dbName);
        List> maps = jdbcTemplate.queryForList(sql);
        List tableNameList = new ArrayList();
        maps.forEach(stringObjectMap -> {
            String tableName = (String) stringObjectMap.get("table_name");
            tableNameList.add(tableName);
        });
        return tableNameList;
    }

    /**
     * 执行dump命令
     *
     * @param archiverTable
     * @return
     */
    @Override
    public String executeDump(ArchiverTable archiverTable) {
        return getSQLBuilder().executeDump(archiverTable);
    }

    /**
     * drop 过期的数据库表
     *
     * @param archiverTable
     * @return
     */
    @Override
    public String dropTable(ArchiverTable archiverTable) {
        String sql = getSQLBuilder().dropTable(archiverTable);
        jdbcTemplate.execute(sql);
        return null;
    }
}