chore: import upstream snapshot with attribution

codec/bytes/bytes.go

@@ -0,0 +1,71 @@
+// Package bytes provides a bytes codec which does not encode or decode anything
+package bytes
+
+import (
+	"fmt"
+	"io"
+
+	"go-micro.dev/v6/codec"
+)
+
+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 := io.ReadAll(c.Conn)
+	if err != nil {
+		return err
+	}
+
+	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)
+	}
+
+	return nil
+}
+
+func (c *Codec) Write(m *codec.Message, b interface{}) error {
+	var v []byte
+	switch vb := b.(type) {
+	case *Frame:
+		v = vb.Data
+	case *[]byte:
+		v = *vb
+	case []byte:
+		v = vb
+	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 {
+	return "bytes"
+}
+
+func NewCodec(c io.ReadWriteCloser) codec.Codec {
+	return &Codec{
+		Conn: c,
+	}
+}

codec/bytes/marshaler.go

@@ -0,0 +1,40 @@
+package bytes
+
+import (
+	"go-micro.dev/v6/codec"
+)
+
+type Marshaler struct{}
+
+type Message struct {
+	Header map[string]string
+	Body   []byte
+}
+
+func (n Marshaler) Marshal(v interface{}) ([]byte, error) {
+	switch ve := v.(type) {
+	case *[]byte:
+		return *ve, nil
+	case []byte:
+		return ve, nil
+	case *Message:
+		return ve.Body, nil
+	}
+	return nil, codec.ErrInvalidMessage
+}
+
+func (n Marshaler) Unmarshal(d []byte, v interface{}) error {
+	switch ve := v.(type) {
+	case *[]byte:
+		*ve = d
+		return nil
+	case *Message:
+		ve.Body = d
+		return nil
+	}
+	return codec.ErrInvalidMessage
+}
+
+func (n Marshaler) String() string {
+	return "bytes"
+}

codec/codec.go

@@ -0,0 +1,69 @@
+// Package codec is an interface for encoding messages
+package codec
+
+import (
+	"errors"
+	"io"
+)
+
+const (
+	Error MessageType = iota
+	Request
+	Response
+	Event
+)
+
+var (
+	ErrInvalidMessage = errors.New("invalid message")
+)
+
+type MessageType int
+
+// Takes in a connection/buffer and returns a new Codec.
+type NewCodec func(io.ReadWriteCloser) Codec
+
+// Codec encodes/decodes various types of messages used within go-micro.
+// ReadHeader and ReadBody are called in pairs to read requests/responses
+// from the connection. Close is called when finished with the
+// connection. ReadBody may be called with a nil argument to force the
+// body to be read and discarded.
+type Codec interface {
+	Reader
+	Writer
+	Close() error
+	String() string
+}
+
+type Reader interface {
+	ReadHeader(*Message, MessageType) error
+	ReadBody(interface{}) error
+}
+
+type Writer interface {
+	Write(*Message, interface{}) error
+}
+
+// Marshaler is a simple encoding interface used for the broker/transport
+// where headers are not supported by the underlying implementation.
+type Marshaler interface {
+	Marshal(interface{}) ([]byte, error)
+	Unmarshal([]byte, interface{}) error
+	String() string
+}
+
+// Message represents detailed information about
+// the communication, likely followed by the body.
+// In the case of an error, body may be nil.
+type Message struct {
+
+	// The values read from the socket
+	Header   map[string]string
+	Id       string
+	Target   string
+	Method   string
+	Endpoint string
+	Error    string
+
+	Body []byte
+	Type MessageType
+}

codec/grpc/grpc.go

@@ -0,0 +1,150 @@
+// Package grpc provides a grpc codec
+package grpc
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"strings"
+
+	"github.com/golang/protobuf/proto"
+	"go-micro.dev/v6/codec"
+	"go-micro.dev/v6/transport/headers"
+)
+
+type Codec struct {
+	Conn        io.ReadWriteCloser
+	ContentType string
+}
+
+func (c *Codec) ReadHeader(m *codec.Message, t codec.MessageType) error {
+	if ct := m.Header["Content-Type"]; len(ct) > 0 {
+		c.ContentType = ct
+	}
+
+	if ct := m.Header["content-type"]; len(ct) > 0 {
+		c.ContentType = ct
+	}
+
+	// service method
+	path := m.Header[":path"]
+	if len(path) == 0 || path[0] != '/' {
+		m.Target = m.Header[headers.Request]
+		m.Endpoint = m.Header[headers.Endpoint]
+	} else {
+		// [ , a.package.Foo, Bar]
+		parts := strings.Split(path, "/")
+		if len(parts) != 3 {
+			return errors.New("unknown request path")
+		}
+		service := strings.Split(parts[1], ".")
+		m.Endpoint = strings.Join([]string{service[len(service)-1], parts[2]}, ".")
+		m.Target = strings.Join(service[:len(service)-1], ".")
+	}
+
+	return nil
+}
+
+func (c *Codec) ReadBody(b interface{}) error {
+	// no body
+	if b == nil {
+		return nil
+	}
+
+	_, buf, err := decode(c.Conn)
+	if err != nil {
+		return err
+	}
+
+	switch c.ContentType {
+	case "application/grpc+json":
+		return json.Unmarshal(buf, b)
+	case "application/grpc+proto", "application/grpc":
+		return proto.Unmarshal(buf, b.(proto.Message))
+	}
+
+	return errors.New("unsupported Content-Type")
+}
+
+func (c *Codec) Write(m *codec.Message, b interface{}) error {
+	var buf []byte
+	var err error
+
+	if ct := m.Header["Content-Type"]; len(ct) > 0 {
+		c.ContentType = ct
+	}
+
+	if ct := m.Header["content-type"]; len(ct) > 0 {
+		c.ContentType = ct
+	}
+
+	switch m.Type {
+	case codec.Request:
+		parts := strings.Split(m.Endpoint, ".")
+		m.Header[":method"] = "POST"
+		m.Header[":path"] = fmt.Sprintf("/%s.%s/%s", m.Target, parts[0], parts[1])
+		m.Header[":proto"] = "HTTP/2.0"
+		m.Header["te"] = "trailers"
+		m.Header["user-agent"] = "grpc-go/1.0.0"
+		m.Header[":authority"] = m.Target
+		m.Header["content-type"] = c.ContentType
+	case codec.Response:
+		m.Header["Trailer"] = "grpc-status" // , grpc-message"
+		m.Header["content-type"] = c.ContentType
+		m.Header[":status"] = "200"
+		m.Header["grpc-status"] = "0"
+		//		m.Header["grpc-message"] = ""
+	case codec.Error:
+		m.Header["Trailer"] = "grpc-status, grpc-message"
+		// micro end of stream
+		if m.Error == "EOS" {
+			m.Header["grpc-status"] = "0"
+		} else {
+			m.Header["grpc-message"] = m.Error
+			m.Header["grpc-status"] = "13"
+		}
+
+		return nil
+	}
+
+	// marshal content
+	switch c.ContentType {
+	case "application/grpc+json":
+		buf, err = json.Marshal(b)
+	case "application/grpc+proto", "application/grpc":
+		pb, ok := b.(proto.Message)
+		if ok {
+			buf, err = proto.Marshal(pb)
+		}
+	default:
+		err = errors.New("unsupported Content-Type")
+	}
+	// check error
+	if err != nil {
+		m.Header["grpc-status"] = "8"
+		m.Header["grpc-message"] = err.Error()
+		return err
+	}
+
+	if len(buf) == 0 {
+		return nil
+	}
+
+	return encode(0, buf, c.Conn)
+}
+
+func (c *Codec) Close() error {
+	return c.Conn.Close()
+}
+
+func (c *Codec) String() string {
+	return "grpc"
+}
+
+func NewCodec(c io.ReadWriteCloser) codec.Codec {
+	return &Codec{
+		Conn:        c,
+		ContentType: "application/grpc",
+	}
+}

codec/grpc/util.go

@@ -0,0 +1,70 @@
+package grpc
+
+import (
+	"encoding/binary"
+	"fmt"
+	"io"
+)
+
+var (
+	MaxMessageSize = 1024 * 1024 * 4 // 4Mb
+	maxInt         = int(^uint(0) >> 1)
+)
+
+func decode(r io.Reader) (uint8, []byte, error) {
+	header := make([]byte, 5)
+
+	// read the header
+	if _, err := r.Read(header); err != nil {
+		return uint8(0), nil, err
+	}
+
+	// get encoding format e.g compressed
+	cf := header[0]
+
+	// get message length
+	length := binary.BigEndian.Uint32(header[1:])
+
+	// no encoding format
+	if length == 0 {
+		return cf, nil, nil
+	}
+
+	//
+	if int64(length) > int64(maxInt) {
+		return cf, nil, fmt.Errorf("grpc: received message larger than max length allowed on current machine (%d vs. %d)", length, maxInt)
+	}
+	if int(length) > MaxMessageSize {
+		return cf, nil, fmt.Errorf("grpc: received message larger than max (%d vs. %d)", length, MaxMessageSize)
+	}
+
+	msg := make([]byte, int(length))
+
+	if _, err := r.Read(msg); err != nil {
+		if err == io.EOF {
+			err = io.ErrUnexpectedEOF
+		}
+		return cf, nil, err
+	}
+
+	return cf, msg, nil
+}
+
+func encode(cf uint8, buf []byte, w io.Writer) error {
+	header := make([]byte, 5)
+
+	// set compression
+	header[0] = cf
+
+	// write length as header
+	binary.BigEndian.PutUint32(header[1:], uint32(len(buf)))
+
+	// read the header
+	if _, err := w.Write(header); err != nil {
+		return err
+	}
+
+	// write the buffer
+	_, err := w.Write(buf)
+	return err
+}

codec/json/any_test.go

@@ -0,0 +1,91 @@
+package json
+
+import (
+	"encoding/json"
+	"testing"
+
+	"google.golang.org/protobuf/types/known/anypb"
+	"google.golang.org/protobuf/types/known/wrapperspb"
+)
+
+// TestAnyTypeMarshaling tests that google.protobuf.Any types are properly marshaled with @type field
+func TestAnyTypeMarshaling(t *testing.T) {
+	marshaler := Marshaler{}
+
+	// Create a StringValue message
+	stringValue := wrapperspb.String("test value")
+
+	// Wrap it in an Any message
+	anyMsg, err := anypb.New(stringValue)
+	if err != nil {
+		t.Fatalf("Failed to create Any message: %v", err)
+	}
+
+	// Marshal using our JSON marshaler
+	data, err := marshaler.Marshal(anyMsg)
+	if err != nil {
+		t.Fatalf("Failed to marshal Any message: %v", err)
+	}
+
+	// Unmarshal into a map to check for @type field
+	var result map[string]interface{}
+	if err := json.Unmarshal(data, &result); err != nil {
+		t.Fatalf("Failed to unmarshal JSON: %v", err)
+	}
+
+	// Check that @type field exists
+	typeURL, ok := result["@type"].(string)
+	if !ok {
+		t.Fatalf("@type field not found in JSON output. Got: %v", string(data))
+	}
+
+	// Verify the type URL is correct
+	expectedTypeURL := "type.googleapis.com/google.protobuf.StringValue"
+	if typeURL != expectedTypeURL {
+		t.Errorf("Expected @type to be %s, got %s", expectedTypeURL, typeURL)
+	}
+
+	// Verify the value field exists
+	if _, ok := result["value"]; !ok {
+		t.Errorf("value field not found in JSON output. Got: %v", string(data))
+	}
+
+	t.Logf("Successfully marshaled Any type with @type field: %s", string(data))
+}
+
+// TestAnyTypeUnmarshaling tests that JSON with @type field can be unmarshaled into google.protobuf.Any
+func TestAnyTypeUnmarshaling(t *testing.T) {
+	marshaler := Marshaler{}
+
+	// JSON representation of an Any message with @type field
+	jsonData := []byte(`{
+		"@type": "type.googleapis.com/google.protobuf.StringValue",
+		"value": "test value"
+	}`)
+
+	// Unmarshal into an Any message
+	anyMsg := &anypb.Any{}
+	if err := marshaler.Unmarshal(jsonData, anyMsg); err != nil {
+		t.Fatalf("Failed to unmarshal Any message: %v", err)
+	}
+
+	// Verify the type URL is set
+	expectedTypeURL := "type.googleapis.com/google.protobuf.StringValue"
+	if anyMsg.TypeUrl != expectedTypeURL {
+		t.Errorf("Expected TypeUrl to be %s, got %s", expectedTypeURL, anyMsg.TypeUrl)
+	}
+
+	// Unmarshal the contained message
+	stringValue := &wrapperspb.StringValue{}
+	if err := anyMsg.UnmarshalTo(stringValue); err != nil {
+		t.Fatalf("Failed to unmarshal contained message: %v", err)
+	}
+
+	// Verify the value
+	expectedValue := "test value"
+	if stringValue.Value != expectedValue {
+		t.Errorf("Expected value to be %s, got %s", expectedValue, stringValue.Value)
+	}
+
+	t.Logf("Successfully unmarshaled Any type from JSON with @type field")
+}

codec/json/codec_test.go

@@ -0,0 +1,98 @@
+package json
+
+import (
+	"bytes"
+	"encoding/json"
+	"testing"
+
+	"go-micro.dev/v6/codec"
+	"google.golang.org/protobuf/types/known/anypb"
+	"google.golang.org/protobuf/types/known/wrapperspb"
+)
+
+// mockReadWriteCloser implements io.ReadWriteCloser for testing
+type mockReadWriteCloser struct {
+	*bytes.Buffer
+}
+
+func (m *mockReadWriteCloser) Close() error {
+	return nil
+}
+
+// TestCodecAnyTypeWrite tests that google.protobuf.Any types are properly written with @type field
+func TestCodecAnyTypeWrite(t *testing.T) {
+	buf := &mockReadWriteCloser{Buffer: bytes.NewBuffer(nil)}
+	c := NewCodec(buf).(*Codec)
+
+	// Create a StringValue message
+	stringValue := wrapperspb.String("test value")
+
+	// Wrap it in an Any message
+	anyMsg, err := anypb.New(stringValue)
+	if err != nil {
+		t.Fatalf("Failed to create Any message: %v", err)
+	}
+
+	// Write the message
+	msg := &codec.Message{
+		Type: codec.Response,
+	}
+	if err := c.Write(msg, anyMsg); err != nil {
+		t.Fatalf("Failed to write Any message: %v", err)
+	}
+
+	// Parse the written JSON
+	var result map[string]interface{}
+	if err := json.Unmarshal(buf.Bytes(), &result); err != nil {
+		t.Fatalf("Failed to unmarshal JSON: %v", err)
+	}
+
+	// Check that @type field exists
+	typeURL, ok := result["@type"].(string)
+	if !ok {
+		t.Fatalf("@type field not found in JSON output. Got: %v", buf.String())
+	}
+
+	// Verify the type URL is correct
+	expectedTypeURL := "type.googleapis.com/google.protobuf.StringValue"
+	if typeURL != expectedTypeURL {
+		t.Errorf("Expected @type to be %s, got %s", expectedTypeURL, typeURL)
+	}
+
+	t.Logf("Successfully wrote Any type with @type field: %s", buf.String())
+}
+
+// TestCodecAnyTypeRead tests that JSON with @type field can be read into google.protobuf.Any
+func TestCodecAnyTypeRead(t *testing.T) {
+	// JSON representation of an Any message with @type field
+	jsonData := `{"@type":"type.googleapis.com/google.protobuf.StringValue","value":"test value"}`
+
+	buf := &mockReadWriteCloser{Buffer: bytes.NewBufferString(jsonData + "\n")}
+	c := NewCodec(buf).(*Codec)
+
+	// Read into an Any message
+	anyMsg := &anypb.Any{}
+	if err := c.ReadBody(anyMsg); err != nil {
+		t.Fatalf("Failed to read Any message: %v", err)
+	}
+
+	// Verify the type URL is set
+	expectedTypeURL := "type.googleapis.com/google.protobuf.StringValue"
+	if anyMsg.TypeUrl != expectedTypeURL {
+		t.Errorf("Expected TypeUrl to be %s, got %s", expectedTypeURL, anyMsg.TypeUrl)
+	}
+
+	// Unmarshal the contained message
+	stringValue := &wrapperspb.StringValue{}
+	if err := anyMsg.UnmarshalTo(stringValue); err != nil {
+		t.Fatalf("Failed to unmarshal contained message: %v", err)
+	}
+
+	// Verify the value
+	expectedValue := "test value"
+	if stringValue.Value != expectedValue {
+		t.Errorf("Expected value to be %s, got %s", expectedValue, stringValue.Value)
+	}
+
+	t.Logf("Successfully read Any type from JSON with @type field")
+}

codec/json/json.go

@@ -0,0 +1,68 @@
+// Package json provides a json codec
+package json
+
+import (
+	"encoding/json"
+	"io"
+
+	"go-micro.dev/v6/codec"
+	"google.golang.org/protobuf/encoding/protojson"
+	"google.golang.org/protobuf/proto"
+)
+
+type Codec struct {
+	Conn    io.ReadWriteCloser
+	Encoder *json.Encoder
+	Decoder *json.Decoder
+}
+
+func (c *Codec) ReadHeader(m *codec.Message, t codec.MessageType) error {
+	return nil
+}
+
+func (c *Codec) ReadBody(b interface{}) error {
+	if b == nil {
+		return nil
+	}
+	if pb, ok := b.(proto.Message); ok {
+		// Read all JSON data from decoder
+		var raw json.RawMessage
+		if err := c.Decoder.Decode(&raw); err != nil {
+			return err
+		}
+		return protojson.Unmarshal(raw, pb)
+	}
+	return c.Decoder.Decode(b)
+}
+
+func (c *Codec) Write(m *codec.Message, b interface{}) error {
+	if b == nil {
+		return nil
+	}
+	if pb, ok := b.(proto.Message); ok {
+		data, err := protojson.Marshal(pb)
+		if err != nil {
+			return err
+		}
+		// Write the marshaled data to the encoder
+		var raw json.RawMessage = data
+		return c.Encoder.Encode(raw)
+	}
+	return c.Encoder.Encode(b)
+}
+
+func (c *Codec) Close() error {
+	return c.Conn.Close()
+}
+
+func (c *Codec) String() string {
+	return "json"
+}
+
+func NewCodec(c io.ReadWriteCloser) codec.Codec {
+	return &Codec{
+		Conn:    c,
+		Decoder: json.NewDecoder(c),
+		Encoder: json.NewEncoder(c),
+	}
+}

codec/json/marshaler.go

@@ -0,0 +1,32 @@
+package json
+
+import (
+	"encoding/json"
+
+	"google.golang.org/protobuf/encoding/protojson"
+	"google.golang.org/protobuf/proto"
+)
+
+var protojsonMarshaler = protojson.MarshalOptions{
+	EmitUnpopulated: false,
+}
+
+type Marshaler struct{}
+
+func (j Marshaler) Marshal(v interface{}) ([]byte, error) {
+	if pb, ok := v.(proto.Message); ok {
+		return protojsonMarshaler.Marshal(pb)
+	}
+	return json.Marshal(v)
+}
+
+func (j Marshaler) Unmarshal(d []byte, v interface{}) error {
+	if pb, ok := v.(proto.Message); ok {
+		return protojson.Unmarshal(d, pb)
+	}
+	return json.Unmarshal(d, v)
+}
+
+func (j Marshaler) String() string {
+	return "json"
+}

codec/jsonrpc/client.go

@@ -0,0 +1,99 @@
+package jsonrpc
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"sync"
+
+	"go-micro.dev/v6/codec"
+)
+
+type clientCodec struct {
+
+	// temporary work space
+	req  clientRequest
+	resp clientResponse
+
+	c io.Closer
+
+	dec     *json.Decoder // for reading JSON values
+	enc     *json.Encoder // for writing JSON values
+	pending map[interface{}]string
+
+	sync.Mutex
+}
+
+type clientRequest struct {
+	Params [1]interface{} `json:"params"`
+	ID     interface{}    `json:"id"`
+	Method string         `json:"method"`
+}
+
+type clientResponse struct {
+	ID     interface{}      `json:"id"`
+	Result *json.RawMessage `json:"result"`
+	Error  interface{}      `json:"error"`
+}
+
+func newClientCodec(conn io.ReadWriteCloser) *clientCodec {
+	return &clientCodec{
+		dec:     json.NewDecoder(conn),
+		enc:     json.NewEncoder(conn),
+		c:       conn,
+		pending: make(map[interface{}]string),
+	}
+}
+
+func (c *clientCodec) Write(m *codec.Message, b interface{}) error {
+	c.Lock()
+	c.pending[m.Id] = m.Method
+	c.Unlock()
+	c.req.Method = m.Method
+	c.req.Params[0] = b
+	c.req.ID = m.Id
+	return c.enc.Encode(&c.req)
+}
+
+func (r *clientResponse) reset() {
+	r.ID = 0
+	r.Result = nil
+	r.Error = nil
+}
+
+func (c *clientCodec) ReadHeader(m *codec.Message) error {
+	c.resp.reset()
+	if err := c.dec.Decode(&c.resp); err != nil {
+		return err
+	}
+
+	c.Lock()
+	m.Method = c.pending[c.resp.ID]
+	delete(c.pending, c.resp.ID)
+	c.Unlock()
+
+	m.Error = ""
+	m.Id = fmt.Sprintf("%v", c.resp.ID)
+	if c.resp.Error != nil {
+		x, ok := c.resp.Error.(string)
+		if !ok {
+			return fmt.Errorf("invalid error %v", c.resp.Error)
+		}
+		if x == "" {
+			x = "unspecified error"
+		}
+		m.Error = x
+	}
+	return nil
+}
+
+func (c *clientCodec) ReadBody(x interface{}) error {
+	if x == nil || c.resp.Result == nil {
+		return nil
+	}
+	return json.Unmarshal(*c.resp.Result, x)
+}
+
+func (c *clientCodec) Close() error {
+	return c.c.Close()
+}

