236 lines
7.2 KiB
Go
236 lines
7.2 KiB
Go
package handler
|
||
|
||
import (
|
||
"encoding/json"
|
||
"log"
|
||
"net/http"
|
||
"strconv"
|
||
|
||
"kfz-goods-pricing/internal/repository"
|
||
)
|
||
|
||
// TokenHandler Token处理器
|
||
type TokenHandler struct {
|
||
tokenRepo *repository.TokenRepository
|
||
}
|
||
|
||
// NewTokenHandler 创建Token处理器实例
|
||
func NewTokenHandler(tokenRepo *repository.TokenRepository) *TokenHandler {
|
||
return &TokenHandler{
|
||
tokenRepo: tokenRepo,
|
||
}
|
||
}
|
||
|
||
// TokenResponse Token响应结构
|
||
type TokenResponse struct {
|
||
Code int `json:"code"`
|
||
Message string `json:"message"`
|
||
Data interface{} `json:"data,omitempty"`
|
||
}
|
||
|
||
// TokenInput Token输入结构
|
||
type TokenInput struct {
|
||
Username string `json:"username"`
|
||
Token string `json:"token"`
|
||
}
|
||
|
||
// BatchAddTokens 批量添加Token（JSON数组: [{"username":"","token":""},...]）
|
||
func (h *TokenHandler) BatchAddTokens(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := r.RemoteAddr
|
||
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
clientIP = forwarded
|
||
}
|
||
|
||
if r.Method == http.MethodOptions {
|
||
return
|
||
}
|
||
|
||
var inputs []TokenInput
|
||
if err := json.NewDecoder(r.Body).Decode(&inputs); err != nil {
|
||
log.Printf("[Token/BatchAdd] 请求体解析失败: %v, 来源IP: %s", err, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 400, Message: "invalid request body"})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/BatchAdd] 收到请求, 来源IP: %s, token数量: %d", clientIP, len(inputs))
|
||
|
||
if len(inputs) == 0 {
|
||
log.Printf("[Token/BatchAdd] 空数组, 来源IP: %s", clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 400, Message: "empty array"})
|
||
return
|
||
}
|
||
|
||
// 逐条插入
|
||
var failed []TokenInput
|
||
for i, input := range inputs {
|
||
if input.Username == "" || input.Token == "" {
|
||
log.Printf("[Token/BatchAdd] 第%d条数据不完整(username或token为空), 跳过", i+1)
|
||
failed = append(failed, input)
|
||
continue
|
||
}
|
||
|
||
id, err := h.tokenRepo.Insert(input.Username, "", input.Token, true)
|
||
if err != nil {
|
||
log.Printf("[Token/BatchAdd] 第%d条插入失败: username=%s, 错误=%v", i+1, input.Username, err)
|
||
failed = append(failed, input)
|
||
} else {
|
||
log.Printf("[Token/BatchAdd] 第%d条插入成功: id=%d, username=%s", i+1, id, input.Username)
|
||
}
|
||
}
|
||
|
||
w.Header().Set("Content-Type", "application/json")
|
||
if len(failed) > 0 {
|
||
log.Printf("[Token/BatchAdd] 部分成功: 成功%d条, 失败%d条, 来源IP: %s", len(inputs)-len(failed), len(failed), clientIP)
|
||
json.NewEncoder(w).Encode(TokenResponse{
|
||
Code: 207,
|
||
Message: "partial success",
|
||
Data: map[string]interface{}{"failed": failed},
|
||
})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/BatchAdd] 全部成功: %d条, 来源IP: %s", len(inputs), clientIP)
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 200, Message: "success"})
|
||
}
|
||
|
||
// GetAllTokens 查询所有Token
|
||
func (h *TokenHandler) GetAllTokens(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := r.RemoteAddr
|
||
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
clientIP = forwarded
|
||
}
|
||
log.Printf("[Token/GetAll] 收到请求, 来源IP: %s", clientIP)
|
||
|
||
if r.Method == http.MethodOptions {
|
||
return
|
||
}
|
||
|
||
records, err := h.tokenRepo.GetAll()
|
||
if err != nil {
|
||
log.Printf("[Token/GetAll] 查询失败: %v, 来源IP: %s", err, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 500, Message: err.Error()})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/GetAll] 查询成功: 共%d条记录, 来源IP: %s", len(records), clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 200, Message: "success", Data: records})
|
||
}
|
||
|
||
// DeleteToken 删除Token
|
||
func (h *TokenHandler) DeleteToken(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := r.RemoteAddr
|
||
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
clientIP = forwarded
|
||
}
|
||
log.Printf("[Token/Delete] 收到请求, 来源IP: %s", clientIP)
|
||
|
||
if r.Method == http.MethodOptions {
|
||
return
|
||
}
|
||
|
||
// 解析 form-data 或 URL 参数
|
||
r.ParseMultipartForm(32 << 20)
|
||
r.ParseForm()
|
||
|
||
idStr := r.PostForm.Get("id")
|
||
if idStr == "" {
|
||
// 尝试从URL获取
|
||
idStr = r.URL.Query().Get("id")
|
||
}
|
||
|
||
id, err := strconv.ParseInt(idStr, 10, 64)
|
||
if err != nil {
|
||
log.Printf("[Token/Delete] 参数id无效: %s, 来源IP: %s", idStr, clientIP)
|
||
http.Error(w, "invalid id", http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
err = h.tokenRepo.Delete(id)
|
||
if err != nil {
|
||
log.Printf("[Token/Delete] 删除失败: id=%d, 错误=%v, 来源IP: %s", id, err, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 500, Message: err.Error()})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/Delete] 删除成功: id=%d, 来源IP: %s", id, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 200, Message: "success"})
|
||
}
|
||
|
||
// UpdateToken 修改Token
|
||
func (h *TokenHandler) UpdateToken(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := r.RemoteAddr
|
||
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
clientIP = forwarded
|
||
}
|
||
log.Printf("[Token/Update] 收到请求, 来源IP: %s", clientIP)
|
||
|
||
if r.Method == http.MethodOptions {
|
||
return
|
||
}
|
||
|
||
// 解析 form-data
|
||
r.ParseMultipartForm(32 << 20)
|
||
r.ParseForm()
|
||
|
||
idStr := r.PostForm.Get("id")
|
||
username := r.PostForm.Get("username")
|
||
token := r.PostForm.Get("token")
|
||
isEnableStr := r.PostForm.Get("is_enable")
|
||
|
||
id, err := strconv.ParseInt(idStr, 10, 64)
|
||
if err != nil {
|
||
log.Printf("[Token/Update] 参数id无效: %s, 来源IP: %s", idStr, clientIP)
|
||
http.Error(w, "invalid id", http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
isEnable := true
|
||
if isEnableStr == "0" || isEnableStr == "false" {
|
||
isEnable = false
|
||
}
|
||
|
||
log.Printf("[Token/Update] 更新: id=%d, username=%s, is_enable=%v, 来源IP: %s", id, username, isEnable, clientIP)
|
||
err = h.tokenRepo.Update(id, username, "", token, isEnable)
|
||
if err != nil {
|
||
log.Printf("[Token/Update] 更新失败: id=%d, 错误=%v, 来源IP: %s", id, err, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 500, Message: err.Error()})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/Update] 更新成功: id=%d, 来源IP: %s", id, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 200, Message: "success"})
|
||
}
|
||
|
||
// GetEnabledTokens 获取所有启用的Token
|
||
func (h *TokenHandler) GetEnabledTokens(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := r.RemoteAddr
|
||
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
clientIP = forwarded
|
||
}
|
||
log.Printf("[Token/GetEnabled] 收到请求, 来源IP: %s", clientIP)
|
||
|
||
if r.Method == http.MethodOptions {
|
||
return
|
||
}
|
||
|
||
records, err := h.tokenRepo.GetEnabledTokens()
|
||
if err != nil {
|
||
log.Printf("[Token/GetEnabled] 查询失败: %v, 来源IP: %s", err, clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 500, Message: err.Error()})
|
||
return
|
||
}
|
||
|
||
log.Printf("[Token/GetEnabled] 查询成功: 共%d条启用token, 来源IP: %s", len(records), clientIP)
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(TokenResponse{Code: 200, Message: "success", Data: records})
|
||
}
|