factory_bro 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0a0d637ecdd62dc71833bfe6d0b98471b8214ba9
-  data.tar.gz: 6351f82583eb758c6f014468e0ca2f8c3a8b3087
+  metadata.gz: 59a75451c5f0e7cb6413f88c1880be8fc307da79
+  data.tar.gz: 78fa6608aa32483b46a6dbae9aae9caddaab8800
 SHA512:
-  metadata.gz: 2351bbf0d25e653fcf949f3f1303e18527ba8271eb76fce6d4582b08e236acf34d8c73109b11a4ad2acf7e89152f4599e440660fc1ad7b899ea0082cdce8dfeb
-  data.tar.gz: d7991aa18411378cdbdd9231799f8ae59dcb091d9508e7a2ced9fc7fbf1cfadf32bbf38a450d3d171c2cfc403cf8b298bcbc1c7645c2f7f2ab38fc47d8591445
+  metadata.gz: b312e1ffd77694d46cc426574dd76a2986b3259c8d90555415272b346009cbcf2a51dfd5a991f040c3e80f44309166114b7d8d6df01f1b2425836cd70457d4a7
+  data.tar.gz: 080a2645b01f82a96b569a4d0786dbdcb485d71e9d70f29d5f32ddabe8a506f1c6dccee52886d42070ee79adcade03c64228a3ad34c32ff050d16cd13f9b4692

data/lib/factory_bro.rb

@@ -3,10 +3,19 @@ require 'faker'
 require 'json'
 
 class FactoryBro
-  def self.connect(db)
+  def self.connect(name, db)
+    unless $conns
+      $conns = {}
+    else
+      $conns.each do |key, conn|
+        if conn.finished?
+          $conns[key] = ''
+        end
+      end
+    end
     if db.include? 'postgres://'
       dbData = parse_url(db)
-      $conn = PG.connect( user:     dbData[:user],
+      $conns[name] = PG.connect( user:     dbData[:user],
                           password: dbData[:pw],
                           host:     dbData[:host],
                           port:     dbData[:port],
@@ -14,35 +23,45 @@ class FactoryBro
                         )
     else
       # take arguments of user and pw
-      $conn = PG.connect( dbname: db)
+      $conns[name] = PG.connect( dbname: db)
     end
   end
 
-  def self.close
-    $conn.finish
+  def self.close(name)
+    $conns[name].finish
+    $conns[name] = ''
+  end
+
+  def self.close_all
+    $conns.each do |key, conn|
+      if conn.class == PG::Connection
+        conn.finish
+      end
+    end
+    $conns = nil
   end
 
-  def self.exec(statement)
-    run = $conn.exec(statement)
+  def self.exec(name, statement)
+    run = $conns[name].exec(statement)
     run.values
   end
 
-  def self.create_bases
-    tables = parse_tables
+  def self.create_bases(name)
+    tables = parse_tables(name)
     system 'mkdir', '-p', 'factories' unless File.directory?(Dir.pwd+"/factories")
     tables.each do |table|
       path = File.join(Dir.pwd, 'factories', "#{table}.rb")
       unless File.exists?(path)
 
-        File.open(path, 'w') {|file| file.write(base_factory_content(table)) }
+        File.open(path, 'w') {|file| file.write(base_factory_content(name, table)) }
         # puts "#{table} factory created at: #{path}"
       end
     end
   end
 
-  def self.generate_data(table, data)
+  def self.generate_data(name, table, data)
     helperData = generate_helper(data[:factoryData])
-    res = $conn.exec("INSERT INTO #{table} (#{helperData[:columns]})
+    res = $conns[name].exec("INSERT INTO #{table} (#{helperData[:columns]})
     VALUES (#{helperData[:values]});")
     data[:meta]
   end
@@ -58,8 +77,8 @@ class FactoryBro
     }
   end
 
-  def self.parse_tables
-    res = $conn.exec("SELECT TABLE_NAME
+  def self.parse_tables(name)
+    res = $conns[name].exec("SELECT TABLE_NAME
                       FROM INFORMATION_SCHEMA.TABLES
                       WHERE TABLE_TYPE = 'BASE TABLE' AND TABLE_SCHEMA='public'
                       ORDER BY TABLE_NAME;")
@@ -68,20 +87,20 @@ class FactoryBro
 
   def self.parse_columns(table)
     # hash this
-    res = $conn.exec(" SELECT COLUMN_NAME, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH
+    res = $conns[name].exec("SELECT COLUMN_NAME, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH
                       FROM INFORMATION_SCHEMA.COLUMNS
                       WHERE TABLE_NAME = '#{table}';")
     res.values # as tuple
   end
 
-  def self.base_factory_content(table)
+  def self.base_factory_content(name, table)
     "module Factory
   class #{table.split('_').map { |word| word.capitalize }.join}
     def self.base
       # remember:
       # to return any needed metadata (i.e. ID)
       # run other required bases before running method below
-      FactoryBro.generate_data('#{table}' , #{base_method_helper(table)})
+      FactoryBro.generate_data(#{name}, '#{table}' , #{base_method_helper(table)})
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: factory_bro
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Nate Reynolds
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-10-11 00:00:00.000000000 Z
+date: 2016-10-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg