138 lines
3.0 KiB
Go
138 lines
3.0 KiB
Go
// Package util ...
|
|
//
|
|
// Description : util ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2022-07-04 11:45
|
|
package util
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/Jeffail/gabs"
|
|
"github.com/pkg/errors"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// MapRule 映射规则
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 12:21 2022/7/4
|
|
type MapRule struct {
|
|
SourcePath string `json:"source_path"` // 原路径
|
|
MapPath string `json:"map_path"` // 映射路径
|
|
Required bool `json:"required"` // 必须存在
|
|
DataType string `json:"data_type"` // 数据类型
|
|
DefaultValue string `json:"default_value"` // 默认值, 以字符串传入, 会转换成 DataType
|
|
}
|
|
|
|
// NewFilter 过滤器实例
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:54 2022/7/4
|
|
func NewFilter(sourceData string, filterRuleList []MapRule) *filter {
|
|
return &filter{
|
|
sourceData: sourceData,
|
|
filterRuleList: filterRuleList,
|
|
jsonObj: &gabs.Container{},
|
|
}
|
|
}
|
|
|
|
// filter 数据过滤
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:58 2022/7/4
|
|
type filter struct {
|
|
sourceData string
|
|
filterRuleList []MapRule
|
|
jsonObj *gabs.Container // 生成的json对象实例
|
|
}
|
|
|
|
// Deal ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:59 2022/7/4
|
|
func (f *filter) Deal() ([]byte, error) {
|
|
for _, rule := range f.filterRuleList {
|
|
sourceResult := gjson.Get(f.sourceData, rule.SourcePath)
|
|
sourceVal := sourceResult.String()
|
|
if !sourceResult.Exists() {
|
|
// 不存在, 使用默认值
|
|
sourceVal = rule.DefaultValue
|
|
}
|
|
formatVal, err := f.getValue(rule.DataType, sourceVal)
|
|
if nil != err {
|
|
return nil, fmt.Errorf("%s = %v can not convert to %s : %s", rule.SourcePath, sourceResult.Value(), rule.DataType, err.Error())
|
|
}
|
|
if _, err := f.jsonObj.SetP(formatVal, rule.MapPath); nil != err {
|
|
return nil, fmt.Errorf("%s set val = %v fail : %s", rule.MapPath, formatVal, err.Error())
|
|
}
|
|
}
|
|
return f.jsonObj.EncodeJSON(), nil
|
|
}
|
|
|
|
// getValue 获取值
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 12:25 2022/7/4
|
|
func (f *filter) getValue(dataType string, defaultValue string) (interface{}, error) {
|
|
switch dataType {
|
|
case "int8":
|
|
fallthrough
|
|
case "int16":
|
|
fallthrough
|
|
case "int32":
|
|
fallthrough
|
|
case "int64":
|
|
fallthrough
|
|
case "int":
|
|
var (
|
|
err error
|
|
val int64
|
|
)
|
|
err = ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "uint8":
|
|
fallthrough
|
|
case "uint16":
|
|
fallthrough
|
|
case "uint32":
|
|
fallthrough
|
|
case "uint64":
|
|
fallthrough
|
|
case "uint":
|
|
var (
|
|
err error
|
|
val uint64
|
|
)
|
|
err = ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "bool":
|
|
var (
|
|
err error
|
|
val bool
|
|
)
|
|
err = ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "float32":
|
|
fallthrough
|
|
case "float64":
|
|
var (
|
|
err error
|
|
val float64
|
|
)
|
|
err = ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "string":
|
|
return defaultValue, nil
|
|
default:
|
|
return nil, errors.New(dataType + " is not support!")
|
|
}
|
|
}
|