common/components/connector.go

package components
import (
	"fmt"
	"net"
	"pro2d/common/logger"
)
type ConnectorOption func(*Connector)
func WithCtorSplitter(splitter ISplitter) ConnectorOption {
	return func(connector *Connector) {
		connector.splitter = splitter
	}
}
func WithCtorCount(count int) ConnectorOption {
	return func(connector *Connector) {
		connector.sum = count
	}
}
type Connector struct {
	IConnector
	IServer
	splitter ISplitter
	ip       string
	port     int
	sum      int
	Conns IConnManage
	ids   uint32
}
func NewConnector(ip string, port int, options ...ConnectorOption) IConnector {
	c := &Connector{
		ids:   0,
		ip:    ip,
		port:  port,
		Conns: NewConnManage(),
	}
	for _, option := range options {
		option(c)
	}
	return c
}
func (c *Connector) Connect() error {
	if c.sum == 0 {
		c.sum = 1
	}
	for i := 0; i < c.sum; i++ {
		conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", c.ip, c.port))
		if err != nil {
			return err
		}
		c.ids++
		cli := NewConn(int(c.ids), conn, c.splitter)
		cli.SetConnectionCallback(c.OnConnect)
		cli.SetMessageCallback(c.OnMessage)
		cli.SetCloseCallback(c.OnClose)
		cli.SetTimerCallback(c.OnTimer)
		cli.Start()
	}
	return nil
}
func (c *Connector) DisConnect() {
	c.Conns.StopAllConns()
}
func (c *Connector) Send(cmd uint32, b []byte) {
	c.Conns.Range(func(key interface{}, value interface{}) bool {
		conn := value.(IConnection)
		conn.Send(0, cmd, b)
		return true
	})
}
func (c *Connector) GetSplitter() ISplitter {
	return c.splitter
}
func (c *Connector) OnConnect(conn IConnection) {
	c.Conns.AddConn(conn.GetID(), conn)
}
func (c *Connector) OnMessage(msg IMessage) {
	logger.Debug("recv msg cmd: %d, conn: %d data: %s", msg.GetHeader().GetMsgID(), msg.GetSID(), msg.GetData())
}
func (c *Connector) OnClose(conn IConnection) {
	logger.Debug("onclose id: %d", conn.GetID())
}
func (c *Connector) OnTimer(conn IConnection) {
	logger.Debug("ontimer id: %d", conn.GetID())
}