feat: BED-5079 - add a count function for efficiently counting ingest…

… tasks

cmd/api/src/api/error.go

@@ -54,7 +54,7 @@ const (
 	ErrorResponseDetailsOTPInvalid                  = "one time password is invalid"
 	ErrorResponseDetailsResourceNotFound            = "resource not found"
 	ErrorResponseDetailsToBeforeFrom                = "to time cannot be before from time"
-	ErrorResponseDetailsTimeRangeInvalid	= "time range provided is invalid"
+	ErrorResponseDetailsTimeRangeInvalid            = "time range provided is invalid"
 	ErrorResponseDetailsToMalformed                 = "to parameter should be formatted as RFC3339 i.e 2021-04-21T07:20:50.52Z"
 	ErrorResponseMultipleCollectionScopesProvided   = "may only scope collection by exactly one of OU, Domain, or All Trusted Domains"
 	ErrorResponsePayloadUnmarshalError              = "error unmarshalling JSON payload"

cmd/api/src/database/db.go

@@ -66,6 +66,7 @@ type Database interface {
 	// Ingest
 	ingest.IngestData
 	GetAllIngestTasks(ctx context.Context) (model.IngestTasks, error)
+	CountAllIngestTasks(ctx context.Context) (int64, error)
 	DeleteIngestTask(ctx context.Context, ingestTask model.IngestTask) error
 	GetIngestTasksForJob(ctx context.Context, jobID int64) (model.IngestTasks, error)
 

cmd/api/src/database/ingest.go

@@ -35,6 +35,16 @@ func (s *BloodhoundDB) GetAllIngestTasks(ctx context.Context) (model.IngestTasks
 	return ingestTasks, CheckError(result)
 }
 
+func (s *BloodhoundDB) CountAllIngestTasks(ctx context.Context) (int64, error) {
+	var (
+		ingestTaskCount  int64
+		ingestTasksModel model.IngestTasks
+	)
+
+	result := s.db.Model(&ingestTasksModel).WithContext(ctx).Count(&ingestTaskCount)
+	return ingestTaskCount, CheckError(result)
+}
+
 func (s *BloodhoundDB) DeleteIngestTask(ctx context.Context, ingestTask model.IngestTask) error {
 	result := s.db.WithContext(ctx).Delete(&ingestTask)
 	return CheckError(result)