Merge pull request #710 from wonderflow/fixsenddata

close on sendData
qiniu · Aug 15, 2018 · 9cda102 · 9cda102
2 parents a97ab6a + 7c1a768
commit 9cda102
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/mgr/dataflow.go b/mgr/dataflow.go
@@ -196,6 +196,11 @@ func SendData(senderConfig map[string]interface{}) error {
 	if err != nil {
 		return err
 	}
+	defer func() {
+		for _, s := range senders {
+			s.Close()
+		}
+	}()
 
 	router, err := getRouter(senderConfig, len(senders))
 	if err != nil {
@@ -213,7 +218,6 @@ func SendData(senderConfig map[string]interface{}) error {
 			return err
 		}
 	}
-
 	return nil
 }
 

diff --git a/reader/sql/sql.go b/reader/sql/sql.go
@@ -1678,6 +1678,7 @@ func (r *Reader) getCheckAll(queryType int) (checkAll bool, err error) {
 
 	return true, nil
 }
+
 //根据数据库类型返回表名
 func getWrappedTableName(dbtype string, table string) (tableName string, err error) {
 	switch dbtype {