github.com/zlyuancn/zstr@v0.0.0-20230412074414-14d6b645962f/sql_template.go

/*
-------------------------------------------------
   Author :       Zhang Fan
   date：         2020/7/18
   Description :
-------------------------------------------------
*/

package zstr

import (
	"bytes"
	"fmt"
	"reflect"
	"strconv"
	"strings"
)

const defaultSqlCompareFlag = "="

var (
	// 操作符
	sqlTemplateOperationMapp = map[int32]struct{}{
		'&': {},
		'|': {},
		'#': {},
		'@': {},
	}
	// 标记
	sqlTemplateFlagMapp = map[string]struct{}{
		">":          {},
		">=":         {},
		"<":          {},
		"<=":         {},
		"!=":         {},
		"<>":         {},
		"=":          {},
		"in":         {},
		"notin":      {},
		"not_in":     {},
		"like":       {},
		"likestart":  {},
		"like_start": {},
		"likeend":    {},
		"like_end":   {},
	}
	// 选项
	sqlTemplateOptsMapp = map[int32]struct{}{
		'a': {}, // attention, 不会忽略参数值为该类型的零值
		'd': {}, // direct, 直接将值写入sql语句
		'm': {}, // must, 必填
	}
)

type sqlTemplate struct {
	data       map[string]interface{}
	names      []string
	values     []interface{}
	keyCounter *counter // key计数器
	sub        int      // 下标计数器
}

func newSqlTemplate(values []interface{}) *sqlTemplate {
	return &sqlTemplate{
		data:       makeMapOfValues(values),
		keyCounter: newCounter(-1),
	}
}

func (m *sqlTemplate) calculateTemplate(ss []rune, start int) (int, int, bool, bool) {
	var crust, has, ok bool
	// 查找开头
	for i := start; i < len(ss); i++ {
		if ss[i] == '{' {
			start, crust, has = i, true, true
			break
		}
		if _, ok = sqlTemplateOperationMapp[ss[i]]; ok {
			start, crust, has = i, false, true
			break
		}
	}
	if !has {
		return 0, 0, false, false
	}

	// 预检
	if crust && (len(ss)-start < 4) || (len(ss)-start < 2) {
		return 0, 0, false, false
	}

	if !crust {
		for i := start + 1; i < len(ss); i++ {
			_, ok = templateVariableNameMap[ss[i]]
			if !ok { // 表示查找变量结束了
				if i-start < 2 || ss[i-1] == '.' { // 操作符占一个位置, 变量长度不可能为0
					return m.calculateTemplate(ss, i)
				}
				return start, i, false, true // 中间的数据就是需要的变量
			}
		}
		// 可能整个字符串都是需要的数据
		return start, len(ss), false, len(ss)-start >= 2 && ss[len(ss)-1] != '.'
	}

	// 以下包含{
	for i := start + 1; i < len(ss); i++ {
		if ss[i] != '}' {
			continue
		}
		return start, i + 1, true, true
	}
	return 0, 0, false, false
}

func (m *sqlTemplate) replaceAllFunc(s string, fn func(s string, crust bool) string) string {
	ss := []rune(s)
	var buff bytes.Buffer
	for offset := 0; offset < len(ss); {
		start, end, crust, has := m.calculateTemplate(ss, offset)
		if !has {
			buff.WriteString(string(ss[offset:]))
			break
		}

		buff.WriteString(string(ss[offset:start]))
		buff.WriteString(fn(string(ss[start:end]), crust))
		offset = end
	}
	return buff.String()
}

func (m *sqlTemplate) addValue(name string, value interface{}) {
	m.names = append(m.names, name)
	m.values = append(m.values, value)
}

func (m *sqlTemplate) Parse(sql_template string) (sql_str string, names []string, args []interface{}) {
	sql_str = m.replaceAllFunc(sql_template, func(s string, crust bool) string {
		if crust {
			s = s[1 : len(s)-1]
		}

		operation, name, flag, opts, err := m.sqlTemplateSyntaxParse(s)
		if err != nil {
			panic(err)
		}
		return m.translate(operation, name, flag, opts)
	})
	return m.repairSql(sql_str), m.names, m.values
}

