You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.4 KiB
82 lines
1.4 KiB
package form
|
|
|
|
import (
|
|
"net/url"
|
|
"reflect"
|
|
|
|
"github.com/go-kratos/kratos/v2/encoding"
|
|
|
|
"github.com/go-playground/form/v4"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
const (
|
|
// Name is form codec name
|
|
Name = "x-www-form-urlencoded"
|
|
// Null value string
|
|
nullStr = "null"
|
|
)
|
|
|
|
var (
|
|
encoder = form.NewEncoder()
|
|
decoder = form.NewDecoder()
|
|
)
|
|
|
|
func init() {
|
|
decoder.SetTagName("json")
|
|
encoder.SetTagName("json")
|
|
encoding.RegisterCodec(codec{encoder: encoder, decoder: decoder})
|
|
}
|
|
|
|
type codec struct {
|
|
encoder *form.Encoder
|
|
decoder *form.Decoder
|
|
}
|
|
|
|
func (c codec) Marshal(v interface{}) ([]byte, error) {
|
|
var vs url.Values
|
|
var err error
|
|
if m, ok := v.(proto.Message); ok {
|
|
vs, err = EncodeValues(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
vs, err = c.encoder.Encode(v)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
for k, v := range vs {
|
|
if len(v) == 0 {
|
|
delete(vs, k)
|
|
}
|
|
}
|
|
return []byte(vs.Encode()), nil
|
|
}
|
|
|
|
func (c codec) Unmarshal(data []byte, v interface{}) error {
|
|
vs, err := url.ParseQuery(string(data))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rv := reflect.ValueOf(v)
|
|
for rv.Kind() == reflect.Ptr {
|
|
if rv.IsNil() {
|
|
rv.Set(reflect.New(rv.Type().Elem()))
|
|
}
|
|
rv = rv.Elem()
|
|
}
|
|
if m, ok := v.(proto.Message); ok {
|
|
return DecodeValues(m, vs)
|
|
} else if m, ok := reflect.Indirect(reflect.ValueOf(v)).Interface().(proto.Message); ok {
|
|
return DecodeValues(m, vs)
|
|
}
|
|
|
|
return c.decoder.Decode(v, vs)
|
|
}
|
|
|
|
func (codec) Name() string {
|
|
return Name
|
|
}
|
|
|