增加简单的数据映射能力

filter.go

@@ -8,8 +8,11 @@
 package util
 
 import (
+	"fmt"
+
 	"github.com/Jeffail/gabs"
 	"github.com/pkg/errors"
+	"github.com/tidwall/gjson"
 )
 
 // MapRule 映射规则
@@ -30,7 +33,7 @@ type MapRule struct {
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 11:54 2022/7/4
-func NewFilter(sourceData []byte, filterRuleList []MapRule) *filter {
+func NewFilter(sourceData string, filterRuleList []MapRule) *filter {
 	return &filter{
 		sourceData:     sourceData,
 		filterRuleList: filterRuleList,
@@ -44,7 +47,7 @@ func NewFilter(sourceData []byte, filterRuleList []MapRule) *filter {
 //
 // Date : 11:58 2022/7/4
 type filter struct {
-	sourceData     []byte
+	sourceData     string
 	filterRuleList []MapRule
 	jsonObj        *gabs.Container // 生成的json对象实例
 }
@@ -55,15 +58,30 @@ type filter struct {
 //
 // Date : 11:59 2022/7/4
 func (f *filter) Deal() ([]byte, error) {
-	return nil, nil
+	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
 }
 
-// getDefaultValue 获取默认值
+// getValue 获取值
 //
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 12:25 2022/7/4
-func (f *filter) getDefaultValue(dataType string, defaultValue string) (interface{}, error) {
+func (f *filter) getValue(dataType string, defaultValue string) (interface{}, error) {
 	switch dataType {
 	case "int8":
 		fallthrough
@@ -111,6 +129,8 @@ func (f *filter) getDefaultValue(dataType string, defaultValue string) (interfac
 		)
 		err = ConvertAssign(&val, defaultValue)
 		return val, err
+	case "string":
+		return defaultValue, nil
 	default:
 		return nil, errors.New(dataType + " is not support!")
 	}

filter_test.go

@@ -0,0 +1,57 @@
+// Package util ...
+//
+// Description : util ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2022-07-04 12:44
+package util
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test_filter_Deal(t *testing.T) {
+	sourceData := `{
+    "name":"zhangsan",
+    "age":"18",
+    "extension":{
+        "sex":"man",
+        "height":"180"
+    }
+}`
+	ruleList := []MapRule{
+		{
+			SourcePath:   "name",
+			MapPath:      "user_name",
+			Required:     true,
+			DataType:     "string",
+			DefaultValue: "lalala",
+		},
+		{
+			SourcePath:   "age",
+			MapPath:      "user_age",
+			Required:     true,
+			DataType:     "int",
+			DefaultValue: "280",
+		},
+		{
+			SourcePath:   "extension.height",
+			MapPath:      "user_height",
+			Required:     true,
+			DataType:     "int",
+			DefaultValue: "359",
+		},
+		{
+			SourcePath:   "extension.sex",
+			MapPath:      "user_sex",
+			Required:     true,
+			DataType:     "string",
+			DefaultValue: "lalala",
+		},
+	}
+	instance := NewFilter(sourceData, ruleList)
+	result, err := instance.Deal()
+	fmt.Println(string(result), err)
+}

go.mod

@@ -11,4 +11,7 @@ require (
 require (
 	github.com/Jeffail/gabs v1.4.0 // indirect
 	github.com/go-ini/ini v1.66.6 // indirect
+	github.com/tidwall/gjson v1.14.1 // indirect
+	github.com/tidwall/match v1.1.1 // indirect
+	github.com/tidwall/pretty v1.2.0 // indirect
 )

go.sum

@@ -6,6 +6,12 @@ github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
 github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
+github.com/tidwall/gjson v1.14.1 h1:iymTbGkQBhveq21bEvAQ81I0LEBork8BFe1CUZXdyuo=
+github.com/tidwall/gjson v1.14.1/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
+github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
+github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
+github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
+github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=