Go 语言网络编程系列（八）—— RPC 编程篇：使用 JSON 对传输数据进行编解码

type ClientCodec interface { 
    WriteRequest(*Request, interface{}) error
    ReadResponseHeader(*Response) error 
    ReadResponseBody(interface{}) error
    Close() error
}

type ServerCodec interface {
    ReadRequestHeader(*Request) error 
    ReadRequestBody(interface{}) error 
    WriteResponse(*Response, interface{}) error
    Close() error
}

package main

type Item struct {
  Id   int `json:"id"`      Name string `json:"name"`
}

type Response struct {
  Ok  bool `json:"ok"`
  Id  int `json:"id"`
  Message string `json:"msg"`
}

package main

import (
  "log"
  "net"
  "net/rpc"
  "net/rpc/jsonrpc"
)

// 服务端的rpc处理器
type ServiceHandler struct {}

func (serviceHandler *ServiceHandler) GetName(id int, item *Item) error {
  log.Printf("receive GetName call, id: %d", id)
  item.Id = id
  item.Name = "学院君"
  return nil
}

func (serviceHandler *ServiceHandler) SaveName(item Item, resp *Response) error {
  log.Printf("receive SaveName call, Item: %v", item)
  resp.Ok = true
  resp.Id = item.Id
  resp.Message = "存储成功"
  return nil
}

func main()  {
  // 初始化 RPC 服务端
  server := rpc.NewServer()

  // 监听端口 8080
  listener, err := net.Listen("tcp", ":8080")
  if err != nil {
    log.Fatalf("监听端口失败：%v", err)
  }
  defer listener.Close()

  log.Println("Start listen on port localhost:8080")

  // 初始化服务处理器
  serviceHandler := &ServiceHandler{}
  // 注册处理器
  err = server.Register(serviceHandler)
  if err != nil {
    log.Fatalf("注册服务处理器失败：%v", err)
  }

  // 等待并处理客户端连接
  for {
    conn, err := listener.Accept()
    if err != nil {
      log.Fatalf("接收客户端连接请求失败: %v", err)
    }

    // 自定义 RPC 编码器：新建一个 jsonrpc 编码器，并将该编码器绑定给 RPC 服务端处理器
    go server.ServeCodec(jsonrpc.NewServerCodec(conn))
  }
}

go server.ServeCodec(jsonrpc.NewServerCodec(conn))

func (server *Server) ServeConn(conn io.ReadWriteCloser) {
  buf := bufio.NewWriter(conn)
  srv := &gobServerCodec{
    rwc:    conn,
    dec:    gob.NewDecoder(conn),
    enc:    gob.NewEncoder(buf),
    encBuf: buf,
  }
  server.ServeCodec(srv)
}

package main

import (
  "log"
  "net"
  "net/rpc/jsonrpc"
  "time"
)

func main() {
  conn, err := net.DialTimeout("tcp", "localhost:8080", 30 * time.Second) // 30秒超时时间
  if err != nil {
    log.Fatalf("客户端发起连接失败：%v", err)
  }
  defer conn.Close()

  client := jsonrpc.NewClient(conn)
  var item Item
  client.Call("ServiceHandler.GetName", 1, &item)
  log.Printf("ServiceHandler.GetName 返回结果：%v\n", item)

  var resp Response
  item = Item{2, "学院君2"}
  client.Call("ServiceHandler.SaveName", item, &resp)
  log.Printf("ServiceHandler.SaveName 返回结果：%v\n", resp)
}

go run server.go utils.go