Class: Telegem::Session::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/session/middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(store = nil) ⇒ Middleware

Returns a new instance of Middleware.



4
5
6
# File 'lib/session/middleware.rb', line 4

def initialize(store = nil)
  @store = store || MemoryStore.new
end

Instance Method Details

#call(ctx, next_middleware) ⇒ Object



8
9
10
11
12
13
14
15
16
17
18
19
20
21
# File 'lib/session/middleware.rb', line 8

def call(ctx, next_middleware)
  user_id = get_user_id(ctx)
  return next_middleware.call(ctx) unless user_id

  ctx.session = @store.get(user_id) || {}

  begin
    result = next_middleware.call(ctx)
    # Handle async result
    result.is_a?(Async::Task) ? result : Async::Task.new(result)
  ensure
    @store.set(user_id, ctx.session)
  end
end