Rename Sequel::Model.DB to Sequel::Model.db

alphagov · Oct 16, 2013 · d8dbb33 · d8dbb33
1 parent f9d25f7
commit d8dbb33
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config/initializers/sequel.rb b/config/initializers/sequel.rb
@@ -8,3 +8,5 @@
 Sequel::Model.plugin :take
 Sequel::Model.plugin :validation_class_methods
 Sequel::Model.plugin :tariff_validation_helper
+
+Sequel::Model.db.extension :server_block
diff --git a/lib/trade_tariff_backend.rb b/lib/trade_tariff_backend.rb
@@ -43,10 +43,8 @@ def data_migration_path
     end
 
     def with_locked_database(&block)
-      Sequel::Model.DB.extension :server_block
-
       # We should use the master database for this
-      Sequel::Model.DB.with_server(:default) do
+      Sequel::Model.db.with_server(:default) do
         begin
           if Sequel::Model.db.get_lock(db_lock_key)
             yield