sidekiq-unique-jobs 6.0.24 → 7.0.4

Sign up to get free protection for your applications and to get access to all the features.

Potentially problematic release.


This version of sidekiq-unique-jobs might be problematic. Click here for more details.

Files changed (122) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +707 -25
  3. data/README.md +516 -105
  4. data/lib/sidekiq_unique_jobs.rb +48 -7
  5. data/lib/sidekiq_unique_jobs/batch_delete.rb +123 -0
  6. data/lib/sidekiq_unique_jobs/changelog.rb +78 -0
  7. data/lib/sidekiq_unique_jobs/cli.rb +34 -31
  8. data/lib/sidekiq_unique_jobs/config.rb +263 -0
  9. data/lib/sidekiq_unique_jobs/connection.rb +6 -5
  10. data/lib/sidekiq_unique_jobs/constants.rb +46 -24
  11. data/lib/sidekiq_unique_jobs/core_ext.rb +80 -0
  12. data/lib/sidekiq_unique_jobs/digests.rb +71 -100
  13. data/lib/sidekiq_unique_jobs/exceptions.rb +78 -12
  14. data/lib/sidekiq_unique_jobs/job.rb +41 -12
  15. data/lib/sidekiq_unique_jobs/json.rb +40 -0
  16. data/lib/sidekiq_unique_jobs/key.rb +93 -0
  17. data/lib/sidekiq_unique_jobs/lock.rb +325 -0
  18. data/lib/sidekiq_unique_jobs/lock/base_lock.rb +66 -50
  19. data/lib/sidekiq_unique_jobs/lock/client_validator.rb +28 -0
  20. data/lib/sidekiq_unique_jobs/lock/server_validator.rb +27 -0
  21. data/lib/sidekiq_unique_jobs/lock/until_and_while_executing.rb +7 -10
  22. data/lib/sidekiq_unique_jobs/lock/until_executed.rb +6 -6
  23. data/lib/sidekiq_unique_jobs/lock/until_executing.rb +1 -1
  24. data/lib/sidekiq_unique_jobs/lock/until_expired.rb +4 -21
  25. data/lib/sidekiq_unique_jobs/lock/validator.rb +96 -0
  26. data/lib/sidekiq_unique_jobs/lock/while_executing.rb +13 -9
  27. data/lib/sidekiq_unique_jobs/lock/while_executing_reject.rb +3 -3
  28. data/lib/sidekiq_unique_jobs/lock_args.rb +123 -0
  29. data/lib/sidekiq_unique_jobs/lock_config.rb +122 -0
  30. data/lib/sidekiq_unique_jobs/lock_digest.rb +79 -0
  31. data/lib/sidekiq_unique_jobs/lock_info.rb +68 -0
  32. data/lib/sidekiq_unique_jobs/lock_timeout.rb +62 -0
  33. data/lib/sidekiq_unique_jobs/lock_ttl.rb +77 -0
  34. data/lib/sidekiq_unique_jobs/locksmith.rb +261 -101
  35. data/lib/sidekiq_unique_jobs/logging.rb +149 -23
  36. data/lib/sidekiq_unique_jobs/logging/middleware_context.rb +44 -0
  37. data/lib/sidekiq_unique_jobs/lua/delete.lua +51 -0
  38. data/lib/sidekiq_unique_jobs/lua/delete_by_digest.lua +42 -0
  39. data/lib/sidekiq_unique_jobs/lua/delete_job_by_digest.lua +38 -0
  40. data/lib/sidekiq_unique_jobs/lua/find_digest_in_queues.lua +26 -0
  41. data/lib/sidekiq_unique_jobs/lua/lock.lua +93 -0
  42. data/lib/sidekiq_unique_jobs/lua/locked.lua +35 -0
  43. data/lib/sidekiq_unique_jobs/lua/queue.lua +87 -0
  44. data/lib/sidekiq_unique_jobs/lua/reap_orphans.lua +94 -0
  45. data/lib/sidekiq_unique_jobs/lua/shared/_common.lua +40 -0
  46. data/lib/sidekiq_unique_jobs/lua/shared/_current_time.lua +8 -0
  47. data/lib/sidekiq_unique_jobs/lua/shared/_delete_from_queue.lua +22 -0
  48. data/lib/sidekiq_unique_jobs/lua/shared/_delete_from_sorted_set.lua +18 -0
  49. data/lib/sidekiq_unique_jobs/lua/shared/_find_digest_in_process_set.lua +53 -0
  50. data/lib/sidekiq_unique_jobs/lua/shared/_find_digest_in_queues.lua +43 -0
  51. data/lib/sidekiq_unique_jobs/lua/shared/_find_digest_in_sorted_set.lua +24 -0
  52. data/lib/sidekiq_unique_jobs/lua/shared/_hgetall.lua +13 -0
  53. data/lib/sidekiq_unique_jobs/lua/shared/_upgrades.lua +3 -0
  54. data/lib/sidekiq_unique_jobs/lua/unlock.lua +95 -0
  55. data/lib/sidekiq_unique_jobs/lua/update_version.lua +40 -0
  56. data/lib/sidekiq_unique_jobs/lua/upgrade.lua +68 -0
  57. data/lib/sidekiq_unique_jobs/middleware.rb +29 -31
  58. data/lib/sidekiq_unique_jobs/middleware/client.rb +42 -0
  59. data/lib/sidekiq_unique_jobs/middleware/server.rb +27 -0
  60. data/lib/sidekiq_unique_jobs/normalizer.rb +4 -4
  61. data/lib/sidekiq_unique_jobs/on_conflict.rb +23 -10
  62. data/lib/sidekiq_unique_jobs/on_conflict/log.rb +9 -5
  63. data/lib/sidekiq_unique_jobs/on_conflict/null_strategy.rb +1 -1
  64. data/lib/sidekiq_unique_jobs/on_conflict/raise.rb +1 -1
  65. data/lib/sidekiq_unique_jobs/on_conflict/reject.rb +61 -15
  66. data/lib/sidekiq_unique_jobs/on_conflict/replace.rb +54 -14
  67. data/lib/sidekiq_unique_jobs/on_conflict/reschedule.rb +12 -5
  68. data/lib/sidekiq_unique_jobs/on_conflict/strategy.rb +25 -6
  69. data/lib/sidekiq_unique_jobs/options_with_fallback.rb +41 -27
  70. data/lib/sidekiq_unique_jobs/orphans/lua_reaper.rb +29 -0
  71. data/lib/sidekiq_unique_jobs/orphans/manager.rb +212 -0
  72. data/lib/sidekiq_unique_jobs/orphans/null_reaper.rb +24 -0
  73. data/lib/sidekiq_unique_jobs/orphans/observer.rb +42 -0
  74. data/lib/sidekiq_unique_jobs/orphans/reaper.rb +114 -0
  75. data/lib/sidekiq_unique_jobs/orphans/ruby_reaper.rb +201 -0
  76. data/lib/sidekiq_unique_jobs/redis.rb +11 -0
  77. data/lib/sidekiq_unique_jobs/redis/entity.rb +106 -0
  78. data/lib/sidekiq_unique_jobs/redis/hash.rb +56 -0
  79. data/lib/sidekiq_unique_jobs/redis/list.rb +32 -0
  80. data/lib/sidekiq_unique_jobs/redis/set.rb +32 -0
  81. data/lib/sidekiq_unique_jobs/redis/sorted_set.rb +86 -0
  82. data/lib/sidekiq_unique_jobs/redis/string.rb +49 -0
  83. data/lib/sidekiq_unique_jobs/rspec/matchers.rb +26 -0
  84. data/lib/sidekiq_unique_jobs/rspec/matchers/have_valid_sidekiq_options.rb +51 -0
  85. data/lib/sidekiq_unique_jobs/script.rb +15 -0
  86. data/lib/sidekiq_unique_jobs/script/caller.rb +125 -0
  87. data/lib/sidekiq_unique_jobs/server.rb +48 -0
  88. data/lib/sidekiq_unique_jobs/sidekiq_unique_ext.rb +92 -65
  89. data/lib/sidekiq_unique_jobs/sidekiq_unique_jobs.rb +185 -34
  90. data/lib/sidekiq_unique_jobs/sidekiq_worker_methods.rb +11 -5
  91. data/lib/sidekiq_unique_jobs/testing.rb +62 -21
  92. data/lib/sidekiq_unique_jobs/timer_task.rb +78 -0
  93. data/lib/sidekiq_unique_jobs/timing.rb +58 -0
  94. data/lib/sidekiq_unique_jobs/unlockable.rb +20 -4
  95. data/lib/sidekiq_unique_jobs/update_version.rb +25 -0
  96. data/lib/sidekiq_unique_jobs/upgrade_locks.rb +155 -0
  97. data/lib/sidekiq_unique_jobs/version.rb +3 -1
  98. data/lib/sidekiq_unique_jobs/version_check.rb +23 -4
  99. data/lib/sidekiq_unique_jobs/web.rb +50 -27
  100. data/lib/sidekiq_unique_jobs/web/helpers.rb +125 -10
  101. data/lib/sidekiq_unique_jobs/web/views/changelogs.erb +54 -0
  102. data/lib/sidekiq_unique_jobs/web/views/lock.erb +108 -0
  103. data/lib/sidekiq_unique_jobs/web/views/locks.erb +52 -0
  104. data/lib/tasks/changelog.rake +5 -5
  105. metadata +117 -177
  106. data/lib/sidekiq_unique_jobs/client/middleware.rb +0 -56
  107. data/lib/sidekiq_unique_jobs/scripts.rb +0 -118
  108. data/lib/sidekiq_unique_jobs/server/middleware.rb +0 -46
  109. data/lib/sidekiq_unique_jobs/timeout.rb +0 -8
  110. data/lib/sidekiq_unique_jobs/timeout/calculator.rb +0 -63
  111. data/lib/sidekiq_unique_jobs/unique_args.rb +0 -150
  112. data/lib/sidekiq_unique_jobs/util.rb +0 -103
  113. data/lib/sidekiq_unique_jobs/web/views/unique_digest.erb +0 -28
  114. data/lib/sidekiq_unique_jobs/web/views/unique_digests.erb +0 -46
  115. data/redis/acquire_lock.lua +0 -21
  116. data/redis/convert_legacy_lock.lua +0 -13
  117. data/redis/delete.lua +0 -14
  118. data/redis/delete_by_digest.lua +0 -23
  119. data/redis/delete_job_by_digest.lua +0 -60
  120. data/redis/lock.lua +0 -62
  121. data/redis/release_stale_locks.lua +0 -90
  122. data/redis/unlock.lua +0 -35
