Make json/protobuf codecs
This commit is contained in:
		
							
								
								
									
										49
									
								
								codec/proto/proto.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								codec/proto/proto.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,49 @@ | ||||
| // Package proto provides a proto codec | ||||
| package proto | ||||
|  | ||||
| import ( | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
|  | ||||
| 	"github.com/golang/protobuf/proto" | ||||
| 	"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 { | ||||
| 	buf, err := ioutil.ReadAll(c.Conn) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return proto.Unmarshal(buf, b.(proto.Message)) | ||||
| } | ||||
|  | ||||
| func (c *Codec) Write(m *codec.Message, b interface{}) error { | ||||
| 	buf, err := proto.Marshal(b.(proto.Message)) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	_, err = c.Conn.Write(buf) | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (c *Codec) Close() error { | ||||
| 	return c.Conn.Close() | ||||
| } | ||||
|  | ||||
| func (c *Codec) String() string { | ||||
| 	return "proto" | ||||
| } | ||||
|  | ||||
| func NewCodec(c io.ReadWriteCloser) codec.Codec { | ||||
| 	return &Codec{ | ||||
| 		Conn: c, | ||||
| 	} | ||||
| } | ||||
		Reference in New Issue
	
	Block a user