func makeAllocator(name string, cidrStr string, quorum uint) (*Allocator, address.Range) { peername, err := router.PeerNameFromString(name) if err != nil { panic(err) } _, cidr, err := address.ParseCIDR(cidrStr) if err != nil { panic(err) } alloc := NewAllocator(peername, router.PeerUID(rand.Int63()), "nick-"+name, cidr.Range(), quorum, func(router.PeerName) bool { return true }) return alloc, cidr.HostRange() }
// Restart a node func (m *Model) restart(node *TestNode) { node.Node = NewNode(router.PeerName(m.nextID), router.PeerUID(m.r.Int63()), m.quorum) m.nextID++ node.Propose() // The node is now ignorant, so we need to mark the links into // the node as ready. for _, l := range node.links { if !l.to.isolated { m.readyLink(l.converse) } } // If a consensus was just accepted due to this node accepting // it, without other nodes hearing of it, and we then restart // this node, then a different consensus can occur later on. // If a tree falls with no one to hear it, does it make a // sound? node.firstConsensus = AcceptedValue{} }
// Make a network of nodes with random topology func makeRandomModel(params *TestParams, r *rand.Rand, t *testing.T) *Model { m := Model{ t: t, r: r, quorum: params.nodeCount/2 + 1, nodes: make([]TestNode, params.nodeCount), readyLinks: []*Link{}, nextID: params.nodeCount + 1, } for i := range m.nodes { m.nodes[i].Node = NewNode(router.PeerName(i/2+1), router.PeerUID(r.Int63()), m.quorum) m.nodes[i].Propose() } for i := 1; i < len(m.nodes); i++ { // was node i connected to the other nodes yet? connected := false for j := 0; j < i; j++ { if r.Float32() < params.connectedProb { connected = true m.addLink(&m.nodes[i], &m.nodes[j]) } } if !connected { // node i must be connected into the graph // somewhere. So if we didn't connect it // already, this is a last resort. m.addLink(&m.nodes[i], &m.nodes[r.Intn(i)]) } } return &m }