add migrations phase and visits table

2026-04-09 14:54:54 +03:00
parent c170950a1f
commit d64d470ef0
2 changed files with 64 additions and 9 deletions
--- a/main.go
+++ b/main.go
@@ -26,7 +26,7 @@ func env(key, fallback string) string {
 	return fallback
 }

-func testPostgres() result {
+func openDB() (*sql.DB, error) {
 	dsn := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
 		env("POSTGRES_HOST", "localhost"),
 		env("POSTGRES_PORT", "5432"),
@@ -35,16 +35,51 @@ func testPostgres() result {
 		env("POSTGRES_DB", "app"),
 	)
 	db, err := sql.Open("postgres", dsn)
+	if err != nil {
+		return nil, err
+	}
+	db.SetConnMaxLifetime(5 * time.Second)
+	return db, nil
+}
+
+// migrate runs database migrations.
+func migrate() error {
+	db, err := openDB()
+	if err != nil {
+		return fmt.Errorf("open db: %w", err)
+	}
+	defer db.Close()
+
+	migrations := []string{
+		`CREATE TABLE IF NOT EXISTS healthcheck (
+			id SERIAL PRIMARY KEY,
+			checked_at TIMESTAMPTZ DEFAULT NOW(),
+			msg TEXT
+		)`,
+		`CREATE TABLE IF NOT EXISTS visits (
+			id SERIAL PRIMARY KEY,
+			path TEXT NOT NULL,
+			visited_at TIMESTAMPTZ DEFAULT NOW()
+		)`,
+	}
+
+	for i, m := range migrations {
+		if _, err := db.Exec(m); err != nil {
+			return fmt.Errorf("migration %d: %w", i, err)
+		}
+		log.Printf("migration %d: OK", i)
+	}
+
+	log.Println("all migrations completed")
+	return nil
+}
+
+func testPostgres() result {
+	db, err := openDB()
 	if err != nil {
 		return result{"PostgreSQL", false, fmt.Sprintf("open: %v", err)}
 	}
 	defer db.Close()
-	db.SetConnMaxLifetime(5 * time.Second)
-
-	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS healthcheck (id SERIAL PRIMARY KEY, checked_at TIMESTAMPTZ DEFAULT NOW(), msg TEXT)`)
-	if err != nil {
-		return result{"PostgreSQL", false, fmt.Sprintf("create table: %v", err)}
-	}

 	_, err = db.Exec(`INSERT INTO healthcheck (msg) VALUES ($1)`, fmt.Sprintf("ping at %s", time.Now().Format(time.RFC3339)))
 	if err != nil {
@@ -57,7 +92,10 @@ func testPostgres() result {
 		return result{"PostgreSQL", false, fmt.Sprintf("count: %v", err)}
 	}

-	return result{"PostgreSQL", true, fmt.Sprintf("OK — %d rows in healthcheck", count)}
+	var visits int
+	db.QueryRow(`SELECT COUNT(*) FROM visits`).Scan(&visits)
+
+	return result{"PostgreSQL", true, fmt.Sprintf("OK — %d healthchecks, %d visits", count, visits)}
 }

 func testRedis() result {
@@ -90,6 +128,12 @@ func handler(w http.ResponseWriter, r *http.Request) {
 		return
 	}

+	// Track visit
+	if db, err := openDB(); err == nil {
+		db.Exec(`INSERT INTO visits (path) VALUES ($1)`, r.URL.Path)
+		db.Close()
+	}
+
 	pg := testPostgres()
 	rd := testRedis()

@@ -109,10 +153,11 @@ func handler(w http.ResponseWriter, r *http.Request) {
    .ok .status { color: #4ade80; }
    .fail .status { color: #f87171; }
    .time { color: #6b7280; font-size: 12px; margin-top: 24px; }
+    .badge { display: inline-block; background: #1e3a5f; color: #60a5fa; font-size: 11px; padding: 2px 8px; border-radius: 4px; margin-left: 8px; }
  </style>
 </head>
 <body>
-  <h1>StackOps Test App</h1>
+  <h1>StackOps Test App <span class="badge">with migrations</span></h1>
  <p style="color:#9ca3af">Dependency connectivity checks</p>
 `)
 	for _, t := range []result{pg, rd} {
@@ -132,6 +177,13 @@ func handler(w http.ResponseWriter, r *http.Request) {
 }

 func main() {
+	if len(os.Args) > 1 && os.Args[1] == "migrate" {
+		if err := migrate(); err != nil {
+			log.Fatalf("migration failed: %v", err)
+		}
+		return
+	}
+
 	http.HandleFunc("/", handler)
 	port := env("PORT", "8080")
 	log.Printf("test-app listening on :%s", port)