78 lines
1.2 KiB
Go
78 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/micro/go-micro/codec"
|
|
"github.com/micro/go-micro/transport"
|
|
)
|
|
|
|
type rpcRequest struct {
|
|
service string
|
|
endpoint string
|
|
contentType string
|
|
socket transport.Socket
|
|
codec codec.Codec
|
|
header map[string]string
|
|
body []byte
|
|
stream bool
|
|
}
|
|
|
|
type rpcMessage struct {
|
|
topic string
|
|
contentType string
|
|
payload interface{}
|
|
}
|
|
|
|
func (r *rpcRequest) Codec() codec.Reader {
|
|
return r.codec
|
|
}
|
|
|
|
func (r *rpcRequest) ContentType() string {
|
|
return r.contentType
|
|
}
|
|
|
|
func (r *rpcRequest) Service() string {
|
|
return r.service
|
|
}
|
|
|
|
func (r *rpcRequest) Endpoint() string {
|
|
return r.endpoint
|
|
}
|
|
|
|
func (r *rpcRequest) Header() map[string]string {
|
|
return r.header
|
|
}
|
|
|
|
func (r *rpcRequest) Read() ([]byte, error) {
|
|
// got a body
|
|
if r.body != nil {
|
|
b := r.body
|
|
r.body = nil
|
|
return b, nil
|
|
}
|
|
|
|
var msg transport.Message
|
|
err := r.socket.Recv(&msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.header = msg.Header
|
|
|
|
return msg.Body, nil
|
|
}
|
|
|
|
func (r *rpcRequest) Stream() bool {
|
|
return r.stream
|
|
}
|
|
|
|
func (r *rpcMessage) ContentType() string {
|
|
return r.contentType
|
|
}
|
|
|
|
func (r *rpcMessage) Topic() string {
|
|
return r.topic
|
|
}
|
|
|
|
func (r *rpcMessage) Payload() interface{} {
|
|
return r.payload
|
|
}
|