codec/jsonrpc/jsonrpc.go

@@ -0,0 +1,88 @@
+// Package jsonrpc provides a json-rpc 1.0 codec
+package jsonrpc
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+
+	"go-micro.dev/v6/codec"
+)
+
+type jsonCodec struct {
+	rwc io.ReadWriteCloser
+	buf *bytes.Buffer
+	c   *clientCodec
+	s   *serverCodec
+	mt  codec.MessageType
+}
+
+func (j *jsonCodec) Close() error {
+	j.buf.Reset()
+	return j.rwc.Close()
+}
+
+func (j *jsonCodec) String() string {
+	return "json-rpc"
+}
+
+func (j *jsonCodec) Write(m *codec.Message, b interface{}) error {
+	switch m.Type {
+	case codec.Request:
+		return j.c.Write(m, b)
+	case codec.Response, codec.Error:
+		return j.s.Write(m, b)
+	case codec.Event:
+		data, err := json.Marshal(b)
+		if err != nil {
+			return err
+		}
+		_, err = j.rwc.Write(data)
+		return err
+	default:
+		return fmt.Errorf("unrecognized message type: %v", m.Type)
+	}
+}
+
+func (j *jsonCodec) ReadHeader(m *codec.Message, mt codec.MessageType) error {
+	j.buf.Reset()
+	j.mt = mt
+
+	switch mt {
+	case codec.Request:
+		return j.s.ReadHeader(m)
+	case codec.Response:
+		return j.c.ReadHeader(m)
+	case codec.Event:
+		_, err := io.Copy(j.buf, j.rwc)
+		return err
+	default:
+		return fmt.Errorf("unrecognized message type: %v", mt)
+	}
+}
+
+func (j *jsonCodec) ReadBody(b interface{}) error {
+	switch j.mt {
+	case codec.Request:
+		return j.s.ReadBody(b)
+	case codec.Response:
+		return j.c.ReadBody(b)
+	case codec.Event:
+		if b != nil {
+			return json.Unmarshal(j.buf.Bytes(), b)
+		}
+	default:
+		return fmt.Errorf("unrecognized message type: %v", j.mt)
+	}
+	return nil
+}
+
+func NewCodec(rwc io.ReadWriteCloser) codec.Codec {
+	return &jsonCodec{
+		buf: bytes.NewBuffer(nil),
+		rwc: rwc,
+		c:   newClientCodec(rwc),
+		s:   newServerCodec(rwc),
+	}
+}

