From c1e078b3a33e70ab6830d1b5bc05324b7bbab1e6 Mon Sep 17 00:00:00 2001
From: Leonor Oliveira <9090754+leonorfmartins@users.noreply.github.com>
Date: Thu, 19 Dec 2024 12:38:08 +0100
Subject: [PATCH] Include library elements on counts endpoint (#98246)

---
 pkg/storage/unified/federated/stats.go      | 6 ++++++
 pkg/storage/unified/federated/stats_test.go | 8 ++++++++
 2 files changed, 14 insertions(+)

diff --git a/pkg/storage/unified/federated/stats.go b/pkg/storage/unified/federated/stats.go
index 8e5f7c0033c..42eb8566d86 100644
--- a/pkg/storage/unified/federated/stats.go
+++ b/pkg/storage/unified/federated/stats.go
@@ -63,6 +63,12 @@ func (s *LegacyStatsGetter) GetStats(ctx context.Context, in *resource.ResourceS
 		if err != nil {
 			return err
 		}
+
+		// Legacy library_elements table
+		err = fn("library_element", "org_id=? AND folder_uid=?", group, "library_elements")
+		if err != nil {
+			return err
+		}
 		return nil
 	})
 
diff --git a/pkg/storage/unified/federated/stats_test.go b/pkg/storage/unified/federated/stats_test.go
index 56181303632..669752b2c29 100644
--- a/pkg/storage/unified/federated/stats_test.go
+++ b/pkg/storage/unified/federated/stats_test.go
@@ -133,6 +133,10 @@ func TestDirectSQLStats(t *testing.T) {
 				"group": "sql-fallback",
 				"resource": "folders",
 				"count": 1
+			},
+			{
+				"group": "sql-fallback",
+				"resource": "library_elements"
 			}
 		]`, string(jj))
 	})
@@ -161,6 +165,10 @@ func TestDirectSQLStats(t *testing.T) {
 			{
 				"group": "sql-fallback",
 				"resource": "folders"
+			},
+						{
+				"group": "sql-fallback",
+				"resource": "library_elements"
 			}
 		]`, string(jj))
 	})