基于go-zero的rpc服务示例

以下是一个基于 go-zero 框架的简单 RPC 服务示例,该示例包括一个服务端和一个客户端通过 gRPC 进行通信。

服务端

1、定义 .proto 文件

在 rpc/add 目录下创建 adder.proto 文件,定义 RPC 服务:

proto 复制代码
syntax = "proto3";

package adder;

service Adder {
  rpc Add(AddRequest) returns (AddResponse) {}
}

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

message AddResponse {
  int32 result = 1;
}

2、生成代码

使用 protoc 工具生成 Go 代码:

sh 复制代码
protoc --go_out=plugins=grpc:. adder.proto

3、实现服务

创建 adder_grpc.go 文件,实现 Adder 服务:

go 复制代码
package adder

import (
	"context"
)

type AdderServer struct{}

func (s *AdderServer) Add(ctx context.Context, req *AddRequest) (*AddResponse, error) {
	result := req.GetA() + req.GetB()
	return &AddResponse{Result: result}, nil
}

4、启动服务

创建 server.go 文件,启动 gRPC 服务:

go 复制代码
package main

import (
	"flag"
	"fmt"
	"net"

	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/zrpc"
	"github.com/your/path/to/rpc/add"
)

var configFile = flag.String("f", "etc/config.yaml", "the config file")

func main() {
	flag.Parse()

	var c zrpc.RpcServerConf
	conf.MustLoad(*configFile, &c)

	server := zrpc.MustNewServer(c)
	defer server.Stop()

	adderServer := &add.AdderServer{}
	add.RegisterAdderServer(server.Server, adderServer)

	fmt.Printf("Starting rpc server on %s...\n", c.ListenOn)
	server.Start()
}

5、配置文件

在 etc 目录下创建 config.yaml 文件,配置服务地址和端口:

yaml 复制代码
Name: adder
ListenOn: 127.0.0.1:8080

6、启动服务端

运行 server.go 文件,启动 RPC 服务:

sh 复制代码
go run server.go

客户端

1、创建客户端

创建 client.go 文件,创建 gRPC 客户端:

go 复制代码
package main

import (
	"context"
	"fmt"
	"time"

	"github.com/zeromicro/go-zero/zrpc"
	"github.com/your/path/to/rpc/add"
)

func main() {
	client := zrpc.MustNewClient("127.0.0.1:8080")
	defer client.Conn.Close()

	adderClient := add.NewAdderClient(client.Conn)
	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()

	req := &add.AddRequest{
		A: 5,
		B: 7,
	}
	resp, err := adderClient.Add(ctx, req)
	if err != nil {
		fmt.Println("Error:", err)
	} else {
		fmt.Println("Result:", resp.GetResult())
	}
}

2、运行客户端

运行 client.go 文件,调用 RPC 服务:

sh 复制代码
go run client.go

输出结果:

复制代码
Result: 12

至此,一个基于 go-zero 框架的简单 RPC 服务示例完成。

相关推荐
徐小黑ACG1 小时前
GO语言 使用protobuf
开发语言·后端·golang·protobuf
能来帮帮蒟蒻吗12 小时前
GO语言学习(16)Gin后端框架
开发语言·笔记·学习·golang·gin
JavaPub-rodert12 小时前
一道go面试题
开发语言·后端·golang
6<713 小时前
【go】静态类型与动态类型
开发语言·后端·golang
weixin_4209476416 小时前
windows golang,consul,grpc学习
windows·golang·consul
Json201131516 小时前
Gin、Echo 和 Beego三个 Go 语言 Web 框架的核心区别及各自的优缺点分析,结合其设计目标、功能特性与适用场景
前端·golang·gin·beego
Zfox_18 小时前
【C++项目】从零实现RPC框架「四」:业务层实现与项目使用
linux·开发语言·c++·rpc·项目
二狗哈21 小时前
go游戏后端开发21:处理nats消息
开发语言·游戏·golang
能来帮帮蒟蒻吗1 天前
Go语言学习(15)结构体标签与反射机制
开发语言·笔记·学习·golang
returnShitBoy1 天前
Go语言中的defer关键字有什么作用?
开发语言·后端·golang