ruby_mongo_x 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8c510b50acd0606b4bdb6482634afcd7e2ed8112f61f7969bf6aa7116bcbf8a8
-  data.tar.gz: 48dbb7b516ef651f626b3eff173813a52f7a6666fd5a671b7732dc86ce60e8e8
+  metadata.gz: a8f573ba827ea6150f64c72810b168a43c3bf5fe415f78778942f1ea37d00d12
+  data.tar.gz: 19ad8c68b5ee1d24369db71157271ee42c44c84faac150a105fcc165640fdb1e
 SHA512:
-  metadata.gz: a76b0d3792e8f214937453e38ce1801ce7e6de70095214c32f5a748dad4e56849509779f3fdbe40e6e95c09294ad061d3048fe51431ae5fd42210b962ca96d55
-  data.tar.gz: 3b062aeeb992b25cbb7fd21aac9559c1cf28359802d13344b57b259ada6209bac7d34d3e44107f799e3512ef67f64e97e94a4a538fd8a9b266805589c95588cd
+  metadata.gz: 0d55880966d5792c766a03f8eaef4ceeb5b0bb403f088af7a9c34364dabd1d0c9cae5bc0bbe525639afabc6cc7343bc5ac67c8b3ec3757401d5e051230b20c5f
+  data.tar.gz: 5632252bb60d51f28a500548fafeebdf52d4e73785e5d085a0db534a6e6c17c719b263ccc6c54fb587349d57e9a39cfaa3f2c9ea57a8a586c27e83248f81a466

data/lib/ruby_mongo_x.rb

@@ -1,5 +1,6 @@
-require 'mongo'
 require 'concurrent'
+require 'mongo'
+require 'openssl'
 require_relative './helpers/loggable'
 
 class RubyMongoX
@@ -22,6 +23,12 @@ class RubyMongoX
         log_debug "\t Connecting to shard #{shard} ...\n"
         begin
           @mongo_clients[idx.to_s] = ::Mongo::Client.new(shard, options)
+        rescue Mongo::Error::SocketError => e
+          log_debug "\t SocketError: #{e.message}...\n"
+          log_debug "\t SocketError: Stopped trying to connect to shard #{shard}... FAIL!\n"
+        rescue OpenSSL::SSL::SSLError => e
+          log_debug "\t SSLError: #{e.message}...\n"
+          log_debug "\t SSLError: Stopped trying to connect to shard #{shard}... FAIL!\n"
         rescue Mongo::Error::NoSRVRecords => e
           log_debug "\t NoSRVRecords: #{e.message}...\n"
           log_debug "\t NoSRVRecords: Retrying connect to shard #{shard}...\n"
@@ -117,12 +124,52 @@ class RubyMongoX
   end
 
 
-  def update_one()
-    # TODO: implement
+  def update_one(query, update_query_or_doc, update_options={})
+    log_debug "\t Update one query: #{query.to_json}, update query or doc: #{update_query_or_doc.to_json}\n"
+
+    promises = @mongo_clients.map do |mongo_client_item|
+      mongo_client = mongo_client_item[1]
+      Concurrent::Promise.execute do
+        update_one_in_shard(mongo_client, query, update_query_or_doc, update_options)
+      end
+    end
+
+    results = {
+      modified_count: 0,
+      shards: {}
+    }
+
+    promises.map { |promise|
+      results[:shards][promise.value[:shard]] = promise.value[:modified_count]
+      results[:modified_count] = results[:modified_count] + promise.value[:modified_count]
+    }
+
+    log_debug "Results: #{results}\n"
+    results
   end
 
-  def update_many()
-    # TODO: implement
+  def update_many(query, update_query, update_options={})
+    log_debug "\t Update many query: #{query.to_json}, update query: #{update_query.to_json}\n"
+
+    promises = @mongo_clients.map do |mongo_client_item|
+      mongo_client = mongo_client_item[1]
+      Concurrent::Promise.execute do
+        update_many_in_shard(mongo_client, query, update_query, update_options)
+      end
+    end
+
+    results = {
+      modified_count: 0,
+      shards: {}
+    }
+
+    promises.map { |promise|
+      results[:shards][promise.value[:shard]] = promise.value[:modified_count]
+      results[:modified_count] = results[:modified_count] + promise.value[:modified_count]
+    }
+
+    log_debug "Results: #{results}\n"
+    results
   end
 
   def fetch_data_from_shard(mongo_client, query)
@@ -154,6 +201,40 @@ class RubyMongoX
     }
   end
 
+  def update_one_in_shard(mongo_client, query, update_query_or_doc, update_options)
+    collection = mongo_client[@collection_name.to_sym]
+    result = collection.update_one(query, update_query_or_doc, update_options)
+    result_document = result.documents.first
+
+    mongo_server = mongo_client&.cluster&.servers&.first
+    mongo_user = mongo_server&.options["user"] || ""
+    mongo_db = mongo_server&.options["database"] || ""
+    log_debug "\t\tupdate_one() with mongo_client user: #{mongo_user}, database: #{mongo_db} returns #{result_document}\n"
+
+    {
+      shard: (mongo_user.nil? || mongo_user == "") ? mongo_db : mongo_user,
+      modified_count: result_document["nModified"],
+      ok: result_document["ok"]
+    }
+  end
+
+  def update_many_in_shard(mongo_client, query, update_query, update_options)
+    collection = mongo_client[@collection_name.to_sym]
+    result = collection.update_many(query, update_query, update_options)
+    result_document = result.documents.first
+
+    mongo_server = mongo_client&.cluster&.servers&.first
+    mongo_user = mongo_server&.options["user"] || ""
+    mongo_db = mongo_server&.options["database"] || ""
+    log_debug "\t\tupdate_many() with mongo_client user: #{mongo_user}, database: #{mongo_db} returns #{result_document}\n"
+
+    {
+      shard: (mongo_user.nil? || mongo_user == "") ? mongo_db : mongo_user,
+      modified_count: result_document["nModified"],
+      ok: result_document["ok"]
+    }
+  end
+
   def disconnect()
     @mongo_clients.values.each do |client|
       log_debug "\t Disconnecting from client #{client.to_s} ..."

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_mongo_x
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Grzegorz Błaszczyk
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-10-28 00:00:00.000000000 Z
+date: 2025-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: config
@@ -30,14 +30,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.19.1
+        version: 2.21.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.19.1
+        version: 2.21.1
 - !ruby/object:Gem::Dependency
   name: pry-byebug
   requirement: !ruby/object:Gem::Requirement
@@ -56,16 +56,16 @@ dependencies:
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 13.2.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 13.2.1
 description: MongoDB client in Ruby that connects to dozens of Mongo databases, performs
   queries simultaneously and gets the results in a unified way.
 email: grzegorz.blaszczyk@gmail.com