@@ -0,0 +1,35 @@
1
+ -------- BEGIN keys ---------
2
+ local digest = KEYS[1]
3
+ local queued = KEYS[2]
4
+ local primed = KEYS[3]
5
+ local locked = KEYS[4]
6
+ local info = KEYS[5]
7
+ local changelog = KEYS[6]
8
+ local digests = KEYS[7]
9
+ -------- END keys ---------
10
+
11
+ -------- BEGIN lock arguments ---------
12
+ local job_id = ARGV[1]
13
+ -------- END lock arguments -----------
14
+
15
+ -------- BEGIN injected arguments --------
16
+ local current_time = tonumber(ARGV[2])
17
+ local debug_lua = ARGV[3] == "true"
18
+ local max_history = tonumber(ARGV[4])
19
+ local script_name = tostring(ARGV[5]) .. ".lua"
20
+ --------- END injected arguments ---------
21
+
22
+ -------- BEGIN local functions --------
23
+ <%= include_partial "shared/_common.lua" %>
24
+ --------- END local functions ---------
25
+
26
+
27
+ -------- BEGIN locked.lua --------
28
+ if redis.call("HEXISTS", locked, job_id) == 1 then
29
+ log_debug("Locked", digest, "job_id:", job_id)
30
+ return 1
31
+ else
32
+ log_debug("NOT Locked", digest, "job_id:", job_id)
33
+ return -1
34
+ end
35
+ --------- END locked.lua ---------
@@ -0,0 +1,87 @@
1
+ -------- BEGIN keys ---------
2
+ local digest = KEYS[1]
3
+ local queued = KEYS[2]
4
+ local primed = KEYS[3]
5
+ local locked = KEYS[4]
6
+ local info = KEYS[5]
7
+ local changelog = KEYS[6]
8
+ local digests = KEYS[7]
9
+ -------- END keys ---------
10
+
11
+
12
+ -------- BEGIN lock arguments ---------
13
+ local job_id = ARGV[1] -- The job_id that was previously primed
14
+ local pttl = tonumber(ARGV[2])
15
+ local lock_type = ARGV[3]
16
+ local limit = tonumber(ARGV[4])
17
+ -------- END lock arguments -----------
18
+
19
+
20
+ -------- BEGIN injected arguments --------
21
+ local current_time = tonumber(ARGV[5])
22
+ local debug_lua = ARGV[6] == "true"
23
+ local max_history = tonumber(ARGV[7])
24
+ local script_name = tostring(ARGV[8]) .. ".lua"
25
+ --------- END injected arguments ---------
26
+
27
+
28
+ -------- BEGIN Variables --------
29
+ local queued_count = redis.call("LLEN", queued)
30
+ local locked_count = redis.call("HLEN", locked)
31
+ local within_limit = limit > locked_count
32
+ local limit_exceeded = not within_limit
33
+ -------- END Variables --------
34
+
35
+
36
+ -------- BEGIN local functions --------
37
+ <%= include_partial "shared/_common.lua" %>
38
+ ---------- END local functions ----------
39
+
40
+
41
+ -------- BEGIN queue.lua --------
42
+ log_debug("BEGIN queue with key:", digest, "for job:", job_id)
43
+
44
+ if redis.call("HEXISTS", locked, job_id) == 1 then
45
+ log_debug("HEXISTS", locked, job_id, "== 1")
46
+ log("Duplicate")
47
+ return job_id
48
+ end
49
+
50
+ local prev_jid = redis.call("GET", digest)
51
+ log_debug("job_id:", job_id, "prev_jid:", prev_jid)
52
+ if not prev_jid or prev_jid == false then
53
+ log_debug("SET", digest, job_id)
54
+ redis.call("SET", digest, job_id)
55
+ elseif prev_jid == job_id then
56
+ log_debug(digest, "already queued with job_id:", job_id)
57
+ log("Duplicate")
58
+ return job_id
59
+ else
60
+ -- TODO: Consider constraining the total count of both locked and queued?
61
+ if within_limit and queued_count < limit then
62
+ log_debug("Within limit:", digest, "(", locked_count, "of", limit, ")", "queued (", queued_count, "of", limit, ")")
63
+ log_debug("SET", digest, job_id, "(was", prev_jid, ")")
64
+ redis.call("SET", digest, job_id)
65
+ else
66
+ log_debug("Limit exceeded:", digest, "(", locked_count, "of", limit, ")")
67
+ log("Limit exceeded", prev_jid)
68
+ return prev_jid
69
+ end
70
+ end
71
+
72
+ log_debug("LPUSH", queued, job_id)
73
+ redis.call("LPUSH", queued, job_id)
74
+
75
+ -- The Sidekiq client should only set pttl for until_expired
76
+ -- The Sidekiq server should set pttl for all other jobs
77
+ if pttl and pttl > 0 then
78
+ log_debug("PEXPIRE", digest, pttl)
79
+ redis.call("PEXPIRE", digest, pttl)
80
+ log_debug("PEXPIRE", queued, pttl)
81
+ redis.call("PEXPIRE", queued, pttl)
82
+ end
83
+
84
+ log("Queued")
85
+ log_debug("END queue with key:", digest, "for job:", job_id)
86
+ return job_id
87
+ -------- END queue.lua --------
@@ -0,0 +1,94 @@
1
+ redis.replicate_commands()
2
+
3
+ -------- BEGIN keys ---------
4
+ local digests_set = KEYS[1]
5
+ local schedule_set = KEYS[2]
6
+ local retry_set = KEYS[3]
7
+ -------- END keys ---------
8
+
9
+ -------- BEGIN argv ---------
10
+ local reaper_count = tonumber(ARGV[1])
11
+ local threshold = tonumber(ARGV[2])
12
+ -------- END argv ---------
13
+
14
+ -------- BEGIN injected arguments --------
15
+ local current_time = tonumber(ARGV[3])
16
+ local debug_lua = ARGV[4] == "true"
17
+ local max_history = tonumber(ARGV[5])
18
+ local script_name = ARGV[6] .. ".lua"
19
+ local redisversion = ARGV[7]
20
+ --------- END injected arguments ---------
21
+
22
+
23
+ -------- BEGIN local functions --------
24
+ <%= include_partial "shared/_common.lua" %>
25
+ <%= include_partial "shared/_find_digest_in_queues.lua" %>
26
+ <%= include_partial "shared/_find_digest_in_sorted_set.lua" %>
27
+ <%= include_partial "shared/_find_digest_in_process_set.lua" %>
28
+ ---------- END local functions ----------
29
+
30
+
31
+ -------- BEGIN delete_orphaned.lua --------
32
+ log_debug("BEGIN")
33
+ local found = false
34
+ local per = 50
35
+ local total = redis.call("ZCARD", digests_set)
36
+ local index = 0
37
+ local del_count = 0
38
+ local redis_ver = toversion(redisversion)
39
+ local del_cmd = "DEL"
40
+
41
+ if tonumber(redis_ver["major"]) >= 4 then del_cmd = "UNLINK"; end
42
+
43
+ repeat
44
+ log_debug("Interating through:", digests_set, "for orphaned locks")
45
+ local digests = redis.call("ZREVRANGE", digests_set, index, index + per -1)
46
+
47
+ for _, digest in pairs(digests) do
48
+ log_debug("Searching for digest:", digest, "in", schedule_set)
49
+ found = find_digest_in_sorted_set(schedule_set, digest)
50
+
51
+ if found ~= true then
52
+ log_debug("Searching for digest:", digest, "in", retry_set)
53
+ found = find_digest_in_sorted_set(retry_set, digest)
54
+ end
55
+
56
+ if found ~= true then
57
+ log_debug("Searching for digest:", digest, "in all queues")
58
+ local queue = find_digest_in_queues(digest)
59
+
60
+ if queue then
61
+ log_debug("found digest:", digest, "in queue:", queue)
62
+ found = true
63
+ end
64
+ end
65
+
66
+ -- TODO: Add check for jobs checked out by process
67
+ if found ~= true then
68
+ log_debug("Searching for digest:", digest, "in process sets")
69
+ found = find_digest_in_process_set(digest, threshold)
70
+ end
71
+
72
+ if found ~= true then
73
+ local queued = digest .. ":QUEUED"
74
+ local primed = digest .. ":PRIMED"
75
+ local locked = digest .. ":LOCKED"
76
+ local info = digest .. ":INFO"
77
+ local run_digest = digest .. ":RUN"
78
+ local run_queued = digest .. ":RUN:QUEUED"
79
+ local run_primed = digest .. ":RUN:PRIMED"
80
+ local run_locked = digest .. ":RUN:LOCKED"
81
+ local run_info = digest .. ":RUN:INFO"
82
+
83
+ redis.call(del_cmd, digest, queued, primed, locked, info, run_digest, run_queued, run_primed, run_locked, run_info)
84
+
85
+ redis.call("ZREM", digests_set, digest)
86
+ del_count = del_count + 1
87
+ end
88
+ end
89
+
90
+ index = index + per
91
+ until index >= total or del_count >= reaper_count
92
+
93
+ log_debug("END")
94
+ return del_count
@@ -0,0 +1,40 @@
1
+ local function toversion(version)
2
+ local _, _, maj, min, pat = string.find(version, "(%d+)%.(%d+)%.(%d+)")
3
+
4
+ return {
5
+ ["version"] = version,
6
+ ["major"] = tonumber(maj),
7
+ ["minor"] = tonumber(min),
8
+ ["patch"] = tonumber(pat)
9
+ }
10
+ end
11
+
12
+ local function toboolean(val)
13
+ val = tostring(val)
14
+ return val == "1" or val == "true"
15
+ end
16
+
17
+ local function log_debug( ... )
18
+ if debug_lua ~= true then return end
19
+
20
+ local result = ""
21
+ for _,v in ipairs(arg) do
22
+ result = result .. " " .. tostring(v)
23
+ end
24
+ redis.log(redis.LOG_DEBUG, script_name .. " -" .. result)
25
+ end
26
+
27
+ local function log(message, prev_jid)
28
+ if not max_history or max_history == 0 then return end
29
+ local entry = cjson.encode({digest = digest, job_id = job_id, script = script_name, message = message, time = current_time, prev_jid = prev_jid })
30
+
31
+ log_debug("ZADD", changelog, current_time, entry);
32
+ redis.call("ZADD", changelog, current_time, entry);
33
+ local total_entries = redis.call("ZCARD", changelog)
34
+ local removed_entries = redis.call("ZREMRANGEBYRANK", changelog, max_history, -1)
35
+ if removed_entries > 0 then
36
+ log_debug("Removing", removed_entries , "entries from changelog (total entries", total_entries, "exceeds max_history:", max_history ..")");
37
+ end
38
+ log_debug("PUBLISH", changelog, entry);
39
+ redis.call("PUBLISH", changelog, entry);
40
+ end
@@ -0,0 +1,8 @@
1
+ local function current_time()
2
+ local time = redis.call("time")
3
+ local s = time[1]
4
+ local ms = time[2]
5
+ local number = tonumber((s .. "." .. ms))
6
+
7
+ return number
8
+ end
@@ -0,0 +1,22 @@
1
+ local function delete_from_queue(queue, digest)
2
+ local per = 50
3
+ local total = redis.call("LLEN", queue)
4
+ local index = 0
5
+ local result = nil
6
+
7
+ while (index < total) do
8
+ local items = redis.call("LRANGE", queue, index, index + per -1)
9
+ if #items == 0 then
10
+ break
11
+ end
12
+ for _, item in pairs(items) do
13
+ if string.find(item, digest) then
14
+ redis.call("LREM", queue, 1, item)
15
+ result = item
16
+ break
17
+ end
18
+ end
19
+ index = index + per
20
+ end
21
+ return result
22
+ end
@@ -0,0 +1,18 @@
1
+ local function delete_from_sorted_set(name, digest)
2
+ local per = 50
3
+ local total = redis.call("zcard", name)
4
+ local index = 0
5
+ local result
6
+ while (index < total) do
7
+ local items = redis.call("ZRANGE", name, index, index + per -1)
8
+ for _, item in pairs(items) do
9
+ if string.find(item, digest) then
10
+ redis.call("ZREM", name, item)
11
+ result = item
12
+ break
13
+ end
14
+ end
15
+ index = index + per
16
+ end
17
+ return result
18
+ end
@@ -0,0 +1,53 @@
1
+ local function find_digest_in_process_set(digest, threshold)
2
+ local process_cursor = 0
3
+ local job_cursor = 0
4
+ local pattern = "*" .. digest .. "*"
5
+ local found = false
6
+
7
+ log_debug("Searching in process list",
8
+ "for digest:", digest,
9
+ "cursor:", process_cursor)
10
+
11
+ repeat
12
+ local process_paginator = redis.call("SSCAN", "processes", process_cursor, "MATCH", "*")
13
+ local next_process_cursor = process_paginator[1]
14
+ local processes = process_paginator[2]
15
+ log_debug("Found number of processes:", #processes, "next cursor:", next_process_cursor)
16
+
17
+ for _, process in ipairs(processes) do
18
+ local workers_key = process .. ":workers"
19
+ log_debug("searching in process set:", process,
20
+ "for digest:", digest,
21
+ "cursor:", process_cursor)
22
+
23
+ local jobs = redis.call("HGETALL", workers_key)
24
+
25
+ if #jobs == 0 then
26
+ log_debug("No entries in:", workers_key)
27
+ else
28
+ for i = 1, #jobs, 2 do
29
+ local jobstr = jobs[i +1]
30
+ if string.find(jobstr, digest) then
31
+ log_debug("Found digest", digest, "in:", workers_key)
32
+ found = true
33
+ break
34
+ end
35
+
36
+ local job = cjson.decode(jobstr)
37
+ if job.payload.created_at > threshold then
38
+ found = true
39
+ break
40
+ end
41
+ end
42
+ end
43
+
44
+ if found == true then
45
+ break
46
+ end
47
+ end
48
+
49
+ process_cursor = next_process_cursor
50
+ until found == true or process_cursor == "0"
51
+
52
+ return found
53
+ end
@@ -0,0 +1,43 @@
1
+ local function find_digest_in_queues(digest)
2
+ local cursor = "0"
3
+ local count = 50
4
+ local result = nil
5
+ local found = false
6
+
7
+ repeat
8
+ log_debug("searching all queues for a matching digest:", digest)
9
+ local pagination = redis.call("SCAN", cursor, "MATCH", "queue:*", "COUNT", count)
10
+ local next_cursor = pagination[1]
11
+ local queues = pagination[2]
12
+
13
+ for _, queue in ipairs(queues) do
14
+ local per = 50
15
+ local total = redis.call("LLEN", queue)
16
+ local index = 0
17
+
18
+ log_debug("searching in:", queue,
19
+ "for digest:", digest,
20
+ "from:", index,
21
+ "to:", total,
22
+ "(per: " .. per .. ")",
23
+ "cursor:", cursor)
24
+
25
+ while (index < total) do
26
+ local items = redis.call("LRANGE", queue, index, index + per -1)
27
+ for _, item in pairs(items) do
28
+ if string.find(item, digest) then
29
+ log_debug("Found digest:", digest, "in queue:", queue)
30
+ result = cjson.decode(item).queue
31
+ found = true
32
+ break
33
+ end
34
+ end
35
+ index = index + per
36
+ end
37
+ end
38
+
39
+ cursor = next_cursor
40
+ until found == true or cursor == "0"
41
+
42
+ return result
43
+ end
@@ -0,0 +1,24 @@
1
+ local function find_digest_in_sorted_set(name, digest)
2
+ local cursor = 0
3
+ local count = 5
4
+ local pattern = "*" .. digest .. "*"
5
+ local found = false
6
+
7
+ log_debug("searching in:", name,
8
+ "for digest:", digest,
9
+ "cursor:", cursor)
10
+ repeat
11
+ local pagination = redis.call("ZSCAN", name, cursor, "MATCH", pattern, "COUNT", count)
12
+ local next_cursor = pagination[1]
13
+ local items = pagination[2]
14
+
15
+ if #items > 0 then
16
+ log_debug("Found digest", digest, "in zset:", name)
17
+ found = true
18
+ end
19
+
20
+ cursor = next_cursor
21
+ until found == true or cursor == "0"
22
+
23
+ return found
24
+ end
@@ -0,0 +1,13 @@
1
+ local hgetall = function (key)
2
+ local bulk = redis.call("HGETALL", key)
3
+ local result = {}
4
+ local nextkey
5
+ for i, v in ipairs(bulk) do
6
+ if i % 2 == 1 then
7
+ nextkey = v
8
+ else
9
+ result[nextkey] = v
10
+ end
11
+ end
12
+ return result
13
+ end