本文以官方提供的入門的demo helloworld（https://github.com/grpc/grpc-go/tree/master/examples/helloworld/helloworld）為例，通過源碼講解實際的grpc調(diào)用執(zhí)行的流程(UnaryRPC)，為了方便理解只列出核心代碼。該篇主要關(guān)注Server端源代碼流程。

RegisterGreeterServer實際將helloworld.proto 中定義的Greeter_Service進(jìn)行注冊，所謂的服務(wù)注冊。

service Greeter {
  // Sends a greeting
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

var Greeter_ServiceDesc = grpc.ServiceDesc{
    ServiceName: "helloworld.Greeter",
    HandlerType: (*GreeterServer)(nil),
    Methods: []grpc.MethodDesc{
        {
            MethodName: "SayHello",
            Handler:    _Greeter_SayHello_Handler,
        },
    },
    Streams:  []grpc.StreamDesc{},
    Metadata: "examples/helloworld/helloworld/helloworld.proto",
}

func RegisterGreeterServer(s grpc.ServiceRegistrar, srv GreeterServer) {
    s.RegisterService(&Greeter_ServiceDesc, srv)
}

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
    }
    s.services[sd.ServiceName] = info
}

Serve方法通過lis net.Listener監(jiān)聽連接，每次起一新的goroutine來處理 新的Conn，每一個新的連接都是在獨立的線程中進(jìn)行處理并發(fā)執(zhí)行。

func (s *Server) Serve(lis net.Listener) error {
    for {
        rawConn, err := lis.Accept()
        go func() {
            s.handleRawConn(lis.Addr().String(), rawConn)
        }()
    }
}

建立一個HTTP2的Transport，用其進(jìn)行Transport層的rpc調(diào)用執(zhí)行

func (s *Server) handleRawConn(lisAddr string, rawConn net.Conn) {
    st := s.newHTTP2Transport(conn, authInfo)
    go func() {
        s.serveStreams(st)
    }()
}

func (s *Server) serveStreams(st transport.ServerTransport) {
    s.handleStream(st, stream, s.traceInfo(st, stream))
}

通過stream.Method()得到需要調(diào)用FullMethod(/helloworld.Greeter/SayHello）從注冊的services列表中找到對應(yīng)serviceInfo對象，進(jìn)而找到MethodDesc，然后執(zhí)行processUnaryRPC

func (s *Server) handleStream(t transport.ServerTransport, stream *transport.Stream, trInfo *traceInfo) {
    srv, knownService := s.services[service]
    if knownService {
        if md, ok := srv.methods[method]; ok {
            s.processUnaryRPC(t, stream, srv, md, trInfo)
            return
        }
    }

}

func (s *Server) processUnaryRPC(t transport.ServerTransport, stream *transport.Stream, info *serviceInfo, md *MethodDesc, trInfo *traceInfo) (err error) {
    // 實際執(zhí)行定義的rpc方法google.golang.org/grpc/examples/helloworld/helloworld._Greeter_SayHello_Handler（如下）
    reply, appErr := md.Handler(info.serviceImpl, ctx, df, s.opts.unaryInt)
    //將rpc方法的值返回給客戶端
    s.sendResponse(t, stream, reply, cp, opts, comp)
}

google.golang.org/grpc/examples/helloworld/helloworld._Greeter_SayHello_Handler

func _Greeter_SayHello_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
    in := new(HelloRequest)
    if err := dec(in); err != nil {
        return nil, err
    }
    if interceptor == nil {
        return srv.(GreeterServer).SayHello(ctx, in)
    }
    info := &grpc.UnaryServerInfo{
        Server:     srv,
        FullMethod: "/helloworld.Greeter/SayHello",
    }
    handler := func(ctx context.Context, req interface{}) (interface{}, error) {
        return srv.(GreeterServer).SayHello(ctx, req.(*HelloRequest))
    }
    return interceptor(ctx, in, info, handler)
}

GRPC源碼實例解析（二）
http://www.itdecent.cn/p/944e3eb59dd8