golang 源碼分析：grpc 服務註冊

    我們寫一個 grpc 服務的時候，grpc 服務註冊流程如下

baseServer = grpc.NewServer
xxpb.RegisterxxServiceServer(baseServer, xxServer)

下面我們以健康檢查爲例，分析下服務註冊的邏輯

func RegisterHealthServer(s grpc.ServiceRegistrar, srv HealthServer) {
  s.RegisterService(&Health_ServiceDesc, srv)
}

其中 Health_ServiceDesc 是一個全局變量，它存儲了服務的元數據：

  var Health_ServiceDesc = grpc.ServiceDesc{
  ServiceName: "grpc.health.v1.Health",
  HandlerType: (*HealthServer)(nil),
  Methods: []grpc.MethodDesc{
    {
      MethodName: "Check",
      Handler:    _Health_Check_Handler,
    },
  },
  Streams: []grpc.StreamDesc{
    {
      StreamName:    "Watch",
      Handler:       _Health_Watch_Handler,
      ServerStreams: true,
    },
  },
  Metadata: "grpc/health/v1/health.proto",
}

服務註冊最終調用了生成的服務端代碼的註冊函數：

func (s *Server) RegisterService(sd *ServiceDesc, ss interface{}) {
        s.register(sd, ss)

func (s *Server) register(sd *ServiceDesc, ss interface{}) {
  info := &serviceInfo{
    serviceImpl: ss,
    methods:     make(map[string]*MethodDesc),
    streams:     make(map[string]*StreamDesc),
    mdata:       sd.Metadata,
  }
  for i := range sd.Methods {
    d := &sd.Methods[i]
    info.methods[d.MethodName] = d
  for i := range sd.Streams {
    d := &sd.Streams[i]
    info.streams[d.StreamName] = d
    s.services[sd.ServiceName] = info

可看到，註冊的過程就是建立元數據到具體實現函數的映射。這個過程和我們寫 http 服務的路由註冊流程類似。註冊完成後就開始進行服務的監聽。

baseServer.Serve(grpcListener)

它的實現和 golang 的 http 包的 Serve 實現類似：

func (s *Server) Serve(lis net.Listener) error
  s.serveWG.Add(1)
       for {
    rawConn, err := lis.Accept()
          go func() {
      s.handleRawConn(lis.Addr().String(), rawConn)
    func (s *Server) handleRawConn(lisAddr string, rawConn net.Conn) {
      st := s.newHTTP2Transport(rawConn)
      if !s.addConn(lisAddr, st) {
          s.serveStreams(st)
    s.removeConn(lisAddr, st)

最終調用了 ServeHTTP 方法

func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
      s.serveStreams(st)
        go func() {
          s.handleStream(st, stream, s.traceInfo(st, stream))

google.golang.org/grpc@v1.45.0/server.go

handleStream 的過程就是根據請求信息，在我們前面註冊的 map 裏面取出對應的處理方法，進行邏輯的處理。

func (s *Server) handleStream(t transport.ServerTransport, stream *transport.Stream, trInfo *traceInfo) {
        srv, knownService := s.services[service]
          if md, ok := srv.methods[method]; ok {
      s.processUnaryRPC(t, stream, srv, md, trInfo)
      func (s *Server) processUnaryRPC(t transport.ServerTransport, stream *transport.Stream, info *serviceInfo, md *MethodDesc, trInfo *traceInfo) (err error) {
        sh.HandleRPC(stream.Context(), statsBegin)

本文由 Readfog 進行 AMP 轉碼，版權歸原作者所有。
來源：https://mp.weixin.qq.com/s/qHQYUlefrSsdQg0IdIx_GQ