diff --git a/server/handlers.go b/server/handlers.go index bd6c20e..8e0d09c 100644 --- a/server/handlers.go +++ b/server/handlers.go @@ -10,6 +10,7 @@ import ( "io/ioutil" "net/http" "time" + "fmt" ) var ( @@ -39,16 +40,6 @@ func eventsHandlerJson(w http.ResponseWriter, r *http.Request) { headerName := "X-Correlation-Id" substring := "abcdabcd-1234" headerValue, ok := r.Header[headerName] - if ok { - if strings.HasPrefix(headerValue[0], substring) { - w.Header().Set("Content-Type", "application/json") - lib.JsonRequestChannel <- &lib.RequestObject{Body: body, Header: r.Header} - var rsp = CustomResponse{Code: 200, Message: "OK"} - json.NewEncoder(w).Encode(rsp) - log.Printf("Dropped blacklisted request") - return - } - } var reader io.Reader // check if body is gzip compressed @@ -74,6 +65,17 @@ func eventsHandlerJson(w http.ResponseWriter, r *http.Request) { metrics.EventHandlerTimeHist.WithLabelValues(JSON, ERROR).Observe(float64(utils.NanosToMillis(time.Now().UnixNano()) - eventHandlerStartTime)) return } + + if ok { + if strings.HasPrefix(headerValue[0], substring) { + w.Header().Set("Content-Type", "application/json") + lib.JsonRequestChannel <- &lib.RequestObject{Body: body, Header: r.Header} + var rsp = CustomResponse{Code: 200, Message: "OK"} + json.NewEncoder(w).Encode(rsp) + fmt.Println("Dropped blacklisted request") + return + } + } w.Header().Set("Content-Type", "application/json") lib.JsonRequestChannel <- &lib.RequestObject{Body: body, Header: r.Header}