You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.1 KiB
80 lines
2.1 KiB
package nodes
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
)
|
|
|
|
// leader node作为server为client注册的方法
|
|
type ServerReply struct{
|
|
Isleader bool
|
|
LeaderAddress string // 自己不是leader则返回leader地址
|
|
HaveValue bool
|
|
Value string
|
|
}
|
|
// RPC call
|
|
func (node *Node) WriteKV(kvCall *LogEntryCall, reply *ServerReply) error {
|
|
log.Sugar().Infof("[%s]收到客户端write请求", node.selfId)
|
|
|
|
// 自己不是leader，转交leader地址回复
|
|
if node.state != Leader {
|
|
reply.Isleader = false
|
|
if (node.leaderId == "") {
|
|
log.Fatal("还没选出第一个leader")
|
|
return nil
|
|
}
|
|
reply.LeaderAddress = node.nodes[node.leaderId].address
|
|
log.Sugar().Infof("[%s]转交给[%s]", node.selfId, node.leaderId)
|
|
return nil
|
|
}
|
|
|
|
// 自己是leader，修改自己的记录并广播
|
|
node.maxLogId++
|
|
logId := node.maxLogId
|
|
rLogE := RaftLogEntry{kvCall.LogE, logId, node.currTerm}
|
|
node.log = append(node.log, rLogE)
|
|
node.storage.AppendLog(rLogE)
|
|
log.Info("leader[" + node.selfId + "]处理请求 : " + kvCall.LogE.print() + ", 模拟方式 : " + strconv.Itoa(int(kvCall.CallState)))
|
|
// 广播给其它节点
|
|
node.BroadCastKV(kvCall.CallState)
|
|
reply.Isleader = true
|
|
return nil
|
|
}
|
|
|
|
// RPC call
|
|
func (node *Node) ReadKey(key *string, reply *ServerReply) error {
|
|
log.Sugar().Infof("[%s]收到客户端read请求", node.selfId)
|
|
// 先只读自己(无论自己是不是leader)，也方便测试
|
|
value, err := node.db.Get([]byte(*key), nil)
|
|
if err == leveldb.ErrNotFound {
|
|
reply.HaveValue = false
|
|
} else {
|
|
reply.HaveValue = true
|
|
reply.Value = string(value)
|
|
}
|
|
reply.Isleader = true
|
|
return nil
|
|
}
|
|
|
|
// RPC call 测试中寻找当前leader
|
|
type FindLeaderReply struct{
|
|
Isleader bool
|
|
LeaderId string
|
|
}
|
|
func (node *Node) FindLeader(_ struct{}, reply *FindLeaderReply) error {
|
|
// 自己不是leader，转交leader地址回复
|
|
if node.state != Leader {
|
|
reply.Isleader = false
|
|
if (node.leaderId == "") {
|
|
log.Fatal("还没选出第一个leader")
|
|
return nil
|
|
}
|
|
reply.LeaderId = node.leaderId
|
|
return nil
|
|
}
|
|
|
|
reply.LeaderId = node.selfId
|
|
reply.Isleader = true
|
|
return nil
|
|
}
|