module BlacklightOaiProvider class SolrDocumentWrapper < ::OAI::Provider::Model attr_reader :document_model, :timestamp_field, :solr_timestamp, :limit def initialize(controller, options = {}) @controller = controller @document_model = options[:model] || ::SolrDocument @solr_timestamp = document_model.timestamp_key @timestamp_field = 'timestamp' # method name used by ruby-oai @limit = options[:limit] || 15 end def sets; end def earliest _response, records = @controller.get_search_results(@controller.params, { fl: solr_timestamp, sort: "#{solr_timestamp} asc", rows: 1 }) records.first.timestamp end def latest _response, records = @controller.get_search_results(@controller.params, { fl: solr_timestamp, sort: "#{solr_timestamp} desc", rows: 1 }) records.first.timestamp end def find(selector, options = {}) return next_set(options[:resumption_token]) if options[:resumption_token] if selector == :all response, records = @controller.get_search_results(@controller.params, conditions(options)) if limit && response.total > limit return select_partial(BlacklightOaiProvider::ResumptionToken.new(options.merge(last: 0), nil, response.total)) end else _response, records = @controller.get_solr_response_for_doc_id selector.split('/', 2).last end records end def select_partial(token) _response, records = @controller.get_search_results(@controller.params, token_conditions(token)) raise ::OAI::ResumptionTokenException unless records OAI::Provider::PartialResult.new(records, token.next(token.last + limit)) end def next_set(token_string) raise ::OAI::ResumptionTokenException unless limit token = BlacklightOaiProvider::ResumptionToken.parse(token_string) select_partial(token) end private def base_conditions { sort: "#{solr_timestamp} asc", rows: limit } end def token_conditions(token) conditions(token.to_conditions_hash).merge(start: token.last) end def conditions(options) # conditions/query derived from options if !(options[:from].blank? && options[:until].blank?) base_conditions.merge( fq: "#{solr_timestamp}:[#{solr_date(options[:from])} TO #{solr_date(options[:until]).gsub('Z', '.999Z')}]" ) else base_conditions end end def solr_date(time) if time.respond_to?(:xmlschema) time.utc.xmlschema # Force UTC. elsif time.blank? '*' else time.to_s end end end end