etcd-io__etcd/etcd/v2_raft.go

package etcd

import (
	"encoding/json"
	"fmt"
	"time"

	"github.com/coreos/etcd/raft"
)

type v2Proposal struct {
	data []byte
	ret  chan interface{}
}

type wait struct {
	index int64
	term  int64
}

type v2Raft struct {
	*raft.Node
	result map[wait]chan interface{}
	term   int64
}

func (r *v2Raft) Propose(p v2Proposal) {
	if !r.Node.IsLeader() {
		p.ret <- fmt.Errorf("not leader")
		return
	}
	r.Node.Propose(p.data)
	r.result[wait{r.Index(), r.Term()}] = p.ret
	return
}

func (r *v2Raft) Sync() {
	if !r.Node.IsLeader() {
		return
	}
	sync := &cmd{Type: "sync", Time: time.Now()}
	data, err := json.Marshal(sync)
	if err != nil {
		panic(err)
	}
	r.Node.Propose(data)
}

func (r *v2Raft) StopProposalWaiters() {
	for k, ch := range r.result {
		ch <- raftStopErr
		delete(r.result, k)
	}
}