You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

116 lines
2.3 KiB

package main
import (
"os"
"os/signal"
"github.com/tendermint/tendermint/config"
//"github.com/tendermint/tendermint/consensus"
"github.com/tendermint/tendermint/p2p"
)
type Node struct {
lz []p2p.Listener
sw *p2p.Switch
book *p2p.AddrBook
pexReactor *p2p.PEXReactor
}
func NewNode() *Node {
sw := p2p.NewSwitch(nil) // XXX create and pass reactors
book := p2p.NewAddrBook(config.RootDir + "/addrbook.json")
pexReactor := p2p.NewPEXReactor(sw, book)
return &Node{
sw: sw,
book: book,
pexReactor: pexReactor,
}
}
func (n *Node) Start() {
log.Info("Starting node")
for _, l := range n.lz {
go n.inboundConnectionRoutine(l)
}
n.sw.Start()
n.book.Start()
n.pexReactor.Start()
}
func (n *Node) Stop() {
log.Info("Stopping node")
// TODO: gracefully disconnect from peers.
n.sw.Stop()
n.book.Stop()
n.pexReactor.Stop()
}
// Add a Listener to accept inbound peer connections.
func (n *Node) AddListener(l p2p.Listener) {
log.Info("Added %v", l)
n.lz = append(n.lz, l)
n.book.AddOurAddress(l.ExternalAddress())
}
func (n *Node) inboundConnectionRoutine(l p2p.Listener) {
for {
inConn, ok := <-l.Connections()
if !ok {
break
}
// New inbound connection!
peer, err := n.sw.AddPeerWithConnection(inConn, false)
if err != nil {
log.Info("Ignoring error from inbound connection: %v\n%v",
peer, err)
continue
}
// NOTE: We don't yet have the external address of the
// remote (if they have a listener at all).
// PEXReactor's pexRoutine will handle that.
}
// cleanup
}
//-----------------------------------------------------------------------------
func main() {
// Create & start node
n := NewNode()
l := p2p.NewDefaultListener("tcp", config.Config.LAddr)
n.AddListener(l)
n.Start()
// Seed?
if config.Config.Seed != "" {
peer, err := n.sw.DialPeerWithAddress(p2p.NewNetAddressString(config.Config.Seed))
if err != nil {
log.Error("Error dialing seed: %v", err)
//n.book.MarkAttempt(addr)
return
} else {
log.Info("Connected to seed: %v", peer)
}
}
// Sleep forever and then...
trapSignal(func() {
n.Stop()
})
}
func trapSignal(cb func()) {
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for sig := range c {
log.Info("captured %v, exiting..", sig)
cb()
os.Exit(1)
}
}()
select {}
}