rpcinfo

type ClientStreamEventHandler struct {
	HandleStreamStartEvent      func(ctx context.Context, ri RPCInfo, evt StreamStartEvent)
	HandleStreamRecvHeaderEvent func(ctx context.Context, ri RPCInfo, evt StreamRecvHeaderEvent)
	HandleStreamRecvEvent       func(ctx context.Context, ri RPCInfo, evt StreamRecvEvent)
	HandleStreamSendEvent       func(ctx context.Context, ri RPCInfo, evt StreamSendEvent)
	HandleStreamFinishEvent     func(ctx context.Context, ri RPCInfo, evt StreamFinishEvent)
}

type EndpointBasicInfo struct {
	ServiceName string
	Method      string
	Tags        map[string]string
}

type EndpointInfo interface {
	ServiceName() string
	Method() string
	Address() net.Addr
	Tag(key string) (value string, exist bool)
	DefaultTag(key, def string) string
}

type Event interface {
	Event() stats.Event
	Status() stats.Status
	Info() string
	Time() time.Time
	IsNil() bool
}

type InteractionMode int32

type Invocation interface {
	PackageName() string
	ServiceName() string
	MethodName() string
	MethodInfo() serviceinfo.MethodInfo
	StreamingMode() serviceinfo.StreamingMode
	SeqID() int32
	BizStatusErr() kerrors.BizStatusErrorIface
	Extra(key string) interface{}
}

type InvocationSetter interface {
	SetPackageName(name string)
	SetServiceName(name string)
	SetMethodName(name string)
	SetMethodInfo(methodInfo serviceinfo.MethodInfo)
	SetStreamingMode(mode serviceinfo.StreamingMode)
	SetSeqID(seqID int32)
	SetBizStatusErr(err kerrors.BizStatusErrorIface)
	SetExtra(key string, value interface{})
	Reset()
}

type MutableEndpointInfo interface {
	SetServiceName(service string) error
	SetMethod(method string) error
	SetAddress(addr net.Addr) error
	SetTag(key, value string) error
	ImmutableView() EndpointInfo
	Reset()
	ResetFromBasicInfo(bi *EndpointBasicInfo)
}

type MutableRPCConfig interface {
	SetRPCTimeout(to time.Duration) error
	IsRPCTimeoutLocked() bool
	SetConnectTimeout(to time.Duration) error
	IsConnectTimeoutLocked() bool
	SetReadWriteTimeout(to time.Duration) error
	IsReadWriteTimeoutLocked() bool
	SetIOBufferSize(sz int) error
	SetTransportProtocol(tp transport.Protocol) error
	SetInteractionMode(mode InteractionMode) error
	LockConfig(bits int)
	Clone() MutableRPCConfig
	CopyFrom(from RPCConfig)
	ImmutableView() RPCConfig
	SetPayloadCodec(codec serviceinfo.PayloadCodec)

	SetStreamRecvTimeout(timeout time.Duration)
	SetStreamRecvTimeoutConfig(cfg streaming.TimeoutConfig)
}

type MutableRPCStats interface {
	SetSendSize(size uint64)
	SetRecvSize(size uint64)
	SetError(err error)
	SetPanicked(x interface{})
	SetLevel(level stats.Level)
	Reset()
	ImmutableView() RPCStats
	IncrSendSize(size uint64)
	IncrRecvSize(size uint64)
}

type RPCConfig interface {
	Timeouts
	StreamConfig
	IOBufferSize() int
	TransportProtocol() transport.Protocol
	InteractionMode() InteractionMode
	PayloadCodec() serviceinfo.PayloadCodec
}

type RPCInfo interface {
	From() EndpointInfo
	To() EndpointInfo
	Invocation() Invocation
	Config() RPCConfig
	Stats() RPCStats
}

type RPCStats interface {
	Record(ctx context.Context, event stats.Event, status stats.Status, info string)
	SendSize() uint64
	// LastSendSize returns the size of the last sent message in a stream.
	LastSendSize() uint64
	RecvSize() uint64
	// LastRecvSize returns the size of the last received message in a stream.
	LastRecvSize() uint64
	Error() error
	Panicked() (bool, interface{})
	GetEvent(event stats.Event) Event
	Level() stats.Level
	CopyForRetry() RPCStats
}

type ServerStreamEventHandler struct {
	HandleStreamStartEvent  func(ctx context.Context, ri RPCInfo, evt StreamStartEvent)
	HandleStreamRecvEvent   func(ctx context.Context, ri RPCInfo, evt StreamRecvEvent)
	HandleStreamSendEvent   func(ctx context.Context, ri RPCInfo, evt StreamSendEvent)
	HandleStreamFinishEvent func(ctx context.Context, ri RPCInfo, evt StreamFinishEvent)
}

type StreamConfig interface {
	StreamRecvTimeout() time.Duration
	StreamRecvTimeoutConfig() streaming.TimeoutConfig
}

type StreamEventReporter interface {
	// ReportStreamEvent is for collecting Recv/Send events on stream
	// NOTE: The callee should NOT hold references to event, which may be recycled later
	ReportStreamEvent(ctx context.Context, ri RPCInfo, event Event)
}

type StreamFinishEvent struct {
	GRPCTrailer     map[string][]string
	TTStreamTrailer map[string]string
}

type StreamRecvEvent struct {
	Time time.Time
	Err  error
}

type StreamRecvHeaderEvent struct {
	GRPCHeader     map[string][]string
	TTStreamHeader map[string]string
}

type StreamSendEvent struct {
	Time time.Time
	Err  error
}

type StreamStartEvent struct{}

type Taggable interface {
	SetTag(key, value string) error
}

type TimeoutProvider interface {
	Timeouts(ri RPCInfo) Timeouts
}

type Timeouts interface {
	RPCTimeout() time.Duration
	ConnectTimeout() time.Duration
	ReadWriteTimeout() time.Duration
}

type TraceController struct {
	// contains filtered or unexported fields
}