コード例 #1
0
ファイル: node_test.go プロジェクト: dengoswei/spaxos
func TestAlone(t *testing.T) {
	n := newNode()
	groupsid := []uint64{1}
	peers := []string{":17001"}
	sp := newSpaxos(groupsid[0], groupsid, 0, 0)
	assert(nil != sp)
	go n.run(sp)

	s := newSNet(groupsid[0], groupsid, peers)
	assert(nil != s)
	err := s.RunRecvMsg()
	assert(nil == err)

	go s.RunSendMsg()
	go attach(n, s)

	pv := pb.ProposeValue{Id: 10001, Value: []byte("dengoswei")}
	value, err := pv.Marshal()
	assert(nil == err)
	err = n.Propose(value)
	assert(nil == err)

	i := 0

	var rd Ready
	for {
		rd = <-n.readyc
		if 0 != len(rd.Chosen) {
			break
		}
		i += 1
		if 0 != len(rd.States) {
			println("== store states")
		}

		if 0 != len(rd.Messages) {
			var msgs []pb.Message
			for _, msg := range rd.Messages {
				if 0 != msg.To {
					msgs = append(msgs, msg)
				} else {
					// broad cast
					for _, id := range groupsid {
						newmsg := msg
						newmsg.To = uint64(id)
						msgs = append(msgs, newmsg)
					}
				}
			}
			s.sendc <- msgs
		}

		n.Advance(true)
	}

	for index, value := range rd.Chosen {
		pv := pb.ProposeValue{}
		err = pv.Unmarshal(value)
		assert(nil == err)
		println("=> index", index, pv.Id, string(pv.Value))
	}
}
コード例 #2
0
ファイル: node_test.go プロジェクト: dengoswei/spaxos
func TestBuild(t *testing.T) {
	// create nodeCnt node
	nodeCnt := 3

	ns := make([]node, nodeCnt)
	sps := make([]*spaxos, nodeCnt)

	groupsid := make([]uint64, nodeCnt)
	for i := 0; i < nodeCnt; i += 1 {
		groupsid[i] = uint64(i + 1)
	}

	for i := 0; i < nodeCnt; i += 1 {
		ns[i] = newNode()
		sps[i] = newSpaxos(groupsid[i], groupsid, 1, 1)
		assert(nil != sps[i])
		// start the node run
		go ns[i].run(sps[i])
	}

	snets := make([]*SNet, nodeCnt)
	addrs := make([]string, nodeCnt)
	for i := 0; i < nodeCnt; i += 1 {
		addrs[i] = ":" + strconv.Itoa(15000+i+1)
		println("addrs", addrs[i])
	}

	for i := 0; i < nodeCnt; i += 1 {
		selfid := groupsid[i]
		snets[i] = newSNet(selfid, groupsid, addrs)
		assert(nil != snets[i])
		err := snets[i].RunRecvMsg()
		assert(nil == err)
	}

	ccs := make([]chan []byte, nodeCnt)
	for i := 0; i < nodeCnt; i += 1 {
		go snets[i].RunSendMsg()
		go attach(ns[i], snets[i])
		ccs[i] = make(chan []byte)
		go runSpaxos(groupsid[i], groupsid, ns[i], snets[i], ccs[i])
	}

	for i := 0; i < 10; i += 1 {
		pv := pb.ProposeValue{
			Id: uint64(2000 + i), Value: []byte(randString(100))}
		value, err := pv.Marshal()
		assert(nil == err)

		pidx := i % len(ns)
		err = ns[pidx].Propose(value)
		assert(nil == err)

		cvalue := <-ccs[pidx]
		newpv := pb.ProposeValue{}
		err = newpv.Unmarshal(cvalue)
		assert(nil == err)
		assert(pv.Id == newpv.Id)
		assert(string(pv.Value) == string(newpv.Value))
		for idx := range groupsid {
			if pidx == idx {
				continue
			}

			cvalue = <-ccs[idx]
			newpv = pb.ProposeValue{}
			err = newpv.Unmarshal(cvalue)
			assert(nil == err)
			assert(pv.Id == newpv.Id)
			assert(string(pv.Value) == string(newpv.Value))
		}
	}
}