factory_bro 0.0.8 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 3dcb2bfceacd76d523581ef27f60a5288191ec73
-  data.tar.gz: ac758cfe634c7478b6db82d37f405be735df0628
+SHA256:
+  metadata.gz: 802a546afa0fc6bf1409edb79b18ab172994c6568a7b64920b1f74d0b801ee76
+  data.tar.gz: 97bee4ca138bed55a28c1567166f7c2e33ed5b7e6bf64404b2540fcadb93362f
 SHA512:
-  metadata.gz: ab87b643dfbf083536445ea15e9a8de3063c725803d6df93fd00c6b67fb5305759effd7faef9debdfbabdf2672bda96772137564213d17fc0f9c9881bb3d861c
-  data.tar.gz: b6cdd259fb39f94fd73be16fc4d46b20f5f21ee6445c473646e44154a7651a19355ce387de503aa3d82c35b83f7296461c4e4e23891c8b2d751ddacd246628ff
+  metadata.gz: f2f56133d25ce735c0829e36b254ac04bf4a002908a92ca5a18c552e207f2c4a82d23002ccf02dd9e7707a3803eeaff4319ef47c1d6d070dcb63841c5fca9d20
+  data.tar.gz: 6e28a0383ae72b3f1f80508da3ef127118812a1a8941f115cac76f9066cb9e822ef505a238307970525fb155d9ad0ad14f4d9191ab4a80d0ada4092c4692d65b

data/lib/factory_bro.rb

@@ -8,28 +8,36 @@ class FactoryBro
       $conns = {}
     else
       $conns.each do |key, conn|
-        if conn.finished?
+        if conn&.finished?
           $conns[key] = ''
         end
       end
     end
+    unless $dbs
+      $dbs = {}
+    end
+    $dbs[name] = db
     if db.include? 'postgres://'
       dbData = parse_url(db)
-      $conns[name] = PG.connect( user:     dbData[:user],
-                          password: dbData[:pw],
-                          host:     dbData[:host],
-                          port:     dbData[:port],
-                          dbname:   dbData[:dbname]
-                        )
+      $conns[name] = safe_connect(name) do
+        PG.connect(
+          user:     dbData[:user],
+          password: dbData[:pw],
+          host:     dbData[:host],
+          port:     dbData[:port],
+          dbname:   dbData[:dbname]
+        )
+      end
     else
       # take arguments of user and pw
-      $conns[name] = PG.connect( dbname: db)
+      $conns[name] = safe_connect { PG.connect(dbname: db) }
     end
   end
 
   def self.close(name)
     $conns[name].finish
     $conns[name] = ''
+    $dbs[name] = ''
   end
 
   def self.close_all
@@ -39,10 +47,12 @@ class FactoryBro
       end
     end
     $conns = nil
+    $dbs = nil
   end
 
   def self.exec(name, statement)
-    run = $conns[name].exec(statement)
+    run = safe_connect(name) { $conns[name].exec(statement) }
+    $try = 0
     run.values
   end
 
@@ -154,4 +164,26 @@ end
     end
     { columns: columns, values: values}
   end
+
+  def self.safe_connect(name)
+    $try ||= 0
+    max_tries = 4
+    wait = 10
+    success = false
+    result = nil
+    until success
+      begin
+        result = yield
+        success = true
+        result
+      rescue PG::UnableToSend, PG::ConnectionBad => error
+        $try += 1
+        raise error if $try > max_tries
+        puts "FactoryBro error: Failed to connect to #{name}, try ##{$try} to reconnect in #{wait}s"
+        sleep wait
+        connect(name, $dbs[name])
+      end
+    end
+    result
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: factory_bro
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.10
 platform: ruby
 authors:
 - Nate Reynolds
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-03-13 00:00:00.000000000 Z
+date: 2023-09-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -16,42 +16,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.1.4
+        version: 1.5.4
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.1.4
+        version: 1.5.4
 - !ruby/object:Gem::Dependency
   name: faker
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.9.3
+        version: 3.2.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.9.3
+        version: 3.2.1
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.10.4
+        version: 0.14.2
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.10.4
+        version: 0.14.2
 description: PSQL Parser and Data Generation similar to factory_girl
 email: nate@n8r.us
 executables: []
@@ -60,7 +60,8 @@ extra_rdoc_files: []
 files:
 - lib/factory_bro.rb
 homepage: https://github.com/nvreynolds/factory_bro
-licenses: []
+licenses:
+- MIT
 metadata: {}
 post_install_message: 
 rdoc_options: []
@@ -77,8 +78,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.6.8
+rubygems_version: 3.0.3.1
 signing_key: 
 specification_version: 4
 summary: Factory Bro