如何使用Rack中间件同时设置cookie并在cookie有效的情况下发送响应

Question

我正在尝试使用Rack Middleware设置cookie，并在相同的请求-响应周期中发送cookie有效的响应。

上下文如下：我正在开发一个有两种模式的网站：美国模式和英国模式（不同的徽标、导航栏、样式等）。当一位英国访问者第一次访问该页面时，我想在他的浏览器上设置一个"英国模式"cookie，但也要呈现该页面的英国版本。这是我到目前为止的代码：

 # middleware/geo_filter_middleware.rb
 def initialize(app)
   @app = app
 end
 def call(env)
   status, headers, body = @app.call(env)
   response = Rack::Response.new(body, status, headers)
   if from_uk?(env)
      response.set_cookie('country', 'UK')
   end
   response.to_a
 end

当英国访问者第一次点击页面时，它会在他们的cookie中设置"英国模式"，但它仍然呈现页面的默认美国版本。只有在第二次请求之后，cookie才会生效，英国游客才会看到英国模式。

有人知道在一个请求-响应周期内同时设置cookie并返回cookie有效的响应吗？

Answer 1

您需要在应用程序中设置中间件.rb

config.middleware.insert_before "ActionDispatch::Cookies", "GeoFilterMiddleware"

在您的中间件中执行以下操作：

  def call(env)
    status, headers, body = @app.call(env)
    if from_uk?(env)
      Rack::Utils.set_cookie_header!(headers, 'country', { :value => 'UK', :path => '/'})
    end
    [status, headers, body]
  end