-
-
Notifications
You must be signed in to change notification settings - Fork 277
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add performance test for replace conflict strategy
- Loading branch information
Showing
4 changed files
with
91 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,55 @@ | ||
# frozen_string_literal: true | ||
|
||
RSpec.describe UniqueJobOnConflictReplace, :perf do | ||
let(:lock_prefix) { described_class.sidekiq_options.fetch("lock_prefix") { SidekiqUniqueJobs.config.lock_prefix } } | ||
let(:lock_timeout) { described_class.sidekiq_options.fetch("lock_timeout") { SidekiqUniqueJobs.config.lock_timeout } } | ||
let(:lock_ttl) { described_class.sidekiq_options.fetch("lock_ttl") { SidekiqUniqueJobs.config.lock_ttl } } | ||
let(:queue) { described_class.sidekiq_options["queue"] } | ||
let(:on_conflict) { described_class.sidekiq_options["on_conflict"] } | ||
let(:lock) { described_class.sidekiq_options["lock"] } | ||
|
||
before do | ||
flushdb | ||
end | ||
|
||
context "when schedule queue is large" do | ||
it "locks and replaces quickly" do | ||
(0..100_000).each_slice(1_000) do |nums| | ||
redis do |conn| | ||
conn.pipelined do |pipeline| | ||
nums.each do |num| | ||
created_at = Time.now.to_f | ||
scheduled_at = created_at + rand(3_600..2_592_000) | ||
|
||
payload = { | ||
"retry" => true, | ||
"queue" => queue, | ||
"lock" => lock, | ||
"on_conflict" => on_conflict, | ||
"class" => described_class.name, | ||
"args" => [num, { "type" => "extremely unique" }], | ||
"jid" => SecureRandom.hex(12), | ||
"created_at" => created_at, | ||
"lock_timeout" => lock_timeout, | ||
"lock_ttl" => lock_ttl, | ||
"lock_prefix" => lock_prefix, | ||
"lock_args" => [num, { "type" => "extremely unique" }], | ||
"lock_digest" => "#{lock_prefix}:#{SecureRandom.hex}", | ||
} | ||
|
||
pipeline.zadd("schedule", scheduled_at, payload.to_json) | ||
end | ||
end | ||
end | ||
end | ||
|
||
# queueing it once at the end of the queue should succeed | ||
expect(described_class.perform_in(2_592_000, 100_000, { "type" => "extremely unique" })).not_to be_nil | ||
|
||
# queueing it again should be performant | ||
expect do | ||
described_class.perform_in(2_592_000, 100_000, { "type" => "extremely unique" }) | ||
end.to perform_under(10).ms | ||
end | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
# frozen_string_literal: true | ||
|
||
# :nocov: | ||
|
||
class UniqueJobOnConflictReplace | ||
include Sidekiq::Worker | ||
sidekiq_options lock: :until_executing, | ||
queue: :customqueue, | ||
on_conflict: :replace | ||
|
||
def perform(one, two) | ||
[one, two] | ||
end | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,18 @@ | ||
# frozen_string_literal: true | ||
|
||
RSpec.describe UniqueJobOnConflictReplace do | ||
it_behaves_like "sidekiq with options" do | ||
let(:options) do | ||
{ | ||
"lock" => :until_executing, | ||
"on_conflict" => :replace, | ||
"queue" => :customqueue, | ||
"retry" => true, | ||
} | ||
end | ||
end | ||
|
||
it_behaves_like "a performing worker" do | ||
let(:args) { ["hundred", { "type" => "extremely unique", "id" => 44 }] } | ||
end | ||
end |