From d71b00e067bcfadaa3956c0abc002d7c3dea7236 Mon Sep 17 00:00:00 2001 From: bel Date: Wed, 6 Nov 2019 18:07:38 -0700 Subject: [PATCH] Fix --- config/config.go | 2 ++ server/server.go | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/config/config.go b/config/config.go index 833318b..1daa7f4 100644 --- a/config/config.go +++ b/config/config.go @@ -2,6 +2,7 @@ package config import ( "fmt" + "log" "strings" "time" ) @@ -42,6 +43,7 @@ func GetPort() string { func GetRate() (int, int) { rate := conf.Get("r").GetInt() burst := conf.Get("b").GetInt() + log.Println("rate/burst:", rate, burst) return rate, burst } diff --git a/server/server.go b/server/server.go index d88be9a..48277f1 100755 --- a/server/server.go +++ b/server/server.go @@ -164,7 +164,7 @@ func pipe(a, b net.Conn) { func (s *Server) Pre(foo http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { - ctx, can := context.WithTimeout(r.Context(), time.Second*time.Duration(config.GetTimeout())) + ctx, can := context.WithTimeout(r.Context(), time.Duration(config.GetTimeout())) defer can() if err := s.limiter.Wait(ctx); err != nil { w.WriteHeader(http.StatusTooManyRequests)