Add new private method to DatabaseCollation to retrieve db encoding

mysociety · Feb 8, 2017 · 13eeb25 · 13eeb25
1 parent 857ee1c
commit 13eeb25
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/database_collation.rb b/lib/database_collation.rb
@@ -59,6 +59,15 @@ def supported_collations
         map { |row| row['collname'] }
   end
 
+  def database_encoding
+    sql = <<-EOF.strip_heredoc.squish
+    SELECT encoding FROM pg_database
+    WHERE datname = '#{ connection.current_database }';
+    EOF
+
+    @database_encoding ||= connection.execute(sql).first["encoding"]
+  end
+
   def adapter_name
     @adapter_name ||= connection.adapter_name
   end