request.go

package grpc

import (
	"github.com/gojek/fiber"
	"github.com/gojek/fiber/protocol"
	"google.golang.org/grpc/metadata"
	"google.golang.org/protobuf/proto"
)

type Request struct {
	// Metadata will hold the grpc headers for request
	Metadata metadata.MD
	Message  []byte
	Proto    proto.Message
}

func NewRequest(metadata metadata.MD, msg []byte, protoMsg proto.Message) *Request {
	return &Request{
		Metadata: metadata,
		Message:  msg,
		Proto:    protoMsg,
	}
}

func (r *Request) Protocol() protocol.Protocol {
	return protocol.GRPC
}

func (r *Request) Payload() []byte {
	return r.Message
}

func (r *Request) Header() map[string][]string {
	return r.Metadata
}

func (r *Request) Clone() (fiber.Request, error) {
	// for grpc, we'll just return itself (no cloning)
	return r, nil
}

// OperationName is naming used in tracing interceptors
func (r *Request) OperationName() string {
	// For grpc implementation, serviceMethod and endpoint is init with dispatcher
	return "grpc"
}

// Transform is use by backend component within a Proxy to abstract endpoint from dispatcher
func (r *Request) Transform(_ fiber.Backend) (fiber.Request, error) {
	// For grpc implementation, endpoint is init with dispatcher
	return r, nil
}

func (r *Request) ProtoMessage() proto.Message {
	return r.Proto
}