com.github.azbh111.jdbclogger.SqlLog Maven / Gradle / Ivy

Go to download
package com.github.azbh111.jdbclogger;

import com.github.azbh111.jdbclogger.stacktrace.StackTraceManager;
import com.github.azbh111.jdbclogger.stacktrace.model.StackTraceInfo;
import com.github.azbh111.jdbclogger.wrappers.QueryWrapper;
import com.github.azbh111.jdbclogger.wrappers.SqlCallable;

import java.sql.SQLException;
import java.util.List;

/**
 * @author: zyp
 * @since: 2021/9/18 13:11
 */
public class SqlLog {
    private static Logger logger = (str) -> System.out.println(str);

    public static void setLogger(Logger logger) {
        SqlLog.logger = logger;
    }

    public static void sqllog(long cost, String sql) {
        if (ignore(sql)) {
            return;
        }
        StackTraceInfo stackTraceInfo = StackTraceManager.getInstance().getStackTraceInfo();
        logger.log(String.format("%s cost %s, %s", stackTraceInfo.toString(), cost, sql));
    }

    public static void log(String str) {
        System.out.println(str);
    }

    public static  T wrapExecute(List queryWrappers, SqlCallable runnable) throws SQLException {
        long start = System.currentTimeMillis();
        try {
            return runnable.call();
        } finally {
            long stop = System.currentTimeMillis();
            long duration = stop - start;
            SqlLog.sqllog(duration, QueryWrapper.toString(queryWrappers));
        }
    }

    public static  T wrapExecute(String sql, SqlCallable runnable) throws SQLException {
        long start = System.currentTimeMillis();
        try {
            return runnable.call();
        } finally {
            long stop = System.currentTimeMillis();
            long duration = stop - start;
            SqlLog.sqllog(duration, sql);
        }
    }

    private static boolean ignore(String sql) {
        if (sql == null || sql.startsWith("/* ping */")) {
            return true; // 不打印ping请求
        }
        return false;
    }
}