Contents

require_relative "adapter"
require "aws-sdk-ssm"

module Blinkist
  class Config
    class AwsSsmAdapter < Adapter
      DEFAULT_PREFIX = "/application/".freeze

      def initialize(env, app_name)
        super env, app_name

        @items_cache = {}
        @client = Aws::SSM::Client.new
      end

      def get(key, default=nil, scope: nil)
        prefix = prefix_for(scope)

        query_ssm_parameter prefix + key
      rescue Aws::SSM::Errors::ParameterNotFound
        default
      end

      def preload(scope: nil)
        query_all_ssm_parameters prefix_for(scope)
      end

      private

      def prefix_for(scope)
        if scope.nil?
          DEFAULT_PREFIX + @app_name + "/"
        else
          DEFAULT_PREFIX + scope + "/"
        end
      end

      def query_ssm_parameter(name)
        @items_cache[name] ||= @client.get_parameter(
          name: name,
          with_decryption: true
        ).parameter.value
      end

      def query_all_ssm_parameters(prefix)
        parameters = []
        next_token = nil

        # SSM limits the results and we need to loop
        loop do
          result = @client.get_parameters_by_path(
            path: prefix,
            recursive: true,
            with_decryption: true,
            next_token: next_token
          )

          parameters += result.parameters
          next_token = result.next_token

          break if next_token.nil?
        end

        parameters.map do |parameter|
          @items_cache[parameter.name] = parameter.value
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
blinkist-config-1.3.1	lib/blinkist/config/adapters/aws_ssm_adapter.rb
blinkist-config-1.2.2	lib/blinkist/config/adapters/aws_ssm_adapter.rb