mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-27 07:16:19 +09:00
bc374fe8cd
* 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>
88 lines
2.3 KiB
Go
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
|
|
}
|