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 ...@@ -26,6 +26,8 @@ module Mattermost
include Doorkeeper::Helpers::Controller include Doorkeeper::Helpers::Controller
include HTTParty include HTTParty
LEASE_TIMEOUT = 60
base_uri Settings.mattermost.host base_uri Settings.mattermost.host
attr_accessor :current_resource_owner, :token attr_accessor :current_resource_owner, :token
...@@ -35,14 +37,16 @@ module Mattermost ...@@ -35,14 +37,16 @@ module Mattermost
end end
def with_session def with_session
raise NoSessionError unless create with_lease do
raise NoSessionError unless create
begin
yield self begin
rescue Errno::ECONNREFUSED yield self
raise NoSessionError rescue Errno::ECONNREFUSED
ensure raise NoSessionError
destroy ensure
destroy
end
end end
end end
...@@ -130,5 +134,25 @@ module Mattermost ...@@ -130,5 +134,25 @@ module Mattermost
response.headers['token'] response.headers['token']
end end
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
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