elastic-util 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 631179e3ea01a206b5ddfe51ac9189a55b4e9988
-  data.tar.gz: 245f484434a40d4a86057e519a3ff93a508160ea
+  metadata.gz: 101dd930f2901e53e2f0e3dd37b9a21ecee18265
+  data.tar.gz: ac928d8d5c722e947446b859f1ac10981783d1af
 SHA512:
-  metadata.gz: c56b125c357a4101a9ee8514435638f3c231c6a10fe2130cb49cd1a2a7ad46229b0ea0d023d92c91915f970b60c2907145f3c4c0ff4d4fc50e824e9839ffe3d4
-  data.tar.gz: db2d9d011bc14e7f6e3c64b21a9a31e30d8844541534a3cdffef8cfd67c799c37b1d42bfb2893c0491734e6cf1e8b8338e04264c0e337cea0bb38c01ba0c0929
+  metadata.gz: d2eb13d33794f70d60a2dbec3186414ba5f5c2c51c0cebb7cf8b8220cd922102374de8e67c4e67bf6fb1738fb0dc1c36b6a80aaeff24c30a8d92fab70e4ca9e1
+  data.tar.gz: fbed8715e752f79c11b2eef136662ca43eef2bc5383cea70ea546877956a0f517a13a432b6c02e168c8c42399b624feaf3a34d04c1ff330616112933012b0421

data/bin/elastic-util

@@ -23,6 +23,15 @@ when "backup"
     opts.on('--exclude-fields x,y,z', Array, "The fields to exclude from backup. Default is '_id'.") do |val|
       options[:exclude_fields] = val.collect {|it| it.strip }
     end
+    opts.on('--replace-types type1:_doc,type2:_doc', Array, "Replace certain types with a different type.") do |val|
+      options[:replace_types] = {}
+      val.each  do |it| 
+        pair = it.split(":").collect {|p| p.strip }
+        if pair.size == 2
+          options[:replace_types][pair[0]] = pair[1]
+        end
+      end
+    end
     opts.on( '-s', '--size NUMBER', "The size api parameter. This dicates the size of the files and api payloads. Default is 1000." ) do |val|
       options[:size] = val.to_i
     end

data/lib/elastic_util.rb

@@ -12,7 +12,7 @@ require 'fileutils'
 #
 module ElasticUtil
 
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 
   # The name of the data directory, relative to the user provided backup directory.
   DUMP_DIR = "es_data"
@@ -43,6 +43,7 @@ module ElasticUtil
   # @option opts [Array] :indices The indices to backup. Default is all.
   # @option opts [Array] :exclude_indices Exclude certain indexes.
   # @option opts [Array] :exclude_fields Exclude certain fields. Default is ['_id'].
+  # @option opts [Array] :replace_types Replace certain types with a different type, separated by a colon. eg. 'type1:type2' or 'stat:_doc'
   # @option opts [String] :scroll The scroll api parameter, Default is '5m'.
   # @option opts [Integer] :size The size api parameter. Default is 1000.
   # @option opts [true] :force Delete existing backup directory instead of erroring. Default is false.
@@ -212,7 +213,7 @@ module ElasticUtil
       payload = File.read(file)
       # uri = URI(url)
       http = Net::HTTP.new(uri.host, uri.port)
-      response = http.post("/_bulk", payload)
+      response = http.post("/_bulk", payload, {"Content-Type" => "application/x-ndjson"})
       if !response.is_a?(Net::HTTPSuccess)
         raise Error, "HTTP request failure!\n#{response.inspect}\n#{response.body.to_s}"
       end
@@ -231,8 +232,12 @@ module ElasticUtil
         FileUtils.mkdir_p(dir_name)
         file_name = File.join(dir_name, index_type) + (file_index ? "_#{file_index}" : "") + ".json.data"
         # prepare record for bulk api injection
+        doc_type = hit['_type']
+        if opts[:replace_types] && opts[:replace_types][doc_type]
+          doc_type = opts[:replace_types][doc_type]
+        end
         action_json = {'index' => {
-          '_index' => hit['_index'], '_type' => hit['_type'], '_id' => hit['_id']
+          '_index' => hit['_index'], '_type' => doc_type, '_id' => hit['_id']
         } }
         source_json = hit['_source']
         if opts[:exclude_fields] && source_json
@@ -240,6 +245,7 @@ module ElasticUtil
             source_json.delete(field)
           end
         end
+
         File.open(file_name, 'a') do |file|
           file.write JSON.generate(action_json) + "\n" + JSON.generate(source_json) + "\n"
         end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: elastic-util
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - James Dickson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-06-12 00:00:00.000000000 Z
+date: 2020-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler