add bytes codec, still unused
This commit is contained in:
parent
461df8d464
commit
4adc31e62d
58
codec/bytes/bytes.go
Normal file
58
codec/bytes/bytes.go
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
// 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/codec"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Codec struct {
|
||||||
|
Conn io.ReadWriteCloser
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Codec) ReadHeader(m *codec.Message, t codec.MessageType) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Codec) ReadBody(b interface{}) error {
|
||||||
|
v, ok := b.(*[]byte)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("failed to read body: %v is not type of *[]byte", b)
|
||||||
|
}
|
||||||
|
|
||||||
|
// read bytes
|
||||||
|
buf, err := ioutil.ReadAll(c.Conn)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// set bytes
|
||||||
|
*v = buf
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Codec) Write(m *codec.Message, b interface{}) error {
|
||||||
|
v, ok := b.(*[]byte)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("failed to write: %v is not type of *[]byte", b)
|
||||||
|
}
|
||||||
|
_, err := c.Conn.Write(*v)
|
||||||
|
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,
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user