machinio/solrb

require 'solr/request/node_selection_strategy'
require 'solr/request/default_node_selection_strategy'
require 'solr/request/cloud/first_shard_leader_node_selection_strategy'
require 'solr/request/cloud/leader_node_selection_strategy'
require 'solr/request/master_slave/master_node_selection_strategy'
require 'solr/errors/solr_query_error'
require 'solr/errors/solr_connection_failed_error'
require 'solr/errors/no_active_solr_nodes_error'

module Solr
  module Request
    # TODO: Add documentation about request running
    class Runner
      include Solr::Support::UrlHelper

      attr_reader :request, :response_parser, :node_selection_strategy

      def self.call(opts)
        new(**opts).call
      end

      def initialize(request:,
                     node_selection_strategy: Solr::Request::DefaultNodeSelectionStrategy,
                     solr_connection: Solr::Connection)
        @request = request
        @response_parser = response_parser
        @node_selection_strategy = node_selection_strategy
        @solr_connection = solr_connection
      end

      def call
        solr_url_errors = {}
        solr_urls.each do |node_url|
          request_url = build_request_url(url: node_url,
                                          path: request.path,
                                          url_params: request.url_params)
          begin
            raw_response = @solr_connection.call(url: request_url.to_s, method: request.method, body: request.body)
            solr_response = Solr::Response::Parser.call(raw_response)
            raise Solr::Errors::SolrQueryError, solr_response.error_message unless solr_response.ok?
            return solr_response
          rescue Faraday::ConnectionFailed, Faraday::TimeoutError, Errno::EADDRNOTAVAIL, Timeout::Error => e
            if Solr.master_slave_enabled?
              Solr.configuration.nodes_gray_list.add(node_url)
            end
            solr_url_errors[node_url] = "#{e.class.name} - #{e.message}"
            # Try next node
          end
        end

        raise Solr::Errors::SolrConnectionFailedError.new(solr_url_errors)
      end

      private

      def solr_urls
        @solr_urls ||= begin
          urls = if Solr.node_url_override
            [File.join(Solr.node_url_override, collection_name.to_s)]
          elsif Solr.cloud_enabled? || Solr.master_slave_enabled?
            node_selection_strategy.call(collection_name)
          else
            [Solr.current_core_config.url]
          end
          unless urls && urls.any?
            raise Solr::Errors::NoActiveSolrNodesError
          end
          urls
        end
      end

      def collection_name
        Solr.current_core_config.name
      end
    end
  end
end