修复对[]byte类型的解析，包括实现encoding.TextMarshaler 接口的类型，规范代码

Author: liu-cn

example/select_omit_any_test.go

@@ -21,15 +21,15 @@ func newAnyway() Anyway {
 }
 
 func TestSelectAny(t *testing.T) {
-	fmt.Println(filter.SelectMarshal("chat", newAnyway()).MustJSON())
+	fmt.Println(filter.Select("chat", newAnyway()))
 	//{"age":10,"name":"boyan"}
-	fmt.Println(filter.SelectMarshal("article", newAnyway()).MustJSON())
+	fmt.Println(filter.Select("article", newAnyway()))
 	//{"name":"boyan","sex":10}
 }
 
 func TestOmitAny(t *testing.T) {
-	fmt.Println(filter.OmitMarshal("user", newAnyway()).MustJSON())
+	fmt.Println(filter.Omit("user", newAnyway()))
 	//{"age":10}
-	fmt.Println(filter.OmitMarshal("profile", newAnyway()).MustJSON())
+	fmt.Println(filter.Omit("profile", newAnyway()))
 	//{"name":"boyan"}
 }

example/slice_test.go

@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"github.com/liu-cn/json-filter/filter"
+
 	"testing"
 )
 

filter/example_test.go

@@ -0,0 +1,61 @@
+package filter
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+func ExampleSelect() {
+	type (
+		Tag struct {
+			Icon string `json:"icon,select(article)"`
+			Name string `json:"name,select(profile)"`
+		}
+		User struct {
+			Age  int     `json:"age,select(article|profile)"`
+			ID   int     `json:"id,select($any)"`                        //$any表示任何场景都选中该字段
+			Name *string `json:"name,omitempty,select(article|profile)"` //为nil忽略
+			Tags []Tag   `json:"tags,select(article|profile)"`
+		}
+	)
+	name := "小北"
+	user := User{ID: 1, Name: &name, Age: 21, Tags: []Tag{{"icon", "foo"}, {"icon", "bar"}}}
+	article := Select("article", &user) //传指针或值均可
+	null := Select("null", user)
+	user.Name = nil
+	profile := Select("profile", user)
+	articleJSON, _ := json.Marshal(article)
+	fmt.Println(string(articleJSON))
+	fmt.Println(profile) //可以直接打印，打印会直接输出过滤后的json
+	fmt.Println(null)
+
+	//Output:
+	//{"id":1,"name":"小北","tags":[{"icon":"icon"},{"icon":"icon"}]}
+	//{"age":21,"id":1,"name":"小北","tags":[{"name":"foo"},{"name":"bar"}]}
+	//{"id":1}
+}
+
+func ExampleOmit() {
+
+	type (
+		Tag struct {
+			Icon string `json:"icon,omit(article)"`
+			Name string `json:"name,omit(profile)"`
+		}
+		User struct {
+			Age      int   `json:"age"`
+			Password int   `json:"password,omit($any)"` //$any表示任何场景都排除该字段
+			Tags     []Tag `json:"tags"`
+		}
+	)
+	user := User{Age: 21, Tags: []Tag{{"icon", "foo"}, {"icon", "bar"}}}
+	article := Omit("article", &user) //传指针或值均可
+	profile := Omit("profile", user)
+	articleJSON, _ := json.Marshal(article)
+	fmt.Println(string(articleJSON))
+	fmt.Println(profile) //可以直接打印，打印会直接输出过滤后的json
+
+	//Output:
+	//{"age":21,"tags":[{"name":"foo"},{"name":"bar"}]}
+	//{"age":21,"tags":[{"icon":"icon"},{"icon":"icon"}]}
+}

filter/filter.go

@@ -6,8 +6,45 @@ type Filter struct {
 	node *fieldNodeTree
 }
 
+// Select 直接返回过滤后的数据结构，它可以被json.Marshal解析，直接打印会以过滤后的json字符串展示
+func Select(selectScene string, el interface{}) interface{} {
+	return jsonFilter(selectScene, el, true)
+}
+
+func jsonFilter(selectScene string, el interface{}, isSelect bool) Filter {
+	tree := &fieldNodeTree{
+		Key:        "",
+		ParentNode: nil,
+	}
+	tree.parseAny("", selectScene, el, isSelect)
+	return Filter{
+		node: tree,
+	}
+}
+
+// Omit 直接返回过滤后的数据结构，它可以被json.Marshal解析，直接打印会以过滤后的json字符串展示
+func Omit(omitScene string, el interface{}) interface{} {
+	return jsonFilter(omitScene, el, false)
+}
+
+// EnableCache 决定是否启用缓存，默认开启（强烈建议，除非万一缓存模式下出现bug，可以关闭缓存退回曾经的无缓存过滤模式），开启缓存后会有30%-40%的性能提升，开启缓存并没有副作用，只是会让结构体的字段tag常驻内存减少tag字符串处理操作
+func EnableCache(enable bool) {
+	enableCache = enable
+}
+
+// Deprecated
+// SelectMarshal 不建议使用，第一个参数填你结构体select标签里的场景，第二个参数是你需要过滤的结构体对象，如果字段的select标签里标注的有该场景那么该字段会被选中。
+func SelectMarshal(selectScene string, el interface{}) Filter {
+	return jsonFilter(selectScene, el, true)
+}
+
+// Deprecated
+// OmitMarshal 不建议使用，第一个参数填你结构体omit标签里的场景，第二个参数是你需要过滤的结构体对象，如果字段的omit标签里标注的有该场景那么该字段会被过滤掉
+func OmitMarshal(omitScene string, el interface{}) Filter {
+	return jsonFilter(omitScene, el, false)
+}
 func (f Filter) MarshalJSON() ([]byte, error) {
-	return f.node.Bytes()
+	return useJSONMarshalFunc(f.node.Marshal())
 }
 
 // Deprecated
