feat(admin): establish core data model

This commit is contained in:
QiuSW
2026-08-03 18:37:48 +08:00
parent 8be6d206dd
commit 87591e84f9
17 changed files with 1081 additions and 6 deletions
+58
View File
@@ -0,0 +1,58 @@
package main
import (
"context"
"errors"
"flag"
"fmt"
"io"
"log"
"os"
"cmbuyer/admin/internal/migrations"
"cmbuyer/admin/internal/storage/sqlite"
)
func main() {
if err := run(context.Background(), os.Args[1:], os.Stderr); err != nil {
log.Print(err)
os.Exit(1)
}
}
func run(ctx context.Context, args []string, stderr io.Writer) error {
flags := flag.NewFlagSet("migrate", flag.ContinueOnError)
flags.SetOutput(stderr)
databaseSource := flags.String("database", "", "SQLite data source")
migrationDirectory := flags.String("dir", "migrations", "migration directory")
if err := flags.Parse(args); err != nil {
return err
}
if *databaseSource == "" {
return errors.New("-database is required")
}
if flags.NArg() != 1 {
return fmt.Errorf("usage: migrate -database <sqlite-data-source> [-dir <migration-directory>] <up|down|status>")
}
command := flags.Arg(0)
if command != "up" && command != "down" && command != "status" {
return fmt.Errorf("unsupported migration command %q", command)
}
database, err := sqlite.Open(*databaseSource)
if err != nil {
return fmt.Errorf("open SQLite database: %w", err)
}
defer func() {
if err := database.Close(); err != nil {
log.Printf("close SQLite database: %v", err)
}
}()
if err := migrations.Run(ctx, database, *migrationDirectory, command); err != nil {
return fmt.Errorf("run migrations: %w", err)
}
return nil
}
+68
View File
@@ -0,0 +1,68 @@
package main
import (
"context"
"database/sql"
"errors"
"io"
"os"
"path/filepath"
"runtime"
"testing"
)
func TestRunUp(t *testing.T) {
databaseSource := filepath.Join(t.TempDir(), "migrate.db")
if err := run(context.Background(), []string{
"-database", databaseSource,
"-dir", migrationDirectory(t),
"up",
}, io.Discard); err != nil {
t.Fatalf("run up migration command: %v", err)
}
database, err := sql.Open("sqlite3", databaseSource)
if err != nil {
t.Fatalf("open migrated database: %v", err)
}
t.Cleanup(func() {
if err := database.Close(); err != nil {
t.Errorf("close migrated database: %v", err)
}
})
var count int
if err := database.QueryRow(`SELECT COUNT(*) FROM sqlite_master WHERE type = 'table' AND name = 'tasks'`).Scan(&count); err != nil {
t.Fatalf("look up tasks table: %v", err)
}
if count != 1 {
t.Fatalf("tasks table count = %d, want 1", count)
}
}
func TestRunRequiresDatabase(t *testing.T) {
if err := run(context.Background(), []string{"up"}, io.Discard); err == nil {
t.Fatal("run without database source succeeded")
}
}
func TestRunRejectsUndeclaredCommand(t *testing.T) {
databaseSource := filepath.Join(t.TempDir(), "migrate.db")
err := run(context.Background(), []string{"-database", databaseSource, "reset"}, io.Discard)
if err == nil {
t.Fatal("run with undeclared command succeeded")
}
if _, err := os.Stat(databaseSource); !errors.Is(err, os.ErrNotExist) {
t.Fatalf("undeclared command opened database source: stat error = %v, want not exist", err)
}
}
func migrationDirectory(t *testing.T) string {
t.Helper()
_, file, _, ok := runtime.Caller(0)
if !ok {
t.Fatal("locate migration command test source")
}
return filepath.Join(filepath.Dir(file), "..", "..", "migrations")
}