network

type BroadcastMsg struct {
	ModelID string                 `json:"model_id"`
	MsgType model.BroadcastMsgType `json:"msg_type"`
	Msg     []byte                 `json:"msg"`
}

type OnReceive func(modelID string, msgBytes []byte, p *Peer)

type Peer struct {
	ID      string // 定义peerid  每个peerid应该是唯一的
	Address string // 地址
}

type PeerBooks struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

type SwitcherI interface {
	// 向所有的节点广播消息
	Broadcast(modelID string, msg *BroadcastMsg) error
	// 广播到指定的peer
	BroadcastToPeer(modelID string, msg *BroadcastMsg, p *Peer) error
	// 广播 除了指定的peer
	BroadcastExceptPeer(modelID string, msg *BroadcastMsg, p *Peer) error
	// 移除某个peer
	RemovePeer(p *Peer) error
	RegisterOnReceive(modelID string, callBack OnReceive) error
	Start() error
	// 返回所有存在的peers
	Peers() ([]*Peer, error)
}