DESKTOP-72TV0V4\caoxiaozhu
aabdf81073
- 添加 Model 实体定义 - 实现 Model CRUD 接口 - 添加 Model 仓储层和服务层 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"x-agents/server/internal/model"
|
|
"x-agents/server/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// ModelHandler 模型处理器
|
|
type ModelHandler struct {
|
|
service *service.ModelService
|
|
}
|
|
|
|
func NewModelHandler(svc *service.ModelService) *ModelHandler {
|
|
return &ModelHandler{service: svc}
|
|
}
|
|
|
|
// List 获取列表
|
|
func (h *ModelHandler) List(c *gin.Context) {
|
|
list, err := h.service.List()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if list == nil {
|
|
list = []model.ModelInfo{}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"list": list})
|
|
}
|
|
|
|
// GetByID 获取详情
|
|
func (h *ModelHandler) GetByID(c *gin.Context) {
|
|
id := c.Param("id")
|
|
model, err := h.service.GetByID(id)
|
|
if err != nil {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "Model not found"})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, model)
|
|
}
|
|
|
|
// Create 创建
|
|
func (h *ModelHandler) Create(c *gin.Context) {
|
|
var req model.CreateModelRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
result, err := h.service.Create(req)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
// Update 更新
|
|
func (h *ModelHandler) Update(c *gin.Context) {
|
|
id := c.Param("id")
|
|
var req model.UpdateModelRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
result, err := h.service.Update(id, req)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
// Delete 删除
|
|
func (h *ModelHandler) Delete(c *gin.Context) {
|
|
id := c.Param("id")
|
|
err := h.service.Delete(id)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"success": true})
|
|
}
|
|
|
|
// Test 测试连接
|
|
func (h *ModelHandler) Test(c *gin.Context) {
|
|
var req model.TestModelRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
result, err := h.service.TestConnection(req)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, result)
|
|
}
|