chai2010 / protorpc

prtorpc(2013-2021): Google Protocol Protobufs RPC for Go.
http://godoc.org/github.com/chai2010/protorpc
BSD 3-Clause "New" or "Revised" License
59 stars 14 forks source link

protorpc

██████╗ ██████╗  ██████╗ ████████╗ ██████╗       ██████╗ ██████╗  ██████╗
██╔══██╗██╔══██╗██╔═══██╗╚══██╔══╝██╔═══██╗      ██╔══██╗██╔══██╗██╔════╝
██████╔╝██████╔╝██║   ██║   ██║   ██║   ██║█████╗██████╔╝██████╔╝██║
██╔═══╝ ██╔══██╗██║   ██║   ██║   ██║   ██║╚════╝██╔══██╗██╔═══╝ ██║
██║     ██║  ██║╚██████╔╝   ██║   ╚██████╔╝      ██║  ██║██║     ╚██████╗
╚═╝     ╚═╝  ╚═╝ ╚═════╝    ╚═╝    ╚═════╝       ╚═╝  ╚═╝╚═╝      ╚═════╝

Build Status GoDoc

Install

Install protorpc package:

  1. go install github.com/golang/protobuf/protoc-gen-go
  2. go get github.com/chai2010/protorpc
  3. go run hello.go

Install protoc-gen-go plugin:

  1. install protoc at first: http://github.com/google/protobuf/releases
  2. go get github.com/golang/protobuf/protoc-gen-go
  3. go get github.com/chai2010/protorpc/protoc-gen-protorpc
  4. go generate github.com/chai2010/protorpc/examples/service.pb
  5. go test github.com/chai2010/protorpc/examples/service.pb

Examples

First, create echo.proto:

syntax = "proto3";

package service;

message EchoRequest {
    string msg = 1;
}

message EchoResponse {
    string msg = 1;
}

service EchoService {
    rpc Echo (EchoRequest) returns (EchoResponse);
    rpc EchoTwice (EchoRequest) returns (EchoResponse);
}

Second, generate echo.pb.go and echo.pb.protorpc.go from echo.proto (we can use go generate to invoke this command, see proto.go).

protoc --go_out=. echo.proto
protoc --protorpc_out=. echo.proto

Now, we can use the stub code like this:

package main

import (
    "fmt"
    "log"

    "github.com/chai2010/protorpc"
    service "github.com/chai2010/protorpc/examples/service.pb"
)

type Echo int

func (t *Echo) Echo(args *service.EchoRequest, reply *service.EchoResponse) error {
    reply.Msg = args.Msg
    return nil
}

func (t *Echo) EchoTwice(args *service.EchoRequest, reply *service.EchoResponse) error {
    reply.Msg = args.Msg + args.Msg
    return nil
}

func init() {
    go service.ListenAndServeEchoService("tcp", `127.0.0.1:9527`, new(Echo))
}

func main() {
    echoClient, err := service.DialEchoService("tcp", `127.0.0.1:9527`)
    if err != nil {
        log.Fatalf("service.DialEchoService: %v", err)
    }
    defer echoClient.Close()

    args := &service.EchoRequest{Msg: "你好, 世界!"}
    reply, err := echoClient.EchoTwice(args)
    if err != nil {
        log.Fatalf("echoClient.EchoTwice: %v", err)
    }
    fmt.Println(reply.Msg)

    // or use normal client
    client, err := protorpc.Dial("tcp", `127.0.0.1:9527`)
    if err != nil {
        log.Fatalf("protorpc.Dial: %v", err)
    }
    defer client.Close()

    echoClient1 := &service.EchoServiceClient{client}
    echoClient2 := &service.EchoServiceClient{client}
    reply, err = echoClient1.EchoTwice(args)
    reply, err = echoClient2.EchoTwice(args)
    _, _ = reply, err

    // Output:
    // 你好, 世界!你好, 世界!
}

More examples.

standard net/rpc

First, create echo.proto:

syntax = "proto3";

package service;

message EchoRequest {
    string msg = 1;
}

message EchoResponse {
    string msg = 1;
}

service EchoService {
    rpc Echo (EchoRequest) returns (EchoResponse);
    rpc EchoTwice (EchoRequest) returns (EchoResponse);
}

Second, generate echo.pb.go from echo.proto with protoc-gen-stdrpc plugin.

protoc --stdrpc_out=. echo.proto

The stdrpc plugin generated code do not depends protorpc package, it use gob as the default rpc encoding.

Add prefix

$ ENV_PROTOC_GEN_PROTORPC_FLAG_PREFIX=abc protoc --protorpc_out=. x.proto

BUGS

Report bugs to chaishushan@gmail.com.

Thanks!