add api to get query list, and support skip invalid query

ast/mapper.go

@@ -66,3 +66,20 @@ func (m *Mapper) GetStmt(ctx *Context) (string, error) {
 	}
 	return strings.TrimSuffix(buff.String(), "\n"), nil
 }
+
+func (m *Mapper) GetStmts(ctx *Context, skipErrorQuery bool) ([]string, error) {
+	var stmts []string
+	ctx.Sqls = m.SqlNodes
+	for _, a := range m.QueryNodes {
+		data, err := a.GetStmt(ctx)
+		if err == nil {
+			stmts = append(stmts, data)
+			continue
+		}
+		if skipErrorQuery {
+			continue
+		}
+		return nil, err
+	}
+	return stmts, nil
+}

parser.go

@@ -2,12 +2,14 @@ package parser
 
 import (
 	"encoding/xml"
+	"fmt"
 	"io"
 	"strings"
 
 	"github.com/actiontech/mybatis-mapper-2-sql/ast"
 )
 
+// ParseXML is a parser for parse all query in XML to string.
 func ParseXML(data string) (string, error) {
 	r := strings.NewReader(data)
 	d := xml.NewDecoder(r)
@@ -25,6 +27,29 @@ func ParseXML(data string) (string, error) {
 	return stmt, nil
 }
 
+// ParseXMLQuery is a parser for parse all query in XML to []string one by one;
+// you can set `skipErrorQuery` true to ignore invalid query.
+func ParseXMLQuery(data string, skipErrorQuery bool) ([]string, error) {
+	r := strings.NewReader(data)
+	d := xml.NewDecoder(r)
+	n, err := parse(d, nil)
+	if err != nil {
+		return nil, err
+	}
+	if n == nil {
+		return nil, nil
+	}
+	m, ok := n.(*ast.Mapper)
+	if !ok {
+		return nil, fmt.Errorf("the mapper is not found")
+	}
+	stmts, err := m.GetStmts(ast.NewContext(), skipErrorQuery)
+	if err != nil {
+		return nil, err
+	}
+	return stmts, nil
+}
+
 func parse(d *xml.Decoder, start *xml.StartElement) (node ast.Node, err error) {
 	if start != nil {
 		node, err = scan(start)