mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2026-01-05 23:15:47 +08:00
增加服务之间拷贝配置的API(开源版本只有定义,没有完全实现)
This commit is contained in:
45
internal/utils/json.go
Normal file
45
internal/utils/json.go
Normal file
@@ -0,0 +1,45 @@
|
||||
// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
||||
|
||||
package utils
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
// JSONClone 使用JSON协议克隆对象
|
||||
func JSONClone[T any](ptr T) (newPtr T, err error) {
|
||||
var ptrType = reflect.TypeOf(ptr)
|
||||
var kind = ptrType.Kind()
|
||||
if kind != reflect.Ptr && kind != reflect.Slice {
|
||||
err = errors.New("JSONClone: input must be a ptr or slice")
|
||||
return
|
||||
}
|
||||
var jsonData []byte
|
||||
jsonData, err = json.Marshal(ptr)
|
||||
if err != nil {
|
||||
return ptr, errors.New("JSONClone: marshal failed: " + err.Error())
|
||||
}
|
||||
|
||||
var newValue any
|
||||
switch kind {
|
||||
case reflect.Ptr:
|
||||
newValue = reflect.New(ptrType.Elem()).Interface()
|
||||
case reflect.Slice:
|
||||
newValue = reflect.New(reflect.SliceOf(ptrType.Elem())).Interface()
|
||||
default:
|
||||
return ptr, errors.New("JSONClone: unknown data type")
|
||||
}
|
||||
err = json.Unmarshal(jsonData, newValue)
|
||||
if err != nil {
|
||||
err = errors.New("JSONClone: unmarshal failed: " + err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
if kind == reflect.Slice {
|
||||
newValue = reflect.Indirect(reflect.ValueOf(newValue)).Interface()
|
||||
}
|
||||
|
||||
return newValue.(T), nil
|
||||
}
|
||||
51
internal/utils/json_test.go
Normal file
51
internal/utils/json_test.go
Normal file
@@ -0,0 +1,51 @@
|
||||
// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
||||
|
||||
package utils_test
|
||||
|
||||
import (
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/utils"
|
||||
"github.com/iwind/TeaGo/logs"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestJSONClone(t *testing.T) {
|
||||
type user struct {
|
||||
Name string
|
||||
Age int
|
||||
}
|
||||
|
||||
var u = &user{
|
||||
Name: "Jack",
|
||||
Age: 20,
|
||||
}
|
||||
|
||||
newU, err := utils.JSONClone[*user](u)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Logf("%#v", newU)
|
||||
}
|
||||
|
||||
func TestJSONClone_Slice(t *testing.T) {
|
||||
type user struct {
|
||||
Name string
|
||||
Age int
|
||||
}
|
||||
|
||||
var u = []*user{
|
||||
{
|
||||
Name: "Jack",
|
||||
Age: 20,
|
||||
},
|
||||
{
|
||||
Name: "Lily",
|
||||
Age: 18,
|
||||
},
|
||||
}
|
||||
|
||||
newU, err := utils.JSONClone[[]*user](u)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
logs.PrintAsJSON(newU, t)
|
||||
}
|
||||
Reference in New Issue
Block a user