Add new private method to DatabaseCollation to retrieve db encoding

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