SWAMPy

✎ CARVE FLESH

EDITING: database_builder.rb

# frozen_string_literal: true

#
# Copyright (C) 2018 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.

module Canvas
  module DynamoDB
    module DatabaseBuilder
      class InvalidConfig < StandardError; end

def self.reset
        @clients = {}
      end

def self.configured?(category, environment = default_environment)
        raise ArgumentError, "config name required" if category.blank?

config = configs(environment)[category]
        !!(config && config[:table_prefix] && config[:region])
      end

def self.from_config(category, environment = default_environment, credentials: nil)
        key = [category, environment]
        @clients ||= {}
        @clients.fetch(key) do
          config = configs(environment)[category]
          unless config
            @clients[key] = nil
            return nil
          end
          validate_config(config)
          opts = {
            credentials: credentials || Canvas::AwsCredentialProvider.new("auditors_creds", config["vault_credential_path"]),
            region: config[:region],
          }
          opts[:endpoint] = config[:endpoint] if config[:endpoint]
          fingerprint = "#{category}:#{environment}"
          begin
            @clients[key] = CanvasDynamoDB::Database.new(
              fingerprint,
              prefix: config[:table_prefix],
              client_opts: opts,
              logger: Rails.logger
            )
          rescue => e
            Rails.logger.error "Failed to create DynamoDB client for #{key}: #{e}"
            nil # don't save this nil into @clients[key], so we can retry later
          end
        end
      end

def self.validate_config(config)
        unless config[:table_prefix].present?
          raise InvalidConfig, "No table prefix specified for: #{category.inspect}"
        end
        unless config[:region].present?
          raise InvalidConfig, "No region specified for: #{category.inspect}"
        end
      end

def self.default_environment
        Rails.env
      end

def self.configs(environment = default_environment)
        ConfigFile.load("dynamodb", environment) || {}
      end

def self.categories
        configs.keys
      end
    end
  end
end

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📄 database_builder.rb	RB	2.8 KB	2023-09-24 03:36	EDIT
📄 dev_utils.rb	RB	2.9 KB	2023-09-24 03:36	EDIT