lib/mihari/commands/database.rb in mihari-5.7.1 vs lib/mihari/commands/database.rb in mihari-5.7.2

@@ -7,18 +7,21 @@
     #
     module Database
       class << self
         def included(thor)
           thor.class_eval do
+            include Mixins
+
             desc "migrate", "Migrate DB schemas"
+            around :with_db_connection
             method_option :verbose, type: :boolean, default: true
             #
             # @param [String] direction
             #
             def migrate(direction = "up")
               ActiveRecord::Migration.verbose = options["verbose"]
 
-              Mihari::Database.with_db_connection { Mihari::Database.migrate direction.to_sym }
+              Mihari::Database.migrate direction.to_sym
             end
           end
         end
       end
     end