Fixes

Author: ZorTik

asql-core/src/main/java/me/zort/sqllib/SQLDatabaseConnection.java

@@ -1,7 +1,6 @@
 package me.zort.sqllib;
 
 import lombok.AllArgsConstructor;
-import lombok.Data;
 import lombok.Getter;
 import me.zort.sqllib.api.Query;
 import me.zort.sqllib.api.SQLConnection;
@@ -376,7 +375,10 @@ protected void logSqlError(Exception e) {
 
   @SuppressWarnings("all")
   protected void notifyHandlers(int code) {
-    errorCount++;
+    if (code >= 100) {
+      // Code is higher than 99, so it's an error
+      errorCount++;
+    }
     this.codeHandlers.forEach(handler -> runCatching(() -> handler.onNotified(code)));
   }
 
@@ -398,9 +400,11 @@ protected static class DefsVals {
   }
 
   public static final class Code {
-    public static final int QUERY_FATAL = 0;
     public static final int CONNECTED = 1;
     public static final int CLOSED = 2;
     public static final int CLEARING = 3;
+
+    // Error codes start on 100
+    public static final int QUERY_FATAL = 100;
   }
 }

asql-core/src/main/java/me/zort/sqllib/pool/SQLConnectionPool.java

@@ -130,18 +130,28 @@ private PooledSQLDatabaseConnection establishObject() throws SQLException {
     if (error != null) throw error;
 
     if (polled instanceof SQLDatabaseConnectionImpl)
-      ((SQLDatabaseConnectionImpl) polled).addCodeHandler(code -> handleConnectionError(polled));
+      ((SQLDatabaseConnectionImpl) polled).addCodeHandler(code -> handleConnectionCode(polled, code));
 
     return polled;
   }
 
-  synchronized void handleConnectionError(PooledSQLDatabaseConnection polled) {
-    errorCount++;
+  synchronized void handleConnectionCode(PooledSQLDatabaseConnection polled, int code) {
+    if (code == SQLDatabaseConnection.Code.CONNECTED) {
+      return;
+    }
+
+    if (code >= 100) {
+      errorCount++;
+    }
     // Remove the connection from the pool and disconnect
     // on fatal errors.
     freeConnections.remove(polled);
     usedConnections.remove(polled);
-    polled.disconnect();
+    if (code != SQLDatabaseConnection.Code.CLOSED) {
+      // I prevent stack overflow by closing only connections
+      // that are not already closed.
+      polled.disconnect();
+    }
   }
 
   void releaseObject(PooledSQLDatabaseConnection connection) {