自动生成代码介绍

protoc-gen-mqant 是一款可以自动生成rpc服务注册和调用的工具，其基于mqant微服务框架作为构建模板，通过读取Protobuffer的message和serveice构建服务注册和调用

构建生成代码工具

第一种方式

1. 下载代码到本地

    git clone https://github.com/GodWY/protoc-gen-mqant.git

2. 执行 make {电脑型号}，构建protoc-gen-mqant 可执行文件

第二种方式

 go get -u github.com/GodWY/protoc-gen-mqant@latest

 go install github.com/GodWY/protoc-gen-mqant

or

go install  github.com/GodWY/protoc-gen-mqant@latest

总结: 下载代码目的是将二进制文件下载到gopath的bin文件下。

构建一个服务

1. 定义请求greeter.proto文件

    syntax = "proto3";
    package examples;
    option go_package = "proto/examples/greeter";


    message  Request {
    }

    message Response {

    }

2. 定义一个rpc请求

    syntax = "proto3";
    package examples1;
    option go_package = "proto/greeter1";
    import "proto/examples/greeter.proto";


    message Response {

    }

    service Greeter {
        rpc Hello(examples.Request) returns (Response) {}
        rpc Stream(examples.Request) returns ( Response) {
        }
    }

3. 执行构建代码

    protoc --plugin=protoc-gen-go=/Users/GodWY/go/bin/protoc-gen-go   --go_out=./  ./proto/examples/*.proto --experimental_allow_proto3_optional
    protoc --plugin=protoc-gen-mqant=protoc-gen-mqant  --proto_path=.  --mqant_out=./   ./proto/examples/*.proto ./proto/examples1/*.proto --experimental_allow_proto3_optional

3. 生成的代码

    // Code generated by protoc-gen-go-hi. DO NOT EDIT.
// versions:

package greeter1

import (
	greeter "proto/examples/greeter"
)

// This is a compile-time assertion to ensure that this generated file
// is compatible with the kratos package it is being compiled against.
import (
	"errors"
	basemodule "github.com/liangdas/mqant/module/base"
	client "github.com/liangdas/mqant/module"
	mqrpc "github.com/liangdas/mqant/rpc"
	"golang.org/x/net/context"
)

// generated mqant method
type Greeter interface {
	//  @[email protected]()
	Hello(in *greeter.Request) (out *Response, err error)
	//  @POST
	Stream(in *greeter.Request) (out *Response, err error)
}

func RegisterGreeterTcpHandler(m *basemodule.BaseModule, ser Greeter) {
	m.GetServer().RegisterGO("hello", ser.Hello)
	m.GetServer().RegisterGO("stream", ser.Stream)
}

// generated proxxy handle
type ClientProxyService struct {
	cli  client.App
	name string
}

func NewLoginClient(cli client.App, name string) *ClientProxyService {
	return &ClientProxyService{
		cli:  cli,
		name: name,
	}
}

var ClientProxyIsNil = errors.New("proxy is nil")

func (proxy *ClientProxyService) Hello(req *greeter.Request) (rsp *Response, err error) {
	if proxy == nil {
		return nil, ClientProxyIsNil
	}
	rsp = &Response{}
	err = mqrpc.Proto(rsp, func() (reply interface{}, err interface{}) {
		return proxy.cli.Call(context.TODO(), proxy.name, "hello", mqrpc.Param(req))
	})
	return rsp, err
}
func (proxy *ClientProxyService) Stream(req *greeter.Request) (rsp *Response, err error) {
	if proxy == nil {
		return nil, ClientProxyIsNil
	}
	rsp = &Response{}
	err = mqrpc.Proto(rsp, func() (reply interface{}, err interface{}) {
		return proxy.cli.Call(context.TODO(), proxy.name, "stream", mqrpc.Param(req))
	})
	return rsp, err
}
 

在mqant 初始化函数里调用RegisterGreeterTcpHandler方法可以注册协议，同时如果想调用rpc协议。创建NewLoginClient代理类调用相应的rpc即可