如何在Rails 7中重写DeviseController的set_flash_message方法？

要全局修改Devise的set_flash_message行为，无需逐个修改控制器，最佳实践是通过扩展Devise核心模块实现，具体步骤如下：

1. 创建初始化器文件 #

在config/initializers目录下新建devise_flash_override.rb，这个文件会在Rails启动时自动加载，实现全局覆盖。

2. 编写重写代码 #

以下代码会保留Devise原有消息逻辑（如翻译、flash.now支持），同时适配自定义的列表式Flash消息，并新增终端打印功能：

# config/initializers/devise_flash_override.rb
module Devise
  module Controllers
    module Helpers
      def set_flash_message(key, kind, options = {})
        # 保留Devise原有的消息查找/翻译逻辑
        message = find_message(kind, options)
        return unless message.present?

        # 新增终端打印Flash消息的逻辑
        puts "[Devise Flash] #{key}: #{message}"

        # 处理flash.now参数，和原Devise行为对齐
        flash_now = options.delete(:now)
        target_flash = flash_now ? flash.now : flash

        # 适配自定义的列表式Flash逻辑：转为数组或追加元素
        if target_flash[key].is_a?(Array)
          target_flash[key] << message
        else
          target_flash[key] = [message]
        end
      end
    end
  end
end

3. 生效方式 #

修改完成后重启Rails服务器，初始化器会覆盖Devise原生的set_flash_message方法，所有控制器调用Devise的Flash消息逻辑时，都会自动使用重写后的版本。

关键说明 #

• 完全保留了Devise原有的消息翻译、flash.now参数支持，不会破坏原有功能
• 强制将Flash消息以数组形式存储，完美适配自定义的批量展示逻辑
• 终端打印逻辑可根据需求调整（如改用Rails.logger.debug替代puts）

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