backend/app/brahma/service/token/internal/pkg/query.go

package pkg

import (
	"fmt"
	"git.icechen.cn/monorepo/backend/app/brahma/service/token/internal/model"
	"git.icechen.cn/monorepo/backend/pkg/proto/brahma/murders"
	ctxLogger "github.com/luizsuper/ctxLoggers"
	"github.com/pkg/errors"
	"reflect"
	"strings"
)

const (
	Normal    = "Normal"
	JsonArray = "JsonArray"
	Array     = "Array"
)
const (
	script = "script"
)

type (
	entityName string
	entityKey  string
	keyTag     string

	entityAttrMap map[entityName]nameAttr
	nameAttr      map[entityKey]keyTag
)

var (
	m1   entityAttrMap
	attr nameAttr
)

func init() {
	m1 = make(entityAttrMap)
	m2 := make(map[entityName]interface{})
	m2[script] = new(model.Scripts)
	attr = make(nameAttr)
	getAttr(m2)
}

type (
	RuleType struct {
		Rule  string
		Value []string
	}
	QueryMap map[string]RuleType
	Kv       map[string]string
)

func GenerateKv(str string) (QueryMap, error) {
	kv := make(QueryMap)
	s := []byte(str)
	i := make([]string, 0)
	spilt := make([]string, 0)

	if str != "" {
		spilt = strings.Split(string(s[1:len(s)-1]), ",")
	}
	for _, values := range spilt {
		if i = strings.Split(values, "="); len(i) != 2 {
			return nil, errors.New("无效字符串")
		}

		s := keyTag("")
		ok := true

		if s, ok = attr[entityKey(i[0])]; !ok {
			return nil, errors.New(fmt.Sprintf("无效的key:%v", i[0]))
		}

		switch s {
		case JsonArray:
			jsonArr := ""
			isArr := true

			if jsonArr, isArr = processJsonArr(i); !isArr {
				return nil, errors.New(fmt.Sprintf("无效的jsonArr:%v", i))
			}

			kv[i[0]] = RuleType{
				Rule: JsonArray,
				Value: []string{
					jsonArr,
				},
			}
		case Array:
			i2 := make([]string, 0)

			i2 = processArr(i[1])

			kv[i[0]] = RuleType{
				Rule:  Array,
				Value: i2,
			}
		case Normal:
			kv[i[0]] = RuleType{
				Rule: Normal,
				Value: []string{
					i[1],
				},
			}
		}

	}
	return kv, nil
}

func GetParam(queryMap *murders.QueryCondition) (page, limit int, query string) {
	if queryMap == nil {
		return
	}
	if q := queryMap.Query; q != nil {
		query = *q
	}
	if p := queryMap.Page; p != nil {
		page = int(*p)
	}
	if l := queryMap.Size; l != nil {
		limit = int(*l)
	}

	return
}

func getAttr(entity map[entityName]interface{}) {
	for name, body := range entity {
		m1[name] = attr

		typ := reflect.TypeOf(body)
		val := reflect.ValueOf(body)

		if val.Kind().String() != reflect.Ptr.String() {
			ctxLogger.Error(nil, "is not ptr")
			panic(errors.New("is not ptr"))
		}
		if val.IsNil() {
			ctxLogger.Error(nil, "nil ptr")
			panic(errors.New("nil ptr"))
		}

		num := val.Elem().NumField()
		for i := 0; i < num; i++ {
			field := typ.Elem().Field(i)
			tag := field.Tag.Get("type")
			json := field.Tag.Get("json")
			attr[entityKey(json)] = keyTag(tag)
			if tag == "" {
				attr[entityKey(json)] = Normal
			}
		}

	}

}

func processJsonArr(pair []string) (string, bool) {
	value := fmt.Sprintf("JSON_CONTAINS(%v,JSON_ARRAY(", pair[0])
	s := []byte(pair[1])

	//处理括号，遍历元素
	arr := strings.Split(string(s[1:len(s)-1]), "|")
	for _, v := range arr {
		//如果大括号里面没有元素
		if v == "" {
			return "", false
		}
		value = fmt.Sprintf("%v%v,", value, v)
	}

	s1 := []byte(value)
	value = string(s1[0:len(s1)-1]) + "))"
	return value, true
}

func processArr(str string) []string {
	s := []byte(str)

	arr := strings.Split(string(s[1:len(s)-1]), "|")
	i := make([]string, len(arr))

	for k, v := range i {
		i[k] = v
	}

	return i
}