mirror of
https://github.com/woodchen-ink/random-api-go.git
synced 2025-07-18 05:42:01 +08:00
feat: refactor main.go and readme.md to improve IP handling and logging
This commit is contained in:
parent
4367782a36
commit
8287d9f726
130
main.go
130
main.go
@ -3,10 +3,10 @@ package main
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"math/rand"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
@ -30,7 +30,7 @@ var (
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 使用当前时间作为种子初始化随机数生成器
|
||||
// 初始化随机数生成器
|
||||
source := rand.NewSource(time.Now().UnixNano())
|
||||
rng = rand.New(source)
|
||||
|
||||
@ -42,10 +42,8 @@ func main() {
|
||||
log.Fatal("Failed to load CSV paths:", err)
|
||||
}
|
||||
|
||||
// 提供静态文件
|
||||
// 设置路由
|
||||
http.Handle("/", http.FileServer(http.Dir("./public")))
|
||||
|
||||
// 动态请求处理
|
||||
http.HandleFunc("/pic/", logRequest(handleDynamicRequest))
|
||||
http.HandleFunc("/video/", logRequest(handleDynamicRequest))
|
||||
|
||||
@ -56,52 +54,68 @@ func main() {
|
||||
}
|
||||
|
||||
func setupLogging() {
|
||||
// 同时输出到标准输出和文件
|
||||
logFile, err := os.OpenFile("server.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
|
||||
logFile, err := os.OpenFile("server.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
multiWriter := io.MultiWriter(os.Stdout, logFile)
|
||||
log.SetOutput(multiWriter)
|
||||
log.SetOutput(logFile)
|
||||
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
||||
}
|
||||
|
||||
// 中间件：记录每个请求
|
||||
func getRealIP(r *http.Request) string {
|
||||
ip := r.Header.Get("X-Real-IP")
|
||||
if ip != "" {
|
||||
return ip
|
||||
}
|
||||
|
||||
ip = r.Header.Get("X-Forwarded-For")
|
||||
if ip != "" {
|
||||
ips := strings.Split(ip, ",")
|
||||
if len(ips) > 0 {
|
||||
return strings.TrimSpace(ips[0])
|
||||
}
|
||||
}
|
||||
|
||||
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
||||
if err != nil {
|
||||
return r.RemoteAddr
|
||||
}
|
||||
return ip
|
||||
}
|
||||
|
||||
func logRequest(handler http.HandlerFunc) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
start := time.Now()
|
||||
handler(w, r)
|
||||
duration := time.Since(start)
|
||||
log.Printf("Request: %s %s from %s - Duration: %v\n", r.Method, r.URL.Path, r.RemoteAddr, duration)
|
||||
|
||||
realIP := getRealIP(r)
|
||||
proto := r.Header.Get("X-Forwarded-Proto")
|
||||
if proto == "" {
|
||||
proto = "http"
|
||||
}
|
||||
host := r.Header.Get("X-Forwarded-Host")
|
||||
if host == "" {
|
||||
host = r.Host
|
||||
}
|
||||
|
||||
log.Printf("Request: %s %s://%s%s from %s - Duration: %v\n",
|
||||
r.Method, proto, host, r.URL.Path, realIP, duration)
|
||||
}
|
||||
}
|
||||
|
||||
// 加载 CSV 路径配置
|
||||
func loadCSVPaths() error {
|
||||
// 获取当前工作目录
|
||||
currentDir, err := os.Getwd()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get current directory: %w", err)
|
||||
}
|
||||
log.Printf("Current working directory: %s", currentDir)
|
||||
|
||||
// 构建 url.json 的完整路径
|
||||
jsonPath := filepath.Join(currentDir, "public", "url.json")
|
||||
jsonPath := filepath.Join("public", "url.json")
|
||||
log.Printf("Attempting to read file: %s", jsonPath)
|
||||
|
||||
// 检查文件是否存在
|
||||
if _, err := os.Stat(jsonPath); os.IsNotExist(err) {
|
||||
return fmt.Errorf("url.json does not exist at %s", jsonPath)
|
||||
}
|
||||
|
||||
data, err := ioutil.ReadFile(jsonPath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to read url.json: %w", err)
|
||||
return fmt.Errorf("failed to read url.json: %w", err)
|
||||
}
|
||||
|
||||
var result map[string]map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
return fmt.Errorf("failed to unmarshal url.json: %w", err)
|
||||
return fmt.Errorf("failed to unmarshal url.json: %w", err)
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
@ -118,45 +132,25 @@ func getCSVContent(path string) ([]string, error) {
|
||||
content, exists := csvCache[path]
|
||||
mu.RUnlock()
|
||||
if exists {
|
||||
log.Printf("CSV content for %s found in cache\n", path)
|
||||
return content, nil
|
||||
log.Printf("CSV content for %s found in cache\n", path)
|
||||
return content, nil
|
||||
}
|
||||
|
||||
var fileContent []byte
|
||||
var err error
|
||||
|
||||
if strings.HasPrefix(path, "http://") || strings.HasPrefix(path, "https://") {
|
||||
// 处理远程 URL
|
||||
client := &http.Client{Timeout: requestTimeout}
|
||||
resp, err := client.Get(path)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("error fetching CSV content: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, fmt.Errorf("failed to fetch CSV content: %s", resp.Status)
|
||||
}
|
||||
|
||||
fileContent, err = ioutil.ReadAll(resp.Body)
|
||||
} else {
|
||||
// 处理本地文件
|
||||
fullPath := filepath.Join("public", path) // 注意这里的更改
|
||||
log.Printf("Attempting to read file: %s", fullPath)
|
||||
fileContent, err = ioutil.ReadFile(fullPath)
|
||||
}
|
||||
fullPath := filepath.Join("public", path)
|
||||
log.Printf("Attempting to read file: %s", fullPath)
|
||||
|
||||
fileContent, err := ioutil.ReadFile(fullPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("error reading CSV content: %w", err)
|
||||
return nil, fmt.Errorf("error reading CSV content: %w", err)
|
||||
}
|
||||
|
||||
lines := strings.Split(string(fileContent), "\n")
|
||||
var fileArray []string
|
||||
for _, line := range lines {
|
||||
trimmed := strings.TrimSpace(line)
|
||||
if trimmed != "" && !strings.HasPrefix(trimmed, "#") {
|
||||
fileArray = append(fileArray, trimmed)
|
||||
}
|
||||
trimmed := strings.TrimSpace(line)
|
||||
if trimmed != "" && !strings.HasPrefix(trimmed, "#") {
|
||||
fileArray = append(fileArray, trimmed)
|
||||
}
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
@ -167,8 +161,10 @@ func getCSVContent(path string) ([]string, error) {
|
||||
return fileArray, nil
|
||||
}
|
||||
|
||||
|
||||
func handleDynamicRequest(w http.ResponseWriter, r *http.Request) {
|
||||
realIP := getRealIP(r)
|
||||
log.Printf("Handling request from IP: %s\n", realIP)
|
||||
|
||||
if time.Since(lastFetchTime) > cacheDuration {
|
||||
if err := loadCSVPaths(); err != nil {
|
||||
http.Error(w, "Failed to load CSV paths", http.StatusInternalServerError)
|
||||
@ -204,6 +200,18 @@ func handleDynamicRequest(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
|
||||
randomURL := fileArray[rng.Intn(len(fileArray))]
|
||||
log.Printf("Redirecting to %s\n", randomURL)
|
||||
http.Redirect(w, r, randomURL, http.StatusFound)
|
||||
|
||||
proto := r.Header.Get("X-Forwarded-Proto")
|
||||
if proto == "" {
|
||||
proto = "http"
|
||||
}
|
||||
host := r.Header.Get("X-Forwarded-Host")
|
||||
if host == "" {
|
||||
host = r.Host
|
||||
}
|
||||
|
||||
redirectURL := fmt.Sprintf("%s://%s%s", proto, host, randomURL)
|
||||
|
||||
log.Printf("Redirecting to %s\n", redirectURL)
|
||||
http.Redirect(w, r, redirectURL, http.StatusFound)
|
||||
}
|
||||
|
||||
21
readme.md
21
readme.md
@ -74,6 +74,27 @@ https://example.com/image3.jpg
|
||||
|
||||
访问 `/pic/example` 或 `/video/example` 将重定向到相应 CSV 文件中的随机 URL。
|
||||
|
||||
|
||||
### 推荐的nginx反代配置
|
||||
|
||||
```
|
||||
location ^~ / {
|
||||
proxy_pass http://127.0.0.1:5003;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header REMOTE-HOST $remote_addr;
|
||||
proxy_set_header Upgrade $http_upgrade;
|
||||
proxy_set_header Connection "upgrade";
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
proxy_http_version 1.1;
|
||||
add_header X-Cache $upstream_cache_status;
|
||||
add_header Cache-Control no-cache;
|
||||
proxy_ssl_server_name off;
|
||||
add_header Strict-Transport-Security "max-age=31536000";
|
||||
}
|
||||
```
|
||||
|
||||
## 日志
|
||||
|
||||
日志文件位于 `logs/server.log`。使用 Docker Compose 时，可以通过卷挂载访问日志。
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user