codec/jsonrpc/server.go

@@ -0,0 +1,84 @@
+package jsonrpc
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+
+	"go-micro.dev/v6/codec"
+)
+
+type serverCodec struct {
+	dec *json.Decoder // for reading JSON values
+	enc *json.Encoder // for writing JSON values
+	c   io.Closer
+
+	// temporary work space
+	req serverRequest
+}
+
+type serverRequest struct {
+	ID     interface{}      `json:"id"`
+	Params *json.RawMessage `json:"params"`
+	Method string           `json:"method"`
+}
+
+type serverResponse struct {
+	ID     interface{} `json:"id"`
+	Result interface{} `json:"result"`
+	Error  interface{} `json:"error"`
+}
+
+func newServerCodec(conn io.ReadWriteCloser) *serverCodec {
+	return &serverCodec{
+		dec: json.NewDecoder(conn),
+		enc: json.NewEncoder(conn),
+		c:   conn,
+	}
+}
+
+func (r *serverRequest) reset() {
+	r.Method = ""
+	if r.Params != nil {
+		*r.Params = (*r.Params)[0:0]
+	}
+	if r.ID != nil {
+		r.ID = nil
+	}
+}
+
+func (c *serverCodec) ReadHeader(m *codec.Message) error {
+	c.req.reset()
+	if err := c.dec.Decode(&c.req); err != nil {
+		return err
+	}
+	m.Method = c.req.Method
+	m.Id = fmt.Sprintf("%v", c.req.ID)
+	c.req.ID = nil
+	return nil
+}
+
+func (c *serverCodec) ReadBody(x interface{}) error {
+	if x == nil {
+		return nil
+	}
+	var params [1]interface{}
+	params[0] = x
+	return json.Unmarshal(*c.req.Params, &params)
+}
+
+func (c *serverCodec) Write(m *codec.Message, x interface{}) error {
+	var resp serverResponse
+	resp.ID = m.Id
+	resp.Result = x
+	if m.Error == "" {
+		resp.Error = nil
+	} else {
+		resp.Error = m.Error
+	}
+	return c.enc.Encode(resp)
+}
+
+func (c *serverCodec) Close() error {
+	return c.c.Close()
+}

