[v1.4.1] Allow disable crash

troydai · Nov 29, 2024 · 56bf17f · 56bf17f
1 parent c85667b
commit 56bf17f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/internal/http/server.go b/internal/http/server.go
@@ -32,10 +32,11 @@ type Server struct {
 
 func (s *Server) HandleHTTP(w http.ResponseWriter, r *http.Request) {
 	s.logger.Info("inbound", "method", r.Method, "url", r.URL.Path, "host", r.Host, "proto", r.Proto, "agent", r.UserAgent(), "peer", r.RemoteAddr, "counter", s.counter.Load())
+	newCount := s.counter.Add(1)
 
-	if newValue := s.counter.Add(1); newValue%s.frequency == 0 {
-		s.logger.Warn("crash", "counter", newValue)
-		panic("crash after 10 requests")
+	if s.frequency != 0 && newCount%s.frequency == 0 {
+		s.logger.Warn("crash", "counter", newCount)
+		panic("crash as requested")
 	}
 
 	s.logger.Info("outbound", "status", 200)