基于Unix域套接字的grpc实现

最新推荐文章于 2024-09-05 19:43:41 发布

LOVETEDA

最新推荐文章于 2024-09-05 19:43:41 发布

阅读量5.1k

点赞数

分类专栏： Others 文章标签： grpc golang

本文链接：https://blog.csdn.net/LOVETEDA/article/details/81027083

版权

本文介绍了一种优化本地gRPC通信的方法，即通过Unix域套接字代替TCP，以提高通信效率。文章详细阐述了如何修改gRPC服务器端和客户端的代码，使其基于Unix域套接字工作，包括创建自定义的Dialer函数，并利用grpc官方API实现这一转换。最终，通过示例展示了完整的工作流程。

摘要由CSDN通过智能技术生成

现在上网搜grpc的教程，绝大多数都是底层基于TCP的grpc实现。grpc默认使用TCP方式进行连接，但是如果是本地进程之间进行相互调用的话，采用TCP将会产生很大的不必要的通信开销。而如果基于Unix域套接字，速度可以比TCP套接字快得多。本文采用golang实现了grpc底层使用unix域套接字。

首先在网上给的grpc示例中，服务器端的代码一般是这种形式：

    lis, err := net.Listen("tcp", port)
  if err != nil {
        log.Fatal("fialed to resolve unix addr")
  }

  s := grpc.NewServer()
  pb.RegisterGreeterServer(s, &server{})//设置服务器端的方法供客户端调用
  s.Serve(lis)

net.Listen方法返回一个net.Listener对象，对应着服务器端的监听套接字，而net.UnixListener实现了net.Listener接口，s.Serve中的参数改为net.UnixListener即可，服务器端底层即可采用Unix域套接字。

修改完后的代码为：

  server_addr, err := net.ResolveUnixAddr("unix", server_file)
  if err != nil {
      log.Fatal("fialed to resolve unix addr")
  }
  lis, err := net.ListenUnix("unix", server_addr)
  if err != nil {
	log.Fatal("failed to listen: %v", err)
  }

  s := grpc.NewServer()
  pb.RegisterGreeterServer(s, &server{})

  s.Serve(lis)

其中server_file为Unix域套接字对应的路径，此时lis类型为*net.UnixListener

客户端的实现相对来说比较复杂，不过只要有耐心去分析以下grpc的源码，其实实现原理也不算太难。首先先介绍一下grpc的代码中定义的一些数据类型：

DialOption数据类型：func(*dialOptions). DialOption数据类型为一个函数指针，指向的函数为返回类型为void，参数为dialOptions指针。

dialOptions数据类型&#