Class: Ddr::Index::CSVQueryResult

Inherits:
AbstractQueryResult show all
Defined in:
lib/ddr/index/csv_query_result.rb

Constant Summary collapse

MAX_ROWS =

Just set to a really high number :)

10**8
CSV_MV_SEPARATOR = 
";"

Instance Attribute Summary

Attributes inherited from AbstractQueryResult

#query

Instance Method Summary collapse

Methods inherited from AbstractQueryResult

#count, #initialize

Constructor Details

This class inherits a constructor from Ddr::Index::AbstractQueryResult

Instance Method Details

#[](index_or_header) ⇒ Object 



19
20
21
 
# File 'lib/ddr/index/csv_query_result.rb', line 19

def [](index_or_header)
  table.by_col_or_row![index_or_header]
end

#csv_headers ⇒ Object 



46
47
48
 
# File 'lib/ddr/index/csv_query_result.rb', line 46

def csv_headers
  query.fields.empty? ? :first_row : query_field_headings
end

#csv_opts ⇒ Object 



27
28
29
30
31
32
 
# File 'lib/ddr/index/csv_query_result.rb', line 27

def csv_opts
  { headers:        csv_headers,
    return_headers: false,
    write_headers:  true,
  }
end

#data ⇒ Object 



62
63
64
65
 
# File 'lib/ddr/index/csv_query_result.rb', line 62

def data
  raw = Connection.get("select", params: solr_csv_params)
  raw.gsub(/\\#{CSV_MV_SEPARATOR}/, CSV_MV_SEPARATOR)
end

#delete_empty_columns! ⇒ Object 



11
12
13
 
# File 'lib/ddr/index/csv_query_result.rb', line 11

def delete_empty_columns!
  table.by_col!.delete_if { |c, vals| vals.all?(&:nil?) }
end

#each(&block) ⇒ Object 



15
16
17
 
# File 'lib/ddr/index/csv_query_result.rb', line 15

def each(&block)
  table.by_row!.each(&block)
end

#query_field_headings ⇒ Object 



42
43
44
 
# File 'lib/ddr/index/csv_query_result.rb', line 42

def query_field_headings
  query.fields.map { |f| f.respond_to?(:heading) ? f.heading : f.to_s }
end

#solr_csv_header? ⇒ Boolean 



50
51
52
 
# File 'lib/ddr/index/csv_query_result.rb', line 50

def solr_csv_header?
  csv_headers == :first_row
end

#solr_csv_opts ⇒ Object 



34
35
36
37
38
39
40
 
# File 'lib/ddr/index/csv_query_result.rb', line 34

def solr_csv_opts
  { "csv.mv.separator" => CSV_MV_SEPARATOR,
    "csv.header"       => solr_csv_header?,
    "rows"             => solr_csv_rows,
    "wt"               => "csv",
  }
end

#solr_csv_params ⇒ Object 



58
59
60
 
# File 'lib/ddr/index/csv_query_result.rb', line 58

def solr_csv_params
  params.merge(solr_csv_opts)
end

#solr_csv_rows ⇒ Object 



54
55
56
 
# File 'lib/ddr/index/csv_query_result.rb', line 54

def solr_csv_rows
  query.rows || MAX_ROWS
end

#table ⇒ Object 



23
24
25
 
# File 'lib/ddr/index/csv_query_result.rb', line 23

def table
  @table ||= CSV.parse(data, csv_opts)
end