xmltag.go

package gobatis

import (
	"fmt"
	"reflect"
	"regexp"
	"strings"
)

type dynamicContext struct {
	sqlStr string
	params map[string]interface{}
}

func newDynamicContext(params map[string]interface{}) *dynamicContext {
	return &dynamicContext{
		params: params,
	}
}

func (d *dynamicContext) appendSql(sqlStr string) {
	d.sqlStr += sqlStr + " "
}

func (d *dynamicContext) toSql() string {
	return strings.TrimSpace(d.sqlStr)
}

// [ref](http://www.mybatis.org/mybatis-3/dynamic-sql.html)
type iSqlNode interface {
	build(ctx *dynamicContext) bool
}

// mixed node
type mixedSqlNode struct {
	sqlNodes []iSqlNode
}

var _ iSqlNode = &mixedSqlNode{}

func (m *mixedSqlNode) build(ctx *dynamicContext) bool {
	for i := 0; i < len(m.sqlNodes); i++ {
		sqlNode := m.sqlNodes[i]
		sqlNode.build(ctx)
	}

	return true
}

// if node
type ifSqlNode struct {
	test    string
	sqlNode iSqlNode
}

var _ iSqlNode = &ifSqlNode{}

func (i *ifSqlNode) build(ctx *dynamicContext) bool {
	if ok := eval(i.test, ctx.params); ok {
		i.sqlNode.build(ctx)
		return true
	}

	return false
}

// text node
type textSqlNode struct {
	content string
}

var _ iSqlNode = &textSqlNode{}

func (t *textSqlNode) build(ctx *dynamicContext) bool {
	ctx.appendSql(t.content)
	return true
}

// for node
const listItemPrefix = "_ls_item_p_"

type foreachSqlNode struct {
	sqlNode    iSqlNode
	collection string
	open       string
	close      string
	separator  string
	item       string
	index      string
}

var _ iSqlNode = &foreachSqlNode{}

func (f *foreachSqlNode) build(ctx *dynamicContext) bool {
	collection, ok := ctx.params[f.collection]
	if !ok {
		LOG.Warn("No collection for foreach tag:%s", f.collection)
		return false
	}

	ctx.appendSql(f.open)

	val := reflect.ValueOf(collection)

	if val.Kind() != reflect.Slice && val.Kind() != reflect.Array {
		LOG.Info("Foreach tag collection must be slice or array")
		return false
	}

	for i := 0; i < val.Len(); i++ {
		v := val.Index(i)
		if v.Kind() == reflect.Ptr {
			v = v.Elem()
		}

		// convert struct map val to params
		keys := make([]string, 0)
		params := make(map[string]interface{})
		switch v.Kind() {
		case reflect.Array, reflect.Slice:
			LOG.Info("Foreach tag collection element must not be slice or array")
			return false
		case reflect.Struct:
			m := f.structToMap(v.Interface())
			for k, v := range m {
				key := f.item + "." + k
				keys = append(keys, key)
				params[key] = v
			}
		case reflect.Map:
			m := v.Interface().(map[string]interface{})
			for k, v := range m {
				key := f.item + "." + k
				keys = append(keys, key)
				params[key] = v
			}
		default:
			keys = append(keys, f.item)
			params[f.item] = v.Interface()
		}

		params[f.item] = v.Interface()

		tempCtx := &dynamicContext{
			params: params,
		}

		f.sqlNode.build(tempCtx)
		f.tokenHandler(tempCtx, i)

		if i != 0 {
			ctx.appendSql(f.separator)
		}

		ctx.appendSql(tempCtx.sqlStr)

		// del temp param
		for _, k := range keys {
			delete(tempCtx.params, k)
		}

		// sync tempCtx params to ctx
		for k, v := range tempCtx.params {
			ctx.params[k] = v
		}
	}
	ctx.appendSql(f.close)

	return true
}

func (f *foreachSqlNode) tokenHandler(ctx *dynamicContext, index int) {
	sqlStr := ctx.sqlStr

	finalSqlStr := ""
	itemStr := ""
	start := 0
	for i := 0; i < len(sqlStr); i++ {
		if start > 0 {
			itemStr += string(sqlStr[i])
		}

		if i != 0 && i < len(sqlStr) {
			if string([]byte{sqlStr[i-1], sqlStr[i]}) == "#{" {
				start = i
			}
		}

		if start != 0 && i < len(sqlStr)-1 && sqlStr[i+1] == '}' {
			finalSqlStr += sqlStr[:start+1]
			sqlStr = sqlStr[i+2:]

			var re = regexp.MustCompile("^\\s*" + f.item + "\\s*")
			itemPrefix := listItemPrefix + f.item + fmt.Sprintf("%d", index)
			s := re.ReplaceAllString(itemStr, itemPrefix)
			s = strings.TrimSpace(s)
			if strings.Contains(s, itemPrefix) {
				itemKey := strings.TrimSpace(itemStr)
				if v, ok := ctx.params[itemKey]; ok {
					ctx.params[s] = v
				}
			}

			finalSqlStr += s + "}"

			i = 0
			start = 0
			itemStr = ""
		}
	}

	if start != 0 {
		LOG.Warn("WARN: token not close, SqlStr:" + ctx.sqlStr + " At:" + fmt.Sprintf("%d", start))
	}

	finalSqlStr += sqlStr
	ctx.sqlStr = finalSqlStr
}

