neo-go/pkg/server/connmgr.go

package server

import (
	"fmt"
	"net"
	"strconv"

	"github.com/CityOfZion/neo-go/pkg/connmgr"

	"github.com/CityOfZion/neo-go/pkg/peer"
	iputils "github.com/CityOfZion/neo-go/pkg/wire/util/ip"
)

func setupConnManager(s *Server, port uint16) *connmgr.Connmgr {
	cfg := connmgr.Config{
		GetAddress:   s.getAddress,
		OnAccept:     s.onAccept,
		OnConnection: s.onConnection,
		AddressPort:  iputils.GetLocalIP().String() + ":" + strconv.FormatUint(uint64(port), 10),
	}
	return connmgr.New(cfg)
}

func (s *Server) onConnection(conn net.Conn, addr string) {
	fmt.Println("We have connected successfully to: ", addr)

	p := peer.NewPeer(conn, false, *s.peerCfg)
	err := p.Run()
	if err != nil {
		fmt.Println("Error running peer" + err.Error())
		return
	}

	s.pmg.AddPeer(p)
}

func (s *Server) onAccept(conn net.Conn) {
	fmt.Println("A peer with address: ", conn.RemoteAddr().String(), "has connect to us")

	p := peer.NewPeer(conn, true, *s.peerCfg)
	err := p.Run()
	if err != nil {
		fmt.Println("Error running peer" + err.Error())
		return
	}
	s.pmg.AddPeer(p)
}