An open API service indexing awesome lists of open source software.

https://github.com/hyman-m/tinyrpc

🚀 Fast, stable, mini RPC framework based on protocol buffer
https://github.com/hyman-m/tinyrpc

golang protobuf rpc

Last synced: 4 months ago
JSON representation

🚀 Fast, stable, mini RPC framework based on protocol buffer

Awesome Lists containing this project

README

        

# tinyrpc

[![Go Report Card](https://goreportcard.com/badge/github.com/wanzoma/tinyrpc)](https://goreportcard.com/report/github.com/wanzoma/tinyrpc) ![GitHub top language](https://img.shields.io/github/languages/top/wanzoma/tinyrpc) ![GitHub](https://img.shields.io/github/license/wanzoma/tinyrpc) [![CodeFactor](https://www.codefactor.io/repository/github/wanzoma/tinyrpc/badge)](https://www.codefactor.io/repository/github/wanzoma/tinyrpc) [![codecov](https://codecov.io/gh/wanzoma/tinyrpc/branch/main/graph/badge.svg)](https://codecov.io/gh/wanzoma/tinyrpc)  ![go_version](https://img.shields.io/badge/go%20version-1.17-yellow)

tinyrpc is a high-performance RPC framework based on `protocol buffer` encoding. It is based on `net/rpc` and supports multiple compression formats (`gzip`, `snappy`, `zlib`).

Language  English

## Install

- install `protoc` at first :http://github.com/google/protobuf/releases

- install `protoc-gen-go` and `protoc-gen-tinyrpc`:

```
go install github.com/golang/protobuf/protoc-gen-go@latest
go install github.com/wanzoma/tinyrpc/protoc-gen-tinyrpc@latest
```

## Quick Start
Fisrt, create a demo and import the tinyrpc package:
```shell
> go mod init demo
> go get github.com/zehuamama/tinyrpc
```
under the path of the project, create a protobuf file `arith.proto`:
```protobuf
syntax = "proto3";

package message;
option go_package="/message";

// ArithService Defining Computational Digital Services
service ArithService {
// Add addition
rpc Add(ArithRequest) returns (ArithResponse);
// Sub subtraction
rpc Sub(ArithRequest) returns (ArithResponse);
// Mul multiplication
rpc Mul(ArithRequest) returns (ArithResponse);
// Div division
rpc Div(ArithRequest) returns (ArithResponse);
}

message ArithRequest {
int32 a = 1;
int32 b = 2;
}

message ArithResponse {
int32 c = 1;
}
```
an arithmetic operation service is defined here, using `protoc` to generate code:
```shell
> protoc --tinyrpc_out=. arith.proto --go_out=. arith.proto
```
at this time, two files will be generated in the directory `message`: `arith.pb.go` and `arith.svr.go`

you can also use
```shell
> protoc --tinyrpc_out=gen-cli=true:. arith.proto --go_out=. arith.proto
```
to generate `arith.cli.go`, this file make it easier to call the server.

the code of `arith.svr.go` is shown below:
```go
// Code generated by protoc-gen-tinyrpc.

package message

// ArithService Defining Computational Digital Services
type ArithService struct{}

// Add addition
func (this *ArithService) Add(args *ArithRequest, reply *ArithResponse) error {
// define your service ...
return nil
}

// Sub subtraction
func (this *ArithService) Sub(args *ArithRequest, reply *ArithResponse) error {
// define your service ...
return nil
}

// Mul multiplication
func (this *ArithService) Mul(args *ArithRequest, reply *ArithResponse) error {
// define your service ...
return nil
}

// Div division
func (this *ArithService) Div(args *ArithRequest, reply *ArithResponse) error {
// define your service ...
return nil
}
```
we need to define our services.

Finally, under the path of the project, we create a file named `main.go`, the code is shown below:
```go
package main

import (
"demo/message"
"log"
"net"

"github.com/zehuamama/tinyrpc"
)

func main() {
lis, err := net.Listen("tcp", ":8082")
if err != nil {
log.Fatal(err)
}

server := tinyrpc.NewServer()
server.RegisterName("ArithService", new(message.ArithService))
server.Serve(lis)
}
```
a tinyrpc server is completed.

## Client
We can create a tinyrpc client and call it synchronously with the `Add` function:
```go
import (
"demo/message"
"github.com/zehuamama/tinyrpc"
)
...

conn, err := net.Dial("tcp", ":8082")
if err != nil {
log.Fatal(err)
}
defer conn.Close()
client := tinyrpc.NewClient(conn)
resq := message.ArithRequest{A: 20, B: 5}
resp := message.ArithResponse{}
err = client.Call("ArithService.Add", &resq, &resp)
log.Printf("Arith.Add(%v, %v): %v ,Error: %v", resq.A, resq.B, resp.C, err)
```
you can also call asynchronously, which will return a channel of type *rpc.Call:
```go

result := client.AsyncCall("ArithService.Add", &resq, &resp)
select {
case call := <-result:
log.Printf("Arith.Add(%v, %v): %v ,Error: %v", resq.A, resq.B, resp.C, call.Error)
case <-time.After(100 * time.Microsecond):
log.Fatal("time out")
}
```
of course, you can also compress with three supported formats `gzip`, `snappy`, `zlib`:
```go
import "github.com/zehuamama/tinyrpc/compressor"

...
client := tinyrpc.NewClient(conn, tinyrpc.WithCompress(compressor.Gzip))

```
## Custom Serializer
If you want to customize the serializer, you must implement the `Serializer` interface:
```go
type Serializer interface {
Marshal(message interface{}) ([]byte, error)
Unmarshal(data []byte, message interface{}) error
}
```
`JsonSerializer` is a serializer based Json:
```go
type JsonSerializer struct{}

func (_ JsonSerializer) Marshal(message interface{}) ([]byte, error) {
return json.Marshal(message)
}

func (_ JsonSerializer) Unmarshal(data []byte, message interface{}) error {
return json.Unmarshal(data, message)
}
```
moving on, we create a HelloService with the following code:
```go
type HelloRequest struct {
Req string `json:"req"`
}

type HelloResponce struct {
Resp string `json:"resp"`
}

type HelloService struct{}

func (_ *HelloService) SayHello(args *HelloRequest, reply *HelloResponce) error {
reply.Resp = args.Req
return nil
}

```
finally, we need to set the serializer on the rpc server:
```go
func main() {
lis, err := net.Listen("tcp", ":8082")
if err != nil {
log.Fatal(err)
}

server := tinyrpc.NewServer(tinyrpc.WithSerializer(JsonSerializer{}))
server.Register(new(HelloService))
server.Serve(lis)
}
```
a rpc server based on json serializer is completed.

Remember that when the rpc client calls the service, it also needs to set the serializer:
```go
tinyrpc.NewClient(conn,tinyrpc.WithSerializer(JsonSerializer{}))
```
## Contributing

If you are intersted in contributing to tinyrpc, please see here: [CONTRIBUTING](https://github.com/zehuamama/tinyrpc/blob/main/CONTRIBUTING.md)

## License

tinyrpc is licensed under the term of the [BSD 2-Clause License](https://github.com/zehuamama/tinyrpc/blob/main/LICENSE)