微服务之GRpc

go get -u github.com/favadi/protoc-go-inject-tag v1.0.0

# micro_prod_server -> Services -> protos -> Models.proto
syntax = "proto3";
option go_package = ".;Services";
package Services;
// 商品模型 通过@inject_tag注解为返回体属性设置了tag，是为了修改符合接口定义的返回体重的json数据标签
message ProdModel {
  // @inject_tag: json:"pid"
  int32 ProdId = 1;
  // @inject_tag: json:"pname"
  string ProdName = 2;
}

# micro_prod_server -> Services -> protos -> ProdService.proto
syntax="proto3";
option go_package = ".;Services";
package Services;
import "Models.proto";

message ProdsRequest {
  // @inject_tag: validate:"required" json:"size"
  int32 size = 1;
}

message ProdListResponse {
  repeated ProdModel data = 1;
}

service ProdService {
  rpc GetProdsList(ProdsRequest) returns (ProdListResponse);
}

# micro_prod_server -> gen.bat
cd Services/protos
protoc -I . --micro_out=../ --go_out=plugins=grpc:../ Models.proto
protoc --proto_path=. --proto_path=.. --micro_out=../ --go_out=plugins=grpc:../ ProdService.proto
protoc-go-inject-tag -input=../Models.pb.go
cd .. && cd ..

Services
-- protos
---- Models.proto
---- ProdService.proto
-- Models.pd.go
-- Models.pd.micro.go   # 该文件其实没有用，可以把批处理protoc第一行的 --micro_out=../ 去除，不再生成这个文件，因为Models.proto文件中，没有service的配置。
-- ProdService.pd.go
-- ProdService.pd.micro.go

package ServiceImpl

import (
	"context"
	"fmt"
	"gitee.com/banyanhouse/distill-micro-grpc/micro_prod_server/Services"
	"strconv"
)

type ProdService struct {

}

func (p *ProdService) GetProdsList(ctx context.Context, request *Services.ProdsRequest,
	response *Services.ProdListResponse) error {
	fmt.Printf("receive request:%d\n", request.Size)
	models := make([]*Services.ProdModel, 0)
    // 模拟数据取数场景
	var i int32
	for i=0; i<request.Size;i++{
		models = append(models, newProd(100+i, "prodName" + strconv.Itoa(100 + int(i))))
	}
	response.Data = models
	return nil
}

// 测试方法
func newProd(id int32, pname string) *Services.ProdModel {
	return &Services.ProdModel{
		ProdId:   id,
		ProdName: pname,
	}
}

package grpcapi


import (
	"fmt"
	"gitee.com/banyanhouse/distill-infra/apigrpc"
	"gitee.com/banyanhouse/distill-infra/register"
	"gitee.com/banyanhouse/distill-micro-grpc/micro_prod_server/ServiceImpl"
	"gitee.com/banyanhouse/distill-micro-grpc/micro_prod_server/Services"
	log "github.com/sirupsen/logrus"
)

func init() {
	apigrpc.RegisterGRpcApi(new(GRpcApi))
}

type GRpcApi struct {
}

func (g *GRpcApi) Init(grpcName, grpcPort string) {
	register.GRpcRegister(grpcName, grpcPort)
	// 获取框架go-micro的micro.Service，其内包含注册中心信息
	grpcService := register.GrpcService()
	// 通过RegisterProdServiceHandler将服务实现方法注册到注册中心中
	err := Services.RegisterProdServiceHandler(grpcService.Server(), new(ServiceImpl.ProdService))
	if err != nil {
		log.Errorf("GRpc Service impl failed:%s\n", err.Error())
	}
	// 打印查看中策中心的服务列表清单
	services, err := register.RegistryCenter().ListServices()
	if err != nil {
		log.Errorf("get registry service list failed. %s", err.Error())
	}
	for i, service := range services {
		fmt.Printf("%d -----> %s\n", i, service.Name)
	}
}

package distill_micro_http

import (
	"fmt"
	infra "gitee.com/banyanhouse/distill-infra"
	"gitee.com/banyanhouse/distill-infra/apigrpc"
	"gitee.com/banyanhouse/distill-infra/base"
	"gitee.com/banyanhouse/distill-infra/hook"
	"gitee.com/banyanhouse/distill-infra/log"
	"gitee.com/banyanhouse/distill-infra/register"
	"gitee.com/banyanhouse/distill-infra/validator"
	_ "gitee.com/banyanhouse/distill-micro-grpc/micro_prod_server/apis/grpcapi" // 将注册grpc服务包在这里进行引入，注册服务。
)

func init() {
	fmt.Println("start app...")
	infra.Register(&base.TomlPropsStarter{})
	infra.Register(&log.LoggerStarter{}) // 如果需要输出日志到文件中，则可以打开这里，去初始化文件日志
	infra.Register(&validator.ValidatorStarter{})
	infra.Register(&register.Etcd3Starter{})
	infra.Register(&apigrpc.GRpcApiStarter{}) // 启动distill-infra框架的GRpc启动器
	infra.Register(&hook.HookStarter{})
	fmt.Println("load end ...")
}

[app]
name = "micro_prod_server"
port = 18186
protocol = "grpc" #grpc http https
time = "10s"
testing = false
...

package web

import (
	"context"
	"gitee.com/banyanhouse/distill-infra/api"
	"gitee.com/banyanhouse/distill-infra/validator"
	"gitee.com/banyanhouse/distill-infra/web"
	"gitee.com/banyanhouse/distill-micro-grpc/micro-gateway/Services"
	irisContext "github.com/kataras/iris/context"
	"github.com/micro/go-micro/v2"
	log "github.com/sirupsen/logrus"
)

func init() {
	api.SetupApi(new(ProdsHandler))
}

type ProdsHandler struct {
}

func (c *ProdsHandler) Init() {
}

func (c *ProdsHandler) Setup() {
	// 路由注册
	groupRouter := web.IrisMainParty().Party("/v1/micro/grpc")
	groupRouter.Post("/prod", ProdHandler)
}

func ProdHandler(ctx irisContext.Context)  {
	var (
		res web.Res
	)
	// 开启一个go-micro服务调用客户端
	myService := micro.NewService(
		micro.Name("prod.client"),
	)
	// 声明注册中心micro_prod_server服务
	prodService := Services.NewProdService(
		"micro_prod_server",
		myService.Client(),
	)

	var prodsRequest Services.ProdsRequest
	// 接收客户端请求，以json形式加载到请求体
	err := ctx.ReadJSON(&prodsRequest)
	// 验证请求体内容
	validatorErr := validator.ValidateStruct(&prodsRequest)
	if err != nil || validatorErr != nil {
		errRes(err, validatorErr, &res)
		ctx.JSON(&res)
		return
	}
	// 调用微服务接口
	prodsList, err := prodService.GetProdsList(context.Background(), &prodsRequest)

	if err != nil {
		errRes(err, nil, &res)
		ctx.JSON(&res)
		return
	}
	// 响应REST数据
	res = web.Res{
		Code: web.ResCodeOk,
		Message: "商品列表",
		Data: prodsList.GetData(),
	}
	ctx.JSON(&res)

}
// 错误异常处理
func errRes(err error, validatorErr error, res *web.Res) {
	var paramErr error
	if err != nil {
		paramErr = err
	}
	if validatorErr != nil {
		paramErr = validatorErr
	}
	log.Error(paramErr)
	res.Code = web.ResCodeRequestParamsError
	res.Message = "参数错误:" + paramErr.Error()
}

[app]
name = "micro_gateway"
port = 18081
protocol = "http" #grpc http https
time = "10s"
testing = false
...