CompleteSqlLogger

最新推荐文章于 2024-11-01 14:48:20 发布

喝酸奶要舔盖儿

最新推荐文章于 2024-11-01 14:48:20 发布

阅读量121

点赞数

文章标签： java 开发语言后端

本文链接：https://blog.csdn.net/qq_39361915/article/details/122495467

版权

记录一下

public class CompleteSqlLogger extends BaseJdbcLogger implements InvocationHandler {
    private final Connection connection;

    private CompleteSqlLogger(Connection conn, Log statementLog, int queryStack) {
        super(statementLog, queryStack);
        this.connection = conn;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] params)
            throws Throwable {
        try {
            if (Object.class.equals(method.getDeclaringClass())) {
                return method.invoke(this, params);
            }
            if ("prepareStatement".equals(method.getName()) || "prepareCall".equals(method.getName())) {
                if (isDebugEnabled()) {
                    debug(" Preparing: " + getSql((String) params[0]), true);
                    debug("Parameters: " + getParameterValueString(), true);
                }
                PreparedStatement stmt = (PreparedStatement) method.invoke(connection, params);
                stmt = PreparedStatementLogger.newInstance(stmt, statementLog, queryStack);
                return stmt;
            } else if ("createStatement".equals(method.getName())) {
                Statement stmt = (Statement) method.invoke(connection, params);
                stmt = StatementLogger.newInstance(stmt, statementLog, queryStack);
                return stmt;
            } else {
                return method.invoke(connection, params);
            }
        } catch (Throwable t) {
            throw ExceptionUtil.unwrapThrowable(t);
        }
    }

    /**
     * Creates a logging version of a connection.
     *
     * @param conn         the original connection
     * @param statementLog the statement log
     * @param queryStack   the query stack
     * @return the connection with logging
     */
    public static Connection newInstance(Connection conn, Log statementLog, int queryStack) {
        InvocationHandler handler = new CompleteSqlLogger(conn, statementLog, queryStack);
        ClassLoader cl = Connection.class.getClassLoader();
        return (Connection) Proxy.newProxyInstance(cl, new Class[]{Connection.class}, handler);
    }

    /**
     * return the wrapped connection.
     *
     * @return the connection
     */
    public Connection getConnection() {
        return connection;
    }

    public String getSql(String sql) {
        String cs = getColumnString();
        String[] colum = cs.substring(1, cs.length() - 1).replaceAll(" ", "").split(",");
        for (int i = 0; i < colum.length; i++) {
            sql=sql.replaceFirst("\\?","'"+objectValueString(getColumn(colum[i]))+"'");
        }
        return sql;
    }
}