使用枚举获取分区商品库存

使用枚举、获取分区商品库存

syntax = "proto3";
package services;
import "google/api/annotations.proto";

message ProdRequest {
    int32 prod_id = 1; //传入id
}

message ProdResponse {
    int32 prod_stock = 1; //商品库存
}

message QuerySize {
    int32 size = 1; //页尺寸,这里的1并不是默认值，而是字段的顺序，如果有其他参数就就写2
}

message ProdResponseList {
    repeated ProdResponse prodres = 1; //返回了一堆商品库存,使用了repeated修饰符
}

enum ProdAreas { //定义了枚举
    A = 0;
    B = 1;
    C = 2;
}

message ProdRequest2 {
    int32 pro_id = 1; //传入的商品id
    ProdAreas prod_area = 2;
}

service ProdService {
    rpc GetProdStock (ProdRequest) returns (ProdResponse) {
        option (google.api.http) = {
            get: "/v1/prod/{prod_id}" //和request中的prod_id对应，不能写错
        };
    }
    rpc GetProdStock2 (ProdRequest2) returns (ProdResponse) {
        option (google.api.http) = {
            get: "/v1/prod/{prod_id}" //和request中的prod_id对应，不能写错
        };
    }
    rpc GetProdStocks (QuerySize) returns (ProdResponseList); //定义了参数是QuerySize
}

service中具体方法的定义

func (this *ProdService) GetProdStock2(ctx context.Context, request2 *ProdRequest2) (*ProdResponse, error) {
    var stock int32 = 0
    if request2.ProdArea == ProdAreas_A { //判断传递过来的枚举值是哪一个再做出相应的判断
        stock = 30
    } else if request2.ProdArea == ProdAreas_B {
        stock = 31
    } else {
        stock = 50
    }
    return &ProdResponse{ProdStock: stock}, nil
}

客户端调用

package main

import (
    "context"
    "fmt"
    "google.golang.org/grpc"
    "grpccli/helper"
    "grpccli/services"
    "log"
)

func main() {
    //creds, err := credentials.NewClientTLSFromFile("keys/server.crt", "localhost")
    //if err != nil {
    //    log.Fatal(err)+
    //}

    creds := helper.GetClientCreds()

    conn, err := grpc.Dial(":8081", grpc.WithTransportCredentials(creds))
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    prodClient := services.NewProdServiceClient(conn)
    response, err := prodClient.GetProdStock2(context.Background(), &services.ProdRequest2{ProId: 22, ProdArea: services.ProdAreas_B})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(response.ProdStock)
}

来自为知笔记(Wiz)