From c4750d61f565488a83a840943b8e16c8263b0c4f Mon Sep 17 00:00:00 2001
From: Tim Lee <timlee@fleetdm.com>
Date: Mon, 11 Mar 2024 14:14:29 -0600
Subject: [PATCH] bugfix: add counts_updated_at  (#17543)

---
 server/service/vulnerabilities.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/server/service/vulnerabilities.go b/server/service/vulnerabilities.go
index 6067187b96..07fd0bf8bb 100644
--- a/server/service/vulnerabilities.go
+++ b/server/service/vulnerabilities.go
@@ -24,6 +24,7 @@ type listVulnerabilitiesRequest struct {
 type listVulnerabilitiesResponse struct {
 	Vulnerabilities []fleet.VulnerabilityWithMetadata `json:"vulnerabilities"`
 	Count           uint                              `json:"count"`
+	CountsUpdatedAt time.Time                         `json:"counts_updated_at"`
 	Meta            *fleet.PaginationMetadata         `json:"meta,omitempty"`
 	Err             error                             `json:"error,omitempty"`
 }
@@ -42,10 +43,18 @@ func listVulnerabilitiesEndpoint(ctx context.Context, req interface{}, svc fleet
 		return listVulnerabilitiesResponse{Err: err}, nil
 	}
 
+	updatedAt := time.Now()
+	for _, vuln := range vulns {
+		if vuln.HostsCountUpdatedAt.Before(updatedAt) {
+			updatedAt = vuln.HostsCountUpdatedAt
+		}
+	}
+
 	return listVulnerabilitiesResponse{
 		Vulnerabilities: vulns,
 		Meta:            meta,
 		Count:           count,
+		CountsUpdatedAt: updatedAt,
 	}, nil
 }