171 lines
3.3 KiB
Go
171 lines
3.3 KiB
Go
// Package filter ...
|
|
//
|
|
// Description : filter ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2022-07-04 17:47
|
|
package filter
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/tidwall/sjson"
|
|
"strings"
|
|
|
|
"git.zhangdeman.cn/zhangdeman/util"
|
|
"github.com/pkg/errors"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// NewFilter 过滤器实例
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:54 2022/7/4
|
|
func NewFilter(sourceData string, filterRuleList []MapRule) *filter {
|
|
return &filter{
|
|
sourceData: sourceData,
|
|
formatResult: "{}",
|
|
filterRuleList: filterRuleList,
|
|
}
|
|
}
|
|
|
|
// filter 数据过滤
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:58 2022/7/4
|
|
type filter struct {
|
|
sourceData string
|
|
formatResult string
|
|
filterRuleList []MapRule
|
|
}
|
|
|
|
// Deal ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 11:59 2022/7/4
|
|
func (f *filter) Deal() error {
|
|
var (
|
|
err error
|
|
formatVal interface{}
|
|
)
|
|
|
|
for _, rule := range f.filterRuleList {
|
|
if strings.Contains(rule.SourcePath, "[]") {
|
|
// 对于list的处理
|
|
continue
|
|
}
|
|
sourceResult := gjson.Get(f.sourceData, rule.SourcePath)
|
|
var (
|
|
sourceVal string
|
|
)
|
|
|
|
if !sourceResult.Exists() {
|
|
// 不存在, 使用默认值
|
|
sourceVal = rule.DefaultValue
|
|
} else {
|
|
sourceVal = sourceResult.String()
|
|
}
|
|
if formatVal, err = f.getValue(rule.DataType, sourceVal); nil != err {
|
|
return fmt.Errorf("%s = %v can not convert to %s : %s", rule.SourcePath, sourceResult.Value(), rule.DataType, err.Error())
|
|
}
|
|
if _, err = sjson.Set(f.formatResult, rule.MapPath, formatVal); nil != err {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// String 获取格式化之后的字符串
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 21:18 2022/12/31
|
|
func (f *filter) String() string {
|
|
return f.formatResult
|
|
}
|
|
|
|
// Byte 获取格式化之后的字节数组
|
|
//
|
|
// Author : zhangdeman001@ke.com<张德满>
|
|
//
|
|
// Date : 21:18 2022/12/31
|
|
func (f *filter) Byte() []byte {
|
|
return []byte(f.String())
|
|
}
|
|
|
|
// Parse 解析返回结果
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 21:20 2022/12/31
|
|
func (f *filter) Parse(receiver interface{}) error {
|
|
if nil == receiver {
|
|
return errors.New("receiver is nil")
|
|
}
|
|
return json.Unmarshal(f.Byte(), receiver)
|
|
}
|
|
|
|
// 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 = util.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 = util.ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "bool":
|
|
var (
|
|
err error
|
|
val bool
|
|
)
|
|
err = util.ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "float32":
|
|
fallthrough
|
|
case "float64":
|
|
var (
|
|
err error
|
|
val float64
|
|
)
|
|
err = util.ConvertAssign(&val, defaultValue)
|
|
return val, err
|
|
case "string":
|
|
return defaultValue, nil
|
|
default:
|
|
return nil, errors.New(dataType + " is not support!")
|
|
}
|
|
}
|