ruby - 每当 Amazon SQS 队列更新时运行 Ruby 任务

Question

我最近一直在做很多 Sinatra，我认为是时候将它提升到一个新的水平了。伙计们，我们将不胜感激任何帮助。所以这就是我想做的事情:

只要用户发出事件，应用程序 A 就会向 Amazon SQS 队列抛出一个事件。每当有任何事件被抛入队列时，应用程序 B 都会执行一项任务(比如通过电子邮件向用户发送“嘿”消息)。除了以下代码之外，我从未真正使用过 Cron Jobs:

$sum = 1

Thread.new do # trivial example work thread
  while true do
     sleep 3
     $sum += 1
  end
end


get '/' do
    "Testing background work thread: sum is #{$sum}"
end

感谢马克沃森 (http://markwatson.com/blog/2011-11/ruby-sinatra-web-apps-with-background-work-threads.html)

这是我的疑惑:

• “在 Amazon 队列更新上执行任务”是否可行？还是我想错了？
• 如何更新上面的代码，使其每 3 秒自动刷新一次，而不会出现“无限循环”错误？

Answer 1

我建议您将应用程序分成两部分。一个是常规的 Sinatra 网络应用程序，另一个是某种后台 worker 。

示例应用

这是您可以使用的简单的非生产工作示例。

创建一些常用的 boot.rb 文件，将“初始化”代码放入其中:

require 'bundler/setup'

# Use this to load my '.env' that contains AWS credentials
require 'dotenv'; Dotenv.load

# We are using Amazon Ruby SDK
require 'aws-sdk'

# We configure AWS Ruby SDK
AWS.config(
  access_key_id: ENV.fetch('AWS_ACCESS_KEY_ID'),
  secret_access_key: ENV.fetch('AWS_SECRET_ACCESS_KEY')
)

# We get SQS Web Service
sqs = AWS::SQS.new

QUEUE = sqs.queues[ENV.fetch('AWS_QUEUE_URL')]

简单的 Sinatra web app.rb，如果您访问/smoke，它会向您的 Amazon SQS 发送消息。

require_relative 'boot'
require 'sinatra'

get '/smoke' do
  message = QUEUE.send_message({email: "my@email.com", at: DateTime.now}.to_json.to_s)
  "Message was sent to Amazon SQS with id #{message.id}.\n"
end

创建另一个文件 - 将其命名为 worker.rb - 它将订阅您的 Amazon SQS 队列并拉取消息。

require_relative 'boot'

QUEUE.poll do |msg|
  object = JSON.load(msg.body) rescue {}
  puts "Got your email #{object['email']} at #{object['at']}."
end

我的本​​地 .env 文件包含凭证和 SQA 队列的 URL。

AWS_ACCESS_KEY_ID=secret
AWS_SECRET_ACCESS_KEY=secret
AWS_QUEUE_URL=https://sqs.us-west-1.amazonaws.com/...

运行它

然后您只需启动您的 Sinatra app 和 worker 像这样

ruby app.rb # in one terminal
ruby worker.rb # in another terminal

如果您随后访问 /smoke，您将看到类似这样的内容 - 意味着该作业已安排。

Message was sent to Amazon SQS with id 788e5e28-8055-4c8f-bb51-c634c327a021.

在你的工作终端你会看到

Got your email my@email.com at 2014-08-29T15:20:38+02:00.

现在，您不必将自己限制在一个 worker 上......您可以拥有任意多个，默认情况下，一条消息将仅在一个 worker 上处理。这里的另一个注意事项是我们将 JSON 作为消息发送，这就是我在 worker 中使用 JSON.load 的原因。 Amazon SQS 发送/接收纯文本消息。