search.go 2.4 KB
package query

import (
	"fmt"
	"gitlab.fjmaimaimai.com/allied-creation/character-library-metadata-bastion/pkg/domain"
	"reflect"
	"strings"

	"github.com/beego/beego/v2/core/validation"
)

type SearchTableQuery struct {
	// 表名称
	Name string `cname:"表名称" json:"name"`
	//ViewType string `cname:"视图类型 full:完整 main:主表关系" json:"viewType"`
	// 表类型 MainTable:主表 SideTable:副表 SubTable:分表  ExcelTable:Excel表 SubProcess:子过程 Schema:方案
	TableTypes []string `cname:"表类型 MainTable:主表 SideTable:副表 SubTable:分表  ExcelTable:Excel表" json:"tableTypes" valid:"Required"`
	// 父级ID
	ParentId int `cname:"父级ID" json:"parentId"`
	// 当前所处的模块
	CurrentModule int `json:"currentModule"`
	// 需要查询的模块 应用于模块 1:数控中心 2:拆解模块 4:计算模块
	Module int `json:"module"`
	// 父级ID
	ParentTableId int `cname:"父级ID" json:"parentTableId"`
	// 返回结构信息
	ReturnDetailStructInfo bool `cname:"返回具体的结构信息 默认不返回" json:"returnDetailStructInfo"`
	// 排除分组项,只返回一级列表;默认 false 不排除,连分组也返回
	ReturnGroupItem *bool `cname:"排除分组" json:"returnGroupItem"`
	// 排除指定表
	ExcludeTables []int `cname:"排除指定表" json:"excludeTables"`
	Context       *domain.Context
	FilterRules   []*FilterRule `json:"filterRules"`
	TableId       int           `cname:"ID" json:"tableId"`
}
type FilterRule struct {
	// *:匹配所有
	TableType string `json:"tableType"`
	Status    int    `json:"status"`
}

func (searchQuery *SearchTableQuery) Valid(validation *validation.Validation) {
	if searchQuery.ParentTableId > 0 && searchQuery.ParentId == 0 {
		searchQuery.ParentId = searchQuery.ParentTableId
	}
	if searchQuery.Module == 4 {
		searchQuery.FilterRules = append(searchQuery.FilterRules, &FilterRule{
			TableType: domain.SchemaTable.ToString(),
			Status:    domain.StatusOn,
		})
	}
}

func (searchQuery *SearchTableQuery) ValidateQuery() error {
	valid := validation.Validation{}
	b, err := valid.Valid(searchQuery)
	if err != nil {
		return err
	}
	if !b {
		elem := reflect.TypeOf(searchQuery).Elem()
		for _, validErr := range valid.Errors {
			field, isExist := elem.FieldByName(validErr.Field)
			if isExist {
				return fmt.Errorf(strings.Replace(validErr.Message, validErr.Field, field.Tag.Get("cname"), -1))
			} else {
				return fmt.Errorf(validErr.Message)
			}
		}
	}
	return nil
}