add listener for run function

Author: windancer2010

src/com/xxdb/DBConnection.java

@@ -56,6 +56,12 @@ public class DBConnection {
     private boolean isReverseStreaming_ = false;
     private int tryReconnectNums = -1;
 
+    private ProgressListener _listener;
+
+    public void set_listener(ProgressListener _listener) {
+        this._listener = _listener;
+    }
+
     private static final Logger log = LoggerFactory.getLogger(DBConnection.class);
 
     private enum ExceptionType{
@@ -1345,7 +1351,7 @@ public Entity run(String function, List<Entity> arguments, int priority, int par
 
                 while (!closed_) {
                     try {
-                        return conn_.run(function, (ProgressListener)null, arguments, priority, parallelism, fetchSize, false, currentSeqNo);
+                        return conn_.run(function, this._listener == null?null:this._listener, arguments, priority, parallelism, fetchSize, false, currentSeqNo);
                     } catch (IOException e) {
                         if (currentSeqNo > 0)
                             currentSeqNo = -currentSeqNo;
@@ -1364,7 +1370,7 @@ else if (type == ExceptionType.ET_UNKNOW)
                 }
                 return null;
             } else {
-                return conn_.run(function, (ProgressListener)null, arguments, priority, parallelism, fetchSize, false, 0);
+                return conn_.run(function, this._listener==null?null:this._listener, arguments, priority, parallelism, fetchSize, false, 0);
             }
         } finally {
             mutex_.unlock();