func (f *foreachSqlNode) structToMap(s interface{}) map[string]interface{} {
	return structToMap(s)
}

// set node
type setSqlNode struct {
	sqlNodes []iSqlNode
}

func (s *setSqlNode) build(ctx *dynamicContext) bool {

	sqlStr := ""
	for _, sqlNode := range s.sqlNodes {
		tempCtx := &dynamicContext{
			params: ctx.params,
		}
		sqlNode.build(tempCtx)
		if sqlStr != "" && tempCtx.sqlStr != "" {
			sqlStr += " , "
		}

		sqlStr += tempCtx.sqlStr

		for k, v := range tempCtx.params {
			ctx.params[k] = v
		}
	}

	if sqlStr != "" {
		ctx.appendSql(" set ")
		sqlStr = strings.TrimSpace(sqlStr)
		sqlStr = strings.TrimSuffix(sqlStr, ",")
		ctx.appendSql(sqlStr)
	}

	return true
}

// trim node
type trimSqlNode struct {
	prefix          string // prefix：前缀
	prefixOverrides string // prefixOverride：去掉第一个出现prefixOverrides字符串
	suffixOverrides string // suffixOverride：去掉最后一个字符串
	suffix          string // suffix：后缀
	sqlNodes        []iSqlNode
}

func (t *trimSqlNode) build(ctx *dynamicContext) bool {
	tempCtx := &dynamicContext{
		params: ctx.params,
	}

	for _, sqlNode := range t.sqlNodes {
		if tempCtx.sqlStr != "" {
			tempCtx.sqlStr += " "
		}
		sqlNode.build(tempCtx)
	}

	if tempCtx.sqlStr != "" {
		sqlStr := strings.TrimSpace(tempCtx.sqlStr)

		preOv := strings.TrimSpace(t.prefixOverrides)
		if preOv != "" {
			sqlStr = strings.TrimPrefix(sqlStr, preOv)
		}

		suffOv := strings.TrimSpace(t.suffixOverrides)
		if suffOv != "" {
			sqlStr = strings.TrimSuffix(sqlStr, suffOv)
		}

		pre := strings.TrimSpace(t.prefix)
		if pre != "" {
			sqlStr = pre + " " + sqlStr
		}

		suff := strings.TrimSpace(t.suffix)
		if suff != "" {
			sqlStr += " " + suff
		}

		ctx.appendSql(sqlStr)
	}

	for k, v := range tempCtx.params {
		ctx.params[k] = v
	}

	return true
}

// where node
type whereSqlNode struct {
	sqlNodes []iSqlNode
}

func (w *whereSqlNode) build(ctx *dynamicContext) bool {
	tempCtx := &dynamicContext{
		params: ctx.params,
	}

	for _, sqlNode := range w.sqlNodes {
		if tempCtx.sqlStr != "" {
			tempCtx.sqlStr += " "
		}
		sqlNode.build(tempCtx)
	}

	if tempCtx.sqlStr != "" {
		sqlStr := strings.TrimSpace(tempCtx.sqlStr)
		sqlStr = strings.TrimPrefix(sqlStr, "and ")
		sqlStr = strings.TrimPrefix(sqlStr, "AND ")
		sqlStr = strings.TrimPrefix(sqlStr, "or ")
		sqlStr = strings.TrimPrefix(sqlStr, "OR ")

		ctx.appendSql("where")
		ctx.appendSql(sqlStr)
	}

	for k, v := range tempCtx.params {
		ctx.params[k] = v
	}

	return true
}

// choose node
type chooseNode struct {
	sqlNodes  []iSqlNode
	otherwise iSqlNode
}

func (c *chooseNode) build(ctx *dynamicContext) bool {
	for _, n := range c.sqlNodes {
		if n.build(ctx) {
			return true
		}
	}
	if nil != c.otherwise {
		c.otherwise.build(ctx)
		return true
	}
	return false
}

// include