com.litongjava.db.activerecord.SqlReporter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-db Show documentation

Java Database

There is a newer version: 1.4.5

package com.litongjava.db.activerecord;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.Connection;

import lombok.extern.slf4j.Slf4j;

/**
 * SqlReporter.
 */
@Slf4j
public class SqlReporter implements InvocationHandler {

  private Connection conn;
  private static boolean logOn = false;

  public SqlReporter(Connection conn) {
    this.conn = conn;
  }

  public static void setLog(boolean on) {
    SqlReporter.logOn = on;
  }

  @SuppressWarnings("rawtypes")
  public Connection getConnection() {
    Class clazz = conn.getClass();
    return (Connection) Proxy.newProxyInstance(clazz.getClassLoader(), new Class[] { Connection.class }, this);
  }

  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    try {
      if (method.getName().equals("prepareStatement")) {
        String info = "Sql: " + args[0];
        if (logOn) {
          log.info(info);
        } else {
          System.out.println(info);
        }
      }
      return method.invoke(conn, args);
    } catch (InvocationTargetException e) {
      throw e.getTargetException();
    }
  }
}