site.sorghum.anno.db.service.wood.AnnoWoodConfig Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of anno-admin Show documentation

The newest version!

package site.sorghum.anno.db.service.wood;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import jakarta.inject.Named;
import org.noear.wood.WoodConfig;
import org.noear.wood.wrap.TypeConverter;
import site.sorghum.anno._common.util.AnnoContextUtil;
import site.sorghum.anno.anno.util.ReentrantStopWatch;

import java.io.IOException;
import java.sql.SQLException;

@Named
public class AnnoWoodConfig {

    /**
     * 初始化wood配置
     */
    public void init(){
        WoodConfig.typeConverter = new TypeConverter(){
            @Override
            public Object convert(Object val, Class target) throws SQLException, IOException {
                return Convert.convert(target, val);
            }
        };
        WoodConfig.onExecuteBef((cmd) -> {
            if (AnnoContextUtil.hasContext()) {
                ReentrantStopWatch stopWatch = AnnoContextUtil.getContext().getStopWatch();
                String taskName;
                if (CollUtil.isEmpty(cmd.paramMap())) {
                    taskName = cmd.text;
                } else {
                    taskName = String.format("%s\n%s params ==> %s", cmd.text, " ".repeat(16), cmd.paramMap());
                }
                stopWatch.start(taskName);
            }
            return true;
        });
        WoodConfig.onExecuteAft((cmd) -> {
            if (AnnoContextUtil.hasContext()) {
                ReentrantStopWatch stopWatch = AnnoContextUtil.getContext().getStopWatch();
                stopWatch.stop();
            }
        });
        WoodConfig.primaryKeyStrategy = new AnnoPrimaryKeyStrategy();
        WoodConfig.namingStrategy = new AnnoNamingStrategy();
    }
}