codec/proto/marshaler.go

@@ -0,0 +1,47 @@
+package proto
+
+import (
+	"bytes"
+
+	"github.com/golang/protobuf/proto"
+	"github.com/oxtoacart/bpool"
+	"go-micro.dev/v6/codec"
+)
+
+// create buffer pool with 16 instances each preallocated with 256 bytes.
+var bufferPool = bpool.NewSizedBufferPool(16, 256)
+
+type Marshaler struct{}
+
+func (Marshaler) Marshal(v interface{}) ([]byte, error) {
+	pb, ok := v.(proto.Message)
+	if !ok {
+		return nil, codec.ErrInvalidMessage
+	}
+
+	// looks not good, but allows to reuse underlining bytes
+	buf := bufferPool.Get()
+	pbuf := proto.NewBuffer(buf.Bytes())
+	defer func() {
+		bufferPool.Put(bytes.NewBuffer(pbuf.Bytes()))
+	}()
+
+	if err := pbuf.Marshal(pb); err != nil {
+		return nil, err
+	}
+
+	return pbuf.Bytes(), nil
+}
+
+func (Marshaler) Unmarshal(data []byte, v interface{}) error {
+	pb, ok := v.(proto.Message)
+	if !ok {
+		return codec.ErrInvalidMessage
+	}
+
+	return proto.Unmarshal(data, pb)
+}
+
+func (Marshaler) String() string {
+	return "proto"
+}

