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.
111 lines
2.6 KiB
111 lines
2.6 KiB
package nodes
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/rpc"
|
|
"time"
|
|
)
|
|
|
|
// 真实rpc通讯的transport类型实现
|
|
type HTTPTransport struct{
|
|
NodeMap map[string]string // id到addr的映射
|
|
}
|
|
|
|
// 封装有超时的dial
|
|
func (t *HTTPTransport) DialHTTPWithTimeout(network string, peerId string) (ClientInterface, error) {
|
|
done := make(chan struct{})
|
|
var client *rpc.Client
|
|
var err error
|
|
|
|
go func() {
|
|
client, err = rpc.DialHTTP(network, t.NodeMap[peerId])
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
return &HTTPClient{rpcClient: client}, err
|
|
case <-time.After(50 * time.Millisecond):
|
|
return nil, fmt.Errorf("dial timeout: %s", t.NodeMap[peerId])
|
|
}
|
|
}
|
|
|
|
func (t *HTTPTransport) CallWithTimeout(clientInterface ClientInterface, serviceMethod string, args interface{}, reply interface{}) error {
|
|
c, ok := clientInterface.(*HTTPClient)
|
|
client := c.rpcClient
|
|
if !ok {
|
|
return fmt.Errorf("invalid client type")
|
|
}
|
|
|
|
done := make(chan error, 1)
|
|
|
|
go func() {
|
|
switch serviceMethod {
|
|
case "Node.AppendEntries":
|
|
arg, ok := args.(*AppendEntriesArg)
|
|
resp, ok2 := reply.(*AppendEntriesReply)
|
|
if !ok || !ok2 {
|
|
done <- errors.New("type assertion failed for AppendEntries")
|
|
return
|
|
}
|
|
done <- client.Call(serviceMethod, arg, resp)
|
|
|
|
case "Node.RequestVote":
|
|
arg, ok := args.(*RequestVoteArgs)
|
|
resp, ok2 := reply.(*RequestVoteReply)
|
|
if !ok || !ok2 {
|
|
done <- errors.New("type assertion failed for RequestVote")
|
|
return
|
|
}
|
|
done <- client.Call(serviceMethod, arg, resp)
|
|
|
|
case "Node.WriteKV":
|
|
arg, ok := args.(*LogEntryCall)
|
|
resp, ok2 := reply.(*ServerReply)
|
|
if !ok || !ok2 {
|
|
done <- errors.New("type assertion failed for WriteKV")
|
|
return
|
|
}
|
|
done <- client.Call(serviceMethod, arg, resp)
|
|
|
|
case "Node.ReadKey":
|
|
arg, ok := args.(*string)
|
|
resp, ok2 := reply.(*ServerReply)
|
|
if !ok || !ok2 {
|
|
done <- errors.New("type assertion failed for ReadKey")
|
|
return
|
|
}
|
|
done <- client.Call(serviceMethod, arg, resp)
|
|
|
|
case "Node.FindLeader":
|
|
arg, ok := args.(struct{})
|
|
resp, ok2 := reply.(*FindLeaderReply)
|
|
if !ok || !ok2 {
|
|
done <- errors.New("type assertion failed for FindLeader")
|
|
return
|
|
}
|
|
done <- client.Call(serviceMethod, arg, resp)
|
|
|
|
default:
|
|
done <- fmt.Errorf("unknown service method: %s", serviceMethod)
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case err := <-done:
|
|
return err
|
|
case <-time.After(50 * time.Millisecond): // 设置超时时间
|
|
return fmt.Errorf("call timeout: %s", serviceMethod)
|
|
}
|
|
}
|
|
|
|
type HTTPClient struct {
|
|
rpcClient *rpc.Client
|
|
}
|
|
|
|
func (h *HTTPClient) Close() error {
|
|
return h.rpcClient.Close()
|
|
}
|
|
|
|
|