You've already forked agentic-coding-workflow
29 lines
943 B
Go
29 lines
943 B
Go
// Package db 提供数据库相关的基础设施工具，包括 golang-migrate 迁移运行器。
|
|
package db
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
// 注册 postgres 数据库驱动（database://postgres）。
|
|
_ "github.com/golang-migrate/migrate/v4/database/postgres"
|
|
// 注册 file:// 迁移源驱动。
|
|
_ "github.com/golang-migrate/migrate/v4/source/file"
|
|
)
|
|
|
|
// Migrate 跑指定路径下的迁移到最新版本。
|
|
// migrationsPath 形如 "file://migrations"，dsn 为 postgres 连接串。
|
|
// 当无新迁移可应用时（migrate.ErrNoChange）视为成功，返回 nil。
|
|
func Migrate(dsn, migrationsPath string) error {
|
|
m, err := migrate.New(migrationsPath, dsn)
|
|
if err != nil {
|
|
return fmt.Errorf("init migrate: %w", err)
|
|
}
|
|
defer func() { _, _ = m.Close() }()
|
|
if err := m.Up(); err != nil && !errors.Is(err, migrate.ErrNoChange) {
|
|
return fmt.Errorf("migrate up: %w", err)
|
|
}
|
|
return nil
|
|
}
|