Pārlūkot izejas kodu

Log enhancement: Use slf4j.Logger to replace System.out and print the exception stack trace. (#1116) (#1124)

Co-authored-by: xiaojunxiang <hellosrc@sina.com>
groot 6 mēneši atpakaļ
vecāks
revīzija
392db891ab

+ 5 - 2
src/main/java/io/milvus/pool/ClientPool.java

@@ -2,10 +2,13 @@ package io.milvus.pool;
 
 import org.apache.commons.pool2.impl.GenericKeyedObjectPool;
 import org.apache.commons.pool2.impl.GenericKeyedObjectPoolConfig;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.time.Duration;
 
 public class ClientPool<C, T> {
+    protected static final Logger logger = LoggerFactory.getLogger(ClientPool.class);
     protected GenericKeyedObjectPool<String, T> clientPool;
     protected PoolConfig config;
     protected PoolClientFactory<C, T> clientFactory;
@@ -48,7 +51,7 @@ public class ClientPool<C, T> {
         try {
             return clientPool.borrowObject(key);
         } catch (Exception e) {
-            System.out.println("Failed to get client, exception: " + e.getMessage());
+            logger.error("Failed to get client, exception: ", e);
             return null;
         }
     }
@@ -65,7 +68,7 @@ public class ClientPool<C, T> {
         try {
             clientPool.returnObject(key, grpcClient);
         } catch (Exception e) {
-            System.out.println("Failed to return client, exception: " + e.getMessage());
+            logger.error("Failed to return client, exception: " + e);
             throw e;
         }
     }

+ 5 - 2
src/main/java/io/milvus/pool/PoolClientFactory.java

@@ -3,11 +3,14 @@ package io.milvus.pool;
 import org.apache.commons.pool2.BaseKeyedPooledObjectFactory;
 import org.apache.commons.pool2.PooledObject;
 import org.apache.commons.pool2.impl.DefaultPooledObject;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Method;
 
 public class PoolClientFactory<C, T> extends BaseKeyedPooledObjectFactory<String, T> {
+    protected static final Logger logger = LoggerFactory.getLogger(PoolClientFactory.class);
     private final C config;
     private Constructor<?> constructor;
     private Method closeMethod;
@@ -22,7 +25,7 @@ public class PoolClientFactory<C, T> extends BaseKeyedPooledObjectFactory<String
             closeMethod = clientCls.getMethod("close", long.class);
             verifyMethod = clientCls.getMethod("clientIsReady");
         } catch (Exception e) {
-            System.out.println("Failed to create client pool factory, exception: " + e.getMessage());
+            logger.error("Failed to create client pool factory, exception: ", e);
             throw e;
         }
     }
@@ -54,7 +57,7 @@ public class PoolClientFactory<C, T> extends BaseKeyedPooledObjectFactory<String
             T client = p.getObject();
             return (boolean) verifyMethod.invoke(client);
         } catch (Exception e) {
-            System.out.println("Failed to validate client, exception: " + e.getMessage());
+            logger.error("Failed to validate client, exception: " + e);
             return true;
         }
     }