hashicorp:raft源码系列(2)--网络层

概述

接着上面–导读，本篇讲解Transport传输层实现。

如下NewRaft函数的参数中，除了Config结构，其他都是以interface形式实现。

接下来几篇我们分别对这5类接口进行详细分析。

func NewRaft(conf *Config, fsm FSM, logs LogStore, stable StableStore, snaps SnapshotStore, trans Transport) (*Raft, error) {
   
   ......
}

传输层涉及哪些文件

• transport.go: 定义传输层接口，下图中的Transport Interface对象。

• inmem_transport.go: 以内存的方式实现Transport接口，用于测试。

• net_transport.go: 网络方式实现Transport接口。

• tcp_transport.go: 以TCP的方式实现了NetworkTransport需要的SteamLayer。

• command.go：这个文件里面定义了各种RPC request和response的结构，eg: AppendEntriesRequest、AppendEntriesResponse

  

总结，NetworkTransport和InmemTransport是对Transport层的具体实现，接下来我们对NetworkTransport进行详细分析。

传输层具体是怎么实现的呢？

从测试用例TestNetworkTransport_AppendEntries开始

测试用例一般能告诉我们怎么玩，而AppendEntries发送日志又是最常用的功能。因此，我们就从这里开始。

1. 初始化消费者(trans1)，这一步后面是要详细看的，所以我们标星。

2. 启动消费者，监听rpcCh，也就是Consumer()返回的只读channel。🤔：rpc.Respond函数只是给respCh添加了一个元素，那谁来消费这个元素呢？

3. 初始化生产者( trans2)，同样使用newTCPTransport实现。

4. 调用AppendEntries消息给消费者trans1，这里直接就获取返回了。为啥看着就是同步返回的呢? 按理说网络调用一般都是异步返回吧。这个星标步骤，我们后面来详细分析。

func TestNetworkTransport_AppendEntries(t *testing.T) {
   
	for _, useAddrProvider := range []bool{
   true, false} {
   
		// ①初始化trans1 -- 消费者
		trans1, err := makeTransport(t, useAddrProvider, "localhost:0")
		
		rpcCh := trans1.Consumer()

		// 定义测试使用的RPC请求
		args := AppendEntriesRequest{
   
			...
		}
		resp := AppendEntriesResponse{
   
			...
		}

		// ② 启动消费者监听 
		go func() {
   
			select {
   
			case rpc := <-rpcCh:
				// 获取消息然后返回，注意Respond函数只是给RPC对象的RespChan添加一个值，那么谁来消费这和chan呢
				rpc.Respond(&resp, nil)
			...
			}
		}()

		// ③ 初始化生产者 trans2
		trans2, err := makeTransport(t, useAddrProvider, string(trans1.LocalAddr()))
	
    // ④ 发送AppendEntries消息给trans1，这里直接就获取返回了。为啥看着就是同步返回的呢
		if err := trans2.AppendEntries("id1", trans1.LocalAddr(), &args, &out); err != nil {
   
			t.Fatalf("err: %v", err)
		}
	}