当前位置： 首页 > article >正文

go-zero 的使用

article 2024/10/29 4:10:51

目录

1. 生成 user api 服务

2. 生成 user rpc 服务

3. 生成 user model 模型

4. 编写 user rpc 服务

1 修改配置文件 user.yaml

2 添加 user model 依赖

3 添加用户登录逻辑 Login

5. 编写 user api 服务

1 修改配置文件user.yaml

2 添加 user rpc 依赖

3 添加用户登录逻辑 Login

6. 启动服务

启动rpc服务

启动api服务

本文章讲解使用go-zero生成一个微服务的过程。其中的一些命令或者名不太熟悉的话，可以参考go-zero官网。

项目名字叫zero。创建zero目录，并在该目录下创建user目录。

1. 生成 user api 服务

在user目录下添加api目录，在api目录添加如下的api文件。 

syntax = "v1"

type LoginReq {
	UserName string `json:"username"`
	Password string `json:"password"`
}

type LoginResponse {
	Token string `json:"token"`
}

@server (
	prefix: v1
)
service user {
	@handler login
	post /user/login (LoginReq) returns (LoginResponse)
}

 在user/api下执行

goctl go api --api ./user.api --dir .

执行生成后，api目录结构体如下

.
├── etc
│   └── user.yaml
├── internal
│   ├── config
│   │   └── config.go
│   ├── handler
│   │   ├── loginhandler.go
│   │   └── routes.go
│   ├── logic
│   │   └── loginlogic.go
│   ├── svc
│   │   └── servicecontext.go
│   └── types
│       └── types.go
├── user.api
└── user.go

2. 生成 user rpc 服务

在user目录下创建rpc目录。在rpc目录下添加如下的proto文件。

syntax="proto3";

package user;

option go_package="./user";

//用户登录

message LoginRequest{
    string username=1;
    string Password=2;
}

message LoginResponse{
    int64 Id=1;
    string Username=2;
}

service User{
    rpc Login(LoginRequest)returns(LoginResponse);
}

在user/rpc下执行

goctl rpc protoc ./user.proto --go_out=. --go-grpc_out=. --zrpc_out=.

 执行生成后，rpc目录结构如下

.
├── etc
│   └── user.yaml
├── internal
│   ├── code
│   │   └── code.go
│   ├── config
│   │   └── config.go
│   ├── db
│   │   └── mysql.go
│   ├── logic
│   │   └── loginlogic.go
│   ├── server
│   │   └── userserver.go
│   └── svc
│       └── servicecontext.go
├── user
│   ├── user_grpc.pb.go
│   └── user.pb.go
├── userclient
│   └── user.go
├── user.go
└── user.proto

3. 生成 user model 模型

