Add tasks totalCount support

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/coredata/task.go

@@ -248,6 +248,38 @@ RETURNING
 	return nil
 }
 
+func (c *Tasks) CountByOrganizationID(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	organizationID gid.GID,
+) (int, error) {
+	q := `
+	SELECT
+		COUNT(id)
+	FROM
+		tasks
+	WHERE
+		%s
+		AND organization_id = @organization_id
+	`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"organization_id": organizationID}
+	maps.Copy(args, scope.SQLArguments())
+
+	row := conn.QueryRow(ctx, q, args)
+
+	var count int
+	err := row.Scan(&count)
+	if err != nil {
+		return 0, fmt.Errorf("cannot collect tasks: %w", err)
+	}
+
+	return count, nil
+}
+
 func (c *Tasks) LoadByOrganizationID(
 	ctx context.Context,
 	conn pg.Conn,
@@ -297,6 +329,38 @@ func (c *Tasks) LoadByOrganizationID(
 	return nil
 }
 
+func (c *Tasks) CountByMeasureID(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+	measureID gid.GID,
+) (int, error) {
+	q := `
+SELECT
+    COUNT(id)
+FROM
+    tasks
+WHERE
+    %s
+    AND measure_id = @measure_id
+`
+
+	q = fmt.Sprintf(q, scope.SQLFragment())
+
+	args := pgx.StrictNamedArgs{"measure_id": measureID}
+	maps.Copy(args, scope.SQLArguments())
+
+	row := conn.QueryRow(ctx, q, args)
+
+	var count int
+	err := row.Scan(&count)
+	if err != nil {
+		return 0, fmt.Errorf("cannot collect tasks: %w", err)
+	}
+
+	return count, nil
+}
+
 func (c *Tasks) LoadByMeasureID(
 	ctx context.Context,
 	conn pg.Conn,