Contents

# frozen_string_literal: true
module Hyrax
  module CustomQueries
    ##
    # @see https://github.com/samvera/valkyrie/wiki/Queries#custom-queries
    class FindCountBy
      def self.queries
        [:find_count_by]
      end

      def initialize(query_service:)
        @query_service = query_service
      end

      attr_reader :query_service
      delegate :resource_factory, to: :query_service
      delegate :orm_class, to: :resource_factory

      ##
      # @note this is an unoptimized default implementation of this custom
      #   query. it's Hyrax's policy to provide such implementations of custom
      #   queries in use for cross-compatibility of Valkyrie query services.
      #   it's advisable to provide an optimized query for the specific adapter.
      #
      # @param hash [Hash] the hash representation of the query
      def find_count_by(hash = {}, models: nil)
        return nil if models.empty? && hash.blank?

        internal_array = ["{ #{hash.map { |k, v| "\"#{k}\": #{v}" }.join(', ')} }"] if hash.present?
        if models.empty?
          query_service.orm_class.count_by_sql(([find_count_by_properties_query] + internal_array))
        elsif hash.blank?
          query_service.orm_class.count_by_sql([find_count_by_models_query] + [models])
        else
          query_service.orm_class.count_by_sql(([find_count_by_properties_and_models_query] + internal_array + [models]))
        end
      end

      def find_count_by_properties_and_models_query
        <<-SQL
          SELECT count(*) FROM orm_resources
          WHERE metadata @> ?
          AND internal_resource IN (?);
        SQL
      end

      def find_count_by_models_query
        <<-SQL
          SELECT count(*) FROM orm_resources
          WHERE internal_resource IN (?);
        SQL
      end

      def find_count_by_properties_query
        <<-SQL
          SELECT count(*) FROM orm_resources
          WHERE metadata @> ?;
        SQL
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
hyrax-5.0.1	app/services/hyrax/custom_queries/find_count_by.rb
hyrax-5.0.0	app/services/hyrax/custom_queries/find_count_by.rb
hyrax-5.0.0.rc3	app/services/hyrax/custom_queries/find_count_by.rb
hyrax-5.0.0.rc2	app/services/hyrax/custom_queries/find_count_by.rb
hyrax-5.0.0.rc1	app/services/hyrax/custom_queries/find_count_by.rb