Class: Spring::Server

Inherits:
Object
  • Object
show all
Defined in:
lib/spring/server.rb

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ Server


19
20
21
22
23
24
25
# File 'lib/spring/server.rb', line 19

def initialize(options = {})
  @foreground   = options.fetch(:foreground, false)
  @env          = options[:env] || default_env
  @applications = Hash.new { |h, k| h[k] = ApplicationManager.new(k, env) }
  @pidfile      = env.pidfile_path.open('a')
  @mutex        = Mutex.new
end

Instance Attribute Details

#envObject (readonly)

Returns the value of attribute env


17
18
19
# File 'lib/spring/server.rb', line 17

def env
  @env
end

Class Method Details

.boot(options = {}) ⇒ Object


13
14
15
# File 'lib/spring/server.rb', line 13

def self.boot(options = {})
  new(options).boot
end

Instance Method Details

#bootObject


35
36
37
38
39
40
41
42
43
44
# File 'lib/spring/server.rb', line 35

def boot
  Spring.verify_environment

  write_pidfile
  set_pgid unless foreground?
  ignore_signals unless foreground?
  set_exit_hook
  set_process_title
  start_server
end

#foreground?Boolean


27
28
29
# File 'lib/spring/server.rb', line 27

def foreground?
  @foreground
end

#ignore_signalsObject

Ignore SIGINT and SIGQUIT otherwise the user typing ^C or ^\ on the command line will kill the server/application.


89
90
91
# File 'lib/spring/server.rb', line 89

def ignore_signals
  IGNORE_SIGNALS.each { |sig| trap(sig, "IGNORE") }
end

#log(message) ⇒ Object


31
32
33
# File 'lib/spring/server.rb', line 31

def log(message)
  env.log "[server] #{message}"
end

#rails_env_for(args, default_rails_env) ⇒ Object


76
77
78
# File 'lib/spring/server.rb', line 76

def rails_env_for(args, default_rails_env)
  Spring.command(args.first).env(args.drop(1)) || default_rails_env
end

#redirect_outputObject

We need to redirect STDOUT and STDERR, otherwise the server will keep the original FDs open which would break piping. (e.g. `spring rake -T | grep db` would hang forever because the server would keep the stdout FD open.)


126
127
128
# File 'lib/spring/server.rb', line 126

def redirect_output
  [STDOUT, STDERR].each { |stream| stream.reopen(env.log_file) }
end

#serve(client) ⇒ Object


53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# File 'lib/spring/server.rb', line 53

def serve(client)
  log "accepted client"
  client.puts env.version

  app_client = client.recv_io
  command    = JSON.load(client.read(client.gets.to_i))

  args, default_rails_env = command.values_at('args', 'default_rails_env')

  if Spring.command?(args.first)
    log "running command #{args.first}"
    client.puts
    client.puts @applications[rails_env_for(args, default_rails_env)].run(app_client)
  else
    log "command not found #{args.first}"
    client.close
  end
rescue SocketError => e
  raise e unless client.eof?
ensure
  redirect_output
end

#set_exit_hookObject


93
94
95
96
97
98
# File 'lib/spring/server.rb', line 93

def set_exit_hook
  server_pid = Process.pid

  # We don't want this hook to run in any forks of the current process
  at_exit { shutdown if Process.pid == server_pid }
end

#set_pgidObject

Boot the server into the process group of the current session. This will cause it to be automatically killed once the session ends (i.e. when the user closes their terminal).


83
84
85
# File 'lib/spring/server.rb', line 83

def set_pgid
  Process.setpgid(0, SID.pgid)
end

#set_process_titleObject


130
131
132
133
134
# File 'lib/spring/server.rb', line 130

def set_process_title
  ProcessTitleUpdater.run { |distance|
    "spring server | #{env.app_name} | started #{distance} ago"
  }
end

#shutdownObject


100
101
102
103
104
105
106
107
108
109
110
# File 'lib/spring/server.rb', line 100

def shutdown
  log "shutting down"

  [env.socket_path, env.pidfile_path].each do |path|
    if path.exist?
      path.unlink rescue nil
    end
  end

  @applications.values.map { |a| Spring.failsafe_thread { a.stop } }.map(&:join)
end

#start_serverObject


46
47
48
49
50
51
# File 'lib/spring/server.rb', line 46

def start_server
  server = UNIXServer.open(env.socket_name)
  log "started on #{env.socket_name}"
  loop { serve server.accept }
rescue Interrupt
end

#write_pidfileObject


112
113
114
115
116
117
118
119
120
# File 'lib/spring/server.rb', line 112

def write_pidfile
  if @pidfile.flock(File::LOCK_EX | File::LOCK_NB)
    @pidfile.truncate(0)
    @pidfile.write("#{Process.pid}\n")
    @pidfile.fsync
  else
    exit 1
  end
end