Class: Yarn::Server

Inherits:
Object
  • Object
show all
Includes:
Logging
Defined in:
lib/yarn/server.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Logging

#debug, #log, #output, #timestamp

Constructor Details

#initialize(options = {}) ⇒ Server

Returns a new instance of Server.



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/yarn/server.rb', line 10

def initialize(options={})
  # merge given options with default values
  opts = { 
    output: $stdout, 
    host: '127.0.0.1', 
    port: 3000,
    workers: 4,
    log: true,
    rack: "off" 
  }.merge(options)

  @app = nil
  @app = load_rack_app(opts[:rack]) unless opts[:rack] == "off"
  @opts = opts

  @host, @port, @num_workers = opts[:host], opts[:port], opts[:workers]
  @workers = []
  $output, $debug = opts[:output], opts[:debug]
  $log = opts[:log] || opts[:debug]
end

Instance Attribute Details

#host ⇒ Object

Returns the value of attribute host.



8
9
10
 
# File 'lib/yarn/server.rb', line 8

def host
  @host
end

#port ⇒ Object

Returns the value of attribute port.



8
9
10
 
# File 'lib/yarn/server.rb', line 8

def port
  @port
end

#socket ⇒ Object

Returns the value of attribute socket.



8
9
10
 
# File 'lib/yarn/server.rb', line 8

def socket
  @socket
end

#workers ⇒ Object

Returns the value of attribute workers.



8
9
10
 
# File 'lib/yarn/server.rb', line 8

def workers
  @workers
end

Instance Method Details

#fork_worker ⇒ Object 



58
59
60
 
# File 'lib/yarn/server.rb', line 58

def fork_worker
  fork { worker }
end

#init_workers ⇒ Object 



52
53
54
55
56
 
# File 'lib/yarn/server.rb', line 52

def init_workers
  @num_workers.times do
    @workers << fork_worker
  end
end

#load_rack_app(app_path) ⇒ Object 



31
32
33
34
35
36
37
38
39
 
# File 'lib/yarn/server.rb', line 31

def load_rack_app(app_path)
  if File.exists?(app_path)
    config_file = File.read(app_path)
    rack_application = eval("Rack::Builder.new { #{config_file} }")
  else
    log "#{app_path} does not exist. Exiting."
    Kernel::exit
  end
end

#start ⇒ Object 



41
42
43
44
45
46
47
48
49
50
 
# File 'lib/yarn/server.rb', line 41

def start
  trap("INT") { stop }
  @socket = TCPServer.new(@host, @port)
  log "Yarn started #{@num_workers} workers and is listening on #{@host}:#{@port}"

  init_workers

  # Waits here for all processes to exit
  Process.waitall
end

#stop ⇒ Object 



71
72
73
74
75
 
# File 'lib/yarn/server.rb', line 71

def stop
  @socket.close if (@socket && !@socket.closed?)

  log "Server stopped. Have a nice day!"
end

#worker ⇒ Object 



62
63
64
65
66
67
68
69
 
# File 'lib/yarn/server.rb', line 62

def worker
  trap("INT") { exit }
  loop do
    @handler ||= @app ? RackHandler.new(@app,@opts) : RequestHandler.new
    session = @socket.accept
    @handler.run session 
  end
end