grpc-POST提交主订单数据(gateway实现http api)

Posted 2020-11-25 孙龙 程序员

编写proto文件：

Models.proto

syntax="proto3";
package services;

import "google/protobuf/timestamp.proto";

message ProdModel{ //商品模型
    int32 prod_id=1;
    string prod_name=2;
    float prod_price=3;
}


message OrderMain{ //主订单模型
      int32 order_id=1;//订单ID，数字自增
      string order_no=2; //订单号
      int32 user_id=3; //购买者ID
      float order_money=4;//商品金额
      google.protobuf.Timestamp order_time=5; //下单时间
      repeated OrderDetail order_details=6;
}

//子订单模型
message OrderDetail{
    int32 detail_id=1;
    string order_no=2;
    int32 prod_id=3;
    float prod_price=4;
    int32 prod_num=5;
}

 

Orders.proto

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

message OrderRequest{
    OrderMain order_main=1;
}

message OrderResponse{
    string status=1;
    string message=2;
}
service OrderSerivce{
    rpc NewOrder(OrderRequest) returns (OrderResponse){
        option (google.api.http) = {
            post : "/v1/orders"
            body : "order_main"
        };
    }
}

 

Prod.proto

syntax="proto3";
package services;


import "google/api/annotations.proto";
import "Models.proto";

enum ProdAreas{
    A=0;
    B=1;
    C=2;
}

message  ProdRequest {
    int32 prod_id =1;   //传入的商品ID
}
message ProdResponse{
    int32 prod_stock=1;//商品库存
}

message QuerySize{
    int32 size = 1;//页尺寸
}

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

service ProdService {
    rpc GetProdStock (ProdRequest) returns (ProdResponse){
        option (google.api.http) = {
            get: "/v1/prod/{prod_id}"
        };

    }
    rpc GetProdStocks(QuerySize) returns (ProdResponseList){

    }

    rpc GetProdInfo(ProdRequest) returns(ProdModel){}
}

 

接下来生成pd.go文件

cd pbfiles && protoc --go_out=plugins=grpc:../services  Prod.proto
 protoc --go_out=plugins=grpc:../services  Orders.proto
protoc --go_out=plugins=grpc:../services  Models.proto

protoc  --grpc-gateway_out=logtostderr=true:../services Prod.proto
protoc  --grpc-gateway_out=logtostderr=true:../services Orders.proto

cd ..

 

 

 

编写服务端Serbices

OrdersService.go

package services

import (
    "context"
    "fmt"
)

type OrdersService struct {

}

func(this *OrdersService)NewOrder(ctx context.Context,orderRequest *OrderRequest) (*OrderResponse, error)   {
    fmt.Println(orderRequest.OrderMain)
    return &OrderResponse{
        Status:"OK",
        Message:"success",
    },nil
}

ProdService.go

package services

import (
    "context"
)

type ProdService struct {

}

func(this *ProdService) GetProdStock(ctx context.Context, request *ProdRequest) (*ProdResponse, error) {

      return &ProdResponse{ProdStock:20},nil
}

func (this *ProdService) GetProdStocks(ctx context.Context,size *QuerySize) (*ProdResponseList, error){
    var Prodres []*ProdResponse
    Prodres = make([]*ProdResponse,0,3)
    Prodres = append(Prodres,&ProdResponse{ProdStock:28,})
    Prodres = append(Prodres,&ProdResponse{ProdStock:29,})
    Prodres = append(Prodres,&ProdResponse{ProdStock:30,})
    return &ProdResponseList{
        Prodres:Prodres,
    },nil
}

func (this *ProdService) GetProdInfo(ctx context.Context, in *ProdRequest) (*ProdModel, error){

    ret:=ProdModel{
        ProdId:101,
        ProdName:"测试商品",
        ProdPrice:20.5,
    }
    return &ret,nil
}

View Code

 

最后注册http

启动server和http

 

httpserver.go

package main

import (
    "context"
    "github.com/grpc-ecosystem/grpc-gateway/runtime"
    "google.golang.org/grpc"
    "grpcpro/services"
    "log"
    "net/http"
)

func main()  {
    gwmux:=runtime.NewServeMux()
    opt := []grpc.DialOption{grpc.WithInsecure()}
    err:=services.RegisterProdServiceHandlerFromEndpoint(context.Background(),
        gwmux,"localhost:8081",opt)
    if err != nil {
        log.Fatal(err)
    }

    err = services.RegisterOrderSerivceHandlerFromEndpoint(context.Background(),
        gwmux,"localhost:8081",opt)
    if err != nil {
        log.Fatal(err)
    }


    httpServer:=&http.Server{
        Addr:":8080",
        Handler:gwmux,
    }
    httpServer.ListenAndServe()

}

 

server.go

package main

import (
    "google.golang.org/grpc"
    "grpcpro/services"
    "net"
)

func main()  {
    rpcServer:=grpc.NewServer()
    services.RegisterProdServiceServer(rpcServer,new(services.ProdService))
    services.RegisterOrderSerivceServer(rpcServer,new(services.OrdersService))

    lis,_:=net.Listen("tcp",":8081")

    rpcServer.Serve(lis)


}

 

go build server.go

go build httpserver.go

 

完成。。。。。。。

 

测试

 

 

 

 源码地址：

https://github.com/sunlongv520/grpc-learn

https://github.com/sunlongv520/grpc-doc