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

batch in second intervals

Browse source
This commit is contained in:
Andre Medeiros 2021-07-08 20:53:09 -04:00
parent 4d2554fba6
commit 45576b4d16
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cmd/apollo-scheduler/main.go
View file

@ -108,6 +108,10 @@ func main() {
func enqueueAccounts(ctx context.Context, logger *logrus.Logger, pool *pgxpool.Pool, redisConn *redis.Client, queue rmq.Queue) {
now := float64(time.Now().UnixNano()/int64(time.Millisecond)) / 1000
start := now
end := now + 1
ids := []int64{}
err := pool.BeginFunc(ctx, func(tx pgx.Tx) error {
@ -115,9 +119,8 @@ func enqueueAccounts(ctx context.Context, logger *logrus.Logger, pool *pgxpool.P
WITH account AS (
SELECT id
FROM accounts
WHERE
last_checked_at + 5 < $1 AND
last_enqueued_at + 5 < $1
WHERE (last_checked_at + 5) BETWEEN $1 AND $2
OR last_checked_at + 60 < $1
ORDER BY last_checked_at
FOR UPDATE SKIP LOCKED
)
@ -125,7 +128,7 @@ func enqueueAccounts(ctx context.Context, logger *logrus.Logger, pool *pgxpool.P
SET last_enqueued_at = $1
WHERE accounts.id IN(SELECT id FROM account)
RETURNING accounts.id`
rows, err := tx.Query(ctx, stmt, now)
rows, err := tx.Query(ctx, stmt, start, end)
if err != nil {
return err
}