如何在多租户场景下结合Milia配置ActionCable租户

我之前在项目里集成Milia和ActionCable做租户隔离的时候，也踩过和你一模一样的坑——照着Apartment的方案改完完全没用，还报那个Could not execute command from...的错误。折腾了大半天终于摸清楚了适配方法，给你整理一下：

1. 重写ActionCable连接类，注入租户上下文 #

ActionCable的连接是独立于请求线程的，所以必须在连接建立时主动设置Milia的租户上下文。修改app/channels/application_cable/connection.rb：

module ApplicationCable
  class Connection < ActionCable::Connection::Base
    identified_by :current_user, :current_tenant

    def connect
      self.current_user = find_verified_user
      set_current_tenant
    end

    private

    # 保留你原有的用户验证逻辑，比如从warden或token获取用户
    def find_verified_user
      if current_user = env['warden'].user
        current_user
      else
        reject_unauthorized_connection
      end
    end

    # 关键：从请求参数/ cookie获取租户标识，切换到对应租户
    def set_current_tenant
      # 这里可以根据你的业务调整获取租户的方式，比如子域名、cookie或query参数
      tenant_identifier = request.params[:tenant_id] || cookies[:tenant_id]
      
      if tenant = Milia::Tenant.find_by(identifier: tenant_identifier)
        self.current_tenant = tenant
        Milia::Tenant.set_current_tenant(tenant)
      else
        reject_unauthorized_connection
      end
    end
  end
end

2. 在频道类中持久化租户上下文 #

因为ActionCable会复用线程池，频道处理消息时可能丢失租户上下文，所以要在关键方法里重新设置：

class ChatChannel < ApplicationCable::Channel
  def subscribed
    # 订阅前强制切换到当前连接的租户
    Milia::Tenant.set_current_tenant(current_tenant)
    stream_from "chat_#{current_tenant.id}_channel"
  end

  def receive(data)
    # 处理消息时再次确认租户上下文
    Milia::Tenant.set_current_tenant(current_tenant)
    Message.create!(content: data['content'], user: current_user)
  end
end

3. 前端连接时传递租户标识 #

前端建立WebSocket连接时，必须把租户ID/标识传递给后端，比如通过URL参数：

// app/javascript/channels/consumer.js
import { createConsumer } from "@rails/actioncable"

// 从页面meta标签获取租户ID
const tenantId = document.querySelector('meta[name="tenant-id"]').content;
// 把租户ID加入连接URL
export default createConsumer(`/cable?tenant_id=${tenantId}`);

记得在布局文件里添加对应的meta标签：

# app/views/layouts/application.html.erb
<meta name="tenant-id" content="<%= current_tenant.id %>">

4. 后台广播作业的租户适配 #

如果用后台作业广播消息，也要确保作业执行时切换到对应租户：

class MessageBroadcastJob < ApplicationJob
  queue_as :default

  def perform(message)
    # 基于消息关联的用户切换租户
    Milia::Tenant.set_current_tenant(message.user.tenant)
    ActionCable.server.broadcast "chat_#{message.user.tenant.id}_channel", message: render_message(message)
  end

  private

  def render_message(message)
    ApplicationController.renderer.render(partial: 'messages/message', locals: { message: message })
  end
end

Milia和Apartment的租户隔离机制完全不同：Apartment是通过切换数据库连接或schema实现隔离，而Milia是依赖全局的current_tenant变量，配合模型层的tenant_id关联来隔离数据。所以Apartment那种修改数据库连接的方案对Milia完全无效，我们需要的是在ActionCable的整个生命周期里主动维护Milia的租户上下文。

如果还是出现Could not execute command from...错误，可以从这几点检查：

• 打开浏览器控制台，查看WebSocket连接的URL是否包含tenant_id参数
• 在连接类的set_current_tenant方法里加日志，确认是否正确找到租户
• 检查频道方法里是否遗漏了租户上下文的设置，避免线程复用导致的上下文丢失

内容的提问来源于stack exchange，提问作者Jonas Johnson