在user目录添加model目录，在model中添加如下的user.sql文件。

 CREATE TABLE `user` (
  `id` bigint NOT NULL AUTO_INCREMENT,
  `username` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci NOT NULL,
  `password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci NOT NULL,
  PRIMARY KEY (`id`) USING BTREE
) ENGINE=InnoDB AUTO_INCREMENT=26;

在model目录中执行

goctl model mysql ddl --src user.sql --dir .

 执行生成后，model目录结构体如下

.
├── usermodel_gen.go
├── usermodel.go
├── user.sql
└── vars.go

4. 编写 user rpc 服务

1 修改配置文件 user.yaml

因为需要使用数据库MySql，所以我们需要配置数据库的参数,所以需要在user.yaml中填写数据库相关的参数。而因为rpc是需要用到Etcd（用于服务注册发现等），所以还需要添加Etcd的配置。

Name: user.rpc
ListenOn: 0.0.0.0:8080

#Etcd部分和Mysql部分是新添加的
Etcd:
  Hosts:
  - 127.0.0.1:2379
  Key: user.rpc

Mysql:
  datasource: "root:wook1847@tcp(127.0.0.1:3306)/zero?charset=utf8mb4&parseTime=True&loc=Local"

2 添加 user model 依赖

  • 添加 Mysql 服务配置的实例化

在生成的config目录中有config.go文件，该文件中是用于解析配置文件的结构体。而这里我们添加了MySql配置，所以需要添加相关结构体来进行解析。而rest.RestConf中就带有etcd的(看源码），所以不用写etcd的。

type Config struct {
	zrpc.RpcServerConf

	MysqlConfig Mysql `json:"mysql"`    //添加mysql的配置
}

type Mysql struct {
	Datasource    string
	ConectTimeout int64
}
  • 注册服务上下文 user model 的依赖

在生成是svc目录中的 servicecontext.go文件，后续我们就是使用该文件中的ServiceContext。而要想可以使用mysql，那就需要在其添加该依赖。

type ServiceContext struct {
	Config config.Config
	Conn   sqlx.SqlConn    //这个就是新添加的依赖
}

func NewServiceContext(c config.Config) *ServiceContext {
	conn := sqlx.NewMysql(c.MysqlConfig.Datasource)
	return &ServiceContext{
		Config: c,
		Conn:   conn,
	}
}

3 添加用户登录逻辑 Login

接着看生成的logic目录中的loginlogic.go文件。这里就是我们要写的业务逻辑地方。logic目录中的文件是我们主要写代码的地方。在Login方法中添加业务逻辑。

type LoginLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic {
	return &LoginLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *LoginLogic) Login(in *user.LoginRequest) (*user.LogResponse, error) {
	// todo: add your logic here and delete this line
	res, err := model.NewUserModel(l.svcCtx.Conn).FindOneByNameAndPwd(l.ctx, in.Username, in.Password)
	if err != nil {
		fmt.Println("login rpc err.........................:", err)
		if err == model.ErrNotFound {
			return nil, err
		}
		return nil, err
	}

	return &user.LogResponse{Id: res.Id, Username: res.Username}, nil
}

5. 编写 user api 服务

其过程和编写user rpc服务是类似的。

1 修改配置文件user.yaml

因为我们在api中需要使用rpc，所以我们就需要用到etcd，所以就需要etcd的配置。

Name: user
Host: 0.0.0.0
Port: 8888


#添加user rpc的etcd配置
UserRpc:
  Etcd:
    Hosts:
    - 127.0.0.1:2379
    Key: user.rpc


#要是还需要其他服务的rpc，那就需要添加的
# PayRpc:
#   Etcd:
#     Hosts:
#     - 127.0.0.1:2379
#     Key: pay.rpc

2 添加 user rpc 依赖

  • 添加 user rpc 服务配置的实例化

在user.yaml中添加了user rpc的etcd配置后，那就需要再config.go文件中添加对应的结构体来解析配置。

而前面不是说了，etcd的结构体是rest.RestConf就带有的，不用填写了吗？

这不是这个意思的。这个是对应user rpc的etcd配置的。要是在api中还需要用其他服务，比如支付服务，而支付服务又是一个微服务，那就又要添加pay rpc的etcd配置的。

type Config struct {
	rest.RestConf
	UserRpc zrpc.RpcClientConf

	// PayRpc zrpc.RpcClientConf	要是需要该服务的话，就需要添加
}
  • 添加 user rpc 服务配置的实例化
type ServiceContext struct {
	Config config.Config

	UserRpc userclient.User    //添加user rpc使用
}

func NewServiceContext(c config.Config) *ServiceContext {
	return &ServiceContext{
		Config:  c,
		UserRpc: userclient.NewUser(zrpc.MustNewClient(c.UserRpc)),
	}
}

3 添加用户登录逻辑 Login

在生成的logic目录中的loginlogic.文件。这里就是我们要写的业务逻辑。

type LoginLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic {
	return &LoginLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

//在该方法中添加我们的业务逻辑
func (l *LoginLogic) Login(req *types.LoginReq) (resp *types.LoginResponse, err error) {
	// todo: add your logic here and delete this line

	_, err = l.svcCtx.UserRpc.Login(l.ctx, &userclient.LoginRequest{
		Username: req.UserName,
		Password: req.Password,
	})

	if err != nil {
		return nil, err
	}

	return &types.LoginResponse{Token: "123445"}, nil
}

6. 启动服务

因为api服务是依赖rpc服务，所以先启动rpc服务，再启动api服务。

启动rpc服务

在user/api中执行

go run user.go

启动api服务

在user/rpc中执行

go run user.go

http://www.kler.cn/a/369624.html

相关文章：

  • 【学术会议论文投稿】Spring全家桶使用深度教程：从入门到精通
  • Vue Filter 使用指南：让数据展示更加灵活
  • Netty的简介与实战
  • MobileNetV2实现实时口罩检测tensorflow
  • ML 系列：机器学习和深度学习的深层次总结(17)从样本空间到概率规则概率
  • Spring-Day2
  • 深入解析 OceanBase 数据库中的局部索引和全局索引
  • 模型训练识别手写数字（一）
  • 深入理解支持向量机：从基本原理到实际应用
  • 数据库中少数民族名字的存储
  • vue子组件修改父组件的值
  • #渗透测试#红蓝对抗#Src漏洞挖掘 介绍-Yakit（3）
  • 深度学习：SGD的缺点
  • redis学习路线和内容
  • Javascript立即执行函数
  • 微服务之间调用，OpenFeign传递用户（RequestInterceptor接口）
  • JavaSet集合
  • [332]基于springboot的养老院管理系统
  • 基于SpringBoot的洗浴管理系统【附源码】
  • 400行程序写一个实时操作系统（十七）：调度算法的实现
  • Sqoop的安装配置及使用
  • 梧桐数据库锁处理过程
  • Cesium基础-（Entity）-（point）
  • （STM32笔记）十二、DMA的基础知识与用法
  • OBOO鸥柏丨液晶拼接大屏KVM分布式输入输出节点控制系统技术
  • C/C++每日一练：实现冒泡排序