Sha256: 2cea8bbd84c8d219885af8598e5ce2f418a0a9b169006952f677ee2db8b9b386

Contents?: true

Size: 1.57 KB

Versions: 131

Compression:

Stored size: 1.57 KB

Contents

# frozen_string_literal: true

require 'thread'

module Aws

  # Module/mixin used by token provider classes that can be refreshed. This
  # provides basic refresh logic in a thread-safe manner. Classes mixing in
  # this module are expected to implement a #refresh method that populates
  # the following instance variable:
  #
  # * `@token` [Token] - {Aws::Token} object with the `expiration` and `token`
  #       fields set.
  #
  # @api private
  module RefreshingToken

    def initialize(options = {})
      @mutex = Mutex.new
      @before_refresh = options.delete(:before_refresh) if Hash === options

      @before_refresh.call(self) if @before_refresh
      refresh
    end

    # @return [Token]
    def token
      refresh_if_near_expiration
      @token
    end

    # @return [Time,nil]
    def expiration
      refresh_if_near_expiration
      @expiration
    end

    # Refresh token.
    # @return [void]
    def refresh!
      @mutex.synchronize do
        @before_refresh.call(self) if @before_refresh
        refresh
      end
    end

    private

    # Refreshes token if it is within
    # 5 minutes of expiration.
    def refresh_if_near_expiration
      if near_expiration?
        @mutex.synchronize do
          if near_expiration?
            @before_refresh.call(self) if @before_refresh
            refresh
          end
        end
      end
    end

    def near_expiration?
      if @token && @token.expiration
        # are we within 5 minutes of expiration?
        (Time.now.to_i + 5 * 60) > @token.expiration.to_i
      else
        true
      end
    end
  end
end

Version data entries

131 entries across 131 versions & 1 rubygems

Version Path
aws-sdk-core-3.220.2 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.220.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.220.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.219.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.218.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.218.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.217.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.217.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.216.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.216.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.215.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.215.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.214.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.214.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.213.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.212.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.211.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.210.0 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.209.1 lib/aws-sdk-core/refreshing_token.rb
aws-sdk-core-3.209.0 lib/aws-sdk-core/refreshing_token.rb