bf5fcf1f1159465e27fc1ac197e2462164d3c54b
[tomcat7.0] /
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements.  See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License.  You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 package org.apache.tomcat.jdbc.pool.interceptor;
18
19 import java.lang.reflect.InvocationHandler;
20 import java.lang.reflect.Method;
21 import java.lang.reflect.Proxy;
22 import java.sql.CallableStatement;
23 import java.sql.SQLException;
24
25 import org.apache.tomcat.jdbc.pool.ConnectionPool;
26 import org.apache.tomcat.jdbc.pool.JdbcInterceptor;
27 import org.apache.tomcat.jdbc.pool.PooledConnection;
28
29 /**
30  * @author Filip Hanik
31  * @version 1.0
32  */
33 public abstract class  AbstractCreateStatementInterceptor extends JdbcInterceptor {
34     public static final String[] statements = {"createStatement","prepareStatement","prepareCall"};
35     public static final String[] executes = {"execute","executeQuery","executeUpdate","executeBatch"};
36
37     public  AbstractCreateStatementInterceptor() {
38         super();
39     }
40
41     public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
42         boolean process = false;
43         process = process(statements, method, process);
44         if (process) {
45             Object statement = super.invoke(proxy,method,args);
46             return createStatement(proxy,method,args,statement);
47         } else {
48             return super.invoke(proxy,method,args);
49         }
50     }
51     
52     /**
53      * This method should return a wrapper object around a
54      * java.sql.Statement, java.sql.PreparedStatement or java.sql.CallableStatement
55      * @param proxy
56      * @param method
57      * @param args
58      * @param statement
59      * @return
60      */
61     public abstract Object createStatement(Object proxy, Method method, Object[] args, Object statement);
62
63     protected boolean process(String[] names, Method method, boolean process) {
64         for (int i=0; (!process) && i<names.length; i++) {
65             process = (method.getName()==names[i]);
66         }
67         return process;
68     }
69     
70     public void reset(ConnectionPool parent, PooledConnection con) {
71
72     }
73 }