fa1f65baf7
Some checks failed
Build & Publish Docker Image / build-and-push (push) Failing after 33s
Implement SQLite support using the pure Go `modernc.org/sqlite` driver and update the migration system to handle driver-specific schemas. Users can now choose between MySQL and SQLite by setting the `DB_DRIVER` environment variable.
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/joho/godotenv"
|
|
|
|
"emly-api-go/internal/config"
|
|
"emly-api-go/internal/database"
|
|
"emly-api-go/internal/database/schema"
|
|
"emly-api-go/internal/routes"
|
|
|
|
emlyMiddleware "emly-api-go/internal/middleware"
|
|
)
|
|
|
|
func main() {
|
|
// Load .env (ignored if not present in production)
|
|
_ = godotenv.Load()
|
|
|
|
if name := os.Getenv("INSTANCE_NAME"); name != "" {
|
|
log.SetPrefix("[" + name + "] ")
|
|
}
|
|
|
|
cfg := config.Load()
|
|
|
|
db, err := database.Connect(cfg)
|
|
if err != nil {
|
|
log.Fatalf("database connection failed: %v", err)
|
|
}
|
|
defer func(db *sqlx.DB) {
|
|
err := db.Close()
|
|
if err != nil {
|
|
log.Fatalf("closing database failed: %v", err)
|
|
}
|
|
}(db)
|
|
|
|
// Run conditional schema migrations
|
|
if err := schema.Migrate(db, cfg.Database, cfg.Driver); err != nil {
|
|
log.Fatalf("schema migration failed: %v", err)
|
|
}
|
|
|
|
r := chi.NewRouter()
|
|
|
|
// Global middlewares
|
|
r.Use(middleware.RequestID)
|
|
r.Use(middleware.RealIP)
|
|
r.Use(middleware.Logger)
|
|
r.Use(middleware.Recoverer)
|
|
r.Use(middleware.Timeout(30 * time.Second))
|
|
|
|
rl := emlyMiddleware.NewRateLimiter(cfg)
|
|
|
|
r.Use(rl.Handler)
|
|
|
|
routes.RegisterAll(r, db)
|
|
|
|
addr := fmt.Sprintf(":%s", cfg.Port)
|
|
log.Printf("server listening on %s", addr)
|
|
if err := http.ListenAndServe(addr, r); err != nil {
|
|
log.Fatalf("server error: %v", err)
|
|
}
|
|
}
|