ruby_mongo_x 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9eaedb3e5a9b24803714ebeecbb8b2e1bb5116724da0ae0ef639127372403531
-  data.tar.gz: 91bd5334eaf9309e714ada4c27b1d427794880ca206e30743c2be253af52bf22
+  metadata.gz: a8f573ba827ea6150f64c72810b168a43c3bf5fe415f78778942f1ea37d00d12
+  data.tar.gz: 19ad8c68b5ee1d24369db71157271ee42c44c84faac150a105fcc165640fdb1e
 SHA512:
-  metadata.gz: 109a0d54f4e697f9aa41caf0f810e6952020926fd75f467cf3734bfc974aca81e719f98196f278d289ff94ed568fc073261625f561cd46b32820bde4892366c7
-  data.tar.gz: 1241abab507078eb3b1b43527a48e12dba103eb1c99b069022250d7874914804f547d90d68d88d20d6a1391392337c8cf926885598c1a3325a74ec0ba4636c75
+  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
@@ -25,6 +26,9 @@ class RubyMongoX
         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"
@@ -120,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)
@@ -157,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.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Grzegorz Błaszczyk
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-09 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