94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"strings"
|
||
"syscall"
|
||
|
||
"github.com/sevlyar/go-daemon"
|
||
)
|
||
|
||
// 获取客户端IP的函数
|
||
func getClientIP(r *http.Request) string {
|
||
// 尝试从X-Forwarded-For头部获取IP
|
||
forwarded := r.Header.Get("X-Forwarded-For")
|
||
if forwarded != "" {
|
||
// X-Forwarded-For可能包含多个IP地址，取第一个
|
||
ips := strings.Split(forwarded, ",")
|
||
return strings.TrimSpace(ips[0])
|
||
}
|
||
// 尝试从X-Real-Ip头部获取IP
|
||
realIP := r.Header.Get("X-Real-Ip")
|
||
if realIP != "" {
|
||
return realIP
|
||
}
|
||
// 最后从RemoteAddr中获取IP
|
||
return r.RemoteAddr
|
||
}
|
||
|
||
// 处理HTTP请求的函数
|
||
func handler(w http.ResponseWriter, r *http.Request) {
|
||
clientIP := getClientIP(r)
|
||
response := fmt.Sprintf("%s", clientIP)
|
||
fmt.Fprintln(w, response)
|
||
log.Printf("Handled request from %s\n", clientIP) // 记录日志
|
||
}
|
||
|
||
func main() {
|
||
// 定义守护进程的启动配置
|
||
cntxt := &daemon.Context{
|
||
PidFileName: "inet-ip.pid",
|
||
PidFilePerm: 0644,
|
||
LogFileName: "inet-ip.log",
|
||
LogFilePerm: 0640,
|
||
WorkDir: "./",
|
||
Umask: 027,
|
||
Args: []string{"[inet-ip]"},
|
||
}
|
||
|
||
// 启动守护进程
|
||
d, err := cntxt.Reborn()
|
||
if err != nil {
|
||
log.Fatal("Unable to run: ", err)
|
||
}
|
||
if d != nil {
|
||
return
|
||
}
|
||
defer cntxt.Release()
|
||
|
||
// 设置日志输出到文件
|
||
logFile, err := os.OpenFile("inet-ip.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
|
||
if err != nil {
|
||
log.Fatalf("Error opening log file: %s", err)
|
||
}
|
||
defer logFile.Close()
|
||
log.SetOutput(logFile)
|
||
|
||
// 启动HTTP服务器
|
||
go serveHTTP()
|
||
|
||
// 创建一个通道来接收操作系统信号
|
||
stop := make(chan os.Signal, 1)
|
||
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
|
||
|
||
log.Println("Server started")
|
||
|
||
// 等待信号
|
||
<-stop
|
||
|
||
// 服务器停止时记录日志
|
||
log.Println("Server stopped")
|
||
}
|
||
|
||
func serveHTTP() {
|
||
http.HandleFunc("/", handler)
|
||
fmt.Println("Starting server on :90")
|
||
if err := http.ListenAndServe(":90", nil); err != nil && err != http.ErrServerClosed {
|
||
log.Fatalf("Error starting server: %s", err)
|
||
}
|
||
}
|