Class: Unicorn::App::Inetd::CatBody

Inherits:
Struct
  • Object
show all
Defined in:
lib/unicorn/app/inetd.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(env, cmd) ⇒ CatBody

Returns a new instance of CatBody.



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/unicorn/app/inetd.rb', line 12

def initialize(env, cmd)
  self.errors = env['rack.errors']
  in_rd, in_wr = IO.pipe
  self.err_rd, err_wr = IO.pipe
  self.out_rd, out_wr = IO.pipe

  cmd_pid = fork {
    inp, out, err = (0..2).map { |i| IO.new(i) }
    inp.reopen(in_rd)
    out.reopen(out_wr)
    err.reopen(err_wr)
    [ in_rd, in_wr, err_rd, err_wr, out_rd, out_wr ].each { |i| i.close }
    exec(*cmd)
  }
  [ in_rd, err_wr, out_wr ].each { |io| io.close }
  [ in_wr, err_rd, out_rd ].each { |io| io.binmode }
  in_wr.sync = true

  # Unfortunately, input here must be processed inside a seperate
  # thread/process using blocking I/O since env['rack.input'] is not
  # IO.select-able and attempting to make it so would trip Rack::Lint
  inp_pid = fork {
    input = env['rack.input']
    [ err_rd, out_rd ].each { |io| io.close }

    # this is dependent on input.read having readpartial semantics:
    buf = input.read(16384)
    begin
      in_wr.write(buf)
    end while input.read(16384, buf)
  }
  in_wr.close
  self.pid_map = {
    inp_pid => 'input streamer',
    cmd_pid => cmd.inspect,
  }
end

Instance Attribute Details

#err_rd ⇒ Object

Returns the value of attribute err_rd

Returns:

  • (Object) —

    the current value of err_rd



11
12
13
 
# File 'lib/unicorn/app/inetd.rb', line 11

def err_rd
  @err_rd
end

#errors ⇒ Object

Returns the value of attribute errors

Returns:

  • (Object) —

    the current value of errors



11
12
13
 
# File 'lib/unicorn/app/inetd.rb', line 11

def errors
  @errors
end

#out_rd ⇒ Object

Returns the value of attribute out_rd

Returns:

  • (Object) —

    the current value of out_rd



11
12
13
 
# File 'lib/unicorn/app/inetd.rb', line 11

def out_rd
  @out_rd
end

#pid_map ⇒ Object

Returns the value of attribute pid_map

Returns:

  • (Object) —

    the current value of pid_map



11
12
13
 
# File 'lib/unicorn/app/inetd.rb', line 11

def pid_map
  @pid_map
end

Instance Method Details

#close ⇒ Object 



79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/unicorn/app/inetd.rb', line 79

def close
  pid_map.each { |pid, str|
    begin
      pid, status = Process.waitpid2(pid)
      status.success? or
        errors.write("#{str}: #{status.inspect} (PID:#{pid})\n")
    rescue Errno::ECHILD
      errors.write("Failed to reap #{str} (PID:#{pid})\n")
    end
  }
  out_rd.close
  err_rd.close
end

#each(&block) ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/unicorn/app/inetd.rb', line 50

def each(&block)
  begin
    rd, = IO.select([err_rd, out_rd])
    rd && rd.first or next

    if rd.include?(err_rd)
      begin
        errors.write(err_rd.read_nonblock(16384))
      rescue Errno::EINTR
      rescue Errno::EAGAIN
        break
      end while true
    end

    rd.include?(out_rd) or next

    begin
      yield out_rd.read_nonblock(16384)
    rescue Errno::EINTR
    rescue Errno::EAGAIN
      break
    end while true
  rescue EOFError,Errno::EPIPE,Errno::EBADF,Errno::EINVAL
    break
  end while true

  self
end