drpc

type AfterAcceptPlugin interface {
	Plugin
	AfterAccept(EarlySession) *Status
}

type AfterCloseEndpointPlugin interface {
	Plugin
	AfterCloseEndpoint(Endpoint, error) error
}

type AfterDialFailPlugin interface {
	Plugin
	AfterDialFail(sess EarlySession, err error, isRedial bool) *Status
}

type AfterDialPlugin interface {
	Plugin
	AfterDial(sess EarlySession, isRedial bool) *Status
}

type AfterDisconnectPlugin interface {
	Plugin
	AfterDisconnect(BaseSession) *Status
}

type AfterListenPlugin interface {
	Plugin
	AfterListen(net.Addr) error
}

type AfterNewEndpointPlugin interface {
	Plugin
	AfterNewEndpoint(EarlyEndpoint) error
}

type AfterReadCallBodyPlugin interface {
	Plugin
	AfterReadCallBody(ReadCtx) *Status
}

type AfterReadCallHeaderPlugin interface {
	Plugin
	AfterReadCallHeader(ReadCtx) *Status
}

type AfterReadPushBodyPlugin interface {
	Plugin
	AfterReadPushBody(ReadCtx) *Status
}

type AfterReadPushHeaderPlugin interface {
	Plugin
	AfterReadPushHeader(ReadCtx) *Status
}

type AfterReadReplyBodyPlugin interface {
	Plugin
	AfterReadReplyBody(ReadCtx) *Status
}

type AfterReadReplyHeaderPlugin interface {
	Plugin
	AfterReadReplyHeader(ReadCtx) *Status
}

type AfterRegRouterPlugin interface {
	Plugin
	AfterRegRouter(*Handler) error
}

type AfterWriteCallPlugin interface {
	Plugin
	AfterWriteCall(WriteCtx) *Status
}

type AfterWritePushPlugin interface {
	Plugin
	AfterWritePush(WriteCtx) *Status
}

type AfterWriteReplyPlugin interface {
	Plugin
	AfterWriteReply(WriteCtx) *Status
}

type BaseEndpoint interface {

	// Close 关闭该端点
	Close() (err error)

	// CountSession 统计该端点上的session数量
	CountSession() int

	// GetSession 获取指定ID的 session
	GetSession(sessionID string) (Session, bool)

	// RangeSession 循环迭代session
	RangeSession(fn func(sess Session) bool)

	// SetTLSConfig 设置证书配置
	SetTLSConfig(tlsConfig *tls.Config)

	// SetTLSConfigFromFile 从文件中读取证书并设置证书配置
	SetTLSConfigFromFile(tlsCertFile, tlsKeyFile string, insecureSkipVerifyForClient ...bool) error

	// TLSConfig tls配置对象
	TLSConfig() *tls.Config

	// PluginContainer 插件容器对象
	PluginContainer() *PluginContainer
}

type BaseSession interface {
	Endpoint() Endpoint

	// ID 获取id
	ID() string

	// LocalAddr 本地地址
	LocalAddr() net.Addr

	// RemoteAddr 远端地址
	RemoteAddr() net.Addr

	// Swap 返回交换区的内容
	Swap() *gmap.Map
}

type BeforeCloseEndpointPlugin interface {
	Plugin
	BeforeCloseEndpoint(Endpoint) error
}

type BeforeDialPlugin interface {
	Plugin
	BeforeDial(addr string, isRedial bool) *Status
}

type BeforeNewEndpointPlugin interface {
	Plugin
	BeforeNewEndpoint(*EndpointConfig, *PluginContainer) error
}

type BeforeReadCallBodyPlugin interface {
	Plugin
	BeforeReadCallBody(ReadCtx) *Status
}

type BeforeReadHeaderPlugin interface {
	Plugin
	BeforeReadHeader(EarlyCtx) error
}

type BeforeReadPushBodyPlugin interface {
	Plugin
	BeforeReadPushBody(ReadCtx) *Status
}

