stackrox · janisz · Jul 7, 2023 · Jun 30, 2023 · rhybrillou · Jul 5, 2023
diff --git a/central/authprovider/datastore/internal/store/postgres/store.go b/central/authprovider/datastore/internal/store/postgres/store.go
diff --git a/central/complianceoperatorintegration/store/postgres/store.go b/central/complianceoperatorintegration/store/postgres/store.go
diff --git a/central/externalbackups/internal/store/postgres/store.go b/central/externalbackups/internal/store/postgres/store.go
diff --git a/central/group/datastore/internal/store/postgres/store.go b/central/group/datastore/internal/store/postgres/store.go
diff --git a/central/imageintegration/store/postgres/store.go b/central/imageintegration/store/postgres/store.go
diff --git a/central/logimbue/store/postgres/store.go b/central/logimbue/store/postgres/store.go
diff --git a/central/notifier/datastore/internal/store/postgres/store.go b/central/notifier/datastore/internal/store/postgres/store.go
diff --git a/central/policycategory/store/postgres/store.go b/central/policycategory/store/postgres/store.go
diff --git a/central/policycategoryedge/store/postgres/store.go b/central/policycategoryedge/store/postgres/store.go
diff --git a/central/serviceidentities/internal/store/postgres/store.go b/central/serviceidentities/internal/store/postgres/store.go
diff --git a/pkg/search/postgres/store.go b/pkg/search/postgres/store.go
@@ -177,3 +177,15 @@ func (s *GenericStore[T, PT]) Walk(ctx context.Context, fn func(obj PT) error) e
 	}
 	return nil
 }
+
+// GetAll retrieves all objects from the store.
+func (s *GenericStore[T, PT]) GetAll(ctx context.Context) ([]PT, error) {
+	defer s.setPostgresOperationDurationTime(time.Now(), ops.GetAll)
+
+	var objs []PT
+	err := s.Walk(ctx, func(obj PT) error {
+		objs = append(objs, obj)
+		return nil
+	})
+	return objs, err
+}
diff --git a/tools/generate-helpers/pg-table-bindings/store.go.tpl b/tools/generate-helpers/pg-table-bindings/store.go.tpl
@@ -727,21 +727,6 @@ func (s *storeImpl) GetIDs(ctx context.Context) ([]{{$singlePK.Type}}, error) {
 }
 {{- end }}
 
-{{- if .GetAll }}
-
-// GetAll retrieves all objects from the store.
-func(s *storeImpl) GetAll(ctx context.Context) ([]*{{.Type}}, error) {
-	defer metrics.SetPostgresOperationDurationTime(time.Now(), ops.GetAll, "{{.TrimmedType}}")
-
-    var objs []*{{.Type}}
-    err := s.Walk(ctx, func(obj *{{.Type}}) error {
-        objs = append(objs, obj)
-        return nil
-    })
-    return objs, err
-}
-{{- end}}
-
 //// Interface functions - END
 
 //// Used for testing

diff --git a/tools/generate-helpers/pg-table-bindings/test/postgres/store.go b/tools/generate-helpers/pg-table-bindings/test/postgres/store.go
diff --git a/tools/generate-helpers/pg-table-bindings/testuuidkey/postgres/store.go b/tools/generate-helpers/pg-table-bindings/testuuidkey/postgres/store.go