BaseConfig.java 9.63 KB
/**
 * Copyright 2015-2025 FLY的狐狸(email:jflyfox@sina.com qq:369191470).
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.jflyfox.component.config;

import com.jfinal.plugin.druid.DruidPlugin;
import com.jflyfox.component.beelt.JFinalBeetlRenderFactory;
import org.beetl.core.GroupTemplate;

import com.beetl.functions.BeetlStrUtils;
import com.jfinal.config.Constants;
import com.jfinal.config.Handlers;
import com.jfinal.config.Interceptors;
import com.jfinal.config.JFinalConfig;
import com.jfinal.config.Plugins;
import com.jfinal.config.Routes;
import com.jfinal.ext.handler.ContextPathHandler;
import com.jfinal.ext.interceptor.SessionInViewInterceptor;
import com.jfinal.kit.PathKit;
import com.jfinal.log.Log4jLogFactory;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.CaseInsensitiveContainerFactory;
import com.jfinal.plugin.activerecord.SqlReporter;
import com.jfinal.plugin.activerecord.dialect.OracleDialect;
import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
import com.jfinal.plugin.activerecord.dialect.Sqlite3Dialect;
import com.jfinal.plugin.c3p0.C3p0Plugin;
import com.jfinal.render.ViewType;
import com.jfinal.template.Engine;
import com.jflyfox.component.beelt.BeeltFunctions;
import com.jflyfox.component.interceptor.CommonInterceptor;
import com.jflyfox.component.interceptor.PageViewInterceptor;
import com.jflyfox.component.interceptor.SiteInterceptor;
import com.jflyfox.component.interceptor.UpdateCacheInterceptor;
import com.jflyfox.component.interceptor.UserKeyInterceptor;
import com.jflyfox.component.util.JFlyFoxCache;
import com.jflyfox.jfinal.component.annotation.AutoBindModels;
import com.jflyfox.jfinal.component.annotation.AutoBindRoutes;
import com.jflyfox.jfinal.component.handler.BasePathHandler;
import com.jflyfox.jfinal.component.handler.CurrentPathHandler;
import com.jflyfox.jfinal.component.handler.HtmlHandler;
import com.jflyfox.jfinal.component.interceptor.ExceptionInterceptor;
import com.jflyfox.jfinal.component.interceptor.JflyfoxInterceptor;
import com.jflyfox.jfinal.component.interceptor.SessionAttrInterceptor;
import com.jflyfox.modules.front.template.TemplateImageService;
import com.jflyfox.modules.front.template.TemplateService;
import com.jflyfox.modules.front.template.TemplateVideoService;
import com.jflyfox.system.user.UserInterceptor;
import com.jflyfox.util.Config;
import com.jflyfox.util.StrUtils;
import com.jflyfox.util.cache.Cache;
import com.jflyfox.util.cache.CacheManager;
import com.jflyfox.util.cache.ICacheManager;
import com.jflyfox.util.cache.RedisCache;
import com.jflyfox.util.cache.impl.MemoryCache;
import com.jflyfox.util.cache.impl.MemorySerializeCache;
import com.jflyfox.util.serializable.FSTSerializer;
import com.jflyfox.util.serializable.SerializerManage;

/**
 * API引导式配置
 */
public class BaseConfig extends JFinalConfig {

    private static final String CONFIG_WEB_ROOT = "{webroot}";

    @Override
    public void configConstant(Constants me) {
        me.setDevMode(isDevMode());
        me.setViewType(ViewType.JSP); // 设置视图类型为Jsp,否则默认为FreeMarker
        me.setLogFactory(new Log4jLogFactory());
        me.setError401View(Config.getStr("PAGES.401"));
        me.setError403View(Config.getStr("PAGES.403"));
        me.setError404View(Config.getStr("PAGES.404"));
        me.setError500View(Config.getStr("PAGES.500"));

        // 开启日志
        SqlReporter.setLog(true);

        JFinalBeetlRenderFactory rf = new JFinalBeetlRenderFactory();
        rf.config();
        me.setRenderFactory(rf);

        // 获取GroupTemplate ,可以设置共享变量等操作
        GroupTemplate groupTemplate = rf.groupTemplate;
        groupTemplate.registerFunctionPackage("strutil", BeetlStrUtils.class);
        groupTemplate.registerFunctionPackage("flyfox", BeeltFunctions.class);
        groupTemplate.registerFunctionPackage("temp", TemplateService.class);
        groupTemplate.registerFunctionPackage("tempImage", TemplateImageService.class);
        groupTemplate.registerFunctionPackage("tempVideo", TemplateVideoService.class);

    }

    ;

    /**
     * 配置路由
     */
    @Override
    public void configRoute(Routes me) {
        me.setBaseViewPath("/pages");
        // 自动绑定
        // 1.如果没用加入注解,必须以Controller结尾,自动截取前半部分为key
        // 2.加入ControllerBind的 获取 key
        me.add(new AutoBindRoutes());
    }

