如何使用Rails通过Web套接字发送二进制文件

时间:2015-10-29 22:33:01

标签: ruby-on-rails ruby sockets websocket faye

我有一个Rails应用程序,用户上传音频文件。我想将它们发送到第三方服务器,我需要使用Web套接字连接到外部服务器,因此,我需要我的Rails应用程序作为websocket客户端。

我试图找出如何正确设置它。我还没有承诺任何宝石,但'faye-websocket'宝石看起来很有希望。我甚至在" Sending large file in websocket before timeout"中找到了类似的答案,但是,使用该代码并不适合我。

以下是我的代码示例:

@message = Array.new
EM.run {
    ws = Faye::WebSocket::Client.new("wss://example_url.com")

    ws.on :open do |event|
      File.open('path/to/audio_file.wav','rb') do |f|
        ws.send(f.gets)
      end
    end

    ws.on :message do |event|
      @message << [event.data]
    end

    ws.on :close do |event|
      ws = nil
      EM.stop
    end
}

当我使用它时,收到来自收件人服务器的错误:

No JSON object could be decoded

这是有道理的,因为我不相信它为faye-websocket正确格式化。他们的documentation说:

  

send(message)接受String或字节大小的数组   整数并通过连接发送文本或二进制消息   其他同行;二进制数据必须编码为数组。

我不确定如何做到这一点。如何使用Ruby将二进制文件加载到整数数组中?

我尝试修改send命令以使用bytes方法:

File.open('path/to/audio_file.wav','rb') do |f|
    ws.send(f.gets.bytes)
end

但现在我收到了这个错误:

Stream was 19 bytes but needs to be at least 100 bytes

我知道我的文件是286KB,所以这里有问题。我对何时使用File.read vs File.openFile.new感到困惑。

此外,也许这个宝石不是发送二进制数据的最佳选择。有没有人成功通过websockets在Rails中发送二进制文件?

更新:我确实找到了一种方法让它工作,但内存却很糟糕。对于想要加载小文件的其他人,您只需File.binreadunpack方法:

ws.on :open do |event|
  f = File.binread 'path/to/audio_file.wav'
  ws.send(f.unpack('C*'))
end

但是,如果我在100MB文件上使用相同的代码,则服务器内存不足。它在我的测试服务器上耗尽了整个可用的1.5GB!有谁知道如何做到这一点是一种记忆安全的方式?

1 个答案:

答案 0 :(得分:1)

这是我的看法:

# do only once when initializing Rails:
require 'iodine/client'
Iodine.force_start!

# this sets the callbacks.
# on_message is always required by Iodine.
options = {}
options[:on_message] = Proc.new do |data|
   # this will never get called
   puts "incoming data ignored? for:\n#{data}"
end
options[:on_open] = Proc.new do
   # believe it or not - this variable belongs to the websocket connection.
   @started_upload = true
   # set a task to send the file,
   # so the on_open initialization doesn't block incoming messages.
   Iodine.run do
      # read the file and write to the websocket.
      File.open('filename','r') do |f|
         buffer = String.new # recycle the String's allocated memory
         write f.read(65_536, buffer) until f.eof?
         @started_upload = :done
      end
      # close the connection
      close
   end
end
options[:on_close] = Proc.new do |data|
   # can we notify the user that the file was uploaded?
   if @started_upload == :done
        # we did it :-)
   else
        # what happened?
   end
end

# will not wait for a connection:
Iodine::Http.ws_connect "wss://example_url.com", options
# OR
# will wait for a connection, raising errors if failed.
Iodine::Http::WebsocketClient.connect "wss://example_url.com", options

我唯一可以提及的是我Iodine's作者,我在Plezi(一个RESTful Websocket实时应用程序框架)中使用,可以单独使用,也可以单独使用在Rails中)...我超级有偏见; - )

我会避免使用gets,因为它的大小可能包含整个文件或单个字节,具体取决于下一行行结束(EOL)标记的位置... {{1} }让你更好地控制每个块的大小。