ability to send multiple resultsets from handleQuery

Author: GoWebProd

mysql/result.go

@@ -19,3 +19,16 @@ func (r *Result) Close() {
 		r.Resultset = nil
 	}
 }
+
+func (r *Result) ChainResultSet(rs *Resultset) {
+	if r.Resultset == nil {
+		r.Resultset = rs
+		return
+	}
+
+	var lastRS *Resultset
+
+	for lastRS = r.Resultset; lastRS.Next != nil; lastRS = lastRS.Next {}
+
+	lastRS.Next = rs
+}

mysql/resultset.go

@@ -17,6 +17,8 @@ type Resultset struct {
 	RawPkg []byte
 
 	RowDatas []RowData
+
+	Next *Resultset
 }
 
 var (

server/resp.go

@@ -181,7 +181,20 @@ func (c *Conn) writeValue(value interface{}) error {
 		return c.writeOK(nil)
 	case *Result:
 		if v != nil && v.Resultset != nil {
-			return c.writeResultset(v.Resultset)
+			for rs := v.Resultset; rs != nil; rs = rs.Next {
+				if rs.Next != nil {
+					c.status |= SERVER_MORE_RESULTS_EXISTS
+				}
+
+				err := c.writeResultset(rs)
+				c.status &= ^SERVER_MORE_RESULTS_EXISTS
+
+				if err != nil {
+					return err
+				}
+			}
+
+			return nil
 		} else {
 			return c.writeOK(v)
 		}