github.com/annwntech/go-micro/v2@v2.9.5/server/grpc/response.go (about)

     1  package grpc
     2  
     3  import (
     4  	"github.com/annwntech/go-micro/v2/codec"
     5  )
     6  
     7  type rpcResponse struct {
     8  	header map[string]string
     9  	codec  codec.Codec
    10  }
    11  
    12  func (r *rpcResponse) Codec() codec.Writer {
    13  	return r.codec
    14  }
    15  
    16  func (r *rpcResponse) WriteHeader(hdr map[string]string) {
    17  	for k, v := range hdr {
    18  		r.header[k] = v
    19  	}
    20  }
    21  
    22  func (r *rpcResponse) Write(b []byte) error {
    23  	return r.codec.Write(&codec.Message{
    24  		Header: r.header,
    25  		Body:   b,
    26  	}, nil)
    27  }