source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube, open source software quality management tool.
   3  * Copyright (C) 2008-2013 SonarSource
   4  * mailto:contact AT sonarsource DOT com
   5  *
   6  * SonarQube is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * SonarQube is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.core.persistence.profiling;
  21
  22 import org.sonar.core.profiling.StopWatch;
  23
  24 import java.lang.reflect.InvocationHandler;
  25 import java.lang.reflect.Method;
  26 import java.sql.PreparedStatement;
  27
  28 class ProfilingPreparedStatementHandler implements InvocationHandler {
  29
  30   private static final SqlProfiling profiling = new SqlProfiling();
  31   private final PreparedStatement statement;
  32   private final String sql;
  33
  34   ProfilingPreparedStatementHandler(PreparedStatement statement, String sql) {
  35     this.statement = statement;
  36     this.sql = sql;
  37   }
  38
  39   @Override
  40   public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  41     if (method.getName().startsWith("execute")) {
  42       StopWatch watch = profiling.start();
  43       Object result = method.invoke(statement, args);
  44       profiling.stop(watch, sql);
  45       return result;
  46     } else {
  47       return method.invoke(statement, args);
  48     }
  49   }
  50
  51 }