codec/proto/message.go

@@ -0,0 +1,37 @@
+package proto
+
+type Message struct {
+	Data []byte
+}
+
+func (m *Message) MarshalJSON() ([]byte, error) {
+	return m.Data, nil
+}
+
+func (m *Message) UnmarshalJSON(data []byte) error {
+	m.Data = data
+	return nil
+}
+
+func (m *Message) ProtoMessage() {}
+
+func (m *Message) Reset() {
+	*m = Message{}
+}
+
+func (m *Message) String() string {
+	return string(m.Data)
+}
+
+func (m *Message) Marshal() ([]byte, error) {
+	return m.Data, nil
+}
+
+func (m *Message) Unmarshal(data []byte) error {
+	m.Data = data
+	return nil
+}
+
+func NewMessage(data []byte) *Message {
+	return &Message{data}
+}

codec/proto/proto.go

@@ -0,0 +1,63 @@
+// Package proto provides a proto codec
+package proto
+
+import (
+	"io"
+
+	"github.com/golang/protobuf/proto"
+	"go-micro.dev/v6/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 {
+	if b == nil {
+		return nil
+	}
+	buf, err := io.ReadAll(c.Conn)
+	if err != nil {
+		return err
+	}
+	m, ok := b.(proto.Message)
+	if !ok {
+		return codec.ErrInvalidMessage
+	}
+	return proto.Unmarshal(buf, m)
+}
+
+func (c *Codec) Write(m *codec.Message, b interface{}) error {
+	if b == nil {
+		// Nothing to write
+		return nil
+	}
+	p, ok := b.(proto.Message)
+	if !ok {
+		return codec.ErrInvalidMessage
+	}
+	buf, err := proto.Marshal(p)
+	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,
+	}
+}