type BeforeReadReplyBodyPlugin interface {
	Plugin
	BeforeReadReplyBody(ReadCtx) *Status
}

type BeforeWriteCallPlugin interface {
	Plugin
	BeforeWriteCall(WriteCtx) *Status
}

type BeforeWritePushPlugin interface {
	Plugin
	BeforeWritePush(WriteCtx) *Status
}

type BeforeWriteReplyPlugin interface {
	Plugin
	BeforeWriteReply(WriteCtx) *Status
}

type CallCmd interface {
	TraceEndpoint() (e Endpoint, found bool)
	TraceSession() (sess Session, found bool)

	// Context 协程上下文
	Context() context.Context
	// Output 发送的消息
	Output() message.Message

	// StatusOK 状态是否是OK
	StatusOK() bool
	// Status 状态
	Status() *Status
	// Done 返回指示是否已经完毕的chan
	Done() <-chan struct{}
	// Reply 返回应答
	Reply() (interface{}, *Status)

	// InputBodyCodec 接收到的消息使用的编码
	InputBodyCodec() byte
	// InputMeta 接收到的消息传入的元数据
	InputMeta() *gmap.Map
	// CostTime 消耗的时间
	CostTime() time.Duration
}

type CallCtx interface {

	// Input 获取传入的消息
	Input() message.Message

	// GetBodyCodec 获取当前消息的编码格式
	GetBodyCodec() byte

	// Output 将要发送的消息对象
	Output() message.Message

	// ReplyBodyCodec 获取响应消息的编码格式
	ReplyBodyCodec() byte

	// SetBodyCodec 设置响应消息的编码格式
	SetBodyCodec(byte)

	// SetMeta 设置指定key的值
	SetMeta(key, value string)

	// AddTFilterId 设置回复消息传输层的编码过滤方法id
	AddTFilterId(filterID ...byte)
	// contains filtered or unexported methods
}

type CtxSession interface {

	// ID 获取id
	ID() string

	// LocalAddr 本地地址
	LocalAddr() net.Addr

	// RemoteAddr 远端地址
	RemoteAddr() net.Addr

	// Swap 返回交换区的内容
	Swap() *gmap.Map

	// CloseNotify 返回该链接被关闭时候的通知
	CloseNotify() <-chan struct{}

	// Health 检查该session是否健康
	Health() bool

	// AsyncCall 发送消息，并异步接收响应
	AsyncCall(serviceMethod string, args interface{}, result interface{}, callCmdChan chan<- CallCmd, setting ...message.MsgSetting) CallCmd

	// Call 发送消息并获得响应值
	Call(serviceMethod string, args interface{}, result interface{}, setting ...message.MsgSetting) CallCmd

	// Push 发送消息，不接收响应，只返回发送状态
	Push(serviceMethod string, args interface{}, setting ...message.MsgSetting) *status.Status

	// SessionAge 获取session最大的生存周期
	SessionAge() time.Duration

	// ContextAge 获取 CALL 和 PUSH 消息的最大生存周期
	ContextAge() time.Duration
}

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

type EarlyCtx interface {

	// Endpoint 获取当前Endpoint
	Endpoint() Endpoint

	// Session 返回当前的session
	Session() CtxSession

	// IP 返回远端ip
	IP() string

	// RealIP 返回远端真实ip
	RealIP() string

	// Swap 返回自定义交换区数据
	Swap() *gmap.Map

	// Context 获取上下文
	Context() context.Context
}

