him

type Acceptor interface {
	// Accept 返回一个握手完成的Channel对象或者一个error。
	// 业务层需要处理不同协议和网络环境的下连接握手协议
	Accept(Conn, time.Duration) (string, error)
}

type Agent interface {
	// ID 返回连接的channelID
	ID() string
	// Push 用于上层业务返回消息
	Push([]byte) error
}

type Channel interface {
	Conn
	Agent
	Close() error //overwrite net.Conn.Close()
	ReadLoop(MessageListener) error
	SetWriteWait(time.Duration)
	SetReadWait(time.Duration)
}

type ChannelMap interface {
	Add(Channel)
	Remove(id string)
	Get(id string) (Channel, bool)
	All() []Channel
}

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

type Client interface {
	Service
	Connect(string) error
	SetDialer(Dialer)
	Send([]byte) error
	Read() (Frame, error)
	Close()
}

type Conn interface {
	net.Conn
	ReadFrame() (Frame, error)
	WriteFrame(OpCode, []byte) error
	Flush() error
}

type Context interface {
	Dispatcher
	SessionStorage
	Header() *pkt.Header
	ReadBody(val proto.Message) error
	Session() Session
	RespWithError(status pkt.Status, err error) error
	Resp(status pkt.Status, body proto.Message) error
	Dispatch(body proto.Message, revs ...*Location) error
	Next()
}

type ContextImpl struct {
	sync.Mutex
	Dispatcher
	SessionStorage
	// contains filtered or unexported fields
}

type Dialer interface {
	DialAndHandshake(DialerContext) (net.Conn, error)
}

type DialerContext struct {
	Id      string
	Name    string
	Address string
	Timeout time.Duration
}

type Dispatcher interface {
	Push(gateway string, channels []string, p *pkt.LogicPkt) error
}

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

type Frame interface {
	SetOpCode(OpCode)
	GetOpCode() OpCode
	SetPayload([]byte)
	GetPayload() []byte
}

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

type Handler interface {
	MessageListener
	Acceptor
	StateListener
}

type HandlerFunc func(ctx Context)

type HandlersChain []HandlerFunc

type Location struct {
	// 通道Id
	ChannelId string
	// 网关Id
	GateId string
}

type MessageListener interface {
	// Receive 参数Agent表示发送方
	Receive(Agent, []byte)
}

type OpCode byte

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

type Server interface {
	ServiceRegistration
	SetAcceptor(Acceptor)
	SetMessageListener(MessageListener)
	SetStateListener(StateListener)
	SetReadWait(time.Duration)
	SetChannelMap(ChannelMap)

	Start() error
	Push(string, []byte) error
	Shutdown(ctx context.Context) error
}

type Service interface {
	ServiceID() string
	ServiceName() string
	GetMeta() map[string]string
}

type ServiceRegistration interface {
	Service
	// PublicAddress ip or domain
	PublicAddress() string
	PublicPort() int
	DialURL() string
	GetProtocol() string
	GetNamespace() string
	GetTags() []string
	// String SetTags(tags []string)
	// SetMeta(meta map[string]string)
	String() string
}

type Session interface {
	GetChannelId() string
	GetGateId() string
	GetAccount() string
	GetZone() string
	GetIsp() string
	GetRemoteIP() string
	GetDevice() string
	GetApp() string
	GetTags() []string
}

type SessionStorage interface {
	// Add a session
	Add(session *pkt.Session) error
	// Delete a session
	Delete(account string, channelId string) error
	// Get session by channelId
	Get(channelId string) (*pkt.Session, error)
	// GetLocations Get Locations by accounts
	GetLocations(account ...string) ([]*Location, error)
	// GetLocation Get Location by account and device
	GetLocation(account string, device string) (*Location, error)
}

type StateListener interface {
	// Disconnect 连接断开回调
	Disconnect(string) error
}