codec/protorpc/envelope.pb.go

@@ -0,0 +1,144 @@
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// source: codec/protorpc/envelope.proto
+
+package protorpc
+
+import (
+	fmt "fmt"
+	proto "github.com/golang/protobuf/proto"
+	math "math"
+)
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = fmt.Errorf
+var _ = math.Inf
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the proto package it is being compiled against.
+// A compilation error at this line likely means your copy of the
+// proto package needs to be updated.
+const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
+
+type Request struct {
+	ServiceMethod        string   `protobuf:"bytes,1,opt,name=service_method,json=serviceMethod,proto3" json:"service_method,omitempty"`
+	Seq                  uint64   `protobuf:"fixed64,2,opt,name=seq,proto3" json:"seq,omitempty"`
+	XXX_NoUnkeyedLiteral struct{} `json:"-"`
+	XXX_unrecognized     []byte   `json:"-"`
+	XXX_sizecache        int32    `json:"-"`
+}
+
+func (m *Request) Reset()         { *m = Request{} }
+func (m *Request) String() string { return proto.CompactTextString(m) }
+func (*Request) ProtoMessage()    {}
+func (*Request) Descriptor() ([]byte, []int) {
+	return fileDescriptor_12fd17ed7ee86a33, []int{0}
+}
+
+func (m *Request) XXX_Unmarshal(b []byte) error {
+	return xxx_messageInfo_Request.Unmarshal(m, b)
+}
+func (m *Request) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
+	return xxx_messageInfo_Request.Marshal(b, m, deterministic)
+}
+func (m *Request) XXX_Merge(src proto.Message) {
+	xxx_messageInfo_Request.Merge(m, src)
+}
+func (m *Request) XXX_Size() int {
+	return xxx_messageInfo_Request.Size(m)
+}
+func (m *Request) XXX_DiscardUnknown() {
+	xxx_messageInfo_Request.DiscardUnknown(m)
+}
+
+var xxx_messageInfo_Request proto.InternalMessageInfo
+
+func (m *Request) GetServiceMethod() string {
+	if m != nil {
+		return m.ServiceMethod
+	}
+	return ""
+}
+
+func (m *Request) GetSeq() uint64 {
+	if m != nil {
+		return m.Seq
+	}
+	return 0
+}
+
+type Response struct {
+	ServiceMethod        string   `protobuf:"bytes,1,opt,name=service_method,json=serviceMethod,proto3" json:"service_method,omitempty"`
+	Seq                  uint64   `protobuf:"fixed64,2,opt,name=seq,proto3" json:"seq,omitempty"`
+	Error                string   `protobuf:"bytes,3,opt,name=error,proto3" json:"error,omitempty"`
+	XXX_NoUnkeyedLiteral struct{} `json:"-"`
+	XXX_unrecognized     []byte   `json:"-"`
+	XXX_sizecache        int32    `json:"-"`
+}
+
+func (m *Response) Reset()         { *m = Response{} }
+func (m *Response) String() string { return proto.CompactTextString(m) }
+func (*Response) ProtoMessage()    {}
+func (*Response) Descriptor() ([]byte, []int) {
+	return fileDescriptor_12fd17ed7ee86a33, []int{1}
+}
+
+func (m *Response) XXX_Unmarshal(b []byte) error {
+	return xxx_messageInfo_Response.Unmarshal(m, b)
+}
+func (m *Response) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
+	return xxx_messageInfo_Response.Marshal(b, m, deterministic)
+}
+func (m *Response) XXX_Merge(src proto.Message) {
+	xxx_messageInfo_Response.Merge(m, src)
+}
+func (m *Response) XXX_Size() int {
+	return xxx_messageInfo_Response.Size(m)
+}
+func (m *Response) XXX_DiscardUnknown() {
+	xxx_messageInfo_Response.DiscardUnknown(m)
+}
+
+var xxx_messageInfo_Response proto.InternalMessageInfo
+
+func (m *Response) GetServiceMethod() string {
+	if m != nil {
+		return m.ServiceMethod
+	}
+	return ""
+}
+
+func (m *Response) GetSeq() uint64 {
+	if m != nil {
+		return m.Seq
+	}
+	return 0
+}
+
+func (m *Response) GetError() string {
+	if m != nil {
+		return m.Error
+	}
+	return ""
+}
+
+func init() {
+	proto.RegisterType((*Request)(nil), "protorpc.Request")
+	proto.RegisterType((*Response)(nil), "protorpc.Response")
+}
+
+func init() { proto.RegisterFile("codec/protorpc/envelope.proto", fileDescriptor_12fd17ed7ee86a33) }
+
+var fileDescriptor_12fd17ed7ee86a33 = []byte{
+	// 148 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xe2, 0x92, 0x4d, 0xce, 0x4f, 0x49,
+	0x4d, 0xd6, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x2f, 0x2a, 0x48, 0xd6, 0x4f, 0xcd, 0x2b, 0x4b, 0xcd,
+	0xc9, 0x2f, 0x48, 0xd5, 0x03, 0x8b, 0x08, 0x71, 0xc0, 0x24, 0x94, 0x9c, 0xb8, 0xd8, 0x83, 0x52,
+	0x0b, 0x4b, 0x53, 0x8b, 0x4b, 0x84, 0x54, 0xb9, 0xf8, 0x8a, 0x53, 0x8b, 0xca, 0x32, 0x93, 0x53,
+	0xe3, 0x73, 0x53, 0x4b, 0x32, 0xf2, 0x53, 0x24, 0x18, 0x15, 0x18, 0x35, 0x38, 0x83, 0x78, 0xa1,
+	0xa2, 0xbe, 0x60, 0x41, 0x21, 0x01, 0x2e, 0xe6, 0xe2, 0xd4, 0x42, 0x09, 0x26, 0x05, 0x46, 0x0d,
+	0xb6, 0x20, 0x10, 0x53, 0x29, 0x92, 0x8b, 0x23, 0x28, 0xb5, 0xb8, 0x20, 0x3f, 0xaf, 0x38, 0x95,
+	0x6c, 0x43, 0x84, 0x44, 0xb8, 0x58, 0x53, 0x8b, 0x8a, 0xf2, 0x8b, 0x24, 0x98, 0xc1, 0xea, 0x21,
+	0x9c, 0x24, 0x36, 0xb0, 0x43, 0x8d, 0x01, 0x01, 0x00, 0x00, 0xff, 0xff, 0xe4, 0x73, 0x3a, 0x4b,
+	0xd0, 0x00, 0x00, 0x00,
+}

