From a3e569d0b38a829cac26bf4710e8c4371aabb46a Mon Sep 17 00:00:00 2001 From: jet tsang zeon-git Date: Tue, 7 Sep 2021 22:26:39 +0800 Subject: add a simple header printer Signed-off-by: jet tsang zeon-git --- main.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/main.go b/main.go index a489c51..fc456d4 100644 --- a/main.go +++ b/main.go @@ -24,6 +24,7 @@ var blockTxt string var shortenMap map[string]string var listen string var reloadURI = "/reload" +var clientHeaderURI = "/header" func init() { const ( @@ -60,6 +61,10 @@ func requestHandler(ctx *fasthttp.RequestCtx) { fmt.Fprint(ctx, "ok") ctx.SetStatusCode(fasthttp.StatusAccepted) return + } else if string(ctx.RequestURI()) == clientHeaderURI { + fmt.Fprint(ctx, string(ctx.Request.Header.Header())) + ctx.SetStatusCode(fasthttp.StatusAccepted) + return } if relayUrl, exist := shortenMap[string(ctx.Path())]; exist { var newRequest fasthttp.Request -- cgit v1.2.3