improve sql-wall

src/main/java/com/alibaba/druid/util/JdbcSqlStatUtils.java

@@ -18,7 +18,9 @@ package com.alibaba.druid.util;
 import java.lang.reflect.Method;
 import java.util.Arrays;
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicLongFieldUpdater;
 
 import com.alibaba.druid.stat.JdbcSqlStat;
@@ -75,4 +77,20 @@ public class JdbcSqlStatUtils {
             return updater.get(stat);
         }
     }
+    
+    public static long get(AtomicLong counter, boolean reset) {
+        if (reset) {
+            return counter.getAndSet(0);
+        } else {
+            return counter.get();
+        }
+    }
+    
+    public static int get(AtomicInteger counter, boolean reset) {
+        if (reset) {
+            return counter.getAndSet(0);
+        } else {
+            return counter.get();
+        }
+    }
 }