Update MapperProxy.java

plugin/datasource/src/main/java/com/alibaba/nacos/plugin/datasource/proxy/MapperProxy.java

@@ -18,6 +18,7 @@
 
 import com.alibaba.nacos.common.utils.JacksonUtils;
 import com.alibaba.nacos.plugin.datasource.mapper.Mapper;
+import com.alibaba.nacos.plugin.datasource.model.MapperResult;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -35,20 +36,20 @@
  * @author hyx
  **/
 public class MapperProxy implements InvocationHandler {
-    
+
     private static final Logger LOGGER = LoggerFactory.getLogger(MapperProxy.class);
-    
+
     private Mapper mapper;
-    
+
     private static final Map<String, MapperProxy> SINGLE_MAPPER_PROXY_MAP = new HashMap<>(16);
-    
+
     private static final ReadWriteLock LOCK = new ReentrantReadWriteLock(true);
-    
+
     public <R> R createProxy(Mapper mapper) {
         this.mapper = mapper;
         return (R) Proxy.newProxyInstance(MapperProxy.class.getClassLoader(), mapper.getClass().getInterfaces(), this);
     }
-    
+
     /**
      * create proxy-mapper single instead of using method createProxy.
      */
@@ -72,15 +73,21 @@ public static <R> R createSingleProxy(Mapper mapper) {
             LOCK.readLock().unlock();
         }
     }
-    
+
     @Override
     public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
         Object invoke = method.invoke(mapper, args);
-        
+
         String className = mapper.getClass().getSimpleName();
         String methodName = method.getName();
-        String sql = invoke.toString();
-
+
+        String sql;
+        if (invoke instanceof MapperResult) {
+            sql = ((MapperResult) invoke).getSql();
+        } else {
+            sql = invoke.toString();
+        }
+
         LOGGER.info("[{}] METHOD : {}, SQL : {}, ARGS : {}", className, methodName, sql, JacksonUtils.toJson(args));
         return invoke;
     }