如何在Ruby Sinatra中读取GZIP有效负载

时间:2019-01-15 03:36:25

标签: ruby curl sinatra gzip zlib

在远程主机上,我有一个bash脚本,将一个简单的gzip压缩YAML文件发送到我的Ruby Sinatra端点:

#!/bin/bash

/bin/gzip -c /tmp/test.yaml > /tmp/test.gz

curl -i <hostname>:<port>/last_run_report -H "Content-Type: application/xml" -H "Content-Encoding: gzip" --data-binary @/tmp/test.gz

我的示例Ruby应用程序是:

require 'sinatra'
require 'zlib'
require 'stringio'

set :port, <port>
set :bind, "<ip>"

post '/last_run_report' do
  sio = StringIO.new(request.body.to_s)
  gz = Zlib::GzipReader.new(sio).read
  test_yaml = YAML.load(gz)
end

这给了我以下错误:

Zlib::GzipFile::Error: not in gzip format

如果我需要'base64'并将端点定义更改为:

post '/last_run_report' do
  raw = Base64.decode64(request.body)
  sio = StringIO.new(raw)
  gz = Zlib::GzipReader.new(sio).read
  test_yaml = YAML.load(gz)
end

我收到以下错误:

NoMethodError: undefined method `unpack1' for #<StringIO:0x000055713e2d51b8>

我无法确定是发送错误数据还是读取错误数据。请帮忙。

1 个答案:

答案 0 :(得分:3)

假设使用如下所示的YAML示例:

martin:
    name: Martin D'vloper
    job: Developer
    skill: Elite

您不需要所有多余的StringIO东西。 request.body已经是StringIO实例,因此无需将其转换为字符串然后再将其转换为StringIO

require 'sinatra'
require 'zlib'

post '/last_run_report' do
  gz = Zlib::GzipReader.new(request.body).read
  puts YAML.load(gz)
end

现在提出您的请求:

curl -i localhost:4567/last_run_report -H "Content-Type: application/xml" -H "Content-Encoding: gzip" --data-binary @test.gz

并查看sinatra输出:

== Sinatra (v2.0.4) has taken the stage on 4567 for development with backup from Thin
Thin web server (v1.7.2 codename Bachmanity)
Maximum connections set to 1024
Listening on localhost:4567, CTRL+C to stop
{"martin"=>{"name"=>"Martin D'vloper", "job"=>"Developer", "skill"=>"Elite"}}
::1 - - [14/Jan/2019:23:24:28 -0700] "POST /last_run_report HTTP/1.1" 200 - 0.0048

请注意,puts已将{"martin"=>{"name"=>"Martin D'vloper", "job"=>"Developer", "skill"=>"Elite"}}写入控制台。

我应该指出,在您的示例中,以下代码无法按您期望的方式工作:

sio = StringIO.new(request.body.to_s)

您希望能够致电sio.read并获得类似的信息:

\x1F\x8B\b\b\xA7z=\\\x00\x03test.yaml\x00SVp\xCCSH\xCD-\xC8\xC9\xAFLMU(JM\xCE/J\xE1\xCAM,*\xC9\xCC\xB3\xE2R\x00\x82\xBC\xC4\xDCT+\x05_\xB0\x88\x82\x8BzYN~Aj\x11X&+?\xC9J\xC1%\xB5,\x15!T\x9C\x9D\x99\x93c\xA5\xE0\x9A\x93Y\x92\n\x00\xFC\xA0\x83yZ\x00\x00\x00

您实际上得到的是:

#<StringIO:0x00007ffd8184bdf0>

请注意,这是文字字符串“ #<StringIO:0x00007ffd8184bdf0>”,而不是对StringIO对象的引用,因为这是在.to_s上调用StringIO时返回的内容对象,例如request.body,因此对sio.read的任何后续调用(在对Zlib::GzipReader.new的调用中都是隐式的)将返回该原义字符串,并且不会返回您希望其返回的压缩数据,从而导致错误Zlib::GzipFile::Error: not in gzip format

当您要返回StringIO的字符串表示形式时,应调用.string.read而不是.to_s。考虑到这一点,使您的第一个示例工作所需的最小更改是更改此内容:

sio = StringIO.new(request.body.to_s)

对此:

sio = StringIO.new(request.body.string)

但同样,这是将StringIO转换为字符串然后再转换回StringIO的不必要操作。