快捷導(dǎo)航

golang jsoniter extension 處理動(dòng)態(tài)字段的實(shí)現(xiàn)方法

更新時(shí)間：2023年04月06日 15:11:35 作者：皿小草

這篇文章主要介紹了golang jsoniter extension 處理動(dòng)態(tài)字段的實(shí)現(xiàn)方法,我們使用實(shí)例級(jí)別的 extension, 而非全局，可以針對(duì)不同業(yè)務(wù)邏輯有所區(qū)分，jsoniter 包提供了比較完善的定制能力，通過例子可以感受一下擴(kuò)展性，需要的朋友可以參考下

1. 背景

golang 原生 json 包，在處理 json 對(duì)象的字段的時(shí)候，是需要嚴(yán)格匹配類型的。但是，實(shí)際上，當(dāng)我們與一些老系統(tǒng)或者腳本語言的系統(tǒng)對(duì)接的時(shí)候，有時(shí)候需要對(duì)類型需要做一下兼容，假設(shè)我們有以下需求

	目標(biāo)類型	輸入	解析后
int	int, string	123, “123”	123
string	int, string	123, “123”	“123”
time	unix_seconds, RFC3339	1680676884, “2023-04-05T14:41:24Z”,	“2023-04-05T14:41:24Z”

2. 可選項(xiàng)

我們以 time 作為一個(gè)樣例

包裝類，然后重新實(shí)現(xiàn) Unmarshal 接口

type MyTime struct {
	t    time.Time
}

功能可以實(shí)現(xiàn)，但是如果使用的地方很多的情況下，就可能要改動(dòng)多處，而且，這是全局級(jí)別的，可能會(huì)影響到很多包的行為

使用 jsonter 的 extension 實(shí)現(xiàn)

jsoniter 的插件文檔參考
我們使用實(shí)例級(jí)別的 extension, 而非全局，可以針對(duì)不同業(yè)務(wù)邏輯有所區(qū)分

package main

import (
	"fmt"
	"reflect"
	"strconv"
	"time"
	"unsafe"

	jsoniter "github.com/json-iterator/go"
	"github.com/modern-go/reflect2"
)

type sampleExtension struct {
	jsoniter.DummyExtension
}

type wrapEncoder struct {
	encodeFunc  func(ptr unsafe.Pointer, stream *jsoniter.Stream)
	isEmptyFunc func(ptr unsafe.Pointer) bool
	decodeFunc  func(ptr unsafe.Pointer, iter *jsoniter.Iterator)
}

func (enc *wrapEncoder) Encode(ptr unsafe.Pointer, stream *jsoniter.Stream) {
	enc.encodeFunc(ptr, stream)
}

func (codec *wrapEncoder) Decode(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
	codec.decodeFunc(ptr, iter)
}

func (enc *wrapEncoder) IsEmpty(ptr unsafe.Pointer) bool {
	if enc.isEmptyFunc == nil {
		return false
	}

	return enc.isEmptyFunc(ptr)
}

// 這里統(tǒng)一改用 unix seconds 進(jìn)行輸出
func (e *sampleExtension) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
	if typ.Kind() == reflect.Struct && typ.Type1().PkgPath() == "time" && typ.String() == "time.Time" {

		return &wrapEncoder{
			func(ptr unsafe.Pointer, stream *jsoniter.Stream) {
				t := *(*time.Time)(ptr)
				data := strconv.Itoa(int(t.Unix()))
				stream.WriteRaw(data)
			},
			nil,
			nil,
		}
	}

	return nil
}

func (e *sampleExtension) CreateDecoder(typ reflect2.Type) jsoniter.ValDecoder {
	if typ.Kind() == reflect.Struct && typ.Type1().PkgPath() == "time" && typ.String() == "time.Time" {
		return &wrapEncoder{
			decodeFunc: func(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
				switch iter.WhatIsNext() {
				case jsoniter.NumberValue: // 兼容 unix 數(shù)字解析
					timeUnix := iter.ReadInt()
					newTime := time.Unix(int64(timeUnix), 0)
					*(*time.Time)(ptr) = newTime

				case jsoniter.NilValue:
					iter.Skip()

				case jsoniter.StringValue:
					timeStr := iter.ReadString()
					newTime, err := time.Parse(time.RFC3339, timeStr)
					if err != nil {
						fmt.Println("Unmarshal err", err)
					}
					*(*time.Time)(ptr) = newTime

				}
			},
		}
	}

	return nil
}

type Person struct {
	Birth time.Time `json:"birth"`
}

func main() {
	extension := &sampleExtension{}
	jsoniterAPI := jsoniter.Config{}.Froze()
	jsoniterAPI.RegisterExtension(extension)
	var p1 = Person{
		Birth: time.Now(),
	}
	j, err := jsoniterAPI.MarshalToString(p1)
	if err != nil {
		panic(err)
	}
	fmt.Println(j)

	var p2 Person
	err = jsoniterAPI.Unmarshal([]byte(`{"birth": 1680254527}`), &p2)
	if err != nil {
		panic(err)
	}
	fmt.Println("p2", p2)

	var p3 Person
	err = jsoniterAPI.Unmarshal([]byte(`{"birth": "2023-03-21T07:20:04+00:00"}`), &p3)
	if err != nil {
		panic(err)
	}
	fmt.Println("p3", p3)

	var p4 Person
	err = jsoniterAPI.Unmarshal([]byte(`{"birth": null}`), &p4)
	if err != nil {
		panic(err)
	}
	fmt.Println("p4", p4)
}

我們?cè)诶又?，?shí)現(xiàn)了：

把 p1 使用了 unix 數(shù)字進(jìn)行序列化
在反序列化 p2/p3/p4的時(shí)候，兼容了 字符串/數(shù)字/null

總結(jié)

jsoniter 包提供了比較完善的定制能力，通過例子可以感受一下擴(kuò)展性。后續(xù)大家可以根據(jù)業(yè)務(wù)需求發(fā)掘更多的能力

到此這篇關(guān)于golang jsoniter extension 處理動(dòng)態(tài)字段的文章就介紹到這了,更多相關(guān)go動(dòng)態(tài)字段內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: