BigW Consortium Gitlab

database.rb 2.93 KB
Newer Older
1 2
module Gitlab
  module Database
3 4 5 6 7
    # The max value of INTEGER type is the same between MySQL and PostgreSQL:
    # https://www.postgresql.org/docs/9.2/static/datatype-numeric.html
    # http://dev.mysql.com/doc/refman/5.7/en/integer-types.html
    MAX_INT_VALUE = 2147483647

8 9 10 11
    def self.config
      ActiveRecord::Base.configurations[Rails.env]
    end

12
    def self.adapter_name
13
      config['adapter']
14 15
    end

16
    def self.mysql?
17
      adapter_name.casecmp('mysql2').zero?
18 19 20
    end

    def self.postgresql?
21
      adapter_name.casecmp('postgresql').zero?
22 23 24 25
    end

    def self.version
      database_version.match(/\A(?:PostgreSQL |)([^\s]+).*\z/)[1]
26
    end
27

28 29 30
    def self.nulls_last_order(field, direction = 'ASC')
      order = "#{field} #{direction}"

31
      if postgresql?
32 33 34 35 36 37 38 39 40 41
        order << ' NULLS LAST'
      else
        # `field IS NULL` will be `0` for non-NULL columns and `1` for NULL
        # columns. In the (default) ascending order, `0` comes first.
        order.prepend("#{field} IS NULL, ") if direction == 'ASC'
      end

      order
    end

42 43 44
    def self.nulls_first_order(field, direction = 'ASC')
      order = "#{field} #{direction}"

45
      if postgresql?
46 47 48 49 50 51 52 53 54 55
        order << ' NULLS FIRST'
      else
        # `field IS NULL` will be `0` for non-NULL columns and `1` for NULL
        # columns. In the (default) ascending order, `0` comes first.
        order.prepend("#{field} IS NULL, ") if direction == 'DESC'
      end

      order
    end

56
    def self.random
57
      postgresql? ? "RANDOM()" : "RAND()"
58 59
    end

60 61
    def self.true_value
      if postgresql?
62 63 64 65 66 67
        "'t'"
      else
        1
      end
    end

68 69
    def self.false_value
      if postgresql?
70 71 72 73 74
        "'f'"
      else
        0
      end
    end
75

76 77 78
    def self.with_connection_pool(pool_size)
      pool = create_connection_pool(pool_size)

79 80 81 82 83
      begin
        yield(pool)
      ensure
        pool.disconnect!
      end
84 85
    end

86 87 88
    # pool_size - The size of the DB pool.
    # host - An optional host name to use instead of the default one.
    def self.create_connection_pool(pool_size, host = nil)
89 90 91
      # See activerecord-4.2.7.1/lib/active_record/connection_adapters/connection_specification.rb
      env = Rails.env
      original_config = ActiveRecord::Base.configurations
92

93
      env_config = original_config[env].merge('pool' => pool_size)
94 95
      env_config['host'] = host if host

96 97 98 99 100 101 102 103 104 105
      config = original_config.merge(env => env_config)

      spec =
        ActiveRecord::
          ConnectionAdapters::
          ConnectionSpecification::Resolver.new(config).spec(env.to_sym)

      ActiveRecord::ConnectionAdapters::ConnectionPool.new(spec)
    end

106 107 108 109
    def self.connection
      ActiveRecord::Base.connection
    end

110 111
    private_class_method :connection

112 113 114 115 116 117 118 119 120
    def self.database_version
      row = connection.execute("SELECT VERSION()").first

      if postgresql?
        row['version']
      else
        row.first
      end
    end
121 122

    private_class_method :database_version
123 124
  end
end