func (m *sqlTemplate) translate(operation, name, flag string, opts string) string {
	// 选项检查
	var attention_opt, direct_opt, must_opt bool
	for _, o := range opts {
		switch o {
		case 'a':
			attention_opt = true
		case 'd':
			direct_opt = true
		case 'm':
			must_opt = true
		default:
			panic(fmt.Sprintf(`syntax error, non-supported option "%s"`, string(o)))
		}
	}
	switch operation {
	case "#":
		attention_opt = true
	case "@":
		attention_opt = false
		direct_opt = true
	}

	vName := name + "[" + strconv.Itoa(m.keyCounter.Incr(name)) + "]"
	value, has := m.data[vName]
	if !has {
		vName = name
		value, has = m.data[name]
	}
	if !has {
		vName = "*[" + strconv.Itoa(m.sub) + "]"
		value, has = m.data[vName]
	}
	m.sub++ // 每次一定+1

	// 无值返回空sql语句
	if !has {
		if must_opt {
			panic(fmt.Sprintf(`"%s" must have a value`, name))
		}
		return ""
	}

	// 非注意模式且值为零值返回空sql语句
	if !attention_opt && IsZero(value) {
		return ""
	}

	// 操作检查
	switch operation {
	case "&":
		operation = "and"
	case "|":
		operation = "or"
	case "#":
		// nil改为null
		if value == nil {
			return "null"
		}
		if direct_opt {
			return anyToSqlString(value, true)
		}
		m.addValue(vName, value)
		return "?"
	case "@": // !attention_opt + direct
		return anyToSqlString(value, false)
	default:
		panic(fmt.Errorf(`syntax error, non-supported operation "%s"`, operation))
	}

	// nil 修改语句
	if value == nil {
		switch flag {
		case "!=", "<>", "notin", "not_in", ">", "<":
			return fmt.Sprintf(`%s %s is not null`, operation, name)
		case "=", "like", "likestart", "like_start", "likeend", "like_end":
			return fmt.Sprintf(`%s %s is null`, operation, name)
		case "in", ">=", "<=":
			return ""
		}
	}

	var makeSqlStr func() string
	var directWrite func() string
	// 标记
	switch flag {
	case ">", ">=", "<", "<=", "!=", "<>", "=":
		makeSqlStr = func() string {
			m.addValue(vName, value)
			return fmt.Sprintf(`%s %s %s ?`, operation, name, flag)
		}
		directWrite = func() string {
			return fmt.Sprintf(`%s %s %s %s`, operation, name, flag, anyToSqlString(value, true))
		}
	case "in":
		values := m.parseToSlice(value)
		if len(values) == 0 {
			return ""
		}
		makeSqlStr = func() string {
			if len(values) == 1 {
				m.addValue(vName, values[0])
				return fmt.Sprintf(`%s %s = ?`, operation, name)
			}
			fs := make([]string, len(values))
			for i, s := range values {
				m.addValue(fmt.Sprintf("%s.in(%d)", vName, i), s)
				fs[i] = "?"
			}
			return fmt.Sprintf(`%s %s in (%s)`, operation, name, strings.Join(fs, ","))
		}
		directWrite = func() string {
			if len(values) == 1 {
				return fmt.Sprintf(`%s %s = %s`, operation, name, anyToSqlString(values[0], true))
			}
			return fmt.Sprintf(`%s %s in %s`, operation, name, anyToSqlString(value, true))
		}
	case "notin", "not_in":
		values := m.parseToSlice(value)
		if len(values) == 0 {
			return ""
		}
		makeSqlStr = func() string {
			if len(values) == 1 {
				m.addValue(vName, values[0])
				return fmt.Sprintf(`%s %s != ?`, operation, name)
			}
			fs := make([]string, len(values))
			for i, s := range values {
				m.addValue(fmt.Sprintf("%s.not_in(%d)", vName, i), s)
				fs[i] = "?"
			}
			return fmt.Sprintf(`%s %s not in (%s)`, operation, name, strings.Join(fs, ","))
		}
		directWrite = func() string {
			if len(values) == 1 {
				return fmt.Sprintf(`%s %s != %s`, operation, name, anyToSqlString(values[0], true))
			}
			return fmt.Sprintf(`%s %s not in %s`, operation, name, anyToSqlString(value, true))
		}
	case "like": // 包含xx
		makeSqlStr = func() string {
			m.addValue(vName, "%"+anyToSqlString(value, false)+"%")
			return fmt.Sprintf(`%s %s like ?`, operation, name)
		}
		directWrite = func() string {
			return fmt.Sprintf(`%s %s like '%%%s%%'`, operation, name, anyToSqlString(value, false))
		}
	case "likestart", "like_start": // 以xx开始
		makeSqlStr = func() string {
			m.addValue(vName, anyToSqlString(value, false)+"%")
			return fmt.Sprintf(`%s %s like ?`, operation, name)
		}
		directWrite = func() string {
			return fmt.Sprintf(`%s %s like '%s%%'`, operation, name, anyToSqlString(value, false))
		}
	case "likeend", "like_end": // 以xx结束
		makeSqlStr = func() string {
			m.addValue(vName, "%"+anyToSqlString(value, false))
			return fmt.Sprintf(`%s %s like ?`, operation, name)
		}
		directWrite = func() string {
			return fmt.Sprintf(`%s %s like '%%%s'`, operation, name, anyToSqlString(value, false))
		}
	default:
		panic(fmt.Errorf(`syntax error, non-supported flag "%s"`, flag))
	}

	// 直接模式, 将值写入sql语句
	if direct_opt {
		return directWrite()
	}
	return makeSqlStr()
}

