package main

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/lib/pq"
)

func main() {
	dsn := "host=localhost port=5432 user=postgres password=postgres dbname=trustlog sslmode=disable"
	
	db, err := sql.Open("postgres", dsn)
	if err != nil {
		log.Fatalf("Failed to connect: %v", err)
	}
	defer db.Close()

	if err := db.Ping(); err != nil {
		log.Fatalf("Failed to ping: %v", err)
	}

	fmt.Println("🧹 Cleaning test data...")

	// 清理所有测试数据
	_, err = db.Exec("DELETE FROM trustlog_retry")
	if err != nil {
		log.Printf("Warning: Failed to clean retry table: %v", err)
	}

	_, err = db.Exec("DELETE FROM operation")
	if err != nil {
		log.Printf("Warning: Failed to clean operation table: %v", err)
	}

	_, err = db.Exec("DELETE FROM trustlog_cursor")
	if err != nil {
		log.Printf("Warning: Failed to clean cursor table: %v", err)
	}

	fmt.Println("✅ All test data cleaned!")
}