当前位置：首页 > news >正文

go语言分布式一致

news 2024/11/25 3:23:12

flowchart TDStart([接收 key]) --> CheckCache{检查是否被缓存}CheckCache -->|是| ReturnCache1[返回缓存值 ⑴]CheckCache -->|否| CheckRemote{是否应当从远程节点获取}CheckRemote -->|是| HashSelect[使用一致性哈希选择节点]HashSelect --> IsRemote{是否是远程节点}IsRemote -->|是| HTTPClient[HTTP 客户端访问远程节点]HTTPClient --> Success{成功？}Success -->|是| ReturnCache2[返回缓存值 ⑵]Success -->|否| FallbackLocal[回退到本地节点处理]IsRemote -->|否| FallbackLocalCheckRemote -->|否| Callback[调用回调函数]Callback --> AddCache[添加到缓存]AddCache --> ReturnCache3[返回缓存值 ⑶]style Start fill:#f9f,stroke:#333,stroke-width:2px
style ReturnCache1,ReturnCache2,ReturnCache3 fill:#9f9,stroke:#333,stroke-width:2px

抽象 PeerPicker

type PeerPicker interface{PickPeer(key string) (peer PeerGetter, ok bool)
}type PeerGetter interface{Get(group string,key string) ([]byte,error)
}

在这里，抽象出 2 个接口，PeerPicker 的 PickPeer() 方法用于根据传入的 key 选择相应节点 PeerGetter。
接口 PeerGetter 的 Get() 方法用于从对应 group 查找缓存值。PeerGetter 就对应于上述流程中的 HTTP 客户端。

节点选择与HTTP客户端

// httpGetter 是一个结构体，包含 baseURL 字段，
// 用作所有 HTTP GET 请求的基础 URL。// Get 是 httpGetter 的一个方法，用于通过特定的 URL 来获取数据，
// URL 由 baseURL、group 和 key 参数组合生成。
func (h *httpGetter) Get(group string, key string) ([]byte, error) {// 构造完整的 URL，将 baseURL 与编码后的 group 和 key 参数拼接起来。// 发送 HTTP GET 请求到构造的 URL。if err != nil {// 如果请求出现错误，则返回错误信息。}// 确保函数退出时关闭响应体。// 检查服务器响应状态是否为 OK（200），如果不是则返回错误。if res.StatusCode != http.StatusOK {}// 读取整个响应体。if err != nil {// 如果读取响应体时出错，则返回错误信息。}// 返回读取到的字节和 nil 错误，表示成功。}// 这行代码确保 httpGetter 实现了 PeerGetter 接口。
// 如果 httpGetter 没有满足 PeerGetter 接口要求，将会引发编译错误。

baseURL 表示将要访问的远程节点的地址，例如 http://example.com/_geecache/。
使用 http.Get() 方式获取返回值，并转换为 []bytes 类型

第二步，为HTTPPool添加节点选择的功能

const (  defaultBasePath = "/_geecache/"  defaultReplicas = 50  
)  
// HTTPPool implements PeerPicker for a pool of HTTP peers.  
type HTTPPool struct {  // this peer's base URL, e.g. "https://example.net:8000"  self        string  basePath    string  mu          sync.Mutex // guards peers and httpGetters  peers       *consistenthash.Map  httpGetters map[string]*httpGetter // keyed by e.g. "http://10.0.0.2:8008"  
}

新增成员变量 peers，类型是一致性哈希算法的 Map，用来根据具体的 key 选择节点。
新增成员变量 httpGetters，映射远程节点与对应的 httpGetter。每一个远程节点对应一个 httpGetter，因为 httpGetter 与远程节点的地址 baseURL 有关

第三步，实现PeerPicker接口

func (p *HTTPPool) Set(peers ...string){p.mu.Lock()defer p.mu.Unlock()p.peers = consistenthash.New(defaultReplicas,nil)p.peers.Add(peers...)p.httpGetters = make(map[string]*httpGetter, len(peers))for _,peer := range peers{p.httpGetters[peer] = &httpGetter{baseURL: peer + p.basePath}}
}