Compare commits
4 Commits
Author | SHA1 | Date | |
---|---|---|---|
551d522616 | |||
|
8b2b346af9 | ||
19f0e4b27e | |||
e48795153d |
@ -4,7 +4,7 @@ import (
|
|||||||
"bytes"
|
"bytes"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"go.unistack.org/micro/v3/codec"
|
"go.unistack.org/micro/v4/codec"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestFrame(t *testing.T) {
|
func TestFrame(t *testing.T) {
|
||||||
@ -28,7 +28,7 @@ func TestFrameFlatten(t *testing.T) {
|
|||||||
Name: &codec.Frame{Data: []byte("test")},
|
Name: &codec.Frame{Data: []byte("test")},
|
||||||
}
|
}
|
||||||
|
|
||||||
buf, err := NewCodec(codec.Flatten(true)).Marshal(s)
|
buf, err := NewCodec().Marshal(s)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@ -37,18 +37,14 @@ func TestFrameFlatten(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestNativeYamlTags(t *testing.T) {
|
func TestReadBody(t *testing.T) {
|
||||||
s := &struct {
|
s := &struct {
|
||||||
One string `yaml:"first"`
|
Name string
|
||||||
}{
|
}{}
|
||||||
One: "",
|
c := NewCodec()
|
||||||
}
|
b := bytes.NewReader(nil)
|
||||||
|
err := c.ReadBody(b, s)
|
||||||
err := NewCodec().Unmarshal([]byte(`first: "val"`), s)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
if s.One != "val" {
|
|
||||||
t.Fatalf("XXX %#+v\n", s)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
19
go.mod
19
go.mod
@ -1,10 +1,17 @@
|
|||||||
module go.unistack.org/micro-codec-yaml/v3
|
module go.unistack.org/micro-codec-yaml/v4
|
||||||
|
|
||||||
go 1.16
|
go 1.19
|
||||||
|
|
||||||
require (
|
require (
|
||||||
go.unistack.org/micro-proto/v3 v3.4.1
|
go.unistack.org/micro-proto/v4 v4.0.0
|
||||||
go.unistack.org/micro/v3 v3.10.94
|
go.unistack.org/micro/v4 v4.0.1
|
||||||
google.golang.org/protobuf v1.34.2 // indirect
|
sigs.k8s.io/yaml v1.3.0
|
||||||
gopkg.in/yaml.v3 v3.0.1
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/kr/pretty v0.2.1 // indirect
|
||||||
|
github.com/kr/text v0.2.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.28.1 // indirect
|
||||||
|
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 // indirect
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||||
)
|
)
|
||||||
|
60
yaml.go
60
yaml.go
@ -1,11 +1,13 @@
|
|||||||
// Package yaml provides a yaml codec
|
// Package yaml provides a yaml codec
|
||||||
package yaml // import "go.unistack.org/micro-codec-yaml/v3"
|
package yaml // import "go.unistack.org/micro-codec-yaml/v4"
|
||||||
|
|
||||||
import (
|
import (
|
||||||
pb "go.unistack.org/micro-proto/v3/codec"
|
"io"
|
||||||
"go.unistack.org/micro/v3/codec"
|
|
||||||
rutil "go.unistack.org/micro/v3/util/reflect"
|
pb "go.unistack.org/micro-proto/v4/codec"
|
||||||
yaml "gopkg.in/yaml.v3"
|
"go.unistack.org/micro/v4/codec"
|
||||||
|
rutil "go.unistack.org/micro/v4/util/reflect"
|
||||||
|
"sigs.k8s.io/yaml"
|
||||||
)
|
)
|
||||||
|
|
||||||
type yamlCodec struct {
|
type yamlCodec struct {
|
||||||
@ -14,6 +16,10 @@ type yamlCodec struct {
|
|||||||
|
|
||||||
var _ codec.Codec = &yamlCodec{}
|
var _ codec.Codec = &yamlCodec{}
|
||||||
|
|
||||||
|
const (
|
||||||
|
flattenTag = "flatten"
|
||||||
|
)
|
||||||
|
|
||||||
func (c *yamlCodec) Marshal(v interface{}, opts ...codec.Option) ([]byte, error) {
|
func (c *yamlCodec) Marshal(v interface{}, opts ...codec.Option) ([]byte, error) {
|
||||||
if v == nil {
|
if v == nil {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
@ -23,12 +29,9 @@ func (c *yamlCodec) Marshal(v interface{}, opts ...codec.Option) ([]byte, error)
|
|||||||
for _, o := range opts {
|
for _, o := range opts {
|
||||||
o(&options)
|
o(&options)
|
||||||
}
|
}
|
||||||
|
if nv, nerr := rutil.StructFieldByTag(v, options.TagName, flattenTag); nerr == nil {
|
||||||
if options.Flatten {
|
|
||||||
if nv, nerr := rutil.StructFieldByTag(v, options.TagName, "flatten"); nerr == nil {
|
|
||||||
v = nv
|
v = nv
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
switch m := v.(type) {
|
switch m := v.(type) {
|
||||||
case *codec.Frame:
|
case *codec.Frame:
|
||||||
@ -50,11 +53,9 @@ func (c *yamlCodec) Unmarshal(b []byte, v interface{}, opts ...codec.Option) err
|
|||||||
o(&options)
|
o(&options)
|
||||||
}
|
}
|
||||||
|
|
||||||
if options.Flatten {
|
if nv, nerr := rutil.StructFieldByTag(v, options.TagName, flattenTag); nerr == nil {
|
||||||
if nv, nerr := rutil.StructFieldByTag(v, options.TagName, "flatten"); nerr == nil {
|
|
||||||
v = nv
|
v = nv
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
switch m := v.(type) {
|
switch m := v.(type) {
|
||||||
case *codec.Frame:
|
case *codec.Frame:
|
||||||
@ -68,6 +69,41 @@ func (c *yamlCodec) Unmarshal(b []byte, v interface{}, opts ...codec.Option) err
|
|||||||
return yaml.Unmarshal(b, v)
|
return yaml.Unmarshal(b, v)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *yamlCodec) ReadHeader(conn io.Reader, m *codec.Message, t codec.MessageType) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *yamlCodec) ReadBody(conn io.Reader, v interface{}) error {
|
||||||
|
if v == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
buf, err := io.ReadAll(conn)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
} else if len(buf) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.Unmarshal(buf, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *yamlCodec) Write(conn io.Writer, m *codec.Message, v interface{}) error {
|
||||||
|
if v == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
buf, err := c.Marshal(v)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
} else if len(buf) == 0 {
|
||||||
|
return codec.ErrInvalidMessage
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = conn.Write(buf)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
func (c *yamlCodec) String() string {
|
func (c *yamlCodec) String() string {
|
||||||
return "yaml"
|
return "yaml"
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user