func main() {
var opts struct {
NumPeer int `short:"n" long:"num-peer" description:"num of peers"`
}
args, err := flags.Parse(&opts)
if err != nil {
os.Exit(1)
}
if len(args) != 1 || args[0] != "T1" {
log.Errorf("Usage: performance_test -n <num-peer> T1")
os.Exit(1)
}
peerMap := make(map[string]*server.Peer)
incoming := make(chan *server.FsmMsg, 1024)
num := opts.NumPeer
start := time.Now()
for i := 0; i < num; i++ {
localAddr := fmt.Sprintf("10.10.%d.%d", (i+2)/255, (i+2)%255)
g := config.Global{
GlobalConfig: config.GlobalConfig{
As: uint32(1001 + i),
RouterId: net.ParseIP(localAddr),
},
}
p := config.Neighbor{
NeighborConfig: config.NeighborConfig{
PeerAs: 1000,
NeighborAddress: net.ParseIP("10.10.0.1"),
},
Transport: config.Transport{
TransportConfig: config.TransportConfig{
LocalAddress: net.ParseIP(localAddr),
},
},
}
peer := newPeer(g, p, incoming)
peerMap[p.Transport.TransportConfig.LocalAddress.String()] = peer
}
established := 0
ticker := time.NewTicker(time.Second * 5)
for {
select {
case msg := <-incoming:
peer := peerMap[msg.MsgDst]
switch msg.MsgType {
case server.FSM_MSG_STATE_CHANGE:
nextState := msg.MsgData.(bgp.FSMState)
fsm := peer.Fsm()
fsm.StateChange(nextState)
server.NewFSMHandler(fsm, incoming, peer.Outgoing())
if nextState == bgp.BGP_FSM_ESTABLISHED {
established++
}
if num == established {
goto END
}
}
case <-ticker.C:
now := time.Now()
log.Infof("[%s] # of established: %d", now.Sub(start), established)
}
}
END:
end := time.Now()
log.Infof("all established. elapsed time: %s", end.Sub(start))
if args[0] == "T1" {
return
}
for {
select {
case msg := <-incoming:
fmt.Println(msg)
}
}
}
func newPeer(g config.Global, p config.Neighbor, incoming chan *server.FsmMsg) *server.Peer {
tbl := table.NewTableManager(g.GlobalConfig.RouterId.String(), []bgp.RouteFamily{bgp.RF_IPv4_UC, bgp.RF_IPv6_UC}, 0, 0)
peer := server.NewPeer(g, p, tbl)
server.NewFSMHandler(peer.Fsm(), incoming, peer.Outgoing())
return peer
}