etcd/raft/node.go

package raft

import "sync"

type Interface interface {
	Step(m Message)
}

type Node struct {
	lk sync.Mutex
	sm *stateMachine
}

func New(k, addr int, next Interface) *Node {
	n := &Node{
		sm: newStateMachine(k, addr),
	}
	return n
}

// Propose asynchronously proposes data be applied to the underlying state machine.
func (n *Node) Propose(data []byte) {
	m := Message{Type: msgProp, Data: data}
	n.Step(m)
}

func (n *Node) Step(m Message) {
	n.lk.Lock()
	defer n.lk.Unlock()
	n.sm.Step(m)
}

// Next advances the commit index and returns any new
// commitable entries.
func (n *Node) Next() []Entry {
	n.lk.Lock()
	defer n.lk.Unlock()
	return n.sm.nextEnts()
}