Class: Specjour::Manager

Inherits:

Object

• Object
• Specjour::Manager

show all

Includes:

DRbUndumped, Fork, SocketHelper

Defined in:

lib/specjour/manager.rb

Instance Attribute Summary (collapse)

• - (Object) dispatcher_uri

  Returns the value of attribute dispatcher_uri.

• - (Object) loader_pid readonly

  Returns the value of attribute loader_pid.

• - (Object) options readonly

  Returns the value of attribute options.

• - (Object) pid

  Returns the value of attribute pid.

• - (Object) project_name

  Returns the value of attribute project_name.

• - (Object) registered_projects readonly

  Returns the value of attribute registered_projects.

• - (Object) rsync_port readonly

  Returns the value of attribute rsync_port.

• - (Object) test_paths

  Returns the value of attribute test_paths.

• - (Object) worker_size readonly

  Returns the value of attribute worker_size.

• - (Object) worker_task

  Returns the value of attribute worker_task.

Class Method Summary (collapse)

• + (Object) start_quietly(options)

Instance Method Summary (collapse)

• - (Boolean) available_for?(project_name)
• - (Object) dispatch
• - (Object) dispatch_loader
• - (Object) drb_start
• - (Object) drb_uri
• - (Object) in_project(&block)
• - (Manager) initialize(options = {}) constructor

  A new instance of Manager.

• - (Object) interrupted=(bool)
• - (Object) kill_loader_process
• - (Object) project_path
• - (Boolean) quiet?
• - (Object) start
• - (Object) sync

Methods included from Fork

#fork, #fork_quietly

Methods included from SocketHelper

#current_uri, #hostname, #ip_from_hostname, #new_uri

19
20
21
22
23
24
25

# File 'lib/specjour/manager.rb', line 19

def initialize(options = {})
  @options = options
  @worker_size = options[:worker_size]
  @worker_task = options[:worker_task]
  @registered_projects = options[:registered_projects]
  @rsync_port = options[:rsync_port]
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def dispatcher_uri
  @dispatcher_uri
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def loader_pid
  @loader_pid
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def options
  @options
end

9
10
11

# File 'lib/specjour/manager.rb', line 9

def pid
  @pid || Process.pid
end

9
10
11

# File 'lib/specjour/manager.rb', line 9

def project_name
  @project_name
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def registered_projects
  @registered_projects
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def rsync_port
  @rsync_port
end

9
10
11

# File 'lib/specjour/manager.rb', line 9

def test_paths
  @test_paths
end

10
11
12

# File 'lib/specjour/manager.rb', line 10

def worker_size
  @worker_size
end

9
10
11

# File 'lib/specjour/manager.rb', line 9

def worker_task
  @worker_task
end

12
13
14
15
16
17

# File 'lib/specjour/manager.rb', line 12

def self.start_quietly(options)
  manager = new options.merge(:quiet => true)
  manager.drb_uri
  manager.pid = Fork.fork_quietly { manager.start }
  manager
end

27
28
29

# File 'lib/specjour/manager.rb', line 27

def available_for?(project_name)
  registered_projects ? registered_projects.include?(project_name) : false
end

36
37
38
39
40
41
42
43
44

# File 'lib/specjour/manager.rb', line 36

def dispatch
  suspend_bonjour do
    sync
    with_clean_env do
      execute_before_fork
      dispatch_loader
    end
  end
end

59
60
61
62
63
64
65
66
67
68
69
70
71
72

# File 'lib/specjour/manager.rb', line 59

def dispatch_loader
  @loader_pid = fork do
    exec_cmd = "load --printer-uri #{dispatcher_uri} --workers #{worker_size} --task #{worker_task} --project-path #{project_path}"
    exec_cmd << " --test-paths #{test_paths.join(" ")}" if test_paths.any?
    exec_cmd << " --log" if Specjour.log?
    exec_cmd << " --quiet" if quiet?
    load_path = $LOAD_PATH.detect {|l| l =~ %r(specjour[^/]*/lib$)}
    bin_path = File.expand_path(File.join(load_path, "../bin"))
    Kernel.exec({"RUBYLIB" => load_path}, "#{bin_path}/specjour #{exec_cmd}")
  end
  Process.waitall
ensure
  kill_loader_process if loader_pid
end

46
47
48
49
50

# File 'lib/specjour/manager.rb', line 46

def drb_start
  $PROGRAM_NAME = "specjour listen" if quiet?
  DRb.start_service drb_uri.to_s, self
  at_exit { DRb.stop_service }
end

52
53
54
55
56
57

# File 'lib/specjour/manager.rb', line 52

def drb_uri
  @drb_uri ||= begin
    current_uri.scheme = "druby"
    current_uri
  end
end

74
75
76

# File 'lib/specjour/manager.rb', line 74

def in_project(&block)
  Dir.chdir(project_path, &block)
end

78
79
80
81

# File 'lib/specjour/manager.rb', line 78

def interrupted=(bool)
  Specjour.interrupted = bool
  kill_loader_process if loader_pid
end

83
84
85
86
87
88
89
90

# File 'lib/specjour/manager.rb', line 83

def kill_loader_process
  if Specjour.interrupted?
    Process.kill('INT', loader_pid) rescue Errno::ESRCH
  else
    Process.kill('TERM', loader_pid) rescue Errno::ESRCH
  end
  @loader_pid = nil
end

96
97
98

# File 'lib/specjour/manager.rb', line 96

def project_path
  File.join("/tmp", project_name)
end

107
108
109

# File 'lib/specjour/manager.rb', line 107

def quiet?
  options.has_key? :quiet
end

100
101
102
103
104
105

# File 'lib/specjour/manager.rb', line 100

def start
  drb_start
  bonjour_announce
  at_exit { stop_bonjour }
  DRb.thread.join
end

111
112
113
114

# File 'lib/specjour/manager.rb', line 111

def sync
  cmd "rsync -aL --delete --ignore-errors --port=#{rsync_port} #{dispatcher_uri.host}::#{project_name} #{project_path}"
  puts "rsync complete"
end