diff --git a/lib/planetscale_rails/migration.rb b/lib/planetscale_rails/migration.rb index 9b2a8aa..acb7454 100644 --- a/lib/planetscale_rails/migration.rb +++ b/lib/planetscale_rails/migration.rb @@ -13,6 +13,13 @@ def create_table(table_name, **options) end super(table_name, **options.except(:keyspace)) end + + def drop_table(table_name, **options) + if ENV["ENABLE_PSDB"] && options[:keyspace].present? + table_name = "#{options[:keyspace]}.#{table_name}" + end + super(table_name, **options.except(:keyspace)) + end end end end