lib/pg_migrate/util.rb in pg_migrate-0.1.11 vs lib/pg_migrate/util.rb in pg_migrate-0.1.12
- old
+ new
@@ -1,73 +1,73 @@
-module PgMigrate
-
- class Util
-
- LOGGER = Logging.logger[self]
-
- # recommended to create all connections via this method,
- # so that we can put NOTICE/CONTEXT into logger instead of stderr
- def self.create_conn(args)
- conn = PG::Connection.open(args)
- conn.set_notice_receiver do |result|
- #result.res_status(result.result_status)
- LOGGER.debug result.error_message
- end
- return conn
- end
-
- def self.get_conn(connection_options)
- if !connection_options[:pgconn].nil?
- return connection_options[:pgconn]
- elsif !connection_options[:connstring].nil?
- create_conn(connection_options[:connstring])
- elsif !connection_options[:connopts].nil?
- return create_conn(connection_options[:connopts])
- else
- return create_conn(connection_options)
- end
- end
-
- # the 'out-of-band' conn is a connection to a database that you aren't
- # interested in modifying; it's basically a landing pad so that you can do:
- # DROP DATABSE BLAH; CREATE DATABASE BLAH -- for testing
- def self.get_oob_conn(connection_options)
-
- if !connection_options[:oob_pgconn].nil?
- return connection_options[:oob_pgconn]
- elsif !connection_options[:oob_connstring].nil?
- return create_conn(connection_options[:oob_connstring])
- elsif !connection_options[:oob_connopts].nil?
- return create_conn(connection_options[:oob_connopts])
- else
- return create_conn(connection_options)
- end
- end
-
- # finds dbname from connection_options
- def self.get_db_name(connection_options)
- dbname = nil
- if !connection_options[:pgconn].nil?
- dbname = connection_options[:pgconn].db
- elsif !connection_options[:connstring].nil?
- connstring = connection_options[:connstring]
- bits = connstring.split(" ")
- bits.each do |bit|
- if bit.start_with? "dbname="
- dbname = bit["dbname=".length..-1]
- break
- end
- end
- elsif !connection_options[:connopts].nil?
- dbname = connection_options[:connopts]["dbname"]
- else
- dbname = connection_options["dbname"]
- end
-
- if dbname.nil?
- raise "db name is null. tried finding dbname in #{connection_options.inspect}"
- end
-
- return dbname
- end
- end
-end
+module PgMigrate
+
+ class Util
+
+ LOGGER = Logging.logger[self]
+
+ # recommended to create all connections via this method,
+ # so that we can put NOTICE/CONTEXT into logger instead of stderr
+ def self.create_conn(args)
+ conn = PG::Connection.open(args)
+ conn.set_notice_receiver do |result|
+ #result.res_status(result.result_status)
+ LOGGER.debug result.error_message
+ end
+ return conn
+ end
+
+ def self.get_conn(connection_options)
+ if !connection_options[:pgconn].nil?
+ return connection_options[:pgconn]
+ elsif !connection_options[:connstring].nil?
+ create_conn(connection_options[:connstring])
+ elsif !connection_options[:connopts].nil?
+ return create_conn(connection_options[:connopts])
+ else
+ return create_conn(connection_options)
+ end
+ end
+
+ # the 'out-of-band' conn is a connection to a database that you aren't
+ # interested in modifying; it's basically a landing pad so that you can do:
+ # DROP DATABSE BLAH; CREATE DATABASE BLAH -- for testing
+ def self.get_oob_conn(connection_options)
+
+ if !connection_options[:oob_pgconn].nil?
+ return connection_options[:oob_pgconn]
+ elsif !connection_options[:oob_connstring].nil?
+ return create_conn(connection_options[:oob_connstring])
+ elsif !connection_options[:oob_connopts].nil?
+ return create_conn(connection_options[:oob_connopts])
+ else
+ return create_conn(connection_options)
+ end
+ end
+
+ # finds dbname from connection_options
+ def self.get_db_name(connection_options)
+ dbname = nil
+ if !connection_options[:pgconn].nil?
+ dbname = connection_options[:pgconn].db
+ elsif !connection_options[:connstring].nil?
+ connstring = connection_options[:connstring]
+ bits = connstring.split(" ")
+ bits.each do |bit|
+ if bit.start_with? "dbname="
+ dbname = bit["dbname=".length..-1]
+ break
+ end
+ end
+ elsif !connection_options[:connopts].nil?
+ dbname = connection_options[:connopts]["dbname"]
+ else
+ dbname = connection_options["dbname"]
+ end
+
+ if dbname.nil?
+ raise "db name is null. tried finding dbname in #{connection_options.inspect}"
+ end
+
+ return dbname
+ end
+ end
+end