func (m *sqlTemplate) Render(sql_template string) string {
	result := m.replaceAllFunc(sql_template, func(s string, crust bool) string {
		if crust {
			s = s[1 : len(s)-1]
		}

		operation, name, flag, opts, err := m.sqlTemplateSyntaxParse(s)
		if err != nil {
			panic(err)
		}
		return m.sqlTranslate(operation, name, flag, opts)
	})
	return m.repairSql(result)
}

func (m *sqlTemplate) sqlTranslate(operation, name, flag string, opts string) string {
	// 选项检查
	var attention_opt, must_opt bool
	for _, o := range opts {
		switch o {
		case 'a':
			attention_opt = true
		case 'd':
		case 'm':
			must_opt = true
		default:
			panic(fmt.Sprintf(`syntax error, non-supported option "%s"`, string(o)))
		}
	}
	switch operation {
	case "#":
		attention_opt = true
	case "@":
		attention_opt = false
	}

	value, has := m.data[name+"["+strconv.Itoa(m.keyCounter.Incr(name))+"]"]
	if !has {
		value, has = m.data[name]
	}
	if !has {
		value, has = m.data["*["+strconv.Itoa(m.sub)+"]"]
	}
	m.sub++ // 每次一定+1

	// 无值返回空sql语句
	if !has {
		if must_opt {
			panic(fmt.Sprintf(`"%s" must have a value`, name))
		}
		return ""
	}

	// 非注意模式, 零值返回空sql语句
	if !attention_opt && IsZero(value) {
		return ""
	}

	switch operation {
	case "&":
		operation = "and"
	case "|":
		operation = "or"
	case "#":
		// nil改为null
		if value == nil {
			return "null"
		}
		return anyToSqlString(value, true)
	case "@":
		return anyToSqlString(value, false)
	default:
		panic(fmt.Errorf(`syntax error, non-supported operation "%s"`, operation))
	}

	// nil 修改语句
	if value == nil {
		switch flag {
		case "!=", "<>", "notin", "not_in", ">", "<":
			return fmt.Sprintf(`%s %s is not null`, operation, name)
		case "=", "like", "likestart", "like_start", "likeend", "like_end":
			return fmt.Sprintf(`%s %s is null`, operation, name)
		case "in", ">=", "<=":
			return ""
		}
	}

	var sql_str string
	switch flag {
	case ">", ">=", "<", "<=", "!=", "<>", "=":
		sql_str = fmt.Sprintf(`%s %s %s %s`, operation, name, flag, anyToSqlString(value, true))
	case "in":
		values := m.parseToSlice(value)
		if len(values) == 0 {
			return ""
		}
		if len(values) == 1 {
			return fmt.Sprintf(`%s %s = %s`, operation, name, anyToSqlString(values[0], true))
		}
		sql_str = fmt.Sprintf(`%s %s in %s`, operation, name, anyToSqlString(value, true))
	case "notin", "not_in":
		values := m.parseToSlice(value)
		if len(values) == 0 {
			return ""
		}
		if len(values) == 1 {
			return fmt.Sprintf(`%s %s != %s`, operation, name, anyToSqlString(values[0], true))
		}
		sql_str = fmt.Sprintf(`%s %s not in %s`, operation, name, anyToSqlString(value, true))
	case "like": // 包含xx
		sql_str = fmt.Sprintf(`%s %s like '%%%s%%'`, operation, name, anyToSqlString(value, false))
	case "likestart", "like_start": // 以xx开始
		sql_str = fmt.Sprintf(`%s %s like '%s%%'`, operation, name, anyToSqlString(value, false))
	case "likeend", "like_end": // 以xx结束
		sql_str = fmt.Sprintf(`%s %s like '%%%s'`, operation, name, anyToSqlString(value, false))
	default:
		panic(fmt.Errorf(`syntax error, non-supported flag "%s"`, flag))
	}

	return sql_str
}

