1
2
3
4
5
6
7
8
9
10
11
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
49
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
| | # -*- encoding: binary -*-
require 'kgio'
require 'unicorn'
# the value passed to TCP_DEFER_ACCEPT actually matters in Linux 2.6.32+
Unicorn::SocketHelper::DEFAULTS[:tcp_defer_accept] = 60
# See https://yhbt.net/rainbows/ for documentation
module Rainbows
# :stopdoc:
O = {}
# map of numeric file descriptors to IO objects to avoid using IO.new
# and potentially causing race conditions when using /dev/fd/
FD_MAP = {}.compare_by_identity
require 'rainbows/const'
require 'rainbows/http_parser'
require 'rainbows/http_server'
autoload :Response, 'rainbows/response'
autoload :ProcessClient, 'rainbows/process_client'
autoload :Client, 'rainbows/client'
autoload :Base, 'rainbows/base'
autoload :Sendfile, 'rainbows/sendfile'
autoload :AppPool, 'rainbows/app_pool'
autoload :DevFdResponse, 'rainbows/dev_fd_response'
autoload :MaxBody, 'rainbows/max_body'
autoload :QueuePool, 'rainbows/queue_pool'
autoload :EvCore, 'rainbows/ev_core'
autoload :SocketProxy, 'rainbows/socket_proxy'
# :startdoc:
# Sleeps the current application dispatch. This will pick the
# optimal method to sleep depending on the concurrency model chosen
# (which may still suck and block the entire process). Using this
# with the basic :Coolio or :EventMachine models is not recommended.
# This should be used within your Rack application.
def self.sleep(seconds)
case Rainbows.server.use
when :FiberPool, :FiberSpawn
Rainbows::Fiber.sleep(seconds)
when :RevFiberSpawn, :CoolioFiberSpawn
Rainbows::Fiber::Coolio::Sleeper.new(seconds)
when :Revactor
Actor.sleep(seconds)
else
Kernel.sleep(seconds)
end
end
# :stopdoc:
class << self
attr_accessor :server
attr_accessor :cur # may not always be used
attr_reader :alive
attr_writer :worker
attr_writer :forked
attr_writer :readers
end
def self.config!(mod, *opts)
@forked or abort "#{mod} should only be loaded in a worker process"
opts.each do |opt|
mod.const_set(opt.to_s.upcase, Rainbows.server.__send__(opt))
end
end
@alive = true
@cur = 0
@expire = nil
@at_quit = []
def self.at_quit(&block)
@at_quit << block
end
def self.tick
@worker.tick = now.to_i
exit!(2) if @expire && now >= @expire
@alive && @server.master_pid == Process.ppid or quit!
end
def self.cur_alive
@alive || @cur > 0
end
def self.quit!
unless @expire
@alive = false
Rainbows::HttpParser.quit
@expire = now + (@server.timeout * 2.0)
tmp = @readers.dup
@readers.clear
tmp.each { |s| s.close rescue nil }.clear
@at_quit.each(&:call)
# XXX hack to break out of IO.select in worker_loop for some models
Process.kill(:QUIT, $$)
end
false
end
# try to use the monotonic clock in Ruby >= 2.1, it is immune to clock
# offset adjustments and generates less garbage (Float vs Time object)
begin
Process.clock_gettime(Process::CLOCK_MONOTONIC)
def self.now
Process.clock_gettime(Process::CLOCK_MONOTONIC)
end
rescue NameError, NoMethodError
def self.now # Ruby <= 2.0
Time.now.to_f
end
end
autoload :Base, "rainbows/base"
autoload :WriterThreadPool, "rainbows/writer_thread_pool"
autoload :WriterThreadSpawn, "rainbows/writer_thread_spawn"
autoload :Revactor, "rainbows/revactor"
autoload :ThreadSpawn, "rainbows/thread_spawn"
autoload :ThreadPool, "rainbows/thread_pool"
autoload :Rev, "rainbows/rev"
autoload :RevThreadSpawn, "rainbows/rev_thread_spawn"
autoload :RevThreadPool, "rainbows/rev_thread_pool"
autoload :RevFiberSpawn, "rainbows/rev_fiber_spawn"
autoload :Coolio, "rainbows/coolio"
autoload :CoolioThreadSpawn, "rainbows/coolio_thread_spawn"
autoload :CoolioThreadPool, "rainbows/coolio_thread_pool"
autoload :CoolioFiberSpawn, "rainbows/coolio_fiber_spawn"
autoload :Epoll, "rainbows/epoll"
autoload :XEpoll, "rainbows/xepoll"
autoload :EventMachine, "rainbows/event_machine"
autoload :FiberSpawn, "rainbows/fiber_spawn"
autoload :FiberPool, "rainbows/fiber_pool"
autoload :ActorSpawn, "rainbows/actor_spawn"
autoload :NeverBlock, "rainbows/never_block"
autoload :XEpollThreadSpawn, "rainbows/xepoll_thread_spawn"
autoload :XEpollThreadPool, "rainbows/xepoll_thread_pool"
autoload :StreamResponseEpoll, "rainbows/stream_response_epoll"
autoload :Fiber, 'rainbows/fiber' # core class
autoload :StreamFile, 'rainbows/stream_file'
autoload :ThreadTimeout, 'rainbows/thread_timeout'
autoload :WorkerYield, 'rainbows/worker_yield'
autoload :SyncClose, 'rainbows/sync_close'
autoload :ReverseProxy, 'rainbows/reverse_proxy'
autoload :JoinThreads, 'rainbows/join_threads'
autoload :PoolSize, 'rainbows/pool_size'
end
require 'rainbows/error'
require 'rainbows/configurator'
module Unicorn
# this interferes with Rainbows::Client creation with unicorn 5.3
begin
remove_const :TCPSrv
TCPSrv = Kgio::TCPServer
rescue NameError # unicorn < 5.3.0
end
end
|