baize/convenient/domain/query_rule/query_rule_parser.go

package query_rule

import (
	"encoding/json"
	"git.sxidc.com/go-framework/baize/framework/core/domain"
	"git.sxidc.com/go-framework/baize/framework/core/infrastructure"
	"git.sxidc.com/go-framework/baize/framework/core/infrastructure/database"
	"git.sxidc.com/go-framework/baize/framework/core/infrastructure/database/sql"
	"git.sxidc.com/go-framework/baize/framework/core/tag/rule"
	"git.sxidc.com/go-tools/utils/reflectutils"
	"git.sxidc.com/go-tools/utils/strutils"
	"github.com/pkg/errors"
	"reflect"
)

type Rule struct {
	FieldName  string `json:"fieldName"`
	FieldType  string `json:"fieldType"`
	ColumnName string `json:"columnName"`
	Operator   string `json:"operator"`
	Value      any    `json:"value"`
}

func (r Rule) Check() error {
	if strutils.IsStringEmpty(r.FieldName) {
		return errors.New("字段名为空")
	}

	if !rule.IsSupportedType(r.FieldType) {
		return errors.New("字段类型不支持")
	}

	if strutils.IsStringEmpty(r.ColumnName) {
		return errors.New("列名为空")
	}

	if !IsSupportedOperator(r.Operator) {
		return errors.New("操作符不支持")
	}

	if r.Value == nil {
		return errors.New("没有传递值")
	}

	return nil
}

func Lint(ruleStr string) error {
	rules := make([]Rule, 0)
	err := json.Unmarshal([]byte(ruleStr), &rules)
	if err != nil {
		return err
	}

	for _, r := range rules {
		err := r.Check()
		if err != nil {
			return err
		}
	}

	return nil
}

func HasRule(dbSchema string, scope string, domainName string, i *infrastructure.Infrastructure) (bool, error) {
	dbExecutor := i.DBExecutor()

	return database.CheckExist(dbExecutor, &sql.CheckExistExecuteParams{
		TableName:  domain.TableName(dbSchema, &ValueObject{}),
		Conditions: sql.NewConditions().Equal(ColumnScope, scope).Equal(ColumnDomainName, domainName),
	})
}

func GetRules(dbSchema string, scope string, domainName string, i *infrastructure.Infrastructure) ([]Rule, error) {
	dbExecutor := i.DBExecutor()

	result, err := database.QueryOne(dbExecutor, &sql.QueryOneExecuteParams{
		TableName:  domain.TableName(dbSchema, &ValueObject{}),
		Conditions: sql.NewConditions().Equal(ColumnScope, scope).Equal(ColumnDomainName, domainName),
	})
	if err != nil {
		return nil, err
	}

	rules := make([]Rule, 0)
	err = json.Unmarshal([]byte(result.ColumnValueString(ColumnRule)), &rules)
	if err != nil {
		return nil, err
	}

	return rules, nil
}

func FormConditions(rules []Rule) (*sql.Conditions, error) {
	conditions := sql.NewConditions()
	for _, r := range rules {
		err := r.Check()
		if err != nil {
			return nil, err
		}

		switch r.Operator {
		case opEqual:
			conditions.Equal(r.ColumnName, r.Value)
		case opNotEqual:
			conditions.Not(r.ColumnName, r.Value)
		case opLessThan:
			conditions.LessThan(r.ColumnName, r.Value)
		case opLessThanOrEqual:
			conditions.LessThanAndEqual(r.ColumnName, r.Value)
		case opGreaterThan:
			conditions.GreaterThan(r.ColumnName, r.Value)
		case opGreaterThanOrEqual:
			conditions.GreaterThanAndEqual(r.ColumnName, r.Value)
		case opIn:
			if reflect.TypeOf(r.Value).Kind() != reflect.Slice {
				return nil, errors.New("使用\"包含在列表\"操作符必须使用列表")
			}

			conditions.In(r.FieldName, r.Value)
		case opNotIn:
			if reflect.TypeOf(r.Value).Kind() != reflect.Slice {
				return nil, errors.New("使用\"不包含在列表\"操作符必须使用列表")
			}

			conditions.NotIn(r.FieldName, r.Value)
		case opPrefix:
			strValue, err := reflectutils.ToString(r.Value)
			if err != nil {
				return nil, err
			}

			conditions.Like(r.FieldName, strValue+"%")
		case opSuffix:
			strValue, err := reflectutils.ToString(r.Value)
			if err != nil {
				return nil, err
			}

			conditions.Like(r.FieldName, "%"+strValue)
		case opContains:
			strValue, err := reflectutils.ToString(r.Value)
			if err != nil {
				return nil, err
			}

			conditions.Like(r.FieldName, "%"+strValue+"%")
		default:
			return nil, errors.Errorf("不支持的操作符%v", r.Operator)
		}
	}

	return conditions, nil
}

func GetRulesAndFormConditions(dbSchema string, scope string, domainName string, i *infrastructure.Infrastructure) (*sql.Conditions, error) {
	rules, err := GetRules(dbSchema, scope, domainName, i)
	if err != nil {
		return nil, err
	}

	conditions, err := FormConditions(rules)
	if err != nil {
		return nil, err
	}

	return conditions, nil
}