新增功能

internal/handler/model.go

@@ -1,7 +1,6 @@
 package handler
 
 import (
-	"database/sql"
 	"net/http"
 
 	"github.com/gin-gonic/gin"
@@ -16,9 +15,9 @@ type ModelHandler struct {
 	log  *zap.Logger
 }
 
-func NewModelHandler(db *sql.DB, log *zap.Logger) *ModelHandler {
+func NewModelHandler(repo *repository.ModelRepository, log *zap.Logger) *ModelHandler {
 	return &ModelHandler{
-		repo: repository.NewModelRepository(db),
+		repo: repo,
 		log:  log,
 	}
 }