解决方案 1:子类
只需创建一个 PreparedStatement 的自定义实现,它将所有调用委托给原始准备语句,仅在 setObject 等方法中添加回调。示例:
public PreparedStatement prepareStatement(String sql) {
final PreparedStatement delegate = conn.prepareStatement(sql);
return new PreparedStatement() {
// TODO: much more methods to delegate
@Override
public void setString(int parameterIndex, String x) throws SQLException {
// TODO: remember value of X
delegate.setString(parameterIndex, x);
}
};
}
如果你想保存参数并在以后获取它们,有很多解决方案,但我更喜欢创建一个像 ParameterAwarePreparedStatement 这样的新类,它在地图中包含参数。结构可能类似于:
public class ParameterAwarePreparedStatement implements PreparedStatement {
private final PreparedStatement delegate;
private final Map<Integer,Object> parameters;
public ParameterAwarePreparedStatement(PreparedStatement delegate) {
this.delegate = delegate;
this.parameters = new HashMap<>();
}
public Map<Integer,Object> getParameters() {
return Collections.unmodifiableMap(parameters);
}
// TODO: many methods to delegate
@Override
public void setString(int parameterIndex, String x) throws SQLException {
delegate.setString(parameterIndex, x);
parameters.put(parameterIndex, x);
}
}
解决方案 2:动态代理
第二个解决方案更短,但看起来更老套。
您可以通过调用 java.lang.reflect.Proxy 上的工厂方法来创建动态代理,并将所有调用委托给原始实例。示例:
public PreparedStatement prepareStatement(String sql) {
final PreparedStatement ps = conn.prepareStatement(sql);
final PreparedStatement psProxy = (PreparedStatement) Proxy.newProxyInstance(ClassLoader.getSystemClassLoader(), new Class<?>[]{PreparedStatement.class}, new InvocationHandler() {
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
if (method.getName().equals("setLong")) {
// ... your code here ...
}
// this invokes the default call
return method.invoke(ps, args);
}
});
return psProxy;
}
然后您通过查看方法名称并查看您的值的第二个方法参数来拦截 setObject 等调用。