Class: ContentServer::FileStreamer
- Inherits:
-
Object
- Object
- ContentServer::FileStreamer
- Defined in:
- lib/content_server/file_streamer.rb
Instance Attribute Summary collapse
-
#thread ⇒ Object
readonly
Returns the value of attribute thread.
Instance Method Summary collapse
- #abort_streaming(checksum) ⇒ Object
- #copy_another_chuck(checksum) ⇒ Object
- #handle(message) ⇒ Object
-
#initialize(send_chunk_clb, abort_streaming_clb = nil) ⇒ FileStreamer
constructor
A new instance of FileStreamer.
- #reset_stream(checksum, path, offset) ⇒ Object
- #reset_streaming(checksum, new_offset) ⇒ Object
- #run ⇒ Object
- #start_streaming(checksum, path) ⇒ Object
Constructor Details
#initialize(send_chunk_clb, abort_streaming_clb = nil) ⇒ FileStreamer
Returns a new instance of FileStreamer.
49 50 51 52 53 54 55 56 57 |
# File 'lib/content_server/file_streamer.rb', line 49 def initialize(send_chunk_clb, abort_streaming_clb=nil) @send_chunk_clb = send_chunk_clb @abort_streaming_clb = abort_streaming_clb @stream_queue = Queue.new # Used from internal thread only. @streams = {} @thread = run end |
Instance Attribute Details
#thread ⇒ Object (readonly)
Returns the value of attribute thread.
42 43 44 |
# File 'lib/content_server/file_streamer.rb', line 42 def thread @thread end |
Instance Method Details
#abort_streaming(checksum) ⇒ Object
69 70 71 72 |
# File 'lib/content_server/file_streamer.rb', line 69 def abort_streaming(checksum) @stream_queue << [:ABORT_STREAM, checksum] $process_vars.set('File Streamer queue', @stream_queue.size) end |
#copy_another_chuck(checksum) ⇒ Object
59 60 61 62 |
# File 'lib/content_server/file_streamer.rb', line 59 def copy_another_chuck(checksum) @stream_queue << [:COPY_CHUNK, checksum] $process_vars.set('File Streamer queue', @stream_queue.size) end |
#handle(message) ⇒ Object
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 |
# File 'lib/content_server/file_streamer.rb', line 89 def handle() type, content = if type == :NEW_STREAM checksum, path = content reset_stream(checksum, path, 0) @stream_queue << [:COPY_CHUNK, checksum] if @streams.key?(checksum) $process_vars.set('File Streamer queue', @stream_queue.size) elsif type == :ABORT_STREAM checksum = content Stream.close_delete_stream(checksum, @streams) elsif type == :RESET_STREAM checksum, new_offset = content reset_stream(checksum, nil, new_offset) @stream_queue << [:COPY_CHUNK, checksum] if @streams.key?(checksum) $process_vars.set('File Streamer queue', @stream_queue.size) elsif type == :COPY_CHUNK checksum = content if @streams.key?(checksum) offset = @streams[checksum].file.pos Log.debug1("Sending chunk for %s, offset %s.",checksum, offset) chunk = @streams[checksum].file.read(Params['streaming_chunk_size']) if chunk.nil? # No more to read, send end of file. @send_chunk_clb.call(checksum, offset, @streams[checksum].size, nil, nil) Stream.close_delete_stream(checksum, @streams) else chunk_checksum = FileIndexing::IndexAgent.get_content_checksum(chunk) @send_chunk_clb.call(checksum, offset, @streams[checksum].size, chunk, chunk_checksum) end else Log.debug1("No stream found for copy chunk checksum %s.", checksum) end end end |
#reset_stream(checksum, path, offset) ⇒ Object
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 |
# File 'lib/content_server/file_streamer.rb', line 125 def reset_stream(checksum, path, offset) if !@streams.key? checksum begin file = File.new(path, 'rb') if offset > 0 file.seek(offset) end Log.debug1("File streamer: %s.", file) @streams[checksum] = Stream.new(checksum, path, file, file.size) $process_vars.set('Streams size', @streams.size) rescue IOError, Errno::ENOENT => e Log.warning("Could not stream local file #{path}. #{e.to_s}") end else @streams[checksum].file.seek(offset) end end |
#reset_streaming(checksum, new_offset) ⇒ Object
74 75 76 77 |
# File 'lib/content_server/file_streamer.rb', line 74 def reset_streaming(checksum, new_offset) @stream_queue << [:RESET_STREAM, [checksum, new_offset]] $process_vars.set('File Streamer queue', @stream_queue.size) end |
#run ⇒ Object
79 80 81 82 83 84 85 86 87 |
# File 'lib/content_server/file_streamer.rb', line 79 def run return Thread.new do loop { stream_pop = @stream_queue.pop $process_vars.set('File Streamer queue', @stream_queue.size) checksum = handle(stream_pop) } end end |
#start_streaming(checksum, path) ⇒ Object
64 65 66 67 |
# File 'lib/content_server/file_streamer.rb', line 64 def start_streaming(checksum, path) @stream_queue << [:NEW_STREAM, [checksum, path]] $process_vars.set('File Streamer queue', @stream_queue.size) end |