Commit 2b7d759a authored by Kamil Trzcinski's avatar Kamil Trzcinski

Add with_lease to session [ci skip]

parent e9c14918
......@@ -26,6 +26,8 @@ module Mattermost
include Doorkeeper::Helpers::Controller
include HTTParty
LEASE_TIMEOUT = 60
base_uri Settings.mattermost.host
attr_accessor :current_resource_owner, :token
......@@ -35,6 +37,7 @@ module Mattermost
end
def with_session
with_lease do
raise NoSessionError unless create
begin
......@@ -45,6 +48,7 @@ module Mattermost
destroy
end
end
end
# Next methods are needed for Doorkeeper
def pre_auth
......@@ -130,5 +134,25 @@ module Mattermost
response.headers['token']
end
end
def with_lease
lease_uuid = lease_try_obtain
raise NoSessionError unless lease_uuid
begin
yield
ensure
Gitlab::ExclusiveLease.cancel(lease_key, lease_uuid)
end
end
def lease_key
"mattermost:session"
end
def lease_try_obtain
lease = ::Gitlab::ExclusiveLease.new(lease_key, timeout: LEASE_TIMEOUT)
lease.try_obtain
end
end
end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment