diff --git a/internal/cmd/scheduler.go b/internal/cmd/scheduler.go index a2afde6..8729857 100644 --- a/internal/cmd/scheduler.go +++ b/internal/cmd/scheduler.go @@ -21,7 +21,7 @@ import ( "github.com/christianselig/apollo-backend/internal/repository" ) -const batchSize = 1000 +const batchSize = 250 func SchedulerCmd(ctx context.Context) *cobra.Command { cmd := &cobra.Command{ @@ -203,7 +203,7 @@ func enqueueUsers(ctx context.Context, logger *zap.Logger, statsd *statsd.Client now := time.Now() next := now.Add(domain.NotificationCheckInterval) - var ids []int64 + ids := []int64{} defer func() { tags := []string{"queue:users"} @@ -229,16 +229,11 @@ func enqueueUsers(ctx context.Context, logger *zap.Logger, statsd *statsd.Client return err } defer rows.Close() - - affected := rows.CommandTag().RowsAffected() - ids = make([]int64, affected) - - var id int64 - for i := 0; rows.Next(); i++ { + for rows.Next() { + var id int64 _ = rows.Scan(&id) - ids[i] = id + ids = append(ids, id) } - return nil }) @@ -388,7 +383,7 @@ func enqueueAccounts(ctx context.Context, logger *zap.Logger, statsd *statsd.Cli now := time.Now() next := now.Add(domain.NotificationCheckInterval) - var ids []int64 + ids := []int64{} enqueued := 0 skipped := 0 @@ -417,15 +412,10 @@ func enqueueAccounts(ctx context.Context, logger *zap.Logger, statsd *statsd.Cli return err } defer rows.Close() - - affected := rows.CommandTag().RowsAffected() - ids = make([]int64, affected) - - var id int64 - for i := 0; rows.Next(); i++ { + for rows.Next() { + var id int64 _ = rows.Scan(&id) - ids[i] = id - i++ + ids = append(ids, id) } return nil })