tendermint/main.go

83 lines
1.5 KiB
Go
Raw Normal View History

2014-07-08 00:02:04 -07:00
package main
import (
2014-07-08 15:33:26 -07:00
"os"
"os/signal"
"github.com/tendermint/tendermint/config"
2014-07-08 00:02:04 -07:00
"github.com/tendermint/tendermint/p2p"
)
func main() {
// Define channels for our app
2014-07-08 15:33:26 -07:00
chDescs := []p2p.ChannelDescriptor{
p2p.ChannelDescriptor{
2014-07-08 00:02:04 -07:00
Name: "PEX",
SendBufferSize: 2,
2014-07-08 15:33:26 -07:00
RecvBufferSize: 2,
2014-07-08 00:02:04 -07:00
},
2014-07-08 15:33:26 -07:00
p2p.ChannelDescriptor{
2014-07-08 00:02:04 -07:00
Name: "block",
SendBufferSize: 10,
RecvBufferSize: 10,
},
2014-07-08 15:33:26 -07:00
p2p.ChannelDescriptor{
2014-07-08 00:02:04 -07:00
Name: "mempool",
SendBufferSize: 100,
RecvBufferSize: 100,
},
2014-07-08 15:33:26 -07:00
p2p.ChannelDescriptor{
2014-07-08 00:02:04 -07:00
Name: "consensus",
SendBufferSize: 1000,
RecvBufferSize: 1000,
},
}
// Create the switch
2014-07-08 15:33:26 -07:00
sw := p2p.NewSwitch(chDescs)
2014-07-08 00:02:04 -07:00
// Create a listener for incoming connections
2014-07-08 15:33:26 -07:00
l := p2p.NewDefaultListener("tcp", ":8001")
2014-07-08 00:02:04 -07:00
go func() {
for {
inConn, ok := <-l.Connections()
if !ok {
break
}
2014-07-08 15:33:26 -07:00
peer, err := sw.AddPeerWithConnection(inConn, false)
if err != nil {
log.Infof("Ignoring error from incoming connection: %v\n%v",
peer, err)
continue
}
initPeer(peer)
2014-07-08 00:02:04 -07:00
}
}()
2014-07-08 15:33:26 -07:00
// Open our address book
book := p2p.NewAddrBook(config.AppDir + "/addrbook.json")
// Start PEX
go p2p.PexHandler(sw, book)
// Sleep forever
2014-07-08 22:39:36 -07:00
trapSignal()
2014-07-08 15:33:26 -07:00
select {}
}
func initPeer(peer *p2p.Peer) {
// TODO: ask for more peers if we need them.
}
func trapSignal() {
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for sig := range c {
2014-07-08 22:39:36 -07:00
log.Infof("captured %v, exiting..", sig)
2014-07-08 15:33:26 -07:00
os.Exit(1)
}
}()
2014-07-08 00:02:04 -07:00
}