Golang手写RPC框架(day1)

我的项目地址包括项目源码和笔记内容。

Day1 服务端与消息编码

具体流程见

这里只介绍一下额外的细节和其他知识点。

1.多主程序项目的启动

Goland对于多主程序的项目，可以使用不同的配置来启动不同的项目。

在这里插入图片描述

build目录是使用go build 后程序生成的exe可执行文件。

run是工作目录，可以进行额外的配置。(这里没有作用)

2.流程

这里说明主程序的流程。

startServeraddrOptionh := &codec.Header{}geerpc req ${h.Seq}reply

3.细节

监听端口

l, err := net.Listen("tcp", ":0")

:0

在这里插入图片描述

比如在我的机器上就是13519。

json编码解码

_ = json.NewEncoder(conn).Encode(geerpc.DefaultOption)

geerpc.DefaultOption

使用方法类似于下面的：

 // 3. 使用 json.NewEncoder 编码
    person3 := Person{"王五", 30}
    // 编码结果暂存到 buffer
    bytes3 := new(bytes.Buffer)
    _ = json.NewEncoder(bytes3).Encode(person3)
    if err == nil {
        fmt.Print("json.NewEncoder 编码结果: ", string(bytes3.Bytes()))
    }


// json.NewEncoder 编码结果: {"name":"王五","age":30}

创建自定义的编码解码器

cc := codec.NewGobCodec(conn)

func NewGobCodec(conn io.ReadWriteCloser) Codec {
	buf := bufio.NewWriter(conn)
	return &GobCodec{
		conn: conn,
		buf:  buf,
		dec:  gob.NewDecoder(conn),
		enc:  gob.NewEncoder(buf),
	}
}

这里使用buffer缓存提高写入的效率，也就是说需要写入的内容先写入到buf中，然后用buf.flush() 写到conn的Writer中，而读取不需要缓存，所以直接使用

gob.NewDecoder(conn)

服务器端处理请求

func (server *Server) serveCodec(cc codec.Codec) {
	sending := new(sync.Mutex) // make sure to send a complete response
	wg := new(sync.WaitGroup)  // wait until all request are handled
	for {
		req, err := server.readRequest(cc) //获取请求的head 和 body
		if err != nil {
			if req == nil {
				break // it's not possible to recover, so close the connection
			}
			req.h.Error = err.Error()
			server.sendResponse(cc, req.h, invalidRequest, sending)
			continue
		}
		wg.Add(1)
		go server.handleRequest(cc, req, sending, wg) //处理请求并返回req.replyv
	}
	wg.Wait()
	_ = cc.Close()
}

sending

处理请求是并发的，但是回复请求的报文必须是逐个发送的，并发容易导致多个回复报文交织在一起，客户端无法解析。在这里使用锁(sending)保证。

sync.WaitGroup

使用Codec进行读写操作

_ = cc.Write(h, fmt.Sprintf("geerpc req %d", h.Seq))

func (c *GobCodec) Write(h *Header, body interface{}) (err error) {
	defer func() {
		_ = c.buf.Flush()
		if err != nil {
			_ = c.Close()
		}
	}()
	if err = c.enc.Encode(h); err != nil {
		log.Println("rpc: gob error encoding header:", err)
		return
	}
	if err = c.enc.Encode(body); err != nil {
		log.Println("rpc: gob error encoding body:", err)
		return
	}
	return
}

func (c *GobCodec) ReadHeader(h *Header) error {
	return c.dec.Decode(h)
}

func (c *GobCodec) ReadBody(body interface{}) error {
	return c.dec.Decode(body)
}

这里conn 拥有一块内存区域，用来读写。 c.enc.Encode 将内容编码到buf里缓存，然后flush到conn里。然后读操作ReadHeader和Body时，使用decoder将conn里的内容解码。

验证接口实现

var _ Codec = (*GobCodec)(nil)

GobCodecCodec

另外这个项目实现Codec接口的工厂模式，不同类型实例有不同的构造函数，比较有价值，可以效仿。

4.输出结果

在这里插入图片描述