1 files changed, 6 insertions, 3 deletions
diff --git a/app/controllers/concerns/token_generation_concern.rb b/app/controllers/concerns/token_generation_concern.rb
index 7f43e4c..beab467 100644
--- a/app/controllers/concerns/token_generation_concern.rb
+++ b/app/controllers/concerns/token_generation_concern.rb
@@ -9,12 +9,15 @@ module TokenGenerationConcern
     @service ||= Services::TokenGenerationService.new(service_params)
   end
 
-  def generate_user_session_key
-    @user_account.session_key = SecureRandom.hex(16)
-    @user_account.save
+  def generate_user_session_key(user)
+    user.update_attribute(:session_key, SecureRandom.hex(16))
   end
 
   def generate_token
     { token: service.call(DateTime.current + 30.minutes), refresh: service.call(DateTime.current + 3.days) }
   end
+
+  def service_params
+    { email: @user_account.email, role: @user_account.role, session_key: @user_account.session_key }
+  end
 end