lib/gitlab_exporter/database/base.rb in gitlab-exporter-9.1.0 vs lib/gitlab_exporter/database/base.rb in gitlab-exporter-10.0.0

@@ -6,15 +6,40 @@
     module Database
       # An abstract class for interacting with DB
       #
       # It takes a connection string (e.g. "dbname=test port=5432")
       class Base
+        POOL_SIZE = 3
+
+        # This timeout is configured to higher interval than scrapping
+        # of Prometheus to ensure that connection is kept instead of
+        # needed to be re-initialized
+        POOL_TIMEOUT = 90
+
         def self.connection_pool
-          @connection_pool ||= Hash.new do |h, connection_string|
-            h[connection_string] = ConnectionPool.new(size: 3, timeout: 5) do
-              PG.connect(connection_string)
+          @@connection_pool ||= Hash.new do |h, connection_string| # rubocop:disable Style/ClassVars
+            h[connection_string] = ConnectionPool.new(size: POOL_SIZE, timeout: POOL_TIMEOUT) do
+              PG.connect(connection_string).tap do |conn|
+                configure_type_map_for_results(conn)
+              end
             end
           end
+        end
+
+        def self.configure_type_map_for_results(conn)
+          tm = PG::BasicTypeMapForResults.new(conn)
+
+          # Remove warning message:
+          # Warning: no type cast defined for type "name" with oid 19.
+          # Please cast this type explicitly to TEXT to be safe for future changes.
+          # Warning: no type cast defined for type "regproc" with oid 24.
+          # Please cast this type explicitly to TEXT to be safe for future changes.
+          [{ "type": "text", "oid": 19 }, { "type": "int4", "oid": 24 }].each do |value|
+            old_coder = tm.coders.find { |c| c.name == value[:type] }
+            tm.add_coder(old_coder.dup.tap { |c| c.oid = value[:oid] })
+          end
+
+          conn.type_map_for_results = tm
         end
 
         def initialize(args, logger: nil)
           @connection_string = args[:connection_string]
           @logger = logger