mirror of
https://github.com/woodchen-ink/random-api-go.git
synced 2025-07-18 05:42:01 +08:00
wood chen
b252b8313e
- Removed custom referer parsing logic and replaced it with direct usage of r.Referer() for cleaner code. - Updated logging to consistently use the referer from the request object across all relevant handlers. - Enhanced readability and maintainability by eliminating redundant code related to referer extraction.
109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"random-api-go/monitoring"
|
|
"random-api-go/services"
|
|
"random-api-go/stats"
|
|
"random-api-go/utils"
|
|
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var statsManager *stats.StatsManager
|
|
|
|
// InitializeHandlers 初始化处理器
|
|
func InitializeHandlers(sm *stats.StatsManager) error {
|
|
statsManager = sm
|
|
return services.InitializeCSVService()
|
|
}
|
|
|
|
func HandleAPIRequest(w http.ResponseWriter, r *http.Request) {
|
|
start := time.Now()
|
|
realIP := utils.GetRealIP(r)
|
|
|
|
path := strings.TrimPrefix(r.URL.Path, "/")
|
|
pathSegments := strings.Split(path, "/")
|
|
|
|
if len(pathSegments) < 2 {
|
|
monitoring.LogRequest(monitoring.RequestLog{
|
|
Time: time.Now().Unix(),
|
|
Path: r.URL.Path,
|
|
Method: r.Method,
|
|
StatusCode: http.StatusNotFound,
|
|
Latency: float64(time.Since(start).Microseconds()) / 1000,
|
|
IP: realIP,
|
|
Referer: r.Referer(),
|
|
})
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
prefix := pathSegments[0]
|
|
suffix := pathSegments[1]
|
|
|
|
services.Mu.RLock()
|
|
csvPath, ok := services.CSVPathsCache[prefix][suffix]
|
|
services.Mu.RUnlock()
|
|
|
|
if !ok {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
selector, err := services.GetCSVContent(csvPath)
|
|
if err != nil {
|
|
http.Error(w, "Failed to fetch CSV content", http.StatusInternalServerError)
|
|
log.Printf("Error fetching CSV content: %v", err)
|
|
return
|
|
}
|
|
|
|
if len(selector.URLs) == 0 {
|
|
http.Error(w, "No content available", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
randomURL := selector.GetRandomURL()
|
|
|
|
// 记录统计
|
|
endpoint := fmt.Sprintf("%s/%s", prefix, suffix)
|
|
statsManager.IncrementCalls(endpoint)
|
|
|
|
duration := time.Since(start)
|
|
|
|
// 记录请求日志
|
|
monitoring.LogRequest(monitoring.RequestLog{
|
|
Time: time.Now().Unix(),
|
|
Path: r.URL.Path,
|
|
Method: r.Method,
|
|
StatusCode: http.StatusFound,
|
|
Latency: float64(duration.Microseconds()) / 1000, // 转换为毫秒
|
|
IP: realIP,
|
|
Referer: r.Referer(),
|
|
})
|
|
|
|
log.Printf(" %-12s | %-15s | %-6s | %-20s | %-20s | %-50s",
|
|
duration, // 持续时间
|
|
realIP, // 真实IP
|
|
r.Method, // HTTP方法
|
|
r.URL.Path, // 请求路径
|
|
r.Referer(), // 来源信息
|
|
randomURL, // 重定向URL
|
|
)
|
|
|
|
http.Redirect(w, r, randomURL, http.StatusFound)
|
|
}
|
|
|
|
func HandleStats(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
stats := statsManager.GetStats()
|
|
if err := json.NewEncoder(w).Encode(stats); err != nil {
|
|
http.Error(w, "Error encoding stats", http.StatusInternalServerError)
|
|
log.Printf("Error encoding stats: %v", err)
|
|
}
|
|
}
|