Class: Rfd::Preview::Server

Inherits:
Object
  • Object
show all
Defined in:
lib/rfd/preview/server.rb

Overview

Forked process that handles preview generation asynchronously

Constant Summary collapse

TIMEOUT = 
(ENV['RFD_PREVIEW_TIMEOUT'] || 10).to_i
WORKERS = 
(ENV['RFD_PREVIEW_WORKERS'] || 2).to_i

Instance Method Summary collapse

Constructor Details

#initialize(socket_path) ⇒ Server

Returns a new instance of Server.



18
19
20
21
22
23
24
 
# File 'lib/rfd/preview/server.rb', line 18

def initialize(socket_path)
  @socket_path = socket_path
  @cancelled = {}
  @cancelled_mutex = Mutex.new
  @queue = Queue.new
  @running = true
end

Instance Method Details

#run ⇒ Object 



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
 
# File 'lib/rfd/preview/server.rb', line 26

def run
  cleanup_socket
  server = UNIXServer.new(@socket_path)
  @client_socket = nil
  @client_mutex = Mutex.new

  # Start worker threads
  workers = WORKERS.times.map { Thread.new { worker_loop } }

  # Main loop: accept connection and read requests
  while @running
    begin
      readable, = IO.select([server], nil, nil, 0.5)
      next unless readable && readable.include?(server)

      client = server.accept
      @client_mutex.synchronize { @client_socket = client }

      Thread.new(client) do |c|
        handle_client(c)
      end
    rescue IOError, Errno::EBADF
      break
    end
  end
rescue => e
  STDERR.puts "Preview::Server error: #{e.message}"
ensure
  @running = false
  WORKERS.times { @queue << :shutdown }
  workers.each(&:join) if workers
  server.close if server
  cleanup_socket
end