KJZH001/nezha-2024
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor code

Browse Source
This commit is contained in:
naiba 2024-11-23 00:02:57 +08:00
parent 58cc8aa5df
commit 68f6da436d
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/dashboard/controller/waf/waf.go
View File

@ -36,12 +36,12 @@ func RealIp(c *gin.Context) {
c.AbortWithStatusJSON(http.StatusOK, model.CommonResponse[any]{Success: false, Error: "real ip header not found"})
return
}
ip, err := netip.ParseAddr(vals)
ip, err := netip.ParseAddrPort(vals)
if err != nil {
c.AbortWithStatusJSON(http.StatusOK, model.CommonResponse[any]{Success: false, Error: err.Error()})
return
}
c.Set(model.CtxKeyRealIPStr, ip.String())
c.Set(model.CtxKeyRealIPStr, ip.Addr().String())
c.Next()
}

4
cmd/dashboard/rpc/rpc.go
View File

@ -51,11 +51,11 @@ func getRealIp(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
}
a := strings.Split(vals[0], ",")
h := strings.TrimSpace(a[len(a)-1])
ip, err := netip.ParseAddr(h)
ip, err := netip.ParseAddrPort(h)
if err != nil {
return nil, err
}
ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.String())
ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.Addr().String())
return handler(ctx, req)
}