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【腳本文件】