This commit is contained in:
38
main.go
38
main.go
@@ -71,6 +71,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Server
|
// Server
|
||||||
|
http.HandleFunc("/whitelist", handleWhitelist)
|
||||||
http.HandleFunc("/check/", handleCheck)
|
http.HandleFunc("/check/", handleCheck)
|
||||||
http.Handle("/metrics", promhttp.Handler())
|
http.Handle("/metrics", promhttp.Handler())
|
||||||
|
|
||||||
@@ -176,6 +177,43 @@ func normalizePrefix(s string) (string, bool) {
|
|||||||
return s, err == nil
|
return s, err == nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func handleWhitelist(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodPost {
|
||||||
|
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var body struct {
|
||||||
|
IP string `json:"ip"`
|
||||||
|
}
|
||||||
|
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
||||||
|
http.Error(w, "bad request", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
addr, err := netip.ParseAddr(body.IP)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "invalid IP", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
key := "wl:" + addr.String()
|
||||||
|
if err := rdb.Set(ctx, key, "1", 0).Err(); err != nil {
|
||||||
|
http.Error(w, "redis error", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Optional: Cache leeren für die IP
|
||||||
|
prefixCacheMu.Lock()
|
||||||
|
defer prefixCacheMu.Unlock()
|
||||||
|
// Kein spezifischer IP-Cache in deinem Design, aber hier könnte man Cache invalidieren falls nötig
|
||||||
|
|
||||||
|
writeJSON(w, map[string]string{
|
||||||
|
"status": "whitelisted",
|
||||||
|
"ip": addr.String(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
// Check-Handler
|
// Check-Handler
|
||||||
func handleCheck(w http.ResponseWriter, r *http.Request) {
|
func handleCheck(w http.ResponseWriter, r *http.Request) {
|
||||||
checkRequests.Inc()
|
checkRequests.Inc()
|
||||||
|
Reference in New Issue
Block a user