From 7706bf250f24432eee128ec6a0a92dc1a5617a36 Mon Sep 17 00:00:00 2001 From: xgdgsc Date: Mon, 22 Aug 2022 12:59:12 +0800 Subject: [PATCH] add slash --- server/server.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/server/server.go b/server/server.go index e27190d..40b2919 100644 --- a/server/server.go +++ b/server/server.go @@ -111,7 +111,12 @@ func (server *Server) Run(ctx context.Context, options ...RunOption) error { if server.options.EnableRandomUrl { path = "/" + randomstring.Generate(server.options.RandomUrlLength) + "/" } - + if !strings.HasPrefix(path, "/") { + path = "/" + path + } + if !strings.HasSuffix(path, "/") { + path = path + "/" + } handlers := server.setupHandlers(cctx, cancel, path, counter) srv, err := server.setupHTTPServer(handlers) if err != nil {