    /**
     * 配置插件
     */
    @Override
    public void configPlugin(Plugins me) {
        // 配置druid数据库连接池插件
        DruidPlugin druidPlugin = null;

        String db_type = Config.getStr("db_type") + ".";

        String webRoot = PathKit.getWebRootPath();
        String DBPath = webRoot + "\\WEB-INF\\";
        DBPath = StrUtils.replace(DBPath, "\\", "/");
        String jdbcUrl = Config.getStr(db_type + "jdbcUrl");
        if (db_type.startsWith("sqlite")) {
            jdbcUrl = StrUtils.replaceOnce(jdbcUrl, CONFIG_WEB_ROOT, DBPath);
        }

        druidPlugin = new DruidPlugin(
                jdbcUrl, Config.getStr(db_type + "user"),
                Config.getStr(db_type + "password").trim(),
                Config.getStr(db_type + "driverClass"));

        me.add(druidPlugin);

        // 配置ActiveRecord插件
        ActiveRecordPlugin arp = new ActiveRecordPlugin(druidPlugin);
        me.add(arp);
        if (isDevMode()) {
            arp.setShowSql(true);
        }

        // 数据库类型
        if (db_type.startsWith("postgre")) {
            arp.setDialect(new PostgreSqlDialect());
        } else if (db_type.startsWith("sqlite")) {
            arp.setDialect(new Sqlite3Dialect());
        } else if (db_type.startsWith("oracle")) {
            arp.setDialect(new OracleDialect());
            arp.setContainerFactory(new CaseInsensitiveContainerFactory(true));
        }

        new AutoBindModels(arp);
    }

    @Override
    public void configHandler(Handlers me) {
        // Beelt
        // me.add(new BeeltHandler());
        // me.add(new ImageHandler());

        me.add(new HtmlHandler());
        // 全路径获取
        me.add(new BasePathHandler(Config.getStr("PATH.BASE_PATH")));
        // 根目录获取
        me.add(new ContextPathHandler(Config.getStr("PATH.CONTEXT_PATH")));
        // 当前获取
        me.add(new CurrentPathHandler(Config.getStr("PATH.CURRENT_PATH")));
    }

    /**
     * 配置全局拦截器
     */
    @Override
    public void configInterceptor(Interceptors me) {
        // 异常拦截器,跳转到500页面
        me.add(new ExceptionInterceptor());
        // session model转换
        me.add(new SessionInViewInterceptor());
        // 设置session属性
        me.add(new SessionAttrInterceptor());
        // 公共拦截器
        me.add(new JflyfoxInterceptor());
        // 用户Key设置
        me.add(new UserKeyInterceptor());
        // page view 统计
        me.add(new PageViewInterceptor());
        // 缓存更新
        me.add(new UpdateCacheInterceptor());
        // 用户认证
        me.add(new UserInterceptor());
        // 站点拦截
        me.add(new SiteInterceptor());
        // 公共属性
        me.add(new CommonInterceptor());
    }

    /**
     * 初始化
     */
    @Override
    public void onStart() {
        super.onStart();

        // 初始化Cache为fst序列化
        SerializerManage.add("fst", new FSTSerializer());

        // 设置序列化工具
        String defaultKey = Config.getStr("CACHE.SERIALIZER.DEFAULT");
        defaultKey = StrUtils.isEmpty(defaultKey) ? "java" : defaultKey;
        SerializerManage.setDefaultKey(defaultKey);


        // 设置缓存
        CacheManager.setCache(new ICacheManager() {

            @Override
            public Cache getCache() {
                String cacheName = Config.getStr("CACHE.NAME");
                cacheName = StrUtils.isEmpty(cacheName) ? "MemorySerializeCache" : cacheName;

                if ("MemorySerializeCache".equals(cacheName)) {
                    return new MemorySerializeCache();
                } else if ("MemoryCache".equals(cacheName)) {
                    return new MemoryCache();
                } else if ("RedisCache".equals(cacheName)) {
                    return new RedisCache();
                } else {
                    throw new RuntimeException("####init cache error!");
                }
            }
        });

        JFlyFoxCache.init();
        System.out.println("##################################");
        System.out.println("############系统启动完成##########");
        System.out.println("##################################");
    }

    @Override
    public void onStop() {

        // 关闭模板
        // BeetlRenderFactory.groupTemplate.close();

        super.onStop();

        System.out.println("##################################");
        System.out.println("############系统停止完成##########");
        System.out.println("##################################");
    }


    /**
     * 配置模板
     */
    @Override
    public void configEngine(Engine engine) {

    }

    private boolean isDevMode() {
        return Config.getToBoolean("CONSTANTS.DEV_MODE");
    }


}