type EarlyEndpoint interface {
	BaseEndpoint

	// Router 获取路由对象
	Router() *Router

	// SubRoute 获取分组路由对象
	SubRoute(pathPrefix string, plugin ...Plugin) *SubRouter

	// RouteCall 通过struct注册CALL类型的处理程序，并且返回注册的路径列表
	RouteCall(ctrlStruct interface{}, plugin ...Plugin) []string
	// RouteCallFunc 通过func注册CALL类型的处理程序，并且返回单个注册路径
	RouteCallFunc(callHandleFunc interface{}, plugin ...Plugin) string
	// RoutePush 通过struct注册PUSH类型的处理程序，并且返回注册的路径列表
	RoutePush(ctrlStruct interface{}, plugin ...Plugin) []string
	// RoutePushFunc 通过func注册PUSH类型的处理程序，并且返回单个注册路径
	RoutePushFunc(pushHandleFunc interface{}, plugin ...Plugin) string
	// SetUnknownCall 设置默认处理程序，当没有找到CALL的处理程序时将调用该处理程序。
	SetUnknownCall(fn func(UnknownCallCtx) (interface{}, *status.Status), plugin ...Plugin)
	// SetUnknownPush 设置默认处理程序，当没有找到PUSH的处理程序时将调用该处理程序。
	SetUnknownPush(fn func(UnknownPushCtx) *status.Status, plugin ...Plugin)
}

type EarlySession interface {
	Endpoint() Endpoint

	// LocalAddr 本地地址
	LocalAddr() net.Addr

	// RemoteAddr 远端地址
	RemoteAddr() net.Addr

	// Swap 临时存储区内容
	Swap() *gmap.Map

	// SetID 设置session id
	SetID(newID string)

	// ControlFD 原始链接的fd
	ControlFD(f func(fd uintptr)) error

	// ModifySocket 修改session的底层socket
	ModifySocket(fn func(conn net.Conn) (modifiedConn net.Conn, newProtoFunc proto.ProtoFunc))

	// GetProtoFunc 获取协议方法
	GetProtoFunc() proto.ProtoFunc

	// EarlySend 在会话刚建立的时候临时发送消息，不执行任何中间件
	EarlySend(mType byte, serviceMethod string, body interface{}, stat *status.Status, setting ...message.MsgSetting) (opStat *status.Status)

	// EarlyReceive 在会话刚建立的时候临时接受信息，不执行任何中间件
	EarlyReceive(newArgs message.NewBodyFunc, ctx ...context.Context) (input message.Message)

	// EarlyCall 在会话刚建立的时候临时调用call发送和接受消息，不执行任何中间件
	EarlyCall(serviceMethod string, args, reply interface{}, callSetting ...message.MsgSetting) (opStat *status.Status)

	// EarlyReply 在会话刚建立的时候临时回复消息，不执行任何中间件
	EarlyReply(req message.Message, body interface{}, stat *status.Status, setting ...message.MsgSetting) (opStat *status.Status)

	// RawPush 发送原始push消息，不执行任何中间件
	RawPush(serviceMethod string, args interface{}, setting ...message.MsgSetting) (opStat *status.Status)

	// SessionAge 获取session最大的生存周期
	SessionAge() time.Duration

	// ContextAge 获取 CALL 和 PUSH 消息的最大生存周期
	ContextAge() time.Duration

	// SetSessionAge 设置session的最大生存周期
	SetSessionAge(duration time.Duration)

	// SetContextAge 设置单个 CALL 和 PUSH 消息的最大生存周期
	SetContextAge(duration time.Duration)
}

type Endpoint interface {
	EarlyEndpoint

	// ListenAndServe 打开服务监听
	ListenAndServe(protoFunc ...socket.ProtoFunc) error

	// Dial 作为客户端链接到指定的服务
	Dial(addr string, protoFunc ...socket.ProtoFunc) (Session, *status.Status)

	// ServeConn 传入指定的conn，生成session
	// 提示：
	// 1. 不支持断开链接后自动重拨
	// 2. 不检查TLS
	// 3. 执行 PostAcceptPlugin 插件
	ServeConn(conn net.Conn, protoFunc ...socket.ProtoFunc) (Session, *status.Status)
}

