demo

cmd/api/main.go

@@ -5,12 +5,10 @@ import (
 	"log"
 	"net/http"
 	"os"
-	"time"
 
-	"gitea.starryskymeow.cn/B309/datamarket/internal/handler"
 	"gitea.starryskymeow.cn/B309/datamarket/internal/repository"
-	"github.com/go-chi/chi/v5"
-	"github.com/go-chi/chi/v5/middleware"
+	"gitea.starryskymeow.cn/B309/datamarket/internal/router"
+	"gitea.starryskymeow.cn/B309/datamarket/internal/service"
 	"github.com/jackc/pgx/v5/pgxpool"
 )
 
@@ -29,25 +27,16 @@ func main() {
 	}
 
 	queries := repository.New(pool)
+	appService := service.New(queries)
+	r := router.New(appService)
 
-	// route
-	r := chi.NewRouter()
+	addr := os.Getenv("SERVER_ADDR")
 
-	r.Use(middleware.RequestID)
-	r.Use(middleware.RealIP)
-	r.Use(middleware.Logger)
-	r.Use(middleware.Recoverer)
-	r.Use(middleware.Timeout(60 * time.Second))
+	if addr == "" {
+		addr = ":8080"
+	}
 
-	r.Route("/api", func(r chi.Router) {
-		r.Route("/assets", func(r chi.Router) {
-			r.Post("/", handler.CreateDataAsset(queries))
-			r.Get("/", handler.ListDataAssets(queries))
-			r.Get("/{id}", handler.GetDataAsset(queries))
-		})
-	})
-
-	err = http.ListenAndServe(":8080", r)
+	err = http.ListenAndServe(addr, r)
 	if err != nil {
 		log.Fatal(err)
 	}