@@ -41,87 +78,3 @@ func (f Filter) String() string {
 	}
 	return json
 }
-
-// Deprecated
-// SelectMarshal 不建议使用，第一个参数填你结构体select标签里的场景，第二个参数是你需要过滤的结构体对象，如果字段的select标签里标注的有该场景那么该字段会被选中。
-func SelectMarshal(selectScene string, el interface{}) Filter {
-	if enableCache {
-		return selectWithCache(selectScene, el)
-	}
-	return selectMarshal(selectScene, el)
-}
-func selectMarshal(selectScene string, el interface{}) Filter {
-	tree := &fieldNodeTree{
-		Key:        "",
-		ParentNode: nil,
-	}
-	tree.ParseSelectValue("", selectScene, el)
-	return Filter{
-		node: tree,
-	}
-}
-
-// Select 直接返回过滤后的数据结构，相当于直接SelectMarshal后再调用Interface方法
-func Select(selectScene string, el interface{}) interface{} {
-	if enableCache {
-		return selectWithCache(selectScene, el)
-	}
-	return selectMarshal(selectScene, el)
-}
-
-// selectWithCache 直接返回过滤后的数据结构，相当于直接SelectMarshal后再调用Interface方法
-func selectWithCache(selectScene string, el interface{}) Filter {
-	tree := &fieldNodeTree{
-		Key:        "",
-		ParentNode: nil,
-	}
-	tree.ParseSelectValueWithCache("", selectScene, el)
-	return Filter{
-		node: tree,
-	}
-}
-
-// Omit 直接返回过滤后的数据结构，相当于直接OmitMarshal后再调用Interface方法
-func Omit(omitScene string, el interface{}) interface{} {
-	if enableCache {
-		return omitWithCache(omitScene, el)
-	}
-	return omitMarshal(omitScene, el)
-}
-
-// Deprecated
-// OmitMarshal 不建议使用，第一个参数填你结构体omit标签里的场景，第二个参数是你需要过滤的结构体对象，如果字段的omit标签里标注的有该场景那么该字段会被过滤掉
-func OmitMarshal(omitScene string, el interface{}) Filter {
-	if enableCache {
-		return omitWithCache(omitScene, el)
-	}
-	return omitMarshal(omitScene, el)
-}
-
-func omitMarshal(omitScene string, el interface{}) Filter {
-	tree := &fieldNodeTree{
-		Key:        "",
-		ParentNode: nil,
-	}
-	tree.ParseOmitValue("", omitScene, el)
-	return Filter{
-		node: tree,
-	}
-}
-
-// omitWithCache 第一个参数填你结构体omit标签里的场景，第二个参数是你需要过滤的结构体对象，如果字段的omit标签里标注的有该场景那么该字段会被过滤掉
-func omitWithCache(omitScene string, el interface{}) Filter {
-	tree := &fieldNodeTree{
-		Key:        "",
-		ParentNode: nil,
-	}
-	tree.ParseOmitValueWithCache("", omitScene, el)
-	return Filter{
-		node: tree,
-	}
-}
-
-// EnableCache 决定是否启用缓存，默认开启（强烈建议，除非万一缓存模式下出现bug，可以关闭缓存退回曾经的无缓存过滤模式），开启缓存后会有30%-40%的性能提升，开启缓存并没有副作用，只是会让结构体的字段tag常驻内存减少tag字符串处理操作
-func EnableCache(enable bool) {
-	enableCache = enable
-}

filter/json_marshal_func.go

@@ -0,0 +1,17 @@
+package filter
+
+import "encoding/json"
+
+type jsonMarshalFunc = func(v interface{}) ([]byte, error)
+
+// var useJSONMarshal = false
+var useJSONMarshalFunc = json.Marshal
+
+// SetJSONMarshal 默认使用官方的json.Marshal 解析过滤后的数据结构，可以自定义json解析方法 比如换成字节的github.com/bytedance/sonic 这个进行json解析，
+// filter.SetJSONMarshal(sonic.Marshal)就可以了，但是sonic(速度是官方的2-3倍)这个库好像需要go 1.15以上，可以根据自己的需要选择自己需要的json解析库
+// 如果对性能要求没有那么高没必要换，这个方法在任意位置调用一次就可以了
+func SetJSONMarshal(fn jsonMarshalFunc) {
+	if fn != nil {
+		useJSONMarshalFunc = fn
+	}
+}

filter/node_decode.go

@@ -5,6 +5,7 @@ import (
 )
 
 type fieldNodeTree struct {
+	isSelect    bool             //是否是select 方法
 	Key         string           //字段名
 	Val         interface{}      //字段值，基础数据类型，int string，bool 等类型直接存在这里面，如果是struct,切片数组map 类型则字段所有k v会存在ChildNodes里
 	IsSlice     bool             //是否是切片，或者数组，