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.
94 lines
1.8 KiB
94 lines
1.8 KiB
package clientPkg
|
|
|
|
import (
|
|
"net/rpc"
|
|
"simple-kv-store/internal/logprovider"
|
|
"simple-kv-store/internal/nodes"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var log, _ = logprovider.CreateDefaultZapLogger(zap.InfoLevel)
|
|
|
|
type Client struct {
|
|
// 连接的server端节点（node1）
|
|
ServerId string
|
|
Address string
|
|
}
|
|
|
|
type Status = uint8
|
|
|
|
const (
|
|
Ok Status = iota + 1
|
|
NotFound
|
|
Fail
|
|
)
|
|
|
|
func (client *Client) Write(kvCall nodes.LogEntryCall) Status {
|
|
log.Info("client write request key :" + kvCall.LogE.Key)
|
|
|
|
var reply nodes.ServerReply
|
|
reply.Isleader = false
|
|
addr := client.Address
|
|
for !reply.Isleader {
|
|
c, err := rpc.DialHTTP("tcp", addr)
|
|
if err != nil {
|
|
log.Error("dialing: ", zap.Error(err))
|
|
return Fail
|
|
}
|
|
|
|
callErr := c.Call("Node.WriteKV", kvCall, &reply) // RPC
|
|
if callErr != nil {
|
|
log.Error("dialing: ", zap.Error(callErr))
|
|
return Fail
|
|
}
|
|
err = c.Close()
|
|
if err != nil {
|
|
log.Error("client close err: ", zap.Error(err))
|
|
}
|
|
|
|
if !reply.Isleader { // 发过去的不是leader
|
|
addr = reply.LeaderAddress
|
|
} else { // 成功
|
|
return Ok
|
|
}
|
|
}
|
|
return Fail
|
|
}
|
|
|
|
func (client *Client) Read(key string, value *string) Status { // 查不到value为空
|
|
log.Info("client read request key :" + key)
|
|
if value == nil {
|
|
return Fail
|
|
}
|
|
|
|
c, err := rpc.DialHTTP("tcp", client.Address)
|
|
if err != nil {
|
|
log.Error("dialing: ", zap.Error(err))
|
|
return Fail
|
|
}
|
|
|
|
defer func(server *rpc.Client) {
|
|
err := c.Close()
|
|
if err != nil {
|
|
log.Error("client close err: ", zap.Error(err))
|
|
}
|
|
}(c)
|
|
|
|
var reply nodes.ServerReply
|
|
callErr := c.Call("Node.ReadKey", key, &reply) // RPC
|
|
if callErr != nil {
|
|
log.Error("dialing: ", zap.Error(callErr))
|
|
return Fail
|
|
}
|
|
|
|
// 目前一定发送成功
|
|
if reply.HaveValue {
|
|
*value = reply.Value
|
|
return Ok
|
|
} else {
|
|
return NotFound
|
|
}
|
|
}
|
|
|
|
|