// 将数据解析为切片
func (m *sqlTemplate) parseToSlice(a interface{}) []interface{} {
	switch v := a.(type) {

	case nil:
		return []interface{}{"null"}

	case string, []byte, bool,
		int, int8, int16, int32, int64,
		uint, uint8, uint16, uint32, uint64,
		float32, float64:
		return []interface{}{v}
	}

	r_v := reflect.Indirect(reflect.ValueOf(a))
	if r_v.Kind() != reflect.Slice && r_v.Kind() != reflect.Array {
		return []interface{}{fmt.Sprint(a)}
	}

	l := r_v.Len()
	out := make([]interface{}, 0, l)
	for i := 0; i < l; i++ {
		v := reflect.Indirect(r_v.Index(i)).Interface()
		out = append(out, m.parseToSlice(v)...)
	}
	return out
}

// sql模板语法解析
//
// 语法格式:   (操作符)(name)
// 语法格式:   {(操作符)(name)}
// 语法格式:   {(操作符)(name) (标志)}
// 语法格式:   {(操作符)(name) (标志) (选项)}
// 语法格式:   {(操作符)(name) (选项)}
//
// 操作符:
//
//	&: 转为 and name flag value
//	|: 转为 or name flag value
//	#: 转为 value, 自带 attention 选项, 仅支持以下格式
//	     (操作符)(name)
//	     {(操作符)(name)}
//	     {(操作符)(name) (选项)}
//	@: attention 选项无效且自带 direct 选项, 且不会为字符串加上引号, 仅支持以下格式, 一般用于写入一条语句
//	     (操作符)(name)
//	     {(操作符)(name)}
//	     {(操作符)(name) (选项)}
//
// name:   示例:    a   a2   a_2   a_2.b   a_2.b_2
//
// 标志:   >   >=   <   <=   !=   <>   =   in   notin   not_in   like   likestart    like_start   likeend   like_end
//
// 选项:
//
//	a:   attention, 不会忽略参数值为该类型的零值
//	d:   direct, 直接将值写入sql语句中
//	m:   must, 必须传值, 值可以为零值
//
// 输入的values必须为：map[string]string, map[string]interface{}，或按顺序传入值
//
// 寻值优先级:
//
//	匹配名下标 > 匹配名 > *下标
//	如:  a[0] > a > *[0]
//
// 注意:
//
//	一般情况下如果name没有传参或为该类型的零值, 则替换为空字符串
//	如果name的值为nil, 不同的标志会转为不同的语句
//	我们不会去检查name是否完全符合变量名标志, 因为这是无意义且消耗资源的
//	    变量名首位可以为数字, 变量中间可以连续出现多个小数点, 如 0..a 是合法的
//
// 示例:
//
//	   s := SqlRender("select * from t where &a {&b} {&c !=} {&d in} {|e} limit 1", map[string]interface{}{
//			"a": 1,
//			"b[0]": "2",
//			"*[2]": 3.3,
//			"d": []string{"4"},
//			"e": nil,
//		  })
func (m *sqlTemplate) sqlTemplateSyntaxParse(text string) (operation, name, flag, opts string, err error) {
	// 去头去尾
	temp := strings.TrimSpace(text)
	// 空数据
	if temp == "" {
		err = fmt.Errorf("syntax error, {%s}, empty data", text)
		return
	}

	// 分离操作符
	operation, temp = temp[:1], temp[1:]

	// 缩进空格
	temp = m.retractAllSpace(temp)

	// 分离数据
	texts := strings.SplitN(temp, " ", 4) // 4为考虑尾部可能有空格
	if len(texts) >= 1 {
		name = texts[0]
	}
	if len(texts) >= 2 {
		flag = texts[1]
	} else {
		flag = defaultSqlCompareFlag
	}
	if len(texts) >= 3 {
		opts = texts[2]
	}
	if len(texts) >= 4 && texts[3] != " " {
		err = fmt.Errorf("syntax error, {%s}, redundant data", text)
		return
	}

	// 检查操作符
	if _, ok := sqlTemplateOperationMapp[int32(operation[0])]; !ok {
		err = fmt.Errorf(`syntax error, {%s}, non-supported operation "%s"`, text, operation)
		return
	}

	// 检查变量名
	if name == "" {
		err = fmt.Errorf("syntax error, {%s}, no variable name", text)
		return
	}

	if name[0] == '.' || name[len(name)-1] == '.' {
		err = fmt.Errorf("syntax error, {%s}, Invalid variable name", text)
		return
	}
	for _, v := range []rune(name) {
		if _, ok := templateVariableNameMap[v]; !ok {
			err = fmt.Errorf("syntax error, {%s}, Invalid variable name", text)
			return
		}
	}

	// 检查标记
	if _, ok := sqlTemplateFlagMapp[flag]; !ok {
		if opts != "" {
			err = fmt.Errorf(`syntax error, {%s}, non-supported flag "%s"`, text, flag)
			return
		}
		flag, opts = defaultSqlCompareFlag, flag
	}

	// 检查选项
	os := make(map[int32]struct{})
	for _, o := range opts {
		if _, ok := sqlTemplateOptsMapp[o]; !ok {
			err = fmt.Errorf(`syntax error, {%s}, non-supported option "%s"`, text, string(o))
			return
		}
		// 重复选项
		if _, ok := os[o]; ok {
			err = fmt.Errorf(`syntax error, {%s}, repetitive option "%s"`, text, string(o))
			return
		}
		os[o] = struct{}{}
	}

	return
}

// sql模板解析, 和 SqlParse 一样, 只是加长了函数名
func SqlTemplateParse(sqlTemplate string, values ...interface{}) (sql_str string, names []string, args []interface{}) {
	return newSqlTemplate(values).Parse(sqlTemplate)
}

// sql模板解析
func SqlParse(sqlTemplate string, values ...interface{}) (sql_str string, names []string, args []interface{}) {
	return newSqlTemplate(values).Parse(sqlTemplate)
}

// sql模板渲染, 和 SqlRender 一样, 只是加长了函数名
func SqlTemplateRender(sqlTemplate string, values ...interface{}) string {
	return newSqlTemplate(values).Render(sqlTemplate)
}

// sql模板渲染(不推荐)
//
// 值会直接写入sql语句中, 不支持sql注入检查
func SqlRender(sqlTemplate string, values ...interface{}) string {
	return newSqlTemplate(values).Render(sqlTemplate)
}