自定义json编解码器

Author: 520MianXiangDuiXiang520

decode/builder.go

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2021 Junebao
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+package decode
+
+import (
+	"unsafe"
+)
+
+type builder struct {
+	data []byte
+	// strings.Builder
+}
+
+func (b *builder) Write(d []byte) {
+	b.data = append(b.data, d...)
+}
+
+func (b *builder) WriteByte(d byte) error {
+	b.data = append(b.data, d)
+	return nil
+}
+
+func (b *builder) Bytes() []byte {
+	return b.data
+}
+
+func (b *builder) String() string {
+	return *(*string)(unsafe.Pointer(&b.data))
+}
+
+// func (b *builder) getData() []byte {
+//     ptr := unsafe.Pointer(b)
+//     addr := (*strings.Builder)(ptr)
+//     offset := unsafe.Sizeof(addr)
+//     return *(*[]byte)(unsafe.Pointer(uintptr(ptr) + offset))
+// }

decode/builder_test.go

@@ -0,0 +1,76 @@
+/*
+ * Copyright 2021 Junebao
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+package decode
+
+import (
+	"bytes"
+	"github.com/valyala/bytebufferpool"
+	"strings"
+	"testing"
+)
+
+// 测试多种[]byte拼接方式的优劣
+func BenchmarkWrite(b *testing.B) {
+	length := 1000000000
+	b.Run("bytes.buffer", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			buffer := bytes.NewBuffer([]byte{})
+			for j := 0; j < length; j++ {
+				buffer.WriteByte(byte('a' + j))
+			}
+			_ = buffer.Bytes()
+		}
+	})
+	b.Run("strings.builder", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			build := strings.Builder{}
+			for j := 0; j < length; j++ {
+				build.WriteByte(byte('a' + j))
+			}
+			_ = []byte(build.String())
+		}
+	})
+	b.Run("decode.builder", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			build := builder{}
+			for j := 0; j < length; j++ {
+				_ = build.WriteByte(byte('a' + j))
+			}
+			_ = build.Bytes()
+		}
+	})
+	b.Run("[]byte", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			db := make([]byte, 0)
+			for j := 0; j < length; j++ {
+				db = append(db, byte('a'+1))
+			}
+			_ = db
+		}
+	})
+
+	b.Run("buffer pool", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			pool := bytebufferpool.Get()
+			for j := 0; j < length; j++ {
+				_ = pool.WriteByte(byte('a' + j))
+			}
+			_ = pool.Bytes()
+		}
+	})
+}

decode/jsonnode_test.go

@@ -0,0 +1,70 @@
+/*
+ * Copyright 2021 Junebao
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+package decode
+
+import (
+	"testing"
+)
+
+func Test_keyReplace(t *testing.T) {
+	type args struct {
+		key string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{"common", args{key: "article~1a~01~001name"}, "article~01a~001~0001name"},
+		{"common1", args{key: "01~1"}, "01~01"},
+		{"common2", args{key: "0101~"}, "0101~"},
+		{"common3", args{key: "0101/01"}, "0101~101"},
+		{"common4", args{key: "0101/01~01"}, "0101~101~001"},
+		{"common5", args{key: "article/name"}, "article~1name"},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := KeyReplace(tt.args.key); got != tt.want {
+				t.Errorf("keyReplace() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func Test_keyRestore(t *testing.T) {
+	type args struct {
+		key string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{"common", args{key: "article~01a~001~0001name"}, "article~1a~01~001name"},
+		{"common1", args{key: "article~1name"}, "article/name"},
+		{"common2", args{key: "article_name"}, "article_name"},
+		{"common3", args{key: "article~1a~001~0001name"}, "article/a~01~001name"},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := KeyRestore(tt.args.key); got != tt.want {
+				t.Errorf("keyRestore() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}