From 4a3ec85686047a8f2b5d01bd17f7547fc6ddc2af Mon Sep 17 00:00:00 2001
From: Shivam Sandbhor <shivam.sandbhor@gmail.com>
Date: Fri, 15 Apr 2022 15:54:01 +0530
Subject: [PATCH] Update bouncer pull in rupture mode (#1445)

* Update bouncer pull in rupture mode

Signed-off-by: Shivam Sandbhor <shivam.sandbhor@gmail.com>
---
 pkg/apiserver/middlewares/v1/api_key.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pkg/apiserver/middlewares/v1/api_key.go b/pkg/apiserver/middlewares/v1/api_key.go
index 2778f4e0d..8e7dc087b 100644
--- a/pkg/apiserver/middlewares/v1/api_key.go
+++ b/pkg/apiserver/middlewares/v1/api_key.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"net/http"
 	"strings"
+	"time"
 
 	"github.com/crowdsecurity/crowdsec/pkg/database"
 	"github.com/gin-gonic/gin"
@@ -109,6 +110,12 @@ func (a *APIKey) MiddlewareFunc() gin.HandlerFunc {
 			}
 		}
 
+		if c.Request.Method != "HEAD" && time.Now().UTC().Sub(bouncer.LastPull) >= time.Minute {
+			if err := a.DbClient.UpdateBouncerLastPull(time.Now().UTC(), bouncer.ID); err != nil {
+				log.Errorf("failed to update bouncer last pull: %v", err)
+			}
+		}
+
 		c.Next()
 	}
 }