作者：从容嘛_100 | 来源：互联网 | 2023-09-11 15:45
编写proto文件：Models.protoOrders.protoProd.proto接下来生成pd.go文件编写服务端SerbicesOrdersService.gopacka
编写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
完成。。。。。。。
测试
grpc-POST提交主订单数据(gateway实现http api)
京公网安备 11010802041100号