package com.company.common.core.annotation;

/**
 * 數(shù)據(jù)庫類型
 *
 * @author kally
 * @date 2023/12/13
 */
public enum DbType {
    /**
     * 數(shù)據(jù)庫類型（類型，描述）
     */
    MYSQL("mysql", "MySql數(shù)據(jù)庫"),
    MARIADB("mariadb", "MariaDB數(shù)據(jù)庫"),
    ORACLE("oracle", "Oracle11g及以下數(shù)據(jù)庫(高版本推薦使用ORACLE_NEW)"),
    ORACLE_12C("oracle12c", "Oracle12c+數(shù)據(jù)庫"),
    DB2("db2", "DB2數(shù)據(jù)庫"),
    H2("h2", "H2數(shù)據(jù)庫"),
    HSQL("hsql", "HSQL數(shù)據(jù)庫"),
    SQLITE("sqlite", "SQLite數(shù)據(jù)庫"),
    POSTGRE_SQL("postgresql", "Postgre數(shù)據(jù)庫"),
    SQL_SERVER2005("sqlserver2005", "SQLServer2005數(shù)據(jù)庫"),
    SQL_SERVER("sqlserver", "SQLServer數(shù)據(jù)庫"),
    DM("dm", "達夢數(shù)據(jù)庫"),
    XU_GU("xugu", "虛谷數(shù)據(jù)庫"),
    KINGBASE_ES("kingbasees", "人大金倉數(shù)據(jù)庫"),
    PHOENIX("phoenix", "Phoenix HBase數(shù)據(jù)庫"),
    GAUSS("zenith", "Gauss 數(shù)據(jù)庫"),
    CLICK_HOUSE("clickhouse", "clickhouse 數(shù)據(jù)庫"),
    GBASE("gbase", "南大通用(華庫)數(shù)據(jù)庫"),
    GBASE_8S("gbase-8s", "南大通用數(shù)據(jù)庫 GBase 8s"),

    @Deprecated
    GBASEDBT("gbasedbt", "南大通用數(shù)據(jù)庫"),

    @Deprecated
    GBASE_INFORMIX("gbase 8s", "南大通用數(shù)據(jù)庫 GBase 8s"),

    SINODB("sinodb", "星瑞格數(shù)據(jù)庫"),
    OSCAR("oscar", "神通數(shù)據(jù)庫"),
    SYBASE("sybase", "Sybase ASE 數(shù)據(jù)庫"),
    OCEAN_BASE("oceanbase", "OceanBase 數(shù)據(jù)庫"),
    FIREBIRD("Firebird", "Firebird 數(shù)據(jù)庫"),
    HIGH_GO("highgo", "瀚高數(shù)據(jù)庫"),
    CUBRID("cubrid", "CUBRID數(shù)據(jù)庫"),
    GOLDILOCKS("goldilocks", "GOLDILOCKS數(shù)據(jù)庫"),
    CSIIDB("csiidb", "CSIIDB數(shù)據(jù)庫"),
    SAP_HANA("hana", "SAP_HANA數(shù)據(jù)庫"),
    IMPALA("impala", "impala數(shù)據(jù)庫"),
    VERTICA("vertica", "vertica數(shù)據(jù)庫"),
    XCloud("xcloud", "行云數(shù)據(jù)庫"),
    REDSHIFT("redshift", "亞馬遜redshift數(shù)據(jù)庫"),
    OPENGAUSS("openGauss", "華為 opengauss 數(shù)據(jù)庫"),
    TDENGINE("TDengine", "TDengine數(shù)據(jù)庫"),
    INFORMIX("informix", "Informix數(shù)據(jù)庫"),
    UXDB("uxdb", "優(yōu)炫數(shù)據(jù)庫"),
    LEALONE("lealone", "Lealone數(shù)據(jù)庫"),
    OTHER("other", "其他數(shù)據(jù)庫");

    private final String db;
    private final String desc;

