feat: config redis expired time

paladinsoftware · Oct 28, 2022 · bd1239f · bd1239f
1 parent ee92b75
commit bd1239f
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/lib/rack-cas/session_store/redis.rb b/lib/rack-cas/session_store/redis.rb
@@ -13,14 +13,14 @@ def self.find_by_id(session_id)
         session ? {'sid' => session_id, 'data' => session} : session
       end
 
-      def self.write(session_id:, data:, cas_ticket: )
+      def self.write(session_id:, data:, cas_ticket:, expires_in: nil)
         #create a row with the session_id and the data
         #create a row with the cas_ticket acting as a reverse index
         results = self.client.pipelined do
           self.client.set("rack_cas_session:#{session_id}",data)
-          self.client.expireat("rack_cas_session:#{session_id}",30.days.from_now.to_i)
+          self.client.expireat("rack_cas_session:#{session_id}",expires_in.present? ? expires_in.from_now.to_i : 30.days.from_now.to_i)
           self.client.set("rack_cas_ticket:#{cas_ticket}","rack_cas_session:#{session_id}")
-          self.client.expireat("rack_cas_ticket:#{cas_ticket}",30.days.from_now.to_i)
+          self.client.expireat("rack_cas_ticket:#{cas_ticket}",expires_in.present? ? expires_in.from_now.to_i : 30.days.from_now.to_i)
         end
 
         results == ["OK",true,"OK",true] ? session_id : false
@@ -72,8 +72,7 @@ def find_session(env, sid)
     # Rack 2.0 method
     def write_session(env, sid, session_data, options)
       cas_ticket = (session_data['cas']['ticket'] unless session_data['cas'].nil?)
-
-      success = Session.write(session_id: sid, data: pack(session_data), cas_ticket: cas_ticket)
+      success = Session.write(session_id: sid, data: pack(session_data), cas_ticket: cas_ticket, expires_in: options[:expire_after])
 
       success ? sid : false
     end