mirrors/apollo-backend
2
0
Fork 0
mirror of https://github.com/christianselig/apollo-backend synced 2024-11-22 03:37:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

Revert "another experiment for scheduling"

Browse source 
This reverts commit c8dd76e1ab.
This commit is contained in:
Andre Medeiros 2022-05-27 15:02:47 -04:00
parent f7516a06b3
commit 01ad9f3d1e
1 changed files with 1 additions and 58 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
internal/cmd/scheduler.go
View file

@ -95,7 +95,7 @@ func SchedulerCmd(ctx context.Context) *cobra.Command {
go func() {
for {
enqueueAccounts(ctx, logger, statsd, db, redis, luaSha, notifQueue)
time.Sleep(5 * time.Second)
time.Sleep(100 * time.Millisecond)
}
}()
//_, _ = s.Every(100).Milliseconds().SingletonMode().Do(func() { enqueueAccounts(ctx, logger, statsd, db, redis, luaSha, notifQueue) })
@ -394,63 +394,6 @@ func enqueueStuckAccounts(ctx context.Context, logger *zap.Logger, statsd *stats
}
func enqueueAccounts(ctx context.Context, logger *zap.Logger, statsd *statsd.Client, pool *pgxpool.Pool, redisConn *redis.Client, luaSha string, queue rmq.Queue) {
ids := []int64{}
now := time.Now()
rows, err := pool.Query(ctx, "SELECT id FROM accounts")
if err != nil {
logger.Error("failed to fetch accounts", zap.Error(err))
return
}
defer rows.Close()
var id int64
for i := 0; rows.Next(); i++ {
_ = rows.Scan(&id)
ids = append(ids, id)
}
logger.Debug("enqueueing account batch", zap.Int("count", len(ids)), zap.Time("start", now))
// Split ids in batches
enqueued := 0
skipped := 0
for i := 0; i < len(ids); i += batchSize {
j := i + batchSize
if j > len(ids) {
j = len(ids)
}
batch := Int64Slice(ids[i:j])
logger.Debug("enqueueing batch", zap.Int("len", len(batch)))
res, err := redisConn.EvalSha(ctx, luaSha, []string{"locks:accounts"}, batch).Result()
if err != nil {
logger.Error("failed to check for locked accounts", zap.Error(err))
}
vals := res.([]interface{})
skipped += len(batch) - len(vals)
enqueued += len(vals)
if len(vals) == 0 {
continue
}
batchIds := make([]string, len(vals))
for k, v := range vals {
batchIds[k] = strconv.FormatInt(v.(int64), 10)
}
if err = queue.Publish(batchIds...); err != nil {
logger.Error("failed to enqueue account batch", zap.Error(err))
}
}
logger.Debug("done enqueueing account batch", zap.Int("count", enqueued), zap.Int("skipped", skipped), zap.Time("start", now))
}
func enqueueAccounts2(ctx context.Context, logger *zap.Logger, statsd *statsd.Client, pool *pgxpool.Pool, redisConn *redis.Client, luaSha string, queue rmq.Queue) {
now := time.Now()
next := now.Add(domain.NotificationCheckInterval)