56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/company/ai-ops/internal/service"
|
|
"github.com/company/ai-ops/pkg/errors"
|
|
"github.com/company/ai-ops/pkg/response"
|
|
)
|
|
|
|
// AuditHandler 是审计日志 HTTP 处理器
|
|
type AuditHandler struct {
|
|
service *service.AuditService
|
|
}
|
|
|
|
func NewAuditHandler(s *service.AuditService) *AuditHandler {
|
|
return &AuditHandler{service: s}
|
|
}
|
|
|
|
func (h *AuditHandler) RegisterRoutes(mux *http.ServeMux) {
|
|
mux.HandleFunc("GET /api/v1/ai-ops/audits", h.ListAudits)
|
|
mux.HandleFunc("POST /api/v1/ai-ops/audits/{id}/rollback", h.Rollback)
|
|
}
|
|
|
|
func (h *AuditHandler) ListAudits(w http.ResponseWriter, r *http.Request) {
|
|
query := r.URL.Query()
|
|
objectType := query.Get("object_type")
|
|
objectID := query.Get("object_id")
|
|
page, _ := strconv.Atoi(query.Get("page"))
|
|
pageSize, _ := strconv.Atoi(query.Get("page_size"))
|
|
if page < 1 {
|
|
page = 1
|
|
}
|
|
if pageSize < 1 || pageSize > 100 {
|
|
pageSize = 20
|
|
}
|
|
|
|
logs, total, err := h.service.List(r.Context(), objectType, objectID, page, pageSize)
|
|
if err != nil {
|
|
response.Error(w, errors.Wrap(err, errors.ErrInternal))
|
|
return
|
|
}
|
|
response.Success(w, map[string]any{"items": logs, "total": total, "page": page, "page_size": pageSize})
|
|
}
|
|
|
|
func (h *AuditHandler) Rollback(w http.ResponseWriter, r *http.Request) {
|
|
id := r.PathValue("id")
|
|
log, err := h.service.Rollback(r.Context(), id)
|
|
if err != nil {
|
|
response.Error(w, errors.Wrap(err, errors.ErrBadRequest).WithDetail(map[string]any{"error": err.Error()}))
|
|
return
|
|
}
|
|
response.Success(w, log)
|
|
}
|