Class: BackgroundQueue::ServerLib::TaskRegistry

Inherits:
Object
  • Object
show all
Defined in:
lib/background_queue/server_lib/task_registry.rb

Overview

keep track if tasks already queued and running so if the same task comes in, we know to remove it.

Instance Method Summary collapse

Constructor Details

#initialize ⇒ TaskRegistry

Returns a new instance of TaskRegistry.



6
7
8
9
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 6

def initialize
  @waiting_tasks = {}
  @tasks = {}
end

Instance Method Details

#de_register(task_id) ⇒ Object 



25
26
27
28
29
30
31
32
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 25

def de_register(task_id)
  @tasks.delete(task_id)
  waiting = get_waiting_task(task_id)
  if waiting
    @tasks[task_id] = waiting
  end
  waiting
end

#get_waiting_task(task_id) ⇒ Object 



38
39
40
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 38

def get_waiting_task(task_id)
  @waiting_tasks.delete(task_id)
end

#register(task) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 11

def register(task)
  existing_task = @tasks[task.id]
  if existing_task.nil? 
    @tasks[task.id] = task
    [:new, nil]
  elsif existing_task.running?
    register_waiting_task(task)
    [:waiting, nil]
  else
    @tasks[task.id] = task
    [:existing, existing_task]
  end
end

#register_waiting_task(task) ⇒ Object 



34
35
36
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 34

def register_waiting_task(task)
  @waiting_tasks[task.id] = task
end

#tasks ⇒ Object 



46
47
48
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 46

def tasks
  @tasks
end

#waiting_tasks ⇒ Object 



42
43
44
 
# File 'lib/background_queue/server_lib/task_registry.rb', line 42

def waiting_tasks
  @waiting_tasks
end