tcp_client.go

package edgerouter

import (
	"errors"
	"fmt"
	"net"
)

type TcpClient struct {
	ctl   Controller
	Port  int
	conns map[string]*net.TCPConn
}

func (c *TcpClient) Init() {
	c.conns = make(map[string]*net.TCPConn)
}

func (c *TcpClient) PacketReceived(bts []byte, conn *net.TCPConn) int {
	return c.ctl.OnReceived(bts, conn)
}

func (c *TcpClient) SetController(ctrl Controller) {
	c.ctl = ctrl
}

func (c *TcpClient) Connect(to string) error {
	if addr, err := net.ResolveTCPAddr("tcp", to); err == nil {
		var conn *net.TCPConn
		var ok bool
		if conn, ok = c.conns[addr.String()]; !ok {
			var localPort *net.TCPAddr
			if c.Port != 0 {
				localPort, _ = net.ResolveTCPAddr("tcp", fmt.Sprintf(":%d", c.Port))
			}
			if conn, err = net.DialTCP("tcp", localPort, addr); err == nil {
				go handleTcpConn(conn, c)
				c.conns[addr.String()] = conn
			}
		}
		return err
	} else {
		return err
	}
}

var NoSuchConnection = errors.New("no such connection")

func (c *TcpClient) GetConn(to string, unexceptedConns []net.Conn) (conn net.Conn, err error) {
	var addr *net.TCPAddr
	if to == "any" {
	OUTER:
		for _, conn := range c.conns {
			for _, unexcepted := range unexceptedConns {
				if conn == unexcepted {
					continue OUTER
				}
			}
			return conn, nil
		}
		return nil, NoSuchConnection
	}
	if addr, err = net.ResolveTCPAddr("tcp", to); err == nil {
		var ok bool
		if conn, ok = c.conns[addr.String()]; ok {
		} else {
			err = NoSuchConnection
		}
	}
	return
}

func (c *TcpClient) Close(toDel *net.TCPConn) {
	c.DeleteConn(toDel)
}

//TODO client should reconnect to server
func (c *TcpClient) DeleteConn(toDel net.Conn) {
	for key, conn := range c.conns {
		if conn == toDel {
			delete(c.conns, key)
		}
	}
}

// func (c *TcpClient) Send(msg *BytesMessage, conn net.Conn) (err error) {
// 	var addr *net.TCPAddr
// 	if addr, err = net.ResolveTCPAddr("tcp", msg.To); err == nil {
// 		if conn, ok := c.conns[addr.String()]; ok {
// 			_, err = conn.Write(msg.Message)
// 		} else {
// 			err = errors.New("no such connection")
// 		}
// 	}

// 	return err
// }

func (c *TcpClient) String() string {
	return fmt.Sprintf("tcp client(%p)", c)
}