go命令中的generate秘密武器-Golang-、刘德华、刘亦菲、谌洪

go generate 命令可以自动生成代码、文件和工件，从而简化重复性任务。其用法如下： 1. go generate [-n] importpath... 2. -n：打印生成代码而不出实际执行 3. importpath：要生成包的导入路径 4. 用例举例：生成数据库连接代码（pgxgen）、生成 grpc 服务桩文件（protoc --go-grpc）。

go命令中的generate秘密武器

如果你已经厌倦了手动执行重复的任务，那么 go generate 命令就是你的救命稻草。这个强大的命令可以帮你生成代码、文件和其他工件，从而节省大量时间和精力。

用法

go generate 命令的基本语法如下：

go generate [-n] importpath...

登录后复制

其中：

-n：表示只打印生成的代码，而不实际执行
importpath：要生成的包的导入路径

用例

go generate 命令有很多实际用途，这里举几个例子。

生成数据库连接代码

如果你正在使用像 github.com/jackc/pgx 这样的数据库库，你可以使用 go generate 来生成数据库连接代码。例如：

//go:generate pgxgen -database pgxgen -skip-columns "ID" -skip-package pgxgen

package pgxgen

import "database/sql"
import _ "github.com/jackc/pgx/stdlib" // pgx driver needed for generate
import _ "github.com/jackc/pgx/gen/dbinit"

var DB *sql.DB

登录后复制

运行 go generate 会根据你的数据库模式生成 db.go 文件，其中包含连接到数据库的 *sql.DB。

生成gRPC服务桩文件

如果你正在使用 github.com/<a style='color:#f60; text-decoration:underline;' href="https://www.php.cn/zt/16009.html" target="_blank">golang</a>/protobuf 库，你可以使用 go generate 来生成 gRPC 服务桩文件。例如：

//go:generate protoc --go-grpc ./*.proto

package main

import (
    "context"
    "log"

    pb "github.com/example/mypackage/api/v1"
    "google.golang.org/grpc"
)

func main() {
    // gRPC server address
    addr := "localhost:5000"

    // Connect to gRPC server
    conn, err := grpc.Dial(addr, grpc.WithInsecure())
    if err != nil {
        log.Fatalf("Did not connect: %v", err)
    }
    defer conn.Close()

    client := pb.NewGreeterClient(conn)

    // Call gRPC service
    resp, err := client.SayHello(context.Background(), &pb.HelloRequest{Name: "John"})
    if err != nil {
        log.Fatalf("Could not greet: %v", err)
    }
    log.Printf("Greeting: %s", resp.Message)
}

登录后复制

运行 go generate 会生成 pb.go 文件，其中包含 gRPC 服务桩的代码。