Initial project scaffold

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-09 14:36:35 +08:00
commit 6323e70e49
39 changed files with 8664 additions and 0 deletions
@@ -0,0 +1,56 @@
+package main
+
+import (
+	"context"
+	"errors"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/config"
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/httpapi"
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/store"
+)
+
+func main() {
+	cfg := config.Load()
+	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
+		Level: cfg.LogLevel,
+	}))
+
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	db, err := store.Connect(ctx, cfg.DatabaseURL)
+	if err != nil {
+		logger.Error("connect postgres failed", "error", err)
+		os.Exit(1)
+	}
+	defer db.Close()
+
+	server := &http.Server{
+		Addr:              cfg.HTTPAddr,
+		Handler:           httpapi.NewServer(cfg, db, logger),
+		ReadHeaderTimeout: 10 * time.Second,
+	}
+
+	go func() {
+		logger.Info("easyai ai gateway api started", "addr", cfg.HTTPAddr)
+		if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			logger.Error("http server failed", "error", err)
+			os.Exit(1)
+		}
+	}()
+
+	<-ctx.Done()
+	shutdownCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancel()
+	if err := server.Shutdown(shutdownCtx); err != nil {
+		logger.Error("http shutdown failed", "error", err)
+		os.Exit(1)
+	}
+	logger.Info("easyai ai gateway api stopped")
+}
@@ -0,0 +1,85 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"os"
+	"path/filepath"
+	"sort"
+	"strings"
+
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/config"
+	"github.com/jackc/pgx/v5"
+)
+
+func main() {
+	cfg := config.Load()
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	ctx := context.Background()
+
+	conn, err := pgx.Connect(ctx, cfg.DatabaseURL)
+	if err != nil {
+		logger.Error("connect postgres failed", "error", err)
+		os.Exit(1)
+	}
+	defer conn.Close(ctx)
+
+	if _, err := conn.Exec(ctx, `
+CREATE TABLE IF NOT EXISTS schema_migrations (
+  version text PRIMARY KEY,
+  applied_at timestamptz NOT NULL DEFAULT now()
+);`); err != nil {
+		logger.Error("ensure schema_migrations failed", "error", err)
+		os.Exit(1)
+	}
+
+	files, err := filepath.Glob("migrations/*.sql")
+	if err != nil {
+		logger.Error("read migrations failed", "error", err)
+		os.Exit(1)
+	}
+	sort.Strings(files)
+
+	for _, file := range files {
+		version := strings.TrimSuffix(filepath.Base(file), filepath.Ext(file))
+		var exists bool
+		if err := conn.QueryRow(ctx, "SELECT EXISTS (SELECT 1 FROM schema_migrations WHERE version=$1)", version).Scan(&exists); err != nil {
+			logger.Error("check migration failed", "version", version, "error", err)
+			os.Exit(1)
+		}
+		if exists {
+			logger.Info("migration skipped", "version", version)
+			continue
+		}
+
+		sqlBytes, err := os.ReadFile(file)
+		if err != nil {
+			logger.Error("read migration file failed", "file", file, "error", err)
+			os.Exit(1)
+		}
+
+		tx, err := conn.Begin(ctx)
+		if err != nil {
+			logger.Error("begin migration failed", "version", version, "error", err)
+			os.Exit(1)
+		}
+		if _, err := tx.Exec(ctx, string(sqlBytes)); err != nil {
+			_ = tx.Rollback(ctx)
+			logger.Error("execute migration failed", "version", version, "error", err)
+			os.Exit(1)
+		}
+		if _, err := tx.Exec(ctx, "INSERT INTO schema_migrations(version) VALUES($1)", version); err != nil {
+			_ = tx.Rollback(ctx)
+			logger.Error("record migration failed", "version", version, "error", err)
+			os.Exit(1)
+		}
+		if err := tx.Commit(ctx); err != nil {
+			logger.Error("commit migration failed", "version", version, "error", err)
+			os.Exit(1)
+		}
+		logger.Info("migration applied", "version", version)
+	}
+
+	fmt.Println("migrations complete")
+}