想要自定義設計 gem 的 current_user 方法

Question

我想使用current_user方法有兩個目的。

我有用戶 gmail 登錄和網站的個人注冊。

但是設計 gem 的current_user方法不適用於 gmail 用戶。

對於我使用以下方法的 gmail 用戶

 helper_method :current_user

 def current_user
   @current_user ||= User.find(session[:user_id]) if session[:user_id]
 end

請幫我解決這個問題。 謝謝

Answer 1

我通過自己的信件得到了解決方案。

下面的代碼是上述問題的解決方案

alias_method :devise_current_user, :current_user
  def current_user
    if session[:user_id].blank?
      devise_current_user
    else
      User.find(session[:user_id])
    end   
  end

這對我來說很好用