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/clause"
	"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"
)

const (
	LogicalOperatorAnd = "and"
	LogicalOperatorOr  = "or"
)

func isSupportedLogicalOperator(logicalOperator string) bool {
	return logicalOperator == LogicalOperatorAnd || logicalOperator == LogicalOperatorOr
}

type Rule struct {
	LogicalOperator string `json:"logicalOperator"`
	Left            *Rule  `json:"left"`
	Right           *Rule  `json:"right"`

	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.IsStringNotEmpty(r.LogicalOperator) {
		return r.checkAsNode()
	} else {
		return r.checkAsLeaf()
	}
}

func (r Rule) checkAsNode() error {
	if !isSupportedLogicalOperator(r.LogicalOperator) {
		return errors.New("不支持的逻辑操作符")
	}

	if r.Left == nil {
		return errors.New("左结点为空")
	}

	if r.Right == nil {
		return errors.New("右结点为空")
	}

	if strutils.IsStringNotEmpty(r.Left.LogicalOperator) {
		err := r.Left.checkAsNode()
		if err != nil {
			return err
		}
	} else {
		err := r.Left.checkAsLeaf()
		if err != nil {
			return err
		}
	}

	if strutils.IsStringNotEmpty(r.Right.LogicalOperator) {
		err := r.Right.checkAsNode()
		if err != nil {
			return err
		}
	} else {
		err := r.Right.checkAsLeaf()
		if err != nil {
			return err
		}
	}

	return nil
}

func (r Rule) checkAsLeaf() 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 {
		switch r.FieldType {
		case rule.TypeString:
			r.Value = ""
		case rule.TypeTime:
			r.Value = ""
		case rule.TypeNumber:
			r.Value = 0
		case rule.TypeBool:
			r.Value = false
		default:
			return errors.New("字段类型不支持")
		}
	}

	return nil
}

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

	return r.Check()
}

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 GetRulesAndFormConditionClause(dbSchema string, scope string, domainName string, i *infrastructure.Infrastructure, ruleParams map[string]any) (clause.Clause, error) {
	r, err := getRule(dbSchema, scope, domainName, i)
	if err != nil {
		return nil, err
	}

	c, err := formConditionClause(r, ruleParams)
	if err != nil {
		return nil, err
	}

	return c, nil
}

func getRule(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 Rule{}, err
	}

	r := new(Rule)
	err = json.Unmarshal([]byte(result.ColumnValueString(ColumnRule)), r)
	if err != nil {
		return Rule{}, err
	}

	return *r, nil
}

func formConditionClause(r Rule, ruleParams map[string]any) (clause.Clause, error) {
	ruleValue := r.Value
	ruleParamValue, ok := ruleParams[r.ColumnName]
	if ok {
		ruleValue = ruleParamValue
	}

	if strutils.IsStringEmpty(r.LogicalOperator) {
		conditions := clause.NewConditions()

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

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

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

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

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

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

		return conditions.And(), nil
	}

	leftClause, err := formConditionClause(*r.Left, ruleParams)
	if err != nil {
		return nil, err
	}

	rightClause, err := formConditionClause(*r.Right, ruleParams)
	if err != nil {
		return nil, err
	}

	switch r.LogicalOperator {
	case LogicalOperatorAnd:
		return clause.NewConditionJoin(leftClause, rightClause).And(), nil
	case LogicalOperatorOr:
		return clause.NewConditionJoin(leftClause, rightClause).Or(), nil
	default:
		return clause.NewConditionJoin(leftClause, rightClause).And(), nil
	}
}