mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package fastjsonserializer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/zijiren233/stream"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
|
|
|
type JSONSerializer struct{}
|
|
|
|
func (*JSONSerializer) Scan(
|
|
ctx context.Context,
|
|
field *schema.Field,
|
|
dst reflect.Value,
|
|
dbValue any,
|
|
) (err error) {
|
|
fieldValue := reflect.New(field.FieldType)
|
|
|
|
if dbValue != nil {
|
|
var bytes []byte
|
|
switch v := dbValue.(type) {
|
|
case []byte:
|
|
bytes = v
|
|
case string:
|
|
bytes = stream.StringToBytes(v)
|
|
default:
|
|
return fmt.Errorf("failed to unmarshal JSONB value: %#v", dbValue)
|
|
}
|
|
|
|
if len(bytes) == 0 {
|
|
field.ReflectValueOf(ctx, dst).Set(reflect.Zero(field.FieldType))
|
|
return nil
|
|
}
|
|
|
|
err = json.Unmarshal(bytes, fieldValue.Interface())
|
|
}
|
|
|
|
field.ReflectValueOf(ctx, dst).Set(fieldValue.Elem())
|
|
return
|
|
}
|
|
|
|
func (*JSONSerializer) Value(
|
|
_ context.Context,
|
|
_ *schema.Field,
|
|
_ reflect.Value,
|
|
fieldValue any,
|
|
) (any, error) {
|
|
return json.Marshal(fieldValue)
|
|
}
|
|
|
|
func init() {
|
|
schema.RegisterSerializer("fastjson", new(JSONSerializer))
|
|
}
|