dsProject/dsRes/src/main/java/com/dsideal/Ai/ResApplication.java

package com.dsideal.Ai;

import cn.hutool.core.io.FileUtil;
import com.dsideal.Config.PropKit;
import com.dsideal.Ai.Base.Controller.BaseController;
import com.dsideal.Ai.Index.IndexController;
import com.dsideal.Ai.Interceptor.*;
import com.dsideal.Ai.Milvus.Controller.MilvusDemoController;
import com.dsideal.Ai.Model._MappingKit;
import com.dsideal.Ai.Neo4j.Controller.Neo4jDemoController;
import com.dsideal.Ai.Plugin.MilvusPlugin;
import com.dsideal.Ai.Plugin.Neo4jPlugin;
import com.dsideal.Ai.Res.Controller.ResourceController;
import com.dsideal.Ai.Util.LogBackLogFactory;
import com.jfinal.config.*;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.druid.DruidPlugin;
import com.jfinal.plugin.redis.RedisPlugin;
import com.jfinal.server.undertow.UndertowServer;
import com.jfinal.template.Engine;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;

public class ResApplication extends JFinalConfig {
    private static final Logger logger = LoggerFactory.getLogger(ResApplication.class);

    public static void main(String[] args) {
        System.out.println("当前环境: " + PropKit.getEnvPrefix());
        UndertowServer.create(ResApplication.class, "undertow.properties").start();
    }


    @Override
    public void configConstant(Constants me) {
        //使用LogBack
        me.setLogFactory(new LogBackLogFactory());
    }

    /**
     * 配置路由
     */
    @Override
    public void configRoute(Routes me) {
        //默认页面
        me.add("/", IndexController.class);
        //资源基础管理
        me.add("/base", BaseController.class);
        //资源管理
        me.add("/res", ResourceController.class);
        //Neo4j测试
        me.add("/neo4j", Neo4jDemoController.class);
        //Milvus测试
        me.add("/milvus", MilvusDemoController.class);
    }

    @Override
    public void configEngine(Engine engine) {

    }


    @Override
    public void configPlugin(Plugins me) {
        DruidPlugin plugin = new DruidPlugin(PropKit.get("mysql.jdbcUrl"), PropKit.get("mysql.user"),
                PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName"));
        me.add(plugin);

        // 配置ActiveRecord插件
        ActiveRecordPlugin arp = new ActiveRecordPlugin("master", plugin);
        arp.setDialect(new MysqlDialect());
        //加入ORM配置
        _MappingKit.mapping(arp);


        //遍历sql目录下所有的sql文件
        File sqlDir;
        String basePath = ResApplication.class.getResource("/").getPath();
        sqlDir = new File(basePath + "/Sql");
        File[] sqlFiles = sqlDir.listFiles();
        for (File sqlFile : sqlFiles != null ? sqlFiles : new File[0]) {
            //只加载.sql文件
            if (sqlFile.getName().indexOf(".sql") > 0) {
                arp.addSqlTemplate("/Sql/" + sqlFile.getName());
            }
        }
        //加载
        me.add(arp);

        // 配置 Redis
        RedisPlugin redis = new RedisPlugin("Redis", PropKit.get("redis.ip"),
                PropKit.getInt("redis.port"), 10 * 1000, null);
        //启动redis组件
        me.add(redis);

        // 配置 Neo4j
        String url = PropKit.get("neo4j.url");
        String username = PropKit.get("neo4j.username");
        String password = PropKit.get("neo4j.password");
        logger.info("正在初始化Neo4j连接: url={}, username={},password={}", url, username, password);
        Neo4jPlugin neo4jPlugin = Neo4jPlugin.getInstance();
        neo4jPlugin.setConfig(url, username, password); // 设置Neo4j连接配置
        me.add(neo4jPlugin); // 添加Neo4j插件

        // 配置 Milvus
        String host = PropKit.get("milvus.ms_host");
        int port = PropKit.getInt("milvus.ms_port");
        int maxConnections = PropKit.getInt("milvus.ms_max_connections");
        logger.info("正在初始化Milvus连接: host={}, port={}", host, port);
        MilvusPlugin milvusPlugin = MilvusPlugin.getInstance();
        milvusPlugin.setConfig(host, port, maxConnections); // 设置Milvus连接配置
        me.add(milvusPlugin); // 添加Milvus插件
    }

    /**
     * 配置全局拦截器
     */
    @Override
    public void configInterceptor(Interceptors me) {
        //注册参数数字检查器
        me.add(new IsNumbericInterceptor());

        //注册非空拦截器
        me.add(new EmptyInterceptor());

        //注册Ids检查是不是数字的检查器
        me.add(new CheckIdsInterceptor());

        //注册一个GUID号的检查器
        me.add(new IsGuidInterceptor());

        //检查是不是系统管理员
        me.add(new IsSysAdminInterceptor());

        //注册一个检查输入文本长度的拦截器
        me.add(new LengthInterceptor());

    }

    /**
     * 配置处理器
     */
    @Override
    public void configHandler(Handlers me) {
    }

    /**
     * 在jfinal启动完成后马上执行
     */
    @Override
    public void onStart() {
        //打印 启动Logo
        System.out.println(FileUtil.readUtf8String("logo.txt"));
    }
}