shuguang/nezha
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"}) c.AbortWithStatusJSON(http.StatusOK, model.CommonResponse[any]{Success: false, Error: "real ip header not found"})
return return
} }
ip, err := netip.ParseAddr(vals) ip, err := netip.ParseAddrPort(vals)
if err != nil { if err != nil {
c.AbortWithStatusJSON(http.StatusOK, model.CommonResponse[any]{Success: false, Error: err.Error()}) c.AbortWithStatusJSON(http.StatusOK, model.CommonResponse[any]{Success: false, Error: err.Error()})
return return
} }
c.Set(model.CtxKeyRealIPStr, ip.String()) c.Set(model.CtxKeyRealIPStr, ip.Addr().String())
c.Next() 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], ",") a := strings.Split(vals[0], ",")
h := strings.TrimSpace(a[len(a)-1]) h := strings.TrimSpace(a[len(a)-1])
ip, err := netip.ParseAddr(h) ip, err := netip.ParseAddrPort(h)
if err != nil { if err != nil {
return nil, err return nil, err
} }
ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.String()) ctx = context.WithValue(ctx, model.CtxKeyRealIP{}, ip.Addr().String())
return handler(ctx, req) return handler(ctx, req)
} }