git.internetallee.de Git - tomcat7.0/blob

   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     @Override
  42     public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  43         if (compare(CLOSE_VAL,method)) {
  44             closeInvoked();
  45             return super.invoke(proxy, method, args);
  46         } else {
  47             boolean process = false;
  48             process = process(statements, method, process);
  49             if (process) {
  50                 Object statement = super.invoke(proxy,method,args);
  51                 return createStatement(proxy,method,args,statement);
  52             } else {
  53                 return super.invoke(proxy,method,args);
  54             }
  55         }
  56     }
  57
  58     /**
  59      * This method should return a wrapper object around a
  60      * java.sql.Statement, java.sql.PreparedStatement or java.sql.CallableStatement
  61      * @param proxy
  62      * @param method
  63      * @param args
  64      * @param statement
  65      * @return
  66      */
  67     public abstract Object createStatement(Object proxy, Method method, Object[] args, Object statement);
  68
  69     public abstract void closeInvoked();
  70
  71     protected boolean process(String[] names, Method method, boolean process) {
  72         for (int i=0; (!process) && i<names.length; i++) {
  73             process = compare(names[i],method);
  74         }
  75         return process;
  76     }
  77
  78     public void reset(ConnectionPool parent, PooledConnection con) {
  79
  80     }
  81 }