基于go-zero二次开发的脚本

param=$2
# 字符串风格格式为:DemoName
model_name=$(echo "${param}" | awk -F '_' '{for(i=1;i<=NF;i++) $i=toupper(substr($i,1,1)) tolower(substr($i,2));}1' | tr -d ' ')
# 字符串风格格式为:demoName
struct_name=$(echo "${model_name}" | awk '{print tolower(substr($0,1,1)) substr($0,2)}')
help()
{cat <<- EOF帮助文档:Desc: 基于goctl指令生成代码程序选项如下:api:生成网关和swagger文档,无参数rpc:生成rpc代码和proto文件,参数1:rpc文件名(不带扩展名)proto 只生成base目录下的proto文件,注意路径,参数1:basemysql:生成model和crud代码,参数1:源文件名(不带扩展名)docker:打包docker镜像,参数1:main文件名(不带扩展名),参数2:版本号Usage:./generate_code.sh api./generate_code.sh rpc user./generate_code.sh proto base./generate_code.sh mysql user./generate_code.sh mysql user v1.0.0
EOFexit 0
}# 生成CRUD的方法
gen()
{# 获取要修改的文件名filename=./dao/mysql/model/$(echo "${param}" | tr -d '_')model.go# 获取行号,用来判断是否需要修改line_number=$(sed -n '$=' ${filename})if [ "$line_number" -lt 30 ]; thencat > "$filename" <<- EOF
package modelimport ("context""database/sql""errors""github.com/doug-martin/goqu/v9"_ "github.com/doug-martin/goqu/v9/dialect/mysql""github.com/zeromicro/go-zero/core/stores/sqlx""upgames-go-microservices/utils/utils"
)var _ ${model_name}Model = (*custom${model_name}Model)(nil)type (// ${model_name}Model is an interface to be customized, add more methods here,// and implement the added methods in custom${model_name}Model.${model_name}Model interface {${struct_name}ModelwithSession(session sqlx.Session) ${model_name}Model// GetTableName 获取表名GetTableName() string// GetCount 根据条件获取数量GetCount(ctx context.Context, ex goqu.Expression) (int64, error)// FindList 根据条件获取列表,排序:map[string]int{"字段":0/1(0-升序(ASC);1-降序(DESC))};分页:[]uint{页码,每页条数}FindList(ctx context.Context, ex goqu.Expression, optionalParams ...any) (*[]${model_name}, error)// FindOnly 根据条件获取单条数据,0-升序(ASC);1-降序(DESC)FindOnly(ctx context.Context, ex goqu.Expression, order ...map[string]int) (*${model_name}, error)// InsertOnly 插入单条数据InsertOnly(ctx context.Context, row *${model_name}, tx ...*sql.Tx) (sql.Result, error)// BatchInsert 批量插入BatchInsert(ctx context.Context, rows []*${model_name}, tx ...*sql.Tx) (sql.Result, error)// UpdateByEx 根据条件更新UpdateByEx(ctx context.Context, record goqu.Record, ex goqu.Expression, tx ...*sql.Tx) (sql.Result, error)// DeleteByEx 根据条件删除数据DeleteByEx(ctx context.Context, ex goqu.Expression, tx ...*sql.Tx) (sql.Result, error)}custom${model_name}Model struct {*default${model_name}Model}
)// New${model_name}Model returns a model for the database table.
func New${model_name}Model(conn sqlx.SqlConn) ${model_name}Model {return &custom${model_name}Model{default${model_name}Model: new${model_name}Model(conn),}
}func (m *custom${model_name}Model) withSession(session sqlx.Session) ${model_name}Model {return New${model_name}Model(sqlx.NewSqlConnFromSession(session))
}// GetTableName 获取表名
func (m *custom${model_name}Model) GetTableName() string {return utils.SetTable(m.table)
}// GetCount 根据条件获取数量
func (m *custom${model_name}Model) GetCount(ctx context.Context, ex goqu.Expression) (int64, error) {query, _, err := goqu.Dialect("mysql").Select(goqu.COUNT(1)).From(utils.SetTable(m.table)).Where(ex).ToSQL()if err != nil {return 0, err}var resp int64err = m.conn.QueryRowCtx(ctx, &resp, query)if err != nil && !errors.Is(err, sqlx.ErrNotFound) {return 0, err}return resp, nil
}// FindList 根据条件获取列表,排序:map[string]int{"字段":0/1(0-升序(ASC);1-降序(DESC))};分页:[]uint{页码,每页条数}
func (m *custom${model_name}Model) FindList(ctx context.Context, ex goqu.Expression, optionalParams ...any) (*[]${model_name}, error) {sql := goqu.Dialect("mysql").Select(&${model_name}{}).From(utils.SetTable(m.table)).Where(ex)if len(optionalParams) > 0 {for _, param := range optionalParams {// 排序if v, ok := param.(map[string]int); ok {for key, value := range v {if value > 0 {sql = sql.Order(goqu.C(key).Desc())} else {sql = sql.Order(goqu.C(key).Asc())}}}// 分页if v, ok := param.([]uint); ok {if len(v) == 2 {sql = sql.Offset((v[0] - 1) * v[1]).Limit(v[1])}}}}query, _, err := sql.ToSQL()if err != nil {return nil, err}var resp []${model_name}err = m.conn.QueryRowsCtx(ctx, &resp, query)if err != nil && !errors.Is(err, sqlx.ErrNotFound) {return nil, err}return &resp, nil
}// FindOnly 根据条件获取单条数据,0-升序(ASC);1-降序(DESC)
func (m *custom${model_name}Model) FindOnly(ctx context.Context, ex goqu.Expression, order ...map[string]int) (*${model_name}, error) {sql := goqu.Dialect("mysql").Select(&${model_name}{}).From(utils.SetTable(m.table)).Where(ex)if len(order) > 0 {for key, value := range order[0] {if value > 0 {sql.Order(goqu.C(key).Desc())} else {sql.Order(goqu.C(key).Asc())}}}query, _, err := sql.Limit(1).ToSQL()if err != nil {return nil, err}var resp ${model_name}err = m.conn.QueryRowCtx(ctx, &resp, query)switch err {case nil:return &resp, nilcase sqlx.ErrNotFound:return nil, ErrNotFounddefault:return nil, err}
}// InsertOnly 插入单条数据
func (m *custom${model_name}Model) InsertOnly(ctx context.Context, row *${model_name}, tx ...*sql.Tx) (sql.Result, error) {query, _, err := goqu.Dialect("mysql").Insert(utils.SetTable(m.table)).Rows(row).ToSQL()if err != nil {return nil, err}var result sql.Resultif len(tx) > 0 {result, err = tx[0].ExecContext(ctx, query)} else {result, err = m.conn.ExecCtx(ctx, query)}return result, err
}// BatchInsert 批量插入
func (m *custom${model_name}Model) BatchInsert(ctx context.Context, rows []*${model_name}, tx ...*sql.Tx) (sql.Result, error) {query, _, err := goqu.Dialect("mysql").Insert(utils.SetTable(m.table)).Rows(rows).ToSQL()if err != nil {return nil, err}var result sql.Resultif len(tx) > 0 {result, err = tx[0].ExecContext(ctx, query)} else {result, err = m.conn.ExecCtx(ctx, query)}return result, err
}// UpdateByEx 根据条件更新
func (m *custom${model_name}Model) UpdateByEx(ctx context.Context, record goqu.Record, ex goqu.Expression, tx ...*sql.Tx) (sql.Result, error) {query, _, err := goqu.Dialect("mysql").Update(utils.SetTable(m.table)).Set(record).Where(ex).ToSQL()if err != nil {return nil, err}var result sql.Resultif len(tx) > 0 {result, err = tx[0].ExecContext(ctx, query)} else {result, err = m.conn.ExecCtx(ctx, query)}return result, err
}// DeleteByEx 根据条件删除数据
func (m *custom${model_name}Model) DeleteByEx(ctx context.Context, ex goqu.Expression, tx ...*sql.Tx) (sql.Result, error) {query, _, err := goqu.Dialect("mysql").Delete(utils.SetTable(m.table)).Where(ex).ToSQL()if err != nil {return nil, err}var result sql.Resultif len(tx) > 0 {result, err = tx[0].ExecContext(ctx, query)} else {result, err = m.conn.ExecCtx(ctx, query)}return result, err
}
EOFelseecho "行数不准确,crud代码已生成"fi
}dockerfile()
{cat > "Dockerfile" <<- EOF
FROM golang:1.22-alpine AS builderLABEL stage=gobuilderENV GOPROXY https://goproxy.cn,direct
RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.aliyun.com/g' /etc/apk/repositoriesRUN apk update --no-cache && apk add --no-cache tzdata# 设置工作目录
WORKDIR /build# 加载依赖
ADD go.mod .
ADD go.sum .
RUN go mod download# 复制源代码
COPY . .# 静态编译Go程序
RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o app ./service/${struct_name}Service/${param}.go# 第二阶段:运行时镜像,使用空镜像
FROM scratchCOPY --from=builder /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt# 设置工作目录
WORKDIR /app# 复制编译好的二进制文件到运行时镜像
COPY --from=builder /build/app .# 运行程序
ENTRYPOINT ["./app"]
# 设置CMD指令来指定参数,默认测试环境的etcd
CMD ["16.162.220.93:2379"]
EOF
}if [ "$1" == "rpc" ]; thengoctl rpc protoc ./proto/source/${param}.proto --go_out=./proto/generate --go-grpc_out=./proto/generate --zrpc_out=./service/${struct_name}Service --style go_zero# 替换omitempty,避免json序列化忽略字段sed -i '' -e '/omitempty/s/,omitempty//g' ./proto/generate/${struct_name}/*.pb.go# 修改客户端文件包名path="./service/${struct_name}Service/${param}_client/"sed -i '' -e "s/package ${param}_client/package client/g" ${path}${param}.go# 将客户端文件移到./proto/client下,删除原来目录mv -f ${path}${param}.go ./proto/clientrm -r ${path}echo "Done."
elif [ "$1" == "api" ]; then# api网关生成goctl api go -api ./proto/source/api/gateway.api -dir ./service/gatewayService -style go_zero# swagger文档生成goctl api plugin -plugin goctl-swagger="swagger -filename gateway.json -host 127.0.0.1:8888" -api ./proto/source/api/gateway.api -dir ./doc/swagger/etcecho "Done."
elif [ "$1" == "proto" ]; then# proto文件生成,base目录下protoc --go_out=.. --go-grpc_out=..  ./proto/source/${param}.protosed -i '' -e '/omitempty/s/,omitempty//g' ./proto/generate/${param}/*.pb.goecho "Done."
elif [ "$1" == "mysql" ]; then# mysql生成代码goctl model mysql ddl --src ./dao/mysql/source/${param}.sql --dir ./dao/mysql/model -i ''genecho "Done."
elif [ "$1" == "docker" ]; thenversion=$3dockerfiledocker build -t "${param}":"${version}" .rm -f Dockerfileecho "Done."
elseecho "参数无效"help
fi

项目目录结构

    • com【业务公共代码】
    • dao【model层】
      • mysql
        • model【生成curd和model的目录】
        • source【表结构】
      • redis
    • doc【文档】
      • swagger【接口文档】
    • proto
      • client【rpc客户端连接】
      • generate【生成rpc目录】
      • source【proto源文件】
    • service【微服务】
      • userService【服务名称】
        • etc【配置文件】
        • internal
          • config【配置结构体】
          • logic【业务逻辑】
          • server【rpc service】
          • svc【初始化依赖】
        • user.go【main入口】
    • utils【公共代码模块】
      • config【配置初始化】
      • consts【常量】
      • gos【协程】
      • kafka【kafka相关】
      • logs【日志相关】
      • mysql【mysql】
      • redis【redis】
      • utils【公共代码】
    • generate_code.sh【脚本文件】

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/web/43166.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

ClickHouse表引擎概述

ClickHouse表引擎概述 表引擎的功能&#xff1a; 数据的存储方式 数据的存储位置 是否可以使用索引 是否可以使用分区 是否支持数据副本 并发数据访问 ClickHouse在建表时必须指定表引擎。 表引擎主要分为四大类&#xff1a;MergeTree系列、Log系列、与其他存储/处理系…

字节码编程bytebuddy之获取方法信息和方法入参信息

写在前面 本文看下通过bytebuddy如何获取方法信息和方法的入参信息。 1&#xff1a;代码 package com.dahuyou.bytebuddy.bb;import com.dahuyou.bytebuddy.TT; import net.bytebuddy.ByteBuddy; import net.bytebuddy.dynamic.DynamicType; import net.bytebuddy.implement…

【高中数学/对数函数】比较a=ln2/2,b=ln5/5的大小

【问题】 比较aln2/2,bln5/5的大小 【解答】 a-bln2/2-ln5/5(5*ln2-2*ln5)/10(ln2^5-ln5^2)/10(ln32-ln25)/10>0 所以a>b 【图像】 如果绘出函数ylnx/x的图像&#xff0c;再标记出a,b的位置&#xff0c;则绘出图像如下&#xff1a; 由上图可以看出&#xff0c;a,b两…

随手记:对比两个对象不一样的值,生成一个新的对象

diffObject(obj1, obj2) {let changeForm {}for (let key in obj1) {if (!obj1.hasOwnProperty(key) || obj1[key] ! obj2[key]) {// 新旧数据不相同的key值changeForm[key] obj1[key]}}console.log(changeForm, changeForm)},

初次用bable遍历vue项目下的中文

利用 babel 找到 AST 中的中文 // vite-plugin-babel-transform.js const parser require(babel/parser) const traverse require(babel/traverse).default // const types require(babel/types) // const generate require(babel/generator).default const fs require(f…

【PHP小课堂】学习PHP中的字符串操作函数(二)

学习PHP中的字符串操作函数&#xff08;二&#xff09; 接下来我们继续 PHP 中字符串函数的学习。今天学习的内容主要是带下划线的一些字符串函数&#xff0c;上篇文章说过&#xff0c;这些系统函数的命名是 PHP 非常令人诟病的&#xff0c;有些东西真的只能靠我们的记忆来强行…

显卡、显卡驱动、cuda、cuDNN之间关系

显卡、显卡驱动、CUDA 和 cuDNN 是构成高性能计算和深度学习环境的关键组件&#xff0c;它们之间有着紧密的联系。下面是对这些组件及其关系的详细介绍&#xff1a; 显卡&#xff08;GPU&#xff09; 显卡&#xff0c;全称为图形处理器&#xff08;Graphics Processing Unit&…

【Unity2D 2022:NPC】制作任务系统

一、接受任务 1. 编辑NPC对话脚本&#xff1a; &#xff08;1&#xff09;创建静态布尔变量用来判断ruby是否接受到任务 public class NPCDialog : MonoBehaviour {// 创建全局变量用来判断ruby是否接到任务public static bool receiveTask false; } &#xff08;2&#xff…

python学习-错误与异常

代码是人的逻辑思维的具体体现&#xff0c;因为没有一个人的逻辑思维是完美无缺的&#xff0c;所以人在编写代码时必然会出现各种错误。既然错误或多或少都会发生&#xff0c;那么如何捕捉错误&#xff0c;并且捕捉到错误后要如何处理&#xff0c;就显得很重要。 语法错误 Py…

SPI通信协议和W25Q64

前言&#xff1a; STM32中的通信接口&#xff1a; UART 单总线 IIC SPI CAN 1. SPI FLASH W25Q64的关系 SPI:一种通信接口&#xff0c;可以用于和搭载SPI接口的设备通信 FLASH:是一种掉电不丢失的存储 -- 手机8256G的256 单片机 64K512K的512 芯片内部flash&…

STM32 GPIO的工作原理

STM32的GPIO管脚有下面8种可能的配置:&#xff08;4输入 2 输出 2 复用输出) &#xff08;1&#xff09;浮空输入_IN_FLOATING 在上图上&#xff0c;阴影的部分处于不工作状态&#xff0c;尤其是下半部分的输出电路&#xff0c;实际上是与端口处于隔离状态。黄色的高亮部分显示…

响应式布局下关于gird栅格布局的一些构思

1、传列数&#xff0c;根据列数计算元素容器宽度 好处是子元素可以写百分比宽度&#xff0c;不用固定某一种宽度&#xff0c;反正知道列数通过计算间距就能得到外层容器的宽度。 举个简单的例子&#xff1a; &#xff08;ps:以下用例皆在html中去模拟&#xff0c;就不另外起r…

Python 获取 SQL 指纹和 HASH 值

前言 本文介绍一个提取 SQL 指纹的方法&#xff0c;就是将 SQL 语句的条件转换为 &#xff1f;可用于脱敏和 SQL 聚类分析的场景。 1. 工具安装 这里用到的工具&#xff0c;就是 pt 工具集中的 pt-fingerprint 含在 Percona Toolkit 中&#xff0c;安装方法可参考 Percona T…

python7:装饰器

目录 1.调用外部程序os.system-阻塞式调用subprocess-python中的模块 2.装饰器前戏作用域&#xff08;1&#xff09;全局和局部-就近原则&#xff08;2&#xff09;嵌套作用域&#xff08;3&#xff09;内置作用域、变量 高阶函数&#xff1a;函数是最高级的对象&#xff08;1&…

海外媒体投稿:5个软文代发经典案例,教大家获得突破

随着互联网的飞速发展&#xff0c;软文代发成为一种高效的推广方法。下面我们就详细介绍五个成功软文代发推广实例&#xff0c;致力于帮助读者把握有关方法&#xff0c;完成突破。 第一实例&#xff1a;社交网络散播在如今社交媒体时代&#xff0c;软文代发能够通过社交平台迅速…

nodejs实现:支付宝订单查询

nodejs实现&#xff1a;支付宝订单查询&#xff1b; 原生http请求&#xff0c;不使用三方库&#xff1b; 代码如下&#xff1a; const https require(https); const crypto require(crypto); const querystring require(querystring);// 支付宝公共参数 const PRIVATE_KE…

[C++] 轻熟类和对象

类的定义 格式规范 class为定义类的关键字&#xff0c;后有类名&#xff0c;类的主体存于{}中&#xff1b;类定义结束时后面的分号不能省略&#xff1b;类体的内容成为类的成员&#xff0c;类中的变量成为成员变量&#xff0c;函数成为方法或成员函数&#xff1b;C兼容C语言的…

微软 Edge 浏览器全解析

微软 Edge 是微软推出的一个现代化浏览器,继承了 Internet Explorer(IE)的部分功能,但在速度、安全性和兼容性方面做出了很大改进。下面是对微软 Edge 浏览器的详细解析,包括其特点、安装、配置和常见问题的解答。 微软 Edge 浏览器的特点 基于 Chromium 内核 Edge 浏览…

SpringBoot配置flyway

背景 目前我们的项目代码都会交由Git、SVN等版本管理工具进行管理&#xff0c;但是我们的sql脚本&#xff0c;尤其是各类ddl脚本并没有进行版本的管理&#xff08;python的web框架Django默认就提供了类似的工具&#xff0c;从一开始就鼓励开发者通过版本管理的方式进行数据库的…

C++中的多重继承和虚继承:横向继承、纵向继承和联合继承;虚继承

多重继承 A.横向多重继承&#xff1a; B.纵向多重继承&#xff1a; C.联合多重继承&#xff1a; 因为 single 和 waiter 都继承了一个 worker 组件&#xff0c;因此 SingingWaiter 将包含两个 worker 组件&#xff0c;那么将派生类对象的地址赋给基类指针将出现二义性 那么如何…