Class: PgMigrate::Util

Inherits:
Object
  • Object
show all
Defined in:
lib/pg_migrate/util.rb

Constant Summary collapse

LOGGER = 
Logging.logger[self]

Class Method Summary collapse

Class Method Details

.create_conn(args) ⇒ Object

recommended to create all connections via this method, so that we can put NOTICE/CONTEXT into logger instead of stderr



9
10
11
12
13
14
15
16
 
# File 'lib/pg_migrate/util.rb', line 9

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

.get_conn(connection_options) ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/pg_migrate/util.rb', line 18

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

.get_db_name(connection_options) ⇒ Object

finds dbname from connection_options



47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/pg_migrate/util.rb', line 47

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

.get_oob_conn(connection_options) ⇒ Object

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



33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/pg_migrate/util.rb', line 33

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