Class: SpooledHttpClient

Inherits:
Object
  • Object
show all
Includes:
Proxy::Log, Singleton
Defined in:
lib/smart_proxy_host_reports/spooled_http_client.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ SpooledHttpClient

Returns a new instance of SpooledHttpClient.



18
19
20
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 18

def initialize
  @worker = nil
end

Instance Method Details

#initialize_directory(spool_dir = Proxy::HostReports::Plugin.settings.spool_dir) ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 22

def initialize_directory(spool_dir = Proxy::HostReports::Plugin.settings.spool_dir)
  raise("Setting spool_dir uninitialized") unless spool_dir
  @spool_dir = spool_dir
  ["temp", "todo", "done", "fail"].each do |state|
    FileUtils.mkdir_p(spool_path(state))
  end
  self
end

#process ⇒ Object 



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
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 54

def process
  processed = 0
  client = ::Proxy::HttpRequest::ForemanRequest.new
  factory = client.request_factory
  # send all files via a single persistent HTTP connection
  logger.debug "Opening HTTP connection to Foreman"
  client.http.start do |http|
    Dir.glob(spool_path("todo", "*")) do |filename|
      basename = File.basename(filename)
      logger.debug "Sending report #{basename}"
      begin
        post = factory.create_post("/api/v2/host_reports", File.read(filename))
        response = http.request(post)
        logger.info "Report #{basename} sent with HTTP response #{response.code}"
        logger.debug { "Response body: #{response.body}" }
        if response.code.start_with?("2")
          if Proxy::HostReports::Plugin.settings.keep_reports
            spool_move("todo", "done", basename)
          else
            FileUtils.rm_f spool_path("todo", basename)
          end
        else
          logger.debug { "Moving failed report to 'fail' spool directory" }
          spool_move("todo", "done", basename)
        end
        processed += 1
      rescue StandardError => e
        logger.warn "Unable to send #{basename}, will try on next request: #{e}", e
        raise if ENV["RACK_ENV"] == "test"
      end
    end
  end
  logger.debug "Finished uploading #{processed} reports, closing connection"
end

#spool(filename, data) ⇒ Object 



93
94
95
96
97
98
99
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 93

def spool(filename, data)
  filename = filename.gsub(/[^0-9a-z]/i, "")
  file = spool_path("temp", filename)
  File.open(file, "w") { |f| f.write(data) }
  spool_move("temp", "todo", filename)
  wakeup
end

#spool_move(from_state, to_state, filename) ⇒ Object 



13
14
15
16
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 13

def spool_move(from_state, to_state, filename)
  # TODO: implement fsync?
  FileUtils.mv(spool_path(from_state, filename), spool_path(to_state, filename))
end

#spool_path(state = "temp", filename = nil) ⇒ Object 



5
6
7
8
9
10
11
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 5

def spool_path(state = "temp", filename = nil)
  if filename
    File.join(@spool_dir, state, filename)
  else
    File.join(@spool_dir, state)
  end
end

#start ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 31

def start
  @worker_running = true
  @worker = Thread.new do
    while @worker_running
      begin
        logger.info "Started host reports spooled http client"
        process
      rescue StandardError => e
        logger.error "Error during spool processing: #{e}", e
      end
      Thread.stop
    end
    logger.info "Stopped host reports spooled http client"
  end
end

#stop ⇒ Object

smart proxy currently does not support stopping services



48
49
50
51
52
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 48

def stop
  @worker_running = false
  wakeup
  @worker.join
end

#wakeup ⇒ Object 



89
90
91
 
# File 'lib/smart_proxy_host_reports/spooled_http_client.rb', line 89

def wakeup
  @worker.wakeup if @worker
end