chore: initial import
This commit is contained in:
59
pkg/response/response.go
Normal file
59
pkg/response/response.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package response
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
|
||||
"github.com/company/ai-ops/pkg/errors"
|
||||
)
|
||||
|
||||
// Response 是统一响应结构
|
||||
type Response struct {
|
||||
Code string `json:"code,omitempty"`
|
||||
Message string `json:"message,omitempty"`
|
||||
Data any `json:"data,omitempty"`
|
||||
}
|
||||
|
||||
// JSON 返回 JSON 响应
|
||||
func JSON(w http.ResponseWriter, status int, data any) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(status)
|
||||
json.NewEncoder(w).Encode(data)
|
||||
}
|
||||
|
||||
// Success 返回成功响应
|
||||
func Success(w http.ResponseWriter, data any) {
|
||||
JSON(w, http.StatusOK, Response{Data: data})
|
||||
}
|
||||
|
||||
// Error 返回错误响应
|
||||
func Error(w http.ResponseWriter, err *errors.AppError) {
|
||||
JSON(w, err.HTTPStatus, Response{
|
||||
Code: err.Code,
|
||||
Message: err.Message,
|
||||
})
|
||||
}
|
||||
|
||||
// Paginated 是分页响应结构
|
||||
type Paginated struct {
|
||||
Items any `json:"items"`
|
||||
Total int `json:"total"`
|
||||
Page int `json:"page"`
|
||||
PageSize int `json:"page_size"`
|
||||
TotalPages int `json:"total_pages"`
|
||||
}
|
||||
|
||||
// PaginatedResponse 返回分页响应
|
||||
func PaginatedResponse(w http.ResponseWriter, items any, total, page, pageSize int) {
|
||||
totalPages := total / pageSize
|
||||
if total%pageSize > 0 {
|
||||
totalPages++
|
||||
}
|
||||
Success(w, Paginated{
|
||||
Items: items,
|
||||
Total: total,
|
||||
Page: page,
|
||||
PageSize: pageSize,
|
||||
TotalPages: totalPages,
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user