type EndpointConfig struct {

	// 网络类型; tcp, tcp4, tcp6, unix, unixpacket, kcp or quic"
	Network string `json:"network" comment:"网络类型; tcp, tcp4, tcp6, unix, unixpacket"`
	//作为服务端角色时，要监听的服务器本地IP
	ListenIP string `json:"listen_ip" comment:"作为服务端角色时，要监听的服务器本地IP"`
	//作为服务端角色时，需要监听的本地端口号
	ListenPort uint16 `json:"listen_port" comment:"作为服务端角色时，需要监听的本地端口号"`

	//作为客户端角色时,请求服务端时候，本地使用的地址
	LocalIP string `json:"local_ip" comment:"作为客户端角色时,请求服务端时候，本地使用的地址"`
	//作为客户端角色时,请求服务端时候，本地使用的地址端口号
	LocalPort uint16 `json:"local_port" comment:"作为客户端角色时,请求服务端时候，本地使用的地址端口号"`

	// 默认的消息体编码格式
	DefaultBodyCodec string `json:"default_body_codec" comment:"默认的消息体编码格式"`
	//默认session会话生命周期
	DefaultSessionAge time.Duration `json:"default_session_age" comment:"默认session会话生命周期"`
	//默认单次请求生命周期
	DefaultContextAge time.Duration `json:"default_context_age" comment:"默认单次请求生命周期"`
	//外部配置慢处理定义时间
	SlowCometDuration time.Duration `json:"slow_comet_duration" comment:"慢处理定义时长"`

	//是否打印会话中请求的 body或 metadata
	PrintDetail bool `json:"print_detail" comment:"是否打印请求的详细信息，body和metadata"`

	// 作为客户端角色时，请求服务端的超时时间
	DialTimeout time.Duration `json:"dial_timeout" comment:"作为客户端角色时，请求服务端的超时时间"`
	// 仅限客户端角色使用,链接中断时候，试图链接服务端的最大重试次数。
	RedialTimes int `json:"redial_times" comment:"仅限客户端角色使用,链接中断时候，试图链接服务端的最大重试次数。"`
	//仅限客户端角色使用 试图链接服务端时候，重试的时间间隔.
	RedialInterval time.Duration `json:"redial_interval" comment:"仅限客户端角色使用 试图链接服务端时候，重试的时间间隔."`
	// contains filtered or unexported fields
}

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

type Plugin interface {
	Name() string
}

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

type PushCtx interface {

	// GetBodyCodec 获取当前消息的编码格式
	GetBodyCodec() byte
	// contains filtered or unexported methods
}

type ReadCtx interface {

	// Input 获取传入的消息
	Input() message.Message

	// StatusOK 状态是否ok
	StatusOK() bool

	// Status 当前步骤的状态
	Status() *status.Status
	// contains filtered or unexported methods
}

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

type ServiceMethodMapper func(prefix, name string) (serviceMethod string)

type Session interface {
	Endpoint() Endpoint

	// SetID 设置session id
	SetID(newID string)

	// Close 关闭session
	Close() error

	CtxSession
}

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

type Status = status.Status

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

type UnknownCallCtx interface {

	// GetBodyCodec 获取当前消息的编码格式
	GetBodyCodec() byte

	// InputBodyBytes 传入消息体
	InputBodyBytes() []byte

	// BuildBody 如果push消息是未知的消息，则使用v对象解析消息内容
	BuildBody(v interface{}) (bodyCodec byte, err error)

	// SetBodyCodec 设置回复消息的编码格式
	SetBodyCodec(byte)

	// SetMeta 设置指定key的值
	SetMeta(key, value string)

	// AddTFilterId 设置回复消息传输层的编码过滤方法id
	AddTFilterId(filterID ...byte)
	// contains filtered or unexported methods
}

type UnknownPushCtx interface {

	// GetBodyCodec 获取当前消息的编码格式
	GetBodyCodec() byte

	// InputBodyBytes 传入消息体
	InputBodyBytes() []byte

	// BuildBody 如果push消息是未知的消息，则使用v对象解析消息内容
	BuildBody(v interface{}) (bodyCodec byte, err error)
	// contains filtered or unexported methods
}

type WriteCtx interface {
	EarlyCtx

	// Output 将要发送的消息对象
	Output() message.Message

	// StatusOK 状态是否ok
	StatusOK() bool

	// Status 当前消息状态
	Status() *status.Status
}