micro/codec/bytes/bytes.go

73 lines
1.2 KiB
Go
Raw Normal View History

2019-01-04 17:07:16 +03:00
// Package bytes provides a bytes codec which does not encode or decode anything
package bytes
import (
"fmt"
"io"
"io/ioutil"
"github.com/micro/go-micro/v2/codec"
2019-01-04 17:07:16 +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
}
2019-01-04 17:07:16 +03:00
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) {
case *[]byte:
*v = buf
case *Frame:
v.Data = buf
default:
return fmt.Errorf("failed to read body: %v is not type of *[]byte", b)
}
2019-01-04 17:07:16 +03:00
return nil
}
func (c *Codec) Write(m *codec.Message, b interface{}) error {
2019-01-15 00:30:43 +03:00
var v []byte
2019-12-03 10:25:58 +03:00
switch vb := b.(type) {
case *Frame:
2019-12-03 10:25:58 +03:00
v = vb.Data
2019-01-15 00:30:43 +03:00
case *[]byte:
2019-12-03 10:25:58 +03:00
v = *vb
2019-01-15 00:30:43 +03:00
case []byte:
2019-12-03 10:25:58 +03:00
v = vb
2019-01-15 00:30:43 +03:00
default:
return fmt.Errorf("failed to write: %v is not type of *[]byte or []byte", b)
2019-01-04 17:07:16 +03:00
}
2019-01-15 00:30:43 +03:00
_, err := c.Conn.Write(v)
2019-01-04 17:07:16 +03:00
return err
}
func (c *Codec) Close() error {
return c.Conn.Close()
}
func (c *Codec) String() string {
return "bytes"
}
func NewCodec(c io.ReadWriteCloser) codec.Codec {
return &Codec{
Conn: c,
}
}