feat: add clientIP for accesslog (#24)

Signed-off-by: Dup4 <[email protected]>
hertz-contrib · Mar 20, 2023 · b153fe7 · b153fe7
1 parent 67a82fa
commit b153fe7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/accesslog/README.md b/accesslog/README.md
@@ -156,7 +156,7 @@ package main
 
 import (
 	"context"
-	
+
 	"github.com/cloudwego/hertz/pkg/app"
 	"github.com/cloudwego/hertz/pkg/app/server"
 	"github.com/cloudwego/hertz/pkg/common/hlog"
@@ -200,6 +200,7 @@ const (
 	TagPort              = "port"
 	TagIP                = "ip"
 	TagIPs               = "ips"
+	TagClientIP          = "clientIP"
 	TagHost              = "host"
 	TagMethod            = "method"
 	TagPath              = "path"
@@ -251,4 +252,4 @@ func main() {
 	h.Spin()
 }
 
-```
+```
diff --git a/accesslog/tag.go b/accesslog/tag.go
@@ -60,6 +60,7 @@ const (
 	TagIP                = "ip"
 	TagIPs               = "ips"
 	TagHost              = "host"
+	TagClientIP          = "clientIP"
 	TagMethod            = "method"
 	TagPath              = "path"
 	TagURL               = "url"
@@ -102,6 +103,9 @@ var Tags = map[string]logTagFunc{
 	TagHost: func(ctx context.Context, c *app.RequestContext, buf *bytebufferpool.ByteBuffer) (int, error) {
 		return buf.WriteString(string(c.Request.URI().Host()))
 	},
+	TagClientIP: func(ctx context.Context, c *app.RequestContext, buf *bytebufferpool.ByteBuffer) (int, error) {
+		return buf.WriteString(c.ClientIP())
+	},
 	TagPath: func(ctx context.Context, c *app.RequestContext, buf *bytebufferpool.ByteBuffer) (int, error) {
 		return buf.WriteString(string(c.Request.Path()))
 	},