Merge pull request #1242 from basecamp/routing-headers

Populate writer and last transaction in responses

app/controllers/application_controller.rb

@@ -1,5 +1,6 @@
 class ApplicationController < ActionController::Base
-  include Authentication, Authorization, CurrentRequest, CurrentTimezone, SetPlatform, TurboFlash, ViewTransitions, WriterAffinity
+  include Authentication, Authorization, CurrentRequest, CurrentTimezone, LoadBalancerRouting,
+    SetPlatform, TurboFlash, ViewTransitions, WriterAffinity
 
   stale_when_importmap_changes
   allow_browser versions: :modern, block: -> { render "errors/not_acceptable", layout: "error" }

app/controllers/concerns/load_balancer_routing.rb

@@ -0,0 +1,39 @@
+module LoadBalancerRouting
+  extend ActiveSupport::Concern
+
+  ALLOWED_PRAGMAS = %w[ beamer_primary beamer_last_txn ]
+
+  included do
+    before_action :set_target_header, :set_writer_header
+    after_action :set_transaction_cookie
+  end
+
+  private
+    def set_target_header
+        response.headers["X-Kamal-Target"] = request.headers["X-Kamal-Target"]
+    end
+
+    def set_writer_header
+      if ApplicationRecord.current_tenant.present?
+        response.headers["X-Writer"] = pragma("beamer_primary")
+      end
+    end
+
+    def set_transaction_cookie
+      unless safe_request?
+        if ApplicationRecord.current_tenant.present? && Account.sole.present?
+          cookies[:last_transaction] = { value: pragma("beamer_last_txn"), path: Account.sole.slug }
+        end
+      end
+    end
+
+    def pragma(name)
+      if ALLOWED_PRAGMAS.include?(name)
+        ApplicationRecord.connection.execute("pragma #{name}").first&.values&.first
+      end
+    end
+
+    def safe_request?
+      request.get? || request.head?
+    end
+end