Fixup null check

src/main/java/com/zaxxer/hikari/proxy/JavassistProxyFactoryFactory.java

@@ -100,19 +100,19 @@ public final class JavassistProxyFactoryFactory
             }
             if ("getProxyStatement".equals(method.getName()))
             {
-                call.append("return new com.zaxxer.hikari.proxy.StatementJavassistProxy($$);");
+                call.append("return $2 != null ? new com.zaxxer.hikari.proxy.StatementJavassistProxy($$) : null;");
             }
             if ("getProxyPreparedStatement".equals(method.getName()))
             {
-                call.append("return new com.zaxxer.hikari.proxy.PreparedStatementJavassistProxy($$);");
+                call.append("return $2 != null ? new com.zaxxer.hikari.proxy.PreparedStatementJavassistProxy($$) : null;");
             }
             if ("getProxyResultSet".equals(method.getName()))
             {
-                call.append("return new com.zaxxer.hikari.proxy.ResultSetJavassistProxy($$);");
+                call.append("return $2 != null ? new com.zaxxer.hikari.proxy.ResultSetJavassistProxy($$) : null;");
             }
             if ("getProxyCallableStatement".equals(method.getName()))
             {
-                call.append("return new com.zaxxer.hikari.proxy.CallableStatementJavassistProxy($$);");
+                call.append("return $2 != null ? new com.zaxxer.hikari.proxy.CallableStatementJavassistProxy($$) : null;");
             }
             call.append('}');
             method.setBody(call.toString());

src/main/java/com/zaxxer/hikari/proxy/PreparedStatementProxy.java

@@ -39,11 +39,23 @@ public class PreparedStatementProxy extends StatementProxy
 
     @HikariInject
     public ResultSet executeQuery() throws SQLException
+    {
+    	try
     	{
 	        IHikariResultSetProxy resultSet = (IHikariResultSetProxy) __executeQuery();
+    		if (resultSet == null)
+    		{
+    			return null;
+    		}
+
     		resultSet.setProxyStatement(this);
 	        return (ResultSet) resultSet;
     	}
+    	catch (SQLException e)
+    	{
+    		throw checkException(e);
+    	}
+    }
 
     // ***********************************************************************
     // These methods contain code we do not want injected into the actual
@@ -55,10 +67,6 @@ public class PreparedStatementProxy extends StatementProxy
     public ResultSet __executeQuery() throws SQLException
     {
         ResultSet resultSet = ((PreparedStatement) delegate).executeQuery();
-        if (resultSet == null)
-        {
-            return null;
-        }
         return PROXY_FACTORY.getProxyResultSet(this, resultSet);
     }
 

src/main/java/com/zaxxer/hikari/proxy/StatementProxy.java

@@ -64,24 +64,55 @@ public class StatementProxy implements IHikariStatementProxy
     public void close() throws SQLException
     {
         _connection.unregisterStatement(this);
+        try
+        {
         	__close();
         }
+        catch (SQLException e)
+        {
+        	throw checkException(e);
+        }
+    }
 
+    @HikariInject
     public ResultSet executeQuery(String sql) throws SQLException
+    {
+    	try
     	{
     		IHikariResultSetProxy resultSet = (IHikariResultSetProxy) __executeQuery(sql);
-        resultSet.setProxyStatement(this);
+    		if (resultSet == null)
+    		{
+    			return null;
+    		}
 
+	        resultSet.setProxyStatement(this);	
 	        return (ResultSet) resultSet;
     	}
+    	catch (SQLException e)
+    	{
+    		throw checkException(e);
+    	}
+    }
 
+    @HikariInject
     public ResultSet getGeneratedKeys() throws SQLException
+    {
+    	try
     	{
 	        IHikariResultSetProxy resultSet = (IHikariResultSetProxy) __getGeneratedKeys();
-        resultSet.setProxyStatement(this);
+    		if (resultSet == null)
+    		{
+    			return null;
+    		}
 
+	        resultSet.setProxyStatement(this);	
 	        return (ResultSet) resultSet;
     	}
+    	catch (SQLException e)
+    	{
+    		throw checkException(e);
+    	}
+    }
 
     // ***********************************************************************
     // These methods contain code we do not want injected into the actual
@@ -111,22 +142,12 @@ public class StatementProxy implements IHikariStatementProxy
     public ResultSet __executeQuery(String sql) throws SQLException
     {
         ResultSet resultSet = delegate.executeQuery(sql);
-        if (resultSet == null)
-        {
-            return null;
-        }
-
         return PROXY_FACTORY.getProxyResultSet(this, resultSet);
     }
 
     public ResultSet __getGeneratedKeys() throws SQLException
     {
         ResultSet generatedKeys = delegate.getGeneratedKeys();
-        if (generatedKeys == null)
-        {
-            return null;
-        }
-
         return PROXY_FACTORY.getProxyResultSet(this, generatedKeys);
     }