catch errors from mysql (#113)

XiaoMi · Oct 12, 2020 · 6295c96 · 6295c96
1 parent 6997a71
commit 6295c96
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/backend/direct_connection.go b/backend/direct_connection.go
@@ -551,6 +551,10 @@ func (dc *DirectConnection) FieldList(table string, wildcard string) ([]*mysql.F
 			return fs, nil
 		}
 
+		if data[0] == mysql.ErrHeader {
+			return nil, dc.handleErrorPacket(data)
+		}
+
 		if f, err = mysql.FieldData(data).Parse(); err != nil {
 			return nil, err
 		}
@@ -626,6 +630,10 @@ func (dc *DirectConnection) readResultColumns(result *mysql.Result) (err error)
 			return
 		}
 
+		if data[0] == mysql.ErrHeader {
+			return dc.handleErrorPacket(data)
+		}
+
 		result.Fields[i], err = mysql.FieldData(data).Parse()
 		if err != nil {
 			return
@@ -643,7 +651,6 @@ func (dc *DirectConnection) readResultRows(result *mysql.Result, isBinary bool)
 
 	for {
 		data, err = dc.readPacket()
-
 		if err != nil {
 			return
 		}
@@ -660,6 +667,10 @@ func (dc *DirectConnection) readResultRows(result *mysql.Result, isBinary bool)
 			break
 		}
 
+		if data[0] == mysql.ErrHeader {
+			return dc.handleErrorPacket(data)
+		}
+
 		result.RowDatas = append(result.RowDatas, data)
 	}