首页 > 解决方案 > 如何在 Ruby Sinatra 中读取 GZIP 有效负载

问题描述

在远程主机上,我有一个 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>

我不知道是我发送数据错误还是读取错误。请帮忙。

标签: rubycurlsinatragzipzlib

解决方案


假设 YAML 示例如下:

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

你不需要所有多余的StringIO东西。request.body已经是一个StringIO实例,因此将其转换为字符串然后将其转换为 aStringIO是不必要的:

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_sStringIO对象时返回的内容request.body,因此任何后续调用sio.read(在调用中隐含Zlib::GzipReader.new)都将返回该文字字符串,并且不会返回您期望它返回的压缩数据,这会导致错误Zlib::GzipFile::Error: not in gzip format

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

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

对此:

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

但同样,这是将 a 转换为StringIO字符串并返回到 a的不必要的转换StringIO


推荐阅读