Class: Dumper::Agent
Constant Summary
collapse
- API_VERSION =
1
Instance Attribute Summary collapse
Class Method Summary
collapse
Instance Method Summary
collapse
#log, #log_last_error, #logger, #stdout_logger
Constructor Details
#initialize(options = {}) ⇒ Agent
Returns a new instance of Agent.
22
23
24
25
26
27
28
29
30
31
|
# File 'lib/dumper/agent.rb', line 22
def initialize(options = {})
log 'app_key is missing' if options[:app_key].blank?
@stack = Dumper::Stack.new
@api_base = options[:api_base] || 'http://dumper.io'
@app_key = options[:app_key]
@app_env = @stack.rails_env
@app_name = ObjectSpace.each_object(Rails::Application).first.class.name.split("::").first
logger.level = stdout_logger.level = options[:loglevel] if options[:loglevel]
end
|
Instance Attribute Details
#stack ⇒ Object
Returns the value of attribute stack.
10
11
12
|
# File 'lib/dumper/agent.rb', line 10
def stack
@stack
end
|
Class Method Details
.start(options = {}) ⇒ Object
13
14
15
|
# File 'lib/dumper/agent.rb', line 13
def start(options = {})
new(options).start
end
|
.start_if(options = {}) ⇒ Object
17
18
19
|
# File 'lib/dumper/agent.rb', line 17
def start_if(options = {})
start(options) if yield
end
|
Instance Method Details
#api_request(method_name, options) ⇒ Object
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
|
# File 'lib/dumper/agent.rb', line 92
def api_request(method_name, options)
uri = URI.parse("#{@api_base}/api/#{method_name}")
http = Net::HTTP.new(uri.host, uri.port)
if uri.is_a? URI::HTTPS
http.use_ssl = true
http.verify_mode = OpenSSL::SSL::VERIFY_NONE
end
request = Net::HTTP::Post.new(uri.request_uri)
request['x-app-key'] = @app_key
request['x-app-env'] = @app_env
request['x-api-version'] = API_VERSION.to_s
request['user-agent'] = "Dumper-RailsAgent/#{Dumper::VERSION} (ruby #{::RUBY_VERSION} #{::RUBY_PLATFORM} / rails #{Rails::VERSION::STRING})"
if options[:params]
request.set_form_data(options[:params])
else
request.body = options[:json] || ''
request['Content-Type'] = 'application/octet-stream'
end
response = http.request(request)
if response.code == '200'
log response.body, :debug
MultiJson.load(response.body).with_indifferent_access
else
log "******** ERROR on api: #{method_name}, resp code: #{response.code} ********", :error
{} end
rescue
log_last_error
{} end
|
#register_hash ⇒ Object
83
84
85
86
87
88
89
90
|
# File 'lib/dumper/agent.rb', line 83
def register_hash
{
:hostname => Socket.gethostname,
:agent_version => Dumper::VERSION,
:app_name => @app_name,
:stack => @stack.to_hash,
}
end
|
#start ⇒ Object
33
34
35
36
37
38
39
|
# File 'lib/dumper/agent.rb', line 33
def start
return unless @stack.supported?
log "stack: dispatcher = #{@stack.dispatcher}, framework = #{@stack.framework}, rackup = #{@stack.rackup}"
@loop_thread = Thread.new { start_loop }
@loop_thread[:name] = 'Loop Thread'
end
|
#start_loop ⇒ Object
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
|
# File 'lib/dumper/agent.rb', line 41
def start_loop
sec = 1
begin
sec *= 2
log "sleeping #{sec} seconds for agent/register", :debug
sleep sec
json = api_request('agent/register', :json => MultiJson.dump(register_hash))
end until json[:status]
return log("agent stopped: #{json.to_s}") if json[:status] == 'error'
@token = json[:token]
log "agent started as #{@token ? 'primary' : 'secondary'}"
sleep 1.hour + rand(10) unless @token
loop do
json = api_request('agent/poll', :params => { :token => @token })
if json[:status] == 'ok'
if json[:token]
@token = json[:token] else
@token = nil end
if json[:job]
if pid = fork
srand Process.detach(pid)
else
Dumper::Job.new(self, json[:job]).run_and_exit
end
end
end
sleep [ json[:interval].to_i, 60 ].max
end
end
|