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 int
	term  int
}

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

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

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)
}