From 582c7a5e5a6d9af20bedfdc27c61f552f24a3372 Mon Sep 17 00:00:00 2001
From: Brett Wooldridge <brett.wooldridge@gmail.com>
Date: Tue, 11 Mar 2014 19:20:29 +0900
Subject: [PATCH] Simplify the modifyProxyFactory() method.

---
 .../hikari/proxy/JavassistProxyFactory.java      | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/zaxxer/hikari/proxy/JavassistProxyFactory.java b/src/main/java/com/zaxxer/hikari/proxy/JavassistProxyFactory.java
index f878a2ef..6018189e 100644
--- a/src/main/java/com/zaxxer/hikari/proxy/JavassistProxyFactory.java
+++ b/src/main/java/com/zaxxer/hikari/proxy/JavassistProxyFactory.java
@@ -102,30 +102,22 @@ public final class JavassistProxyFactory
         for (CtMethod method : proxyCt.getMethods())
         {
             String methodName = method.getName();
-            StringBuilder call = new StringBuilder("{");
             if ("getProxyConnection".equals(methodName))
             {
-                call.append("return new ").append(packageName).append(".ConnectionJavassistProxy($$);");
+                method.setBody("{return new " + packageName + ".ConnectionJavassistProxy($$);}");
             }
             else if ("getProxyStatement".equals(methodName))
             {
-                call.append("return new ").append(packageName).append(".StatementJavassistProxy($$);");
+                method.setBody("{return new " + packageName + ".StatementJavassistProxy($$);}");
             }
             else if ("getProxyPreparedStatement".equals(methodName))
             {
-                call.append("return new ").append(packageName).append(".PreparedStatementJavassistProxy($$);");
+                method.setBody("{return new " + packageName + ".PreparedStatementJavassistProxy($$);}");
             }
             else if ("getProxyCallableStatement".equals(methodName))
             {
-                call.append("return new ").append(packageName).append(".CallableStatementJavassistProxy($$);");
+                method.setBody("{return new " + packageName + ".CallableStatementJavassistProxy($$);}");
             }
-            else
-            {
-                continue;
-            }
-
-            call.append('}');
-            method.setBody(call.toString());
         }
 
         proxyCt.toClass(classPool.getClassLoader(), null);