mirror of
https://github.com/ent/ent.git
synced 2026-05-22 09:31:45 +03:00
7988d3084d
* entc/hooks: initial work for mutations and hooks * ent/schema: adding policy to schema * ent: change op string to uint * entc: move entschema to runtime and enable smooth transition * entc/privacy: adding privacy template * all: goimports * intg/hooks: mutation client/tx and basic schema tests * ent/privacy: adding more verbose decisions * entc/gen: edge-ids getter and additional tests * all: regen assets * entc/gen: fix client hookd propagation * intg: add deletion example * intg/privacy: remove old entschema package * typed privacy * ent/privacy: hooks shouldn't be called on privacy deny * entc/gen: fix schema hooks invocation order * remove read policy from public api * update circleci go orb Co-authored-by: Ariel Mashraki <ariel@mashraki.co.il>
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
package graphson
|
|
|
|
import (
|
|
"io"
|
|
"reflect"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/modern-go/reflect2"
|
|
)
|
|
|
|
type encodeExtension struct {
|
|
jsoniter.DummyExtension
|
|
}
|
|
|
|
// Marshal returns the graphson encoding of v.
|
|
func Marshal(v interface{}) ([]byte, error) {
|
|
return config.Marshal(v)
|
|
}
|
|
|
|
// MarshalToString returns the graphson encoding of v as string.
|
|
func MarshalToString(v interface{}) (string, error) {
|
|
return config.MarshalToString(v)
|
|
}
|
|
|
|
// Encoder defines a graphson encoder.
|
|
type Encoder interface {
|
|
Encode(interface{}) error
|
|
}
|
|
|
|
// NewEncoder create a graphson encoder.
|
|
func NewEncoder(w io.Writer) Encoder {
|
|
return config.NewEncoder(w)
|
|
}
|
|
|
|
// Marshaler is the interface implemented by types that
|
|
// can marshal themselves as graphson.
|
|
type Marshaler interface {
|
|
MarshalGraphson() ([]byte, error)
|
|
}
|
|
|
|
// UpdateStructDescriptor decorates struct field encoders for graphson tags.
|
|
func (ext encodeExtension) UpdateStructDescriptor(desc *jsoniter.StructDescriptor) {
|
|
for _, binding := range desc.Fields {
|
|
if tag, ok := binding.Field.Tag().Lookup("graphson"); ok && tag != "-" {
|
|
if enc := ext.DecoratorOfStructField(binding.Encoder, tag); enc != nil {
|
|
binding.Encoder = enc
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// CreateEncoder returns a value encoder for type.
|
|
func (ext encodeExtension) CreateEncoder(typ reflect2.Type) jsoniter.ValEncoder {
|
|
if enc := ext.EncoderOfRegistered(typ); enc != nil {
|
|
return enc
|
|
}
|
|
if enc := ext.EncoderOfNative(typ); enc != nil {
|
|
return enc
|
|
}
|
|
switch typ.Kind() {
|
|
case reflect.Map:
|
|
return ext.EncoderOfMap(typ)
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// DecorateEncoder decorates an passed in value encoder for type.
|
|
func (ext encodeExtension) DecorateEncoder(typ reflect2.Type, enc jsoniter.ValEncoder) jsoniter.ValEncoder {
|
|
if enc := ext.DecoratorOfRegistered(enc); enc != nil {
|
|
return enc
|
|
}
|
|
if enc := ext.DecoratorOfMarshaler(typ, enc); enc != nil {
|
|
return enc
|
|
}
|
|
if enc := ext.DecoratorOfTyper(typ, enc); enc != nil {
|
|
return enc
|
|
}
|
|
if enc := ext.DecoratorOfNative(typ, enc); enc != nil {
|
|
return enc
|
|
}
|
|
switch typ.Kind() {
|
|
case reflect.Ptr, reflect.Interface, reflect.Struct:
|
|
return enc
|
|
case reflect.Array:
|
|
return ext.DecoratorOfArray(enc)
|
|
case reflect.Slice:
|
|
return ext.DecoratorOfSlice(typ, enc)
|
|
case reflect.Map:
|
|
return ext.DecoratorOfMap(enc)
|
|
default:
|
|
return ext.EncoderOfError("graphson: unsupported type: " + typ.String())
|
|
}
|
|
}
|