forked from TrueCloudLab/neoneo-go
48 lines
1 KiB
Go
48 lines
1 KiB
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)
|
||
|
|
}
|