codec/protorpc/envelope.pb.micro.go

@@ -0,0 +1,21 @@
+// Code generated by protoc-gen-micro. DO NOT EDIT.
+// source: codec/protorpc/envelope.proto
+
+package protorpc
+
+import (
+	fmt "fmt"
+	proto "github.com/golang/protobuf/proto"
+	math "math"
+)
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = fmt.Errorf
+var _ = math.Inf
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the proto package it is being compiled against.
+// A compilation error at this line likely means your copy of the
+// proto package needs to be updated.
+const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package

codec/protorpc/envelope.proto

@@ -0,0 +1,14 @@
+syntax = "proto3";
+
+package protorpc;
+
+message Request {
+	string service_method = 1;
+	fixed64 seq = 2;
+}
+
+message Response {
+	string service_method = 1;
+	fixed64 seq = 2;
+	string error = 3;
+}

codec/protorpc/netstring.go

@@ -0,0 +1,36 @@
+package protorpc
+
+import (
+	"encoding/binary"
+	"io"
+)
+
+// WriteNetString writes data to a big-endian netstring on a Writer.
+// Size is always a 32-bit unsigned int.
+func WriteNetString(w io.Writer, data []byte) (written int, err error) {
+	size := make([]byte, 4)
+	binary.BigEndian.PutUint32(size, uint32(len(data)))
+	if written, err = w.Write(size); err != nil {
+		return
+	}
+	return w.Write(data)
+}
+
+// ReadNetString reads data from a big-endian netstring.
+func ReadNetString(r io.Reader) (data []byte, err error) {
+	sizeBuf := make([]byte, 4)
+	_, err = r.Read(sizeBuf)
+	if err != nil {
+		return nil, err
+	}
+	size := binary.BigEndian.Uint32(sizeBuf)
+	if size == 0 {
+		return nil, nil
+	}
+	data = make([]byte, size)
+	_, err = r.Read(data)
+	if err != nil {
+		return nil, err
+	}
+	return
+}

codec/protorpc/protorpc.go

