io.github.kiryu1223.plugin.aot.DrinkRuntimeNativeRegistrar Maven / Gradle / Ivy

Go to download
package io.github.kiryu1223.plugin.aot;

import io.github.kiryu1223.drink.api.crud.read.Empty;
import io.github.kiryu1223.drink.api.crud.read.group.*;
import io.github.kiryu1223.drink.ext.BaseSqlExtension;
import io.github.kiryu1223.drink.ext.NoConverter;
import io.github.kiryu1223.drink.ext.mysql.MySqlDateTimeDiffExtension;
import io.github.kiryu1223.drink.ext.oracle.OracleAddOrSubDateExtension;
import io.github.kiryu1223.drink.ext.oracle.OracleCastExtension;
import io.github.kiryu1223.drink.ext.oracle.OracleDateTimeDiffExtension;
import io.github.kiryu1223.drink.ext.oracle.OracleJoinExtension;
import io.github.kiryu1223.drink.ext.types.Char;
import io.github.kiryu1223.drink.ext.types.SqlTypes;
import io.github.kiryu1223.drink.ext.types.Varchar2;
import io.github.kiryu1223.plugin.configuration.DrinkProperties;
import org.noear.solon.aot.RuntimeNativeMetadata;
import org.noear.solon.aot.RuntimeNativeRegistrar;
import org.noear.solon.aot.hint.MemberCategory;
import org.noear.solon.core.AppContext;

import java.util.Arrays;
import java.util.List;


public class DrinkRuntimeNativeRegistrar implements RuntimeNativeRegistrar
{
    @Override
    public void register(AppContext context, RuntimeNativeMetadata metadata)
    {
        //配置文件
        metadata.registerReflection(DrinkProperties.class, MemberCategory.DECLARED_FIELDS, MemberCategory.INVOKE_PUBLIC_METHODS, MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS);
        //转换器
        metadata.registerReflection(NoConverter.class, MemberCategory.DECLARED_FIELDS, MemberCategory.INVOKE_PUBLIC_METHODS, MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS);
        //空表
        metadata.registerReflection(Empty.class, MemberCategory.DECLARED_FIELDS, MemberCategory.INVOKE_PUBLIC_METHODS, MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS);
        //Group类
        metadata.registerReflection(Group.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group2.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group3.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group4.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group5.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group6.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group7.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group8.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group9.class, MemberCategory.PUBLIC_FIELDS);
        metadata.registerReflection(Group10.class, MemberCategory.PUBLIC_FIELDS);
        //Group的聚合函数类
        metadata.registerReflection(SqlAggregation.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation2.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation3.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation4.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation5.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation6.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation7.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation8.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation9.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        metadata.registerReflection(SqlAggregation10.class, MemberCategory.INTROSPECT_PUBLIC_METHODS);
        //表达式扩展
//        registerExtension(metadata, Arrays.asList(
//                MySqlDateTimeDiffExtension.class,
//                OracleAddOrSubDateExtension.class,
//                OracleCastExtension.class,
//                OracleDateTimeDiffExtension.class,
//                OracleJoinExtension.class
//        ));
        //sql类型
        registerSqlType(metadata, Arrays.asList(
                Char.class,
                Varchar2.class
        ));
    }

//    private void registerExtension(RuntimeNativeMetadata metadata, List> extensions)
//    {
//        for (Class extension : extensions)
//        {
//            metadata.registerReflection(extension, MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS, MemberCategory.INVOKE_PUBLIC_METHODS);
//        }
//    }

    private void registerSqlType(RuntimeNativeMetadata metadata, List>> sqlTypes)
    {
        for (Class> sqlType : sqlTypes)
        {
            metadata.registerReflection(sqlType, MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS, MemberCategory.INVOKE_PUBLIC_METHODS);
        }
    }
}