SWAMPy

✎ CARVE FLESH

EDITING: database_builder.rb

# frozen_string_literal: true

#
# Copyright (C) 2021 - 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/>.

require "config_file"

module CanvasCassandra
  module DatabaseBuilder
    def self.configured?(config_name, environment = ::Rails.env)
      raise ArgumentError, "config name required" if config_name.blank?

config = ConfigFile.load("cassandra", environment)
      config &&= config[config_name]
      config && config["servers"] && config["keyspace"]
    end

# If for a migration or a support week one has reason to override the settings that would come from cassandra.yml,
    # you can pass them in overrride_options. Note, the keys from the yml will be strings so it'll be easier to hash
    # rocket it.
    #
    # Example: from_config("auditors", override_options: { 'timeout' => 360 })
    def self.from_config(config_name, environment = :current, override_options: nil)
      @connections ||= {}
      environment = Rails.env if environment == :current
      key = [config_name, environment]
      @connections.fetch(key) do
        config = ConfigFile.load("cassandra", environment).dup
        config &&= config[config_name]
        unless config
          @connections[key] = nil
          return nil
        end
        config = config.merge(override_options) if override_options
        servers = Array(config["servers"])
        raise "No Cassandra servers defined for: #{config_name.inspect}" unless servers.present?

keyspace = config["keyspace"]
        raise "No keyspace specified for: #{config_name.inspect}" unless keyspace.present?

opts = { keyspace:, cql_version: "3.0.0" }
        opts[:retries] = config["retries"] if config["retries"]
        opts[:connect_timeout] = config["connect_timeout"] if config["connect_timeout"]
        opts[:timeout] = config["timeout"] if config["timeout"]
        fingerprint = "#{config_name}:#{environment}"
        Bundler.require "cassandra"
        begin
          @connections[key] = CanvasCassandra::Database.new(fingerprint, servers, opts, logger)
        rescue => e
          logger.error "Failed to create cassandra connection for #{key}: #{e}"
          nil # don't save this nil into @connections[key], so we can retry later
        end
      end
    end

def self.logger
      CanvasCassandra.logger
    end

def self.configs
      ConfigFile.load("cassandra") || {}
    end

def self.reset_connections!
      @connections = {}
    end

def self.config_names
      configs.keys
    end

def self.read_consistency_setting(database_name = nil)
      setting_key = "event_stream.read_consistency"
      setting_value = settings_store.get("#{setting_key}.#{database_name}", nil) || settings_store.get(setting_key, nil)

setting_value.presence
    end

def self.settings_store
      CanvasCassandra.settings_store
    end
  end
end

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📄 database.rb	RB	9.6 KB	2023-09-24 03:36	EDIT
📄 database_builder.rb	RB	3.4 KB	2023-09-24 03:36	EDIT