@@ -0,0 +1,186 @@
+// Protorpc provides a net/rpc proto-rpc codec. See envelope.proto for the format.
+package protorpc
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"strconv"
+	"sync"
+
+	"github.com/golang/protobuf/proto"
+	"go-micro.dev/v6/codec"
+)
+
+type flusher interface {
+	Flush() error
+}
+
+type protoCodec struct {
+	rwc io.ReadWriteCloser
+	buf *bytes.Buffer
+	mt  codec.MessageType
+	sync.Mutex
+}
+
+func (c *protoCodec) Close() error {
+	c.buf.Reset()
+	return c.rwc.Close()
+}
+
+func (c *protoCodec) String() string {
+	return "proto-rpc"
+}
+
+func id(id string) uint64 {
+	p, err := strconv.ParseInt(id, 10, 64)
+	if err != nil {
+		p = 0
+	}
+	i := uint64(p)
+	return i
+}
+
+func (c *protoCodec) Write(m *codec.Message, b interface{}) error {
+	switch m.Type {
+	case codec.Request:
+		c.Lock()
+		defer c.Unlock()
+		// This is protobuf, of course we copy it.
+		pbr := &Request{ServiceMethod: m.Method, Seq: id(m.Id)}
+		data, err := proto.Marshal(pbr)
+		if err != nil {
+			return err
+		}
+		_, err = WriteNetString(c.rwc, data)
+		if err != nil {
+			return err
+		}
+		// dont trust or incoming message
+		m, ok := b.(proto.Message)
+		if !ok {
+			return codec.ErrInvalidMessage
+		}
+		data, err = proto.Marshal(m)
+		if err != nil {
+			return err
+		}
+		_, err = WriteNetString(c.rwc, data)
+		if err != nil {
+			return err
+		}
+		if flusher, ok := c.rwc.(flusher); ok {
+			if err = flusher.Flush(); err != nil {
+				return err
+			}
+		}
+	case codec.Response, codec.Error:
+		c.Lock()
+		defer c.Unlock()
+		rtmp := &Response{ServiceMethod: m.Method, Seq: id(m.Id), Error: m.Error}
+		data, err := proto.Marshal(rtmp)
+		if err != nil {
+			return err
+		}
+		_, err = WriteNetString(c.rwc, data)
+		if err != nil {
+			return err
+		}
+		if pb, ok := b.(proto.Message); ok {
+			data, err = proto.Marshal(pb)
+			if err != nil {
+				return err
+			}
+		} else {
+			data = nil
+		}
+		_, err = WriteNetString(c.rwc, data)
+		if err != nil {
+			return err
+		}
+		if flusher, ok := c.rwc.(flusher); ok {
+			if err = flusher.Flush(); err != nil {
+				return err
+			}
+		}
+	case codec.Event:
+		m, ok := b.(proto.Message)
+		if !ok {
+			return codec.ErrInvalidMessage
+		}
+		data, err := proto.Marshal(m)
+		if err != nil {
+			return err
+		}
+		_, _ = c.rwc.Write(data)
+	default:
+		return fmt.Errorf("unrecognized message type: %v", m.Type)
+	}
+	return nil
+}
+
+func (c *protoCodec) ReadHeader(m *codec.Message, mt codec.MessageType) error {
+	c.buf.Reset()
+	c.mt = mt
+
+	switch mt {
+	case codec.Request:
+		data, err := ReadNetString(c.rwc)
+		if err != nil {
+			return err
+		}
+		rtmp := new(Request)
+		err = proto.Unmarshal(data, rtmp)
+		if err != nil {
+			return err
+		}
+		m.Method = rtmp.GetServiceMethod()
+		m.Id = fmt.Sprintf("%d", rtmp.GetSeq())
+	case codec.Response:
+		data, err := ReadNetString(c.rwc)
+		if err != nil {
+			return err
+		}
+		rtmp := new(Response)
+		err = proto.Unmarshal(data, rtmp)
+		if err != nil {
+			return err
+		}
+		m.Method = rtmp.GetServiceMethod()
+		m.Id = fmt.Sprintf("%d", rtmp.GetSeq())
+		m.Error = rtmp.GetError()
+	case codec.Event:
+		_, err := io.Copy(c.buf, c.rwc)
+		return err
+	default:
+		return fmt.Errorf("unrecognized message type: %v", mt)
+	}
+	return nil
+}
+
+func (c *protoCodec) ReadBody(b interface{}) error {
+	var data []byte
+	switch c.mt {
+	case codec.Request, codec.Response:
+		var err error
+		data, err = ReadNetString(c.rwc)
+		if err != nil {
+			return err
+		}
+	case codec.Event:
+		data = c.buf.Bytes()
+	default:
+		return fmt.Errorf("unrecognized message type: %v", c.mt)
+	}
+	if b != nil {
+		return proto.Unmarshal(data, b.(proto.Message))
+	}
+	return nil
+}
+
+func NewCodec(rwc io.ReadWriteCloser) codec.Codec {
+	return &protoCodec{
+		buf: bytes.NewBuffer(nil),
+		rwc: rwc,
+	}
+}

codec/text/text.go

@@ -0,0 +1,77 @@
+// Package text reads any text/* content-type
+package text
+
+import (
+	"fmt"
+	"io"
+
+	"go-micro.dev/v6/codec"
+)
+
+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 := io.ReadAll(c.Conn)
+	if err != nil {
+		return err
+	}
+
+	switch v := b.(type) {
+	case *string:
+		*v = string(buf)
+	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
+	switch ve := b.(type) {
+	case *Frame:
+		v = ve.Data
+	case *[]byte:
+		v = *ve
+	case *string:
+		v = []byte(*ve)
+	case string:
+		v = []byte(ve)
+	case []byte:
+		v = ve
+	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 {
+	return "text"
+}
+
+func NewCodec(c io.ReadWriteCloser) codec.Codec {
+	return &Codec{
+		Conn: c,
+	}
+}