github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2024-11-29 16:35:43 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

DEV: fix hanging spec (#29974)

Browse Source
This commit is contained in:
Sam 2024-11-28 14:06:19 +11:00 committed by GitHub
parent 2d945e2373
commit 07813ba83c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/scheduler/defer.rb
View File

@ -76,6 +76,7 @@ module Scheduler
def stop!(finish_work: false)
if finish_work
@finish = true
@queue.push({ finish: true }, force: true)
@thread&.join
end
@thread.kill if @thread&.alive?
@ -108,7 +109,9 @@ module Scheduler
# using non_block to match Ruby #deq
def do_work(non_block = false)
db, job, desc = @queue.shift(block: !non_block).values_at(:db, :job, :desc)
db, job, desc, finish = @queue.shift(block: !non_block).values_at(:db, :job, :desc, :finish)
return if finish
start = Process.clock_gettime(Process::CLOCK_MONOTONIC)
db ||= RailsMultisite::ConnectionManagement::DEFAULT

6
spec/lib/scheduler/defer_spec.rb
View File

@ -22,6 +22,12 @@ RSpec.describe Scheduler::Defer do
Discourse.reset_catch_job_exceptions!
end
it "can finish work properly without crashing" do
@defer.later {}
sleep 0.005
@defer.stop!(finish_work: true)
end
it "supports basic instrumentation" do
@defer.later("first") {}
@defer.later("first") {}