Files
Memoh/internal/db/migrate.go
T
BBQ bc374fe8cd refactor: content-addressed assets, cross-channel multimodal, infra simplification (#63)
* refactor(attachment): multimodal attachment refactor with snapshot schema and storage layer

- Add snapshot schema migration (0008) and update init/versions/snapshots
- Add internal/attachment and internal/channel normalize for unified attachment handling
- Move containerfs provider from internal/media to internal/storage
- Update agent types, channel adapters (Telegram/Feishu), inbound and handlers
- Add containerd snapshot lineage and local_channel tests
- Regenerate sqlc, swagger and SDK

* refactor(media): content-addressed asset system with unified naming

- Replace asset_id foreign key with content_hash as sole identifier
  for bot_history_message_assets (pure soft-link model)
- Remove mime, size_bytes, storage_key from DB; derive at read time
  via media.Resolve from actual storage
- Merge migrations 0008/0009 into single 0008; keep 0001 as canonical schema
- Add Docker initdb script for deterministic migration execution order
- Fix cross-channel real-time image display (Telegram → WebUI SSE)
- Fix message disappearing on refresh (null assets fallback)
- Fix file icon instead of image preview (mime derivation from storage)
- Unify AssetID → ContentHash naming across Go, Agent, and Frontend
- Change storage key prefix from 4-char to 2-char for directory sharding
- Add server-entrypoint.sh for Docker deployment migration handling

* refactor(infra): embedded migrations, Docker simplification, and config consolidation

- Embed SQL migrations into Go binary, removing shell-based migration scripts
- Consolidate config files into conf/ directory (app.example.toml, app.docker.toml, app.dev.toml)
- Simplify Docker setup: remove initdb.d scripts, streamline nginx config and entrypoint
- Remove legacy CLI, feishu-echo commands, and obsolete incremental migration files
- Update install script and docs to require sudo for one-click install
- Add mise tasks for dev environment orchestration

* chore: recover migrations

---------

Co-authored-by: Acbox <acbox0328@gmail.com>
2026-02-19 00:20:27 +08:00

88 lines
2.3 KiB
Go

package db
import (
"fmt"
"io/fs"
"log/slog"
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/postgres"
"github.com/golang-migrate/migrate/v4/source/iofs"
"github.com/memohai/memoh/internal/config"
)
// RunMigrate applies or rolls back database migrations.
// The migrationsFS should contain .sql files at its root (not in a subdirectory).
// Supported commands: "up", "down", "version", "force N".
func RunMigrate(logger *slog.Logger, cfg config.PostgresConfig, migrationsFS fs.FS, command string, args []string) error {
switch command {
case "up", "down", "version", "force":
default:
return fmt.Errorf("unknown migrate command: %s (use: up, down, version, force)", command)
}
if command == "force" && len(args) == 0 {
return fmt.Errorf("force requires a version number argument")
}
dsn := DSN(cfg)
sourceDriver, err := iofs.New(migrationsFS, ".")
if err != nil {
return fmt.Errorf("migration source: %w", err)
}
m, err := migrate.NewWithSourceInstance("iofs", sourceDriver, dsn)
if err != nil {
return fmt.Errorf("migrate init: %w", err)
}
defer m.Close()
m.Log = &migrateLogger{logger: logger}
switch command {
case "up":
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("migrate up: %w", err)
}
ver, dirty, _ := m.Version()
logger.Info("migration complete", slog.Uint64("version", uint64(ver)), slog.Bool("dirty", dirty))
case "down":
if err := m.Down(); err != nil && err != migrate.ErrNoChange {
return fmt.Errorf("migrate down: %w", err)
}
logger.Info("all migrations rolled back")
case "version":
ver, dirty, err := m.Version()
if err != nil {
return fmt.Errorf("migrate version: %w", err)
}
logger.Info("current version", slog.Uint64("version", uint64(ver)), slog.Bool("dirty", dirty))
case "force":
var version int
if _, err := fmt.Sscanf(args[0], "%d", &version); err != nil {
return fmt.Errorf("invalid version: %w", err)
}
if err := m.Force(version); err != nil {
return fmt.Errorf("migrate force: %w", err)
}
logger.Info("forced version", slog.Int("version", version))
}
return nil
}
type migrateLogger struct {
logger *slog.Logger
}
func (l *migrateLogger) Printf(format string, v ...interface{}) {
l.logger.Info(fmt.Sprintf(format, v...))
}
func (l *migrateLogger) Verbose() bool {
return false
}