    public static DbType getDbType(String dbType) {
        DbType[] var1 = values();
        int var2 = var1.length;

        for (int var3 = 0; var3 < var2; ++var3) {
            DbType type = var1[var3];
            if (type.db.equalsIgnoreCase(dbType)) {
                return type;
            }
        }

        return OTHER;
    }

    public String getDb() {
        return this.db;
    }

    public String getDesc() {
        return this.desc;
    }

    /**
     * 數(shù)據(jù)庫類型
     *
     * @param db   數(shù)據(jù)庫
     * @param desc 描述
     */
    private DbType(final String db, final String desc) {
        this.db = db;
        this.desc = desc;
    }

}

@Bean
    public DmFieldCustomizer getDmFieldCustomizer() {
        return new DmFieldCustomizer();
    }

    /**
     * 在mybatisPlus加載進spring容器之前自定義某些配置
     *
     * @author HetFrame
     */
    public static class DmFieldCustomizer implements MybatisPlusPropertiesCustomizer {
        public DmFieldCustomizer() {
            log.info("加載DmFieldCustomizer...");
        }

        @SneakyThrows
        @Override
        public void customize(MybatisPlusProperties properties) {
            // 使用達夢數(shù)據(jù)庫
            if (Arrays.toString(properties.getMapperLocations()).contains("dm")) {
                log.info("使用達夢數(shù)據(jù)庫");
                //實體類的class
                List<Class<?>> classList = new ArrayList<>();
                PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
                //找到所有實體類的class
                Resource[] resources = resolver.getResources("classpath*:com/**/entity/**/*.class");
                for (Resource res : resources) {
                    // 先獲取resource的元信息，然后獲取class元信息，最后得到 class 全路徑
                    String clsName = new SimpleMetadataReaderFactory().getMetadataReader(res).getClassMetadata().getClassName();
                    // 通過名稱加載
                    Class<?> clazz = Class.forName(clsName);
                    classList.add(clazz);
                }

                classList.forEach(e -> {
                    List<Field> list = TableInfoHelper.getAllFields(e);
                    list.forEach(field -> {
                        TableField tableField = field.getAnnotation(TableField.class);
                        String metaColName;
                        if (tableField != null && StringUtils.isNotBlank(metaColName = tableField.value()) && metaColName.contains("`")) {

                            String newColName = metaColName.replace("`", """);

                            InvocationHandler invocationHandler = Proxy.getInvocationHandler(tableField);
                            try {
                                Field memberValues = invocationHandler.getClass().getDeclaredField("memberValues");
                                memberValues.setAccessible(true);
                                Map memberValuesMap = (Map) memberValues.get(invocationHandler);
                                memberValuesMap.put("value", newColName);
                                log.info("將實體類映射字段{}修改為{}", metaColName, newColName);
                            } catch (NoSuchFieldException | IllegalAccessException exception) {
                                throw new RuntimeException(exception);
                            }
                        }
                    });
                });

            } else {
                log.info("使用mysql數(shù)據(jù)庫");
            }
        }
    }

package com.company.common.framework.mybatis;


import com.baomidou.mybatisplus.autoconfigure.MybatisPlusProperties;

/**
 * MybatisPlusPropertiesCustomizer 用于在讀取 properties 之后進行一些自定義操作
 *
 * @author kally
 * @date 2023/12/13
 */
public interface MybatisPlusPropertiesCustomizer {
    /**
     * mysql中我們使用反引號來區(qū)分列名和關鍵字；但在達夢數(shù)據(jù)庫中，使用雙引號來區(qū)分。
     * 而如果創(chuàng)建實體類在注解 @TableField 中使用了反引號例如 @TableField(`range`)，
     * 這種我們改了的話，mysql 就用不了，不改達夢就用不了。所以從代碼入手，在啟動時用反射修改值。
     * 達夢數(shù)據(jù)庫某些關鍵字例如 audit，在 mysql 中不是關鍵字的，需要手動設置 @TableField(`audit`)。
     */

    /**
     * MybatisPlusPropertiesCustomizer 用于在讀取 properties 之后進行一些自定義操作
     *
     * @param properties {@link MybatisPlusProperties}
     */
    void customize(MybatisPlusProperties properties);
}