monq 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bdf67f2d8ba19bd24e12b8d7a5397ca27a800a37925eebb1c584f40f05b9af1f
-  data.tar.gz: aa19a699c514ede037e1663f326a74cd6dd9ab77096255125eaa5a77c5583629
+  metadata.gz: 549e91e0850eef71c43ad692ffb4ad92c6c5bbb4b43a7d834bf2ee690a3384f4
+  data.tar.gz: 5e2c84dacf04271206add867dd75df819d2bc5ccfc37d429e991ad271e3f3487
 SHA512:
-  metadata.gz: 446339b7b3cd119046fce6473ff03ae6ca294bfe642d0ea1e637306b9bbe4eb1674bf5b9d369bcc00ea4c0cb8a276efbe96e571bb146b5258765fe34e8e3851d
-  data.tar.gz: 73da4e7fb09de97340e482641198ee3a5635413a385cb45813250ce807ccd8d6483eb8a9bbfd74ecd149fbabfc7062b0b4457ba9daa89dbee4bc93f8bbeaed94
+  metadata.gz: d762f5a51e8a52f47c2deb2ef4ef1d5f6b09d41f492f1c483aa8b73574fcf436603fa694ff68b791717c491a9a19689dd14cb288ca2f966aed714e23ea742464
+  data.tar.gz: f0dcabaa63dcdc853594b30312ae091e35a84af2c980a47840f2801882f266f04462b79b9c4fe31e22a47d813f9d2275b8d2e5404e7f37f12afd88f506583a11

data/lib/monq/ext/array.rb

@@ -22,7 +22,7 @@ class Array
       csv = CSV.new(file)
       return if docs.empty?
 
-      keys = docs.first.keys
+      keys = Monq::Presenter.column_names(docs)
       csv << keys.map(&:to_s)
       docs.each { |row| csv << keys.map { |k| row[k].to_s } }
     end
@@ -35,7 +35,7 @@ class Array
     package.workbook.add_worksheet(name: sheet_name) do |sheet|
       next if docs.empty?
 
-      keys = docs.first.keys
+      keys = Monq::Presenter.column_names(docs)
       sheet.add_row(keys.map(&:to_s))
       docs.each { |row| sheet.add_row(keys.map { |k| row[k].to_s }) }
     end

data/lib/monq/ext.rb

@@ -34,12 +34,14 @@ module Monq
     end
 
     def self.filter_docs(docs, attrs = [], options = {})
-      unless docs.empty? || docs.first.is_a?(Hash) || !docs.first.respond_to?(:attributes)
+      return docs if docs.empty?
+
+      column_names = Presenter.column_names(docs)
+      unless docs.first.is_a?(Hash) || !docs.first.respond_to?(:attributes)
         docs = docs.map(&:attributes)
       end
-      return docs if docs.empty? || (attrs.empty? && !options[:except])
+      return docs if attrs.empty? && !options[:except]
 
-      column_names = docs.first.keys
       selected = attrs.flat_map { |e| e.is_a?(Regexp) ? column_names.grep(e) : e }.uniq
 
       if options[:except]

data/lib/monq/presenter.rb

@@ -29,11 +29,18 @@ module Monq
       return [] if docs.nil? || docs.empty?
 
       docs = docs.map { |doc| doc.is_a?(Hash) || !doc.respond_to?(:attributes) ? doc : doc.attributes }
-      keys = docs.first.keys
+      keys = column_names(docs)
       body = docs.map { |doc| keys.map { |key| scalar(doc[key]) } }
       [keys, *body]
     end
 
+    # Union of keys across all docs: Mongoid documents only carry the fields
+    # actually stored on each one, so any single doc (e.g. the oldest row in a
+    # natural-order cursor) can miss fields present in the rest.
+    def column_names(docs)
+      docs.flat_map { |doc| doc.is_a?(Hash) || !doc.respond_to?(:attributes) ? doc.keys : doc.attributes.keys }.uniq
+    end
+
     def as_json_tree(value)
       case value
       when BSON::ObjectId, BSON::Decimal128

data/lib/monq/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Monq
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: monq
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Liu Xiang