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.
38 lines
665 B
38 lines
665 B
2 years ago
|
package toml
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"github.com/BurntSushi/toml"
|
||
|
|
"github.com/go-kratos/kratos/v2/encoding"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Name is the name registered for the toml compressor.
|
||
|
|
const Name = "toml"
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
encoding.RegisterCodec(codec{})
|
||
|
|
}
|
||
|
|
|
||
|
|
// codec is a Codec implementation with toml.
|
||
|
|
type codec struct{}
|
||
|
|
|
||
|
|
func (c codec) Marshal(v interface{}) ([]byte, error) {
|
||
|
|
buf := &bytes.Buffer{}
|
||
|
|
encoder := toml.NewEncoder(buf)
|
||
|
|
|
||
|
|
if err := encoder.Encode(v); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
data := buf.Bytes()
|
||
|
|
return data, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c codec) Unmarshal(data []byte, v interface{}) error {
|
||
|
|
return toml.Unmarshal(data, v)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c codec) Name() string {
|
||
|
|
return Name
|
||
|
|
}
|