Go gRPC Hello World

概述

開始 gRPC 了，這篇文章學習使用 gRPC，輸出一個 Hello World，

    用 Go 實作 gRPC 的服務端，

    用 Go 實作 gRPC 的客戶端，

gRPC 支持 4 類服務方法，咱們這次實作 單項 RPC 和 服務端流式 RPC，

四類服務方法

單項 RPC

服務端發送一個請求給服務端，從服務端獲取一個應答，就像一次普通的函式呼叫，

rpc SayHello(HelloRequest) returns (HelloResponse){}

服務端流式 RPC

客戶端發送一個請求給服務端，可獲取一個資料流用來讀取一系列訊息，客戶端從回傳的資料流里一直讀取直到沒有更多訊息為止，

rpc LotsOfReplies(HelloRequest) returns (stream HelloResponse){}

客戶端流式 RPC

客戶端用提供的一個資料流寫入并發送一系列訊息給服務端，一旦客戶端完成訊息寫入，就等待服務端讀取這些訊息并回傳應答，

rpc LotsOfGreetings(stream HelloRequest) returns (HelloResponse) {}

雙向流式 RPC

兩邊都可以分別通過一個讀寫資料流來發送一系列訊息，這兩個資料流操作是相互獨立的，所以客戶端和服務端能按其希望的任意順序讀寫，例如：服務端可以在寫應答前等待所有的客戶端訊息，或者它可以先讀一個訊息再寫一個訊息，或者是讀寫相結合的其他方式，每個資料流里訊息的順序會被保持，

rpc BidiHello(stream HelloRequest) returns (stream HelloResponse){}

安裝

安裝 protobuf 編譯器

brew install protobuf

驗證：

    protoc --version    
    //輸出：libprotoc 3.7.1

安裝 Go protobuf 插件

    go get -u github.com/golang/protobuf/proto    
    go get -u github.com/golang/protobuf/protoc-gen-go

安裝 grpc-go

go get -u google.golang.org/grpc

寫個 Hello World 服務

    撰寫服務端 .proto 檔案

    生成服務端 .pb.go 檔案并同步給客戶端

    撰寫服務端提供介面的代碼

    撰寫客戶端呼叫介面的代碼

目錄結構

    ├─ hello  -- 代碼根目錄    
    │  ├─ go_client    
    │     ├── main.go    
    │     ├── proto    
    │         ├── hello    
    │            ├── hello.pb.go    
    │  ├─ go_server    
    │     ├── main.go    
    │     ├── controller    
    │         ├── hello_controller    
    │            ├── hello_server.go    
    │     ├── proto    
    │         ├── hello    
    │            ├── hello.pb.go    
    │            ├── hello.proto

這樣創建目錄是為了 go_client 和 go_server 后期可以拆成兩個專案，

撰寫服務端 hello.proto 檔案

    syntax = "proto3"; // 指定 proto 版本    
    package hello;     // 指定包名    
    // 定義 Hello 服務    
    service Hello {    
        // 定義 SayHello 方法    
        rpc SayHello(HelloRequest) returns (HelloResponse) {}    
        // 定義 LotsOfReplies 方法    
        rpc LotsOfReplies(HelloRequest) returns (stream HelloResponse){}    
    }    
    // HelloRequest 請求結構    
    message HelloRequest {    
        string name = 1;    
    }    
    // HelloResponse 回應結構    
    message HelloResponse {    
        string message = 1;    
    }

了解更多 Protobuf 語法，請查看：

https://developers.google.com/protocol-buffers/

生成服務端 .pb.go

protoc -I . --go_out=plugins=grpc:. ./hello.proto

同時將生成的 hello.pb.go 復制到客戶端一份，

查看更多命令引數，執行 protoc，查看 OPTION ，

撰寫服務端提供介面的代碼

 

  // hello_server.go    
    package hello_controller    
    import (    
        "fmt"    
        "golang.org/x/net/context"    
        "hello/go_server/proto/hello"    
    )    
    type HelloController struct{}    
    func (h *HelloController) SayHello(ctx context.Context, in *hello.HelloRequest) (*hello.HelloResponse, error) {    
        return &hello.HelloResponse{Message : fmt.Sprintf("%s", in.Name)}, nil    
    }    
    func (h *HelloController) LotsOfReplies(in *hello.HelloRequest, stream hello.Hello_LotsOfRepliesServer)  error {    
        for i := 0; i < 10; i++ {    
            stream.Send(&hello.HelloResponse{Message : fmt.Sprintf("%s %s %d", in.Name, "Reply", i)})    
        }    
        return nil    
    }

    // main.go    
    package main    
    import (    
        "log"    
        "net"    
        "hello/go_server/proto/hello"    
        "hello/go_server/controller/hello_controller"    
        "google.golang.org/grpc"    
    )    
    const (    
        Address = "0.0.0.0:9090"    
    )    
    func main() {    
        listen, err := net.Listen("tcp", Address)    
        if err != nil {    
            log.Fatalf("Failed to listen: %v", err)    
        }    
        s := grpc.NewServer()    
        // 服務注冊    
        hello.RegisterHelloServer(s, &hello_controller.HelloController{})    
        log.Println("Listen on " + Address)    
        if err := s.Serve(listen); err != nil {    
            log.Fatalf("Failed to serve: %v", err)    
        }    
    }

運行：

 

  go run main.go    
    2019/07/28 17:51:20 Listen on 0.0.0.0:9090

撰寫客戶端請求介面的代碼

  
 

package main   
import (    
        "hello/go_client/proto/hello"    
        "io"    
        "log"    
        "golang.org/x/net/context"    
        "google.golang.org/grpc"    
    )    
    const (    
        // gRPC 服務地址    
        Address = "0.0.0.0:9090"    
    )    
    func main() {    
        conn, err := grpc.Dial(Address, grpc.WithInsecure())    
        if err != nil {    
            log.Fatalln(err)    
        }    
        defer conn.Close()    
        // 初始化客戶端    
        c := hello.NewHelloClient(conn)    
        // 呼叫 SayHello 方法    
        res, err := c.SayHello(context.Background(), &hello.HelloRequest{Name: "Hello World"})    
        if err != nil {    
            log.Fatalln(err)    
        }    
        log.Println(res.Message)    
        // 呼叫 LotsOfReplies 方法    
        stream, err := c.LotsOfReplies(context.Background(),&hello.HelloRequest{Name: "Hello World"})    
        if err != nil {    
            log.Fatalln(err)    
        }    
        for {    
            res, err := stream.Recv()    
            if err == io.EOF {    
                break    
            }    
            if err != nil {    
                log.Printf("stream.Recv: %v", err)    
            }    
            log.Printf("%s", res.Message)    
        }    
    }

運行：

    go run main.go    
    2019/07/28 17:58:13 Hello World    
    2019/07/28 17:58:13 Hello World Reply 0    
    2019/07/28 17:58:13 Hello World Reply 1    
    2019/07/28 17:58:13 Hello World Reply 2    
    2019/07/28 17:58:13 Hello World Reply 3    
    2019/07/28 17:58:13 Hello World Reply 4    
    2019/07/28 17:58:13 Hello World Reply 5    
    2019/07/28 17:58:13 Hello World Reply 6    
    2019/07/28 17:58:13 Hello World Reply 7    
    2019/07/28 17:58:13 Hello World Reply 8    

標籤：Go

上一篇：[Go] gocron原始碼閱讀-go語言的結構體

下一篇：[Go] gocron原始碼閱讀-go語言中的切片介面和型別綜合