WithGetRawValue

README.md

@@ -285,3 +285,15 @@ getValue := map[string]string{
 _, err := json.Marshal([]byte(obj), option.WithStructName("reqName"), option.WithTagName("json"), option.WithGetValue(getValue))
 fmt.Println(getValue[".a"], "b"))
 ```
+
+### 2.5 option.WithGetRawValue 支持(json/yaml/header/query string)
+把json串序列化成struct字符串的时候，顺带提取json字符串里面指定的key值, value是any(int/string/float64)类型
+```go
+obj := `{"a":"b"}`
+getValue := map[string]any{
+  ".a": "",
+}
+
+_, err := json.Marshal([]byte(obj), option.WithStructName("reqName"), option.WithTagName("json"), option.WithGetRawValue(getValue))
+fmt.Println(getValue[".a"], "b"))
+```

internal/guesstype/guesstype.go

@@ -42,3 +42,22 @@ func TypeOf(s string) string {
 
 	return "string"
 }
+
+func ToAny(s string) any {
+	if IsFloat(s) {
+		f, _ := strconv.ParseFloat(s, 0)
+		return f
+	}
+
+	if IsInt(s) {
+		i, _ := strconv.Atoi(s)
+		return i
+	}
+
+	if IsBool(s) {
+		b, _ := strconv.ParseBool(s)
+		return b
+	}
+
+	return s
+}

internal/map2struct/map2struct.go

@@ -29,6 +29,14 @@ func MapGenStruct(m map[string][]string, opt option.Option) (res []byte, err err
 
 	for _, k := range ks {
 		v := m[k]
+		if opt.GetRawValue != nil {
+			_, ok := opt.GetRawValue[k]
+			if ok {
+				opt.GetRawValue[k] = guesstype.ToAny(v[0])
+
+			}
+		}
+
 		fieldName, tagName := name.GetFieldAndTagName(k)
 		if opt.TagNameFromKey {
 			tagName = k

json/json.go

@@ -284,7 +284,13 @@ func (f *JSON) marshalValue(key string, obj interface{}, fromArray bool, depth i
 		if ok {
 			f.GetValue[keyPath] = fmt.Sprintf("%s", obj)
 		}
+	}
 
+	if f.GetRawValue != nil {
+		_, ok := f.GetRawValue[keyPath]
+		if ok {
+			f.GetRawValue[keyPath] = obj
+		}
 	}
 
 	tmpFieldName := strings.ToUpper(fieldName)

option/option.go

@@ -15,6 +15,7 @@ type Option struct {
 	StructName      string
 	TypeMap         map[string]string
 	GetValue        map[string]string
+	GetRawValue     map[string]any
 	TagNameFromKey  bool
 	OutputFmtBefore io.Writer //需要format之前的数据
 }
@@ -78,6 +79,13 @@ func WithGetValue(getValue map[string]string) OptionFunc {
 	}
 }
 
+// 目前支持json/yaml/http header/query string
+func WithGetRawValue(getValue map[string]any) OptionFunc {
+	return func(c *Option) {
+		c.GetRawValue = getValue
+	}
+}
+
 // tag使用变量名, http header特殊一点
 // 目前仅仅支持http header marshal
 func WithTagNameFromKey() OptionFunc {