fix/production patch (#11)

anilibria · Jul 14, 2024 · 73d6aa1 · 73d6aa1
2 parents 9215747 + 0efcc8e
commit 73d6aa1
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/internal/cache/cache.go b/internal/cache/cache.go
@@ -33,7 +33,7 @@ func NewCache(c context.Context) (cache *Cache, e error) {
 		CleanWindow: cli.Duration("cache-clean-window"),
 
 		MaxEntriesInWindow: 1000 * 10 * 60,
-		MaxEntrySize:       32 * 1024,
+		MaxEntrySize:       128 * 1024,
 
 		// not worked?
 		Verbose: zerolog.GlobalLevel() == zerolog.TraceLevel,

diff --git a/internal/proxy/proxy.go b/internal/proxy/proxy.go
@@ -47,7 +47,7 @@ func (m *Proxy) ProxyFiberRequest(c *fiber.Ctx) (e error) {
 	rsp := fasthttp.AcquireResponse()
 	defer fasthttp.ReleaseResponse(rsp)
 
-	if e = m.doRequest(req, rsp); e != nil {
+	if e = m.doRequest(c, req, rsp); e != nil {
 		return
 	}
 
@@ -79,25 +79,36 @@ func (m *Proxy) acquireRewritedRequest(c *fiber.Ctx) *fasthttp.Request {
 	return req
 }
 
-func (m *Proxy) doRequest(req *fasthttp.Request, rsp *fasthttp.Response) (e error) {
+func (m *Proxy) doRequest(c *fiber.Ctx, req *fasthttp.Request, rsp *fasthttp.Response) (e error) {
 	if e = m.client.Do(req, rsp); e != nil {
 		return
 	}
 
 	status, body := rsp.StatusCode(), rsp.Body()
 
-	if status < fasthttp.StatusOK && status >= fasthttp.StatusInternalServerError {
+	if status < fasthttp.StatusOK || status >= fasthttp.StatusInternalServerError {
 		e = fmt.Errorf("proxy server respond with status %d", status)
 		return
+	} else if status >= fiber.StatusBadRequest {
+		rlog(c).Warn().Msgf("status %d detected for request, bypass cache", status)
+
+		m.bypassCache(c)
+		return
 	}
 
 	if len(body) == 0 {
 		e = errors.New("proxy server respond with nil body")
+		rlog(c).Trace().Msg(req.String())
 	}
 
 	return
 }
 
+func (*Proxy) bypassCache(c *fiber.Ctx) {
+	key := c.Context().UserValue(utils.UVCacheKey).(*Key)
+	key.Reset()
+}
+
 func (m *Proxy) cacheResponse(c *fiber.Ctx, rsp *fasthttp.Response) (e error) {
 	key := c.Context().UserValue(utils.UVCacheKey).(*Key)
 

diff --git a/internal/service/router.go b/internal/service/router.go
@@ -135,7 +135,7 @@ func (m *Service) fiberRouterInitialization() {
 	cacheapi := m.fb.Group("/internal/cache", m.proxy.MiddlewareInternalApi)
 
 	cacheapi.Get("/stats", m.proxy.HandleCacheStats)
-	cacheapi.Post("/stats/reset", m.proxy.HandleCacheStats)
+	cacheapi.Post("/stats/reset", m.proxy.HandleCacheStatsReset)
 	cacheapi.Get("/dump", m.proxy.HandleCacheDump)
 	cacheapi.Get("/dumpkeys", m.proxy.HandleCacheDumpKeys)
 	cacheapi.Post("/purge", m.proxy.HandleCachePurge)