micro/codec/text/text.go

79 lines
1.3 KiB
Go
Raw Normal View History

2019-05-31 02:38:05 +03:00
// Package text reads any text/* content-type
package text
import (
"fmt"
"io"
"io/ioutil"
"github.com/micro/go-micro/v2/codec"
2019-05-31 02:38:05 +03:00
)
type Codec struct {
Conn io.ReadWriteCloser
}
// Frame gives us the ability to define raw data to send over the pipes
type Frame struct {
Data []byte
}
func (c *Codec) ReadHeader(m *codec.Message, t codec.MessageType) error {
return nil
}
func (c *Codec) ReadBody(b interface{}) error {
// read bytes
buf, err := ioutil.ReadAll(c.Conn)
if err != nil {
return err
}
2019-12-03 10:25:58 +03:00
switch v := b.(type) {
2019-06-10 14:42:43 +03:00
case *string:
*v = string(buf)
2019-05-31 02:38:05 +03:00
case *[]byte:
*v = buf
case *Frame:
v.Data = buf
default:
return fmt.Errorf("failed to read body: %v is not type of *[]byte", b)
}
return nil
}
func (c *Codec) Write(m *codec.Message, b interface{}) error {
var v []byte
2019-12-03 10:25:58 +03:00
switch ve := b.(type) {
2019-05-31 02:38:05 +03:00
case *Frame:
2019-12-03 10:25:58 +03:00
v = ve.Data
2019-05-31 02:38:05 +03:00
case *[]byte:
v = *ve
2019-06-10 14:42:43 +03:00
case *string:
v = []byte(*ve)
case string:
v = []byte(ve)
2019-05-31 02:38:05 +03:00
case []byte:
2019-12-03 10:25:58 +03:00
v = ve
2019-05-31 02:38:05 +03:00
default:
return fmt.Errorf("failed to write: %v is not type of *[]byte or []byte", b)
}
_, err := c.Conn.Write(v)
return err
}
func (c *Codec) Close() error {
return c.Conn.Close()
}
func (c *Codec) String() string {
2019-06-10 14:42:43 +03:00
return "text"
2019-05-31 02:38:05 +03:00
}
func NewCodec(c io.ReadWriteCloser) codec.Codec {
return &Codec{
Conn: c,
}
}