如何用 Golang 编写一个数据库迁移工具_Golang 脚本自动化项目开发-Golang-PHP中文网

如何用 Golang 编写一个数据库迁移工具_Golang 脚本自动化项目开发

P粉602998670

发布： 2025-11-19 12:27:07

原创

738人浏览过

答案是使用Golang实现数据库迁移工具需通过版本控制与可重复执行机制。首先按时间戳命名迁移文件，如20240101120000_create_users_table.up.sql和.down.sql，存于migrations目录；创建schema_migrations表记录已执行版本。程序扫描目录解析版本号，读取数据库已应用版本，跳过旧迁移，对未执行的按升序运行.up.sql，并插入执行记录。核心逻辑包括：用database/sql连接数据库，读取文件、解析版本、排序并执行新增迁移，确保每步原子性与一致性，从而实现安全可靠的数据库版本管理。

如何用 golang 编写一个数据库迁移工具_golang 脚本自动化项目开发

编写数据库迁移工具的核心是版本控制和可重复执行。Golang 适合做这类脚本，因为编译后为单二进制文件，部署方便，无需依赖运行时环境。下面介绍如何用 Golang 实现一个简单但实用的数据库迁移工具。

设计迁移结构

每个迁移文件应包含版本号、描述、升级 SQL 和回滚 SQL。建议按时间戳命名文件，如 20240101120000_create_users_table.sql，前缀为时间戳，便于排序。

目录结构示例如下：

migrations/

20240101120000_create_users_table.up.sql
20240101120000_create_users_table.down.sql
20240102100000_add_email_index.up.sql
20240102100000_add_email_index.down.sql

其中 .up.sql 是升级语句，.down.sql 是回滚语句。

立即学习“go语言免费学习笔记（深入）”；

维护迁移状态

需要一张表记录已执行的迁移版本，例如 schema_migrations：

CREATE TABLE schema_migrations (version BIGINT PRIMARY KEY, applied_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP);

每次执行迁移前，读取该表获取已应用的版本，跳过旧文件，只执行新文件。

Clipfly

一站式AI视频生成和编辑平台，提供多种AI视频处理、AI图像处理工具。

查看详情

核心逻辑如下：

扫描 migrations 目录，解析文件名中的时间戳作为版本号
从数据库读取已执行的版本列表
对未执行的迁移文件按版本升序执行 .up.sql
每成功执行一个，插入一条记录到 schema_migrations 表

实现 Golang 核心代码

使用 database/sql 包连接数据库（以 PostgreSQL 为例）：

package main import ( "database/sql" "io/ioutil" "log" "path/filepath" "sort" "strconv" "strings" _ "github.com/lib/pq" ) func runMigrations(db *sql.DB, migrationDir string) error { // 获取已应用的版本 applied, err := getAppliedVersions(db) if err != nil { return err } files, err := ioutil.ReadDir(migrationDir) if err != nil { return err } var migrations []int64 for _, f := range files { if strings.HasSuffix(f.Name(), ".up.sql") { ver, _ := strconv.ParseInt(f.Name()[:14], 10, 64) if !contains(applied, ver) { migrations = append(migrations, ver) } } } sort.Slice(migrations, func(i, j int) bool { return migrations[i]