Gmarker689
0a9a923d55
- inference.go: GenerateImages 改为 API 优先（callImageGenAPI），无 key 回退 mock；新增 EditImages/callImageEditAPI（multipart 上传编辑）；提取 parseImageResponse 共享响应解析 - handler/generate.go: POST /api/v1/generate 触发生成管线，返回 base64 图片 - handler/edit.go: POST /api/v1/images/edit 图片编辑端点 - mildware/auth.go: JWT Bearer token 认证中间件 - main.go: 路由拆分公开/认证组，generate 与 images/edit 需鉴权
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
// gen2d backend 入口
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"gen2d/internal/config"
|
|
"gen2d/internal/db"
|
|
"gen2d/internal/handler"
|
|
"gen2d/internal/mildware"
|
|
"gen2d/internal/model"
|
|
"gen2d/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
cfg := config.Load()
|
|
|
|
gin.SetMode(cfg.Server.Mode)
|
|
|
|
// 初始化 SQLite 数据库
|
|
if err := db.Init(cfg.Database.DSN, &model.User{}); err != nil {
|
|
log.Fatalf("db init failed: %v", err)
|
|
}
|
|
|
|
// 初始化 AuthService，注入 JWT 配置
|
|
handler.InitAuthService(cfg.JWT.Secret, cfg.JWT.Expire)
|
|
|
|
// 注入 LLM 和文生图配置到 service 层
|
|
service.InitLLMConfig(cfg.LLM)
|
|
service.InitImageGenConfig(cfg.ImageGen)
|
|
|
|
r := gin.New()
|
|
r.Use(gin.Recovery()) // panic 恢复中间件，防止服务因未捕获异常宕机
|
|
|
|
// API v1 路由组 — 公开接口
|
|
v1 := r.Group("/api/v1")
|
|
{
|
|
v1.GET("/health", handler.Health) // 健康检查
|
|
v1.POST("/prompt/optimize", handler.PromptOptimize) // 提示词优化
|
|
}
|
|
|
|
// API v1 路由组 — 需认证
|
|
v1Auth := r.Group("/api/v1")
|
|
v1Auth.Use(mildware.AuthMiddleware(cfg.JWT.Secret))
|
|
{
|
|
v1Auth.POST("/generate", handler.Generate) // 素材生成管线
|
|
v1Auth.POST("/images/edit", handler.EditImage) // 图片编辑
|
|
}
|
|
|
|
// Auth 路由组
|
|
auth := r.Group("/auth")
|
|
{
|
|
auth.POST("/register", handler.Register) // 用户注册
|
|
auth.POST("/login", handler.Login) // 用户登录
|
|
}
|
|
|
|
addr := fmt.Sprintf(":%d", cfg.Server.Port)
|
|
log.Printf("gen2d backend starting on %s", addr)
|
|
if err := r.Run(addr); err != nil {
|
|
log.Fatalf("server failed: %v", err)
|
|
}
|
|
}
|