ability to send multiple resultsets from handleQuery

Author: GoWebProd

server/command.go

@@ -13,7 +13,7 @@ type Handler interface {
 	UseDB(dbName string) error
 	//handle COM_QUERY command, like SELECT, INSERT, UPDATE, etc...
 	//If Result has a Resultset (SELECT, SHOW, etc...), we will send this as the response, otherwise, we will send Result
-	HandleQuery(query string) (*Result, error)
+	HandleQuery(query string) ([]*Result, error)
 	//handle COM_FILED_LIST command
 	HandleFieldList(table string, fieldWildcard string) ([]*Field, error)
 	//handle COM_STMT_PREPARE, params is the param number for this statement, columns is the column number

server/resp.go

@@ -185,6 +185,26 @@ func (c *Conn) writeValue(value interface{}) error {
 		} else {
 			return c.writeOK(v)
 		}
+	case []*Result:
+		if len(v) == 0 {
+			return c.writeValue(nil)
+		}
+
+		c.status |= SERVER_MORE_RESULTS_EXISTS
+
+		for i, res := range v {
+			if i == len(v) - 1 {
+				c.status &= ^SERVER_MORE_RESULTS_EXISTS
+			}
+
+			if err := c.writeValue(res); err != nil {
+				c.status &= ^SERVER_MORE_RESULTS_EXISTS
+
+				return err
+			}
+		}
+
+		return nil
 	case []*Field:
 		return c.writeFieldList(v)
 	case *Stmt: