database_cleaner-spanner 0.1.0 → 0.1.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. checksums.yaml +4 -4
  2. data/.standard.yml +3 -0
  3. data/CHANGELOG.md +4 -2
  4. data/LICENSE +21 -0
  5. data/README.md +6 -0
  6. data/database_cleaner-spanner.gemspec +3 -0
  7. data/example/rails/.gitattributes +7 -0
  8. data/example/rails/.gitignore +39 -0
  9. data/example/rails/.rspec +1 -0
  10. data/example/rails/.ruby-version +1 -0
  11. data/example/rails/Gemfile +69 -0
  12. data/example/rails/README.md +24 -0
  13. data/example/rails/Rakefile +6 -0
  14. data/example/rails/app/assets/config/manifest.js +4 -0
  15. data/example/rails/app/assets/images/.keep +0 -0
  16. data/example/rails/app/assets/stylesheets/application.css +15 -0
  17. data/example/rails/app/channels/application_cable/channel.rb +4 -0
  18. data/example/rails/app/channels/application_cable/connection.rb +4 -0
  19. data/example/rails/app/controllers/application_controller.rb +2 -0
  20. data/example/rails/app/controllers/concerns/.keep +0 -0
  21. data/example/rails/app/helpers/application_helper.rb +2 -0
  22. data/example/rails/app/javascript/application.js +3 -0
  23. data/example/rails/app/javascript/controllers/application.js +9 -0
  24. data/example/rails/app/javascript/controllers/hello_controller.js +7 -0
  25. data/example/rails/app/javascript/controllers/index.js +11 -0
  26. data/example/rails/app/jobs/application_job.rb +7 -0
  27. data/example/rails/app/mailers/application_mailer.rb +4 -0
  28. data/example/rails/app/models/album.rb +5 -0
  29. data/example/rails/app/models/application_record.rb +5 -0
  30. data/example/rails/app/models/concerns/.keep +0 -0
  31. data/example/rails/app/models/customer.rb +3 -0
  32. data/example/rails/app/models/order.rb +4 -0
  33. data/example/rails/app/models/product.rb +3 -0
  34. data/example/rails/app/models/singer.rb +4 -0
  35. data/example/rails/app/models/song.rb +15 -0
  36. data/example/rails/app/views/layouts/application.html.erb +16 -0
  37. data/example/rails/app/views/layouts/mailer.html.erb +13 -0
  38. data/example/rails/app/views/layouts/mailer.text.erb +1 -0
  39. data/example/rails/bin/bundle +114 -0
  40. data/example/rails/bin/importmap +4 -0
  41. data/example/rails/bin/rails +4 -0
  42. data/example/rails/bin/rake +4 -0
  43. data/example/rails/bin/setup +33 -0
  44. data/example/rails/config/application.rb +37 -0
  45. data/example/rails/config/boot.rb +4 -0
  46. data/example/rails/config/cable.yml +10 -0
  47. data/example/rails/config/credentials.yml.enc +1 -0
  48. data/example/rails/config/database.yml +18 -0
  49. data/example/rails/config/environment.rb +5 -0
  50. data/example/rails/config/environments/development.rb +70 -0
  51. data/example/rails/config/environments/production.rb +93 -0
  52. data/example/rails/config/environments/test.rb +60 -0
  53. data/example/rails/config/importmap.rb +7 -0
  54. data/example/rails/config/initializers/assets.rb +12 -0
  55. data/example/rails/config/initializers/content_security_policy.rb +25 -0
  56. data/example/rails/config/initializers/filter_parameter_logging.rb +8 -0
  57. data/example/rails/config/initializers/inflections.rb +16 -0
  58. data/example/rails/config/initializers/permissions_policy.rb +11 -0
  59. data/example/rails/config/locales/en.yml +33 -0
  60. data/example/rails/config/puma.rb +43 -0
  61. data/example/rails/config/routes.rb +6 -0
  62. data/example/rails/config/storage.yml +34 -0
  63. data/example/rails/config.ru +6 -0
  64. data/example/rails/db/migrate/20221113100815_create_customers.rb +9 -0
  65. data/example/rails/db/migrate/20221113101016_create_products.rb +10 -0
  66. data/example/rails/db/migrate/20221113101027_create_orders.rb +11 -0
  67. data/example/rails/db/migrate/20221113101034_create_singers.rb +10 -0
  68. data/example/rails/db/migrate/20221113101039_create_albums.rb +12 -0
  69. data/example/rails/db/migrate/20221113101044_create_songs.rb +13 -0
  70. data/example/rails/db/schema.rb +58 -0
  71. data/example/rails/db/seeds.rb +7 -0
  72. data/example/rails/lib/assets/.keep +0 -0
  73. data/example/rails/lib/tasks/.keep +0 -0
  74. data/example/rails/log/.keep +0 -0
  75. data/example/rails/public/404.html +67 -0
  76. data/example/rails/public/422.html +67 -0
  77. data/example/rails/public/500.html +66 -0
  78. data/example/rails/public/apple-touch-icon-precomposed.png +0 -0
  79. data/example/rails/public/apple-touch-icon.png +0 -0
  80. data/example/rails/public/favicon.ico +0 -0
  81. data/example/rails/public/robots.txt +1 -0
  82. data/example/rails/spec/factories/factory.rb +33 -0
  83. data/example/rails/spec/models/album_spec.rb +5 -0
  84. data/example/rails/spec/models/customer_spec.rb +5 -0
  85. data/example/rails/spec/models/order_spec.rb +5 -0
  86. data/example/rails/spec/models/product_spec.rb +5 -0
  87. data/example/rails/spec/models/singer_spec.rb +5 -0
  88. data/example/rails/spec/models/song_spec.rb +5 -0
  89. data/example/rails/spec/rails_helper.rb +70 -0
  90. data/example/rails/spec/sample_spec.rb +14 -0
  91. data/example/rails/spec/spec_helper.rb +94 -0
  92. data/example/rails/spec/support/factory_bot.rb +3 -0
  93. data/example/rails/storage/.keep +0 -0
  94. data/example/rails/tmp/.keep +0 -0
  95. data/example/rails/tmp/pids/.keep +0 -0
  96. data/example/rails/tmp/storage/.keep +0 -0
  97. data/example/rails/vendor/.keep +0 -0
  98. data/example/rails/vendor/javascript/.keep +0 -0
  99. data/lib/database_cleaner/spanner/deletion.rb +31 -5
  100. data/lib/database_cleaner/spanner/version.rb +1 -1
  101. data/scripts/performance_test.rb +127 -0
  102. data/scripts/schema.sql +93 -0
  103. metadata +125 -2
@@ -0,0 +1,67 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>The page you were looking for doesn't exist (404)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/404.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>The page you were looking for doesn't exist.</h1>
62
+ <p>You may have mistyped the address or the page may have moved.</p>
63
+ </div>
64
+ <p>If you are the application owner check the logs for more information.</p>
65
+ </div>
66
+ </body>
67
+ </html>
@@ -0,0 +1,67 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>The change you wanted was rejected (422)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/422.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>The change you wanted was rejected.</h1>
62
+ <p>Maybe you tried to change something you didn't have access to.</p>
63
+ </div>
64
+ <p>If you are the application owner check the logs for more information.</p>
65
+ </div>
66
+ </body>
67
+ </html>
@@ -0,0 +1,66 @@
1
+ <!DOCTYPE html>
2
+ <html>
3
+ <head>
4
+ <title>We're sorry, but something went wrong (500)</title>
5
+ <meta name="viewport" content="width=device-width,initial-scale=1">
6
+ <style>
7
+ .rails-default-error-page {
8
+ background-color: #EFEFEF;
9
+ color: #2E2F30;
10
+ text-align: center;
11
+ font-family: arial, sans-serif;
12
+ margin: 0;
13
+ }
14
+
15
+ .rails-default-error-page div.dialog {
16
+ width: 95%;
17
+ max-width: 33em;
18
+ margin: 4em auto 0;
19
+ }
20
+
21
+ .rails-default-error-page div.dialog > div {
22
+ border: 1px solid #CCC;
23
+ border-right-color: #999;
24
+ border-left-color: #999;
25
+ border-bottom-color: #BBB;
26
+ border-top: #B00100 solid 4px;
27
+ border-top-left-radius: 9px;
28
+ border-top-right-radius: 9px;
29
+ background-color: white;
30
+ padding: 7px 12% 0;
31
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
32
+ }
33
+
34
+ .rails-default-error-page h1 {
35
+ font-size: 100%;
36
+ color: #730E15;
37
+ line-height: 1.5em;
38
+ }
39
+
40
+ .rails-default-error-page div.dialog > p {
41
+ margin: 0 0 1em;
42
+ padding: 1em;
43
+ background-color: #F7F7F7;
44
+ border: 1px solid #CCC;
45
+ border-right-color: #999;
46
+ border-left-color: #999;
47
+ border-bottom-color: #999;
48
+ border-bottom-left-radius: 4px;
49
+ border-bottom-right-radius: 4px;
50
+ border-top-color: #DADADA;
51
+ color: #666;
52
+ box-shadow: 0 3px 8px rgba(50, 50, 50, 0.17);
53
+ }
54
+ </style>
55
+ </head>
56
+
57
+ <body class="rails-default-error-page">
58
+ <!-- This file lives in public/500.html -->
59
+ <div class="dialog">
60
+ <div>
61
+ <h1>We're sorry, but something went wrong.</h1>
62
+ </div>
63
+ <p>If you are the application owner check the logs for more information.</p>
64
+ </div>
65
+ </body>
66
+ </html>
File without changes
File without changes
@@ -0,0 +1 @@
1
+ # See https://www.robotstxt.org/robotstxt.html for documentation on how to use the robots.txt file
@@ -0,0 +1,33 @@
1
+ FactoryBot.define do
2
+ factory :customer do
3
+ name { "customer" }
4
+ end
5
+
6
+ factory :product do
7
+ name { "product" }
8
+ price { 1.0 }
9
+ end
10
+
11
+ factory :order do
12
+ customer
13
+ product
14
+ quantity { 1 }
15
+ end
16
+
17
+ factory :singer do
18
+ name { "singer" }
19
+ end
20
+
21
+ factory :album do
22
+ singer
23
+ albumid { 1 }
24
+ title { "album" }
25
+ end
26
+
27
+ factory :song do
28
+ singer
29
+ album
30
+ songid { 1 }
31
+ title { "song" }
32
+ end
33
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Album, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Customer, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Order, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Product, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Singer, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,5 @@
1
+ require 'rails_helper'
2
+
3
+ RSpec.describe Song, type: :model do
4
+ pending "add some examples to (or delete) #{__FILE__}"
5
+ end
@@ -0,0 +1,70 @@
1
+ # This file is copied to spec/ when you run 'rails generate rspec:install'
2
+ require 'spec_helper'
3
+ ENV['RAILS_ENV'] ||= 'test'
4
+ require_relative '../config/environment'
5
+ # Prevent database truncation if the environment is production
6
+ abort("The Rails environment is running in production mode!") if Rails.env.production?
7
+ require 'rspec/rails'
8
+ # Add additional requires below this line. Rails is not loaded until this point!
9
+
10
+ # Requires supporting ruby files with custom matchers and macros, etc, in
11
+ # spec/support/ and its subdirectories. Files matching `spec/**/*_spec.rb` are
12
+ # run as spec files by default. This means that files in spec/support that end
13
+ # in _spec.rb will both be required and run as specs, causing the specs to be
14
+ # run twice. It is recommended that you do not name files matching this glob to
15
+ # end with _spec.rb. You can configure this pattern with the --pattern
16
+ # option on the command line or in ~/.rspec, .rspec or `.rspec-local`.
17
+ #
18
+ # The following line is provided for convenience purposes. It has the downside
19
+ # of increasing the boot-up time by auto-requiring all files in the support
20
+ # directory. Alternatively, in the individual `*_spec.rb` files, manually
21
+ # require only the support files necessary.
22
+ #
23
+
24
+ Dir[Rails.root.join('spec', 'support', '**', '*.rb')].sort.each { |f| require f }
25
+
26
+ # Checks for pending migrations and applies them before tests are run.
27
+ # If you are not using ActiveRecord, you can remove these lines.
28
+ # begin
29
+ # ActiveRecord::Migration.maintain_test_schema!
30
+ # rescue ActiveRecord::PendingMigrationError => e
31
+ # abort e.to_s.strip
32
+ # end
33
+ RSpec.configure do |config|
34
+ # Remove this line if you're not using ActiveRecord or ActiveRecord fixtures
35
+ config.fixture_path = "#{::Rails.root}/spec/fixtures"
36
+
37
+ # If you're not using ActiveRecord, or you'd prefer not to run each of your
38
+ # examples within a transaction, remove the following line or assign false
39
+ # instead of true.
40
+ config.use_transactional_fixtures = false
41
+
42
+ # You can uncomment this line to turn off ActiveRecord support entirely.
43
+ # config.use_active_record = false
44
+
45
+ # RSpec Rails can automatically mix in different behaviours to your tests
46
+ # based on their file location, for example enabling you to call `get` and
47
+ # `post` in specs under `spec/controllers`.
48
+ #
49
+ # You can disable this behaviour by removing the line below, and instead
50
+ # explicitly tag your specs with their type, e.g.:
51
+ #
52
+ # RSpec.describe UsersController, type: :controller do
53
+ # # ...
54
+ # end
55
+ #
56
+ # The different available types are documented in the features, such as in
57
+ # https://relishapp.com/rspec/rspec-rails/docs
58
+ config.infer_spec_type_from_file_location!
59
+
60
+ # Filter lines from Rails gems in backtraces.
61
+ config.filter_rails_from_backtrace!
62
+ # arbitrary gems may also be filtered via:
63
+ # config.filter_gems_from_backtrace("gem name")
64
+
65
+ config.around(:each) do |example|
66
+ DatabaseCleaner[:spanner].cleaning do
67
+ example.run
68
+ end
69
+ end
70
+ end
@@ -0,0 +1,14 @@
1
+ RSpec.describe "sample" do
2
+ before do
3
+ customer = create(:customer)
4
+ product = create(:product)
5
+ create(:order, customer: customer, product: product)
6
+ singer = create(:singer)
7
+ album = create(:album, singer: singer)
8
+ create(:song, singer: singer, album: album)
9
+ end
10
+
11
+ it "sample" do
12
+ expect(true).to be true
13
+ end
14
+ end
@@ -0,0 +1,94 @@
1
+ # This file was generated by the `rails generate rspec:install` command. Conventionally, all
2
+ # specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
3
+ # The generated `.rspec` file contains `--require spec_helper` which will cause
4
+ # this file to always be loaded, without a need to explicitly require it in any
5
+ # files.
6
+ #
7
+ # Given that it is always loaded, you are encouraged to keep this file as
8
+ # light-weight as possible. Requiring heavyweight dependencies from this file
9
+ # will add to the boot time of your test suite on EVERY test run, even for an
10
+ # individual file that may not need all of that loaded. Instead, consider making
11
+ # a separate helper file that requires the additional dependencies and performs
12
+ # the additional setup, and require it from the spec files that actually need
13
+ # it.
14
+ #
15
+ # See https://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
16
+ RSpec.configure do |config|
17
+ # rspec-expectations config goes here. You can use an alternate
18
+ # assertion/expectation library such as wrong or the stdlib/minitest
19
+ # assertions if you prefer.
20
+ config.expect_with :rspec do |expectations|
21
+ # This option will default to `true` in RSpec 4. It makes the `description`
22
+ # and `failure_message` of custom matchers include text for helper methods
23
+ # defined using `chain`, e.g.:
24
+ # be_bigger_than(2).and_smaller_than(4).description
25
+ # # => "be bigger than 2 and smaller than 4"
26
+ # ...rather than:
27
+ # # => "be bigger than 2"
28
+ expectations.include_chain_clauses_in_custom_matcher_descriptions = true
29
+ end
30
+
31
+ # rspec-mocks config goes here. You can use an alternate test double
32
+ # library (such as bogus or mocha) by changing the `mock_with` option here.
33
+ config.mock_with :rspec do |mocks|
34
+ # Prevents you from mocking or stubbing a method that does not exist on
35
+ # a real object. This is generally recommended, and will default to
36
+ # `true` in RSpec 4.
37
+ mocks.verify_partial_doubles = true
38
+ end
39
+
40
+ # This option will default to `:apply_to_host_groups` in RSpec 4 (and will
41
+ # have no way to turn it off -- the option exists only for backwards
42
+ # compatibility in RSpec 3). It causes shared context metadata to be
43
+ # inherited by the metadata hash of host groups and examples, rather than
44
+ # triggering implicit auto-inclusion in groups with matching metadata.
45
+ config.shared_context_metadata_behavior = :apply_to_host_groups
46
+
47
+ # The settings below are suggested to provide a good initial experience
48
+ # with RSpec, but feel free to customize to your heart's content.
49
+ =begin
50
+ # This allows you to limit a spec run to individual examples or groups
51
+ # you care about by tagging them with `:focus` metadata. When nothing
52
+ # is tagged with `:focus`, all examples get run. RSpec also provides
53
+ # aliases for `it`, `describe`, and `context` that include `:focus`
54
+ # metadata: `fit`, `fdescribe` and `fcontext`, respectively.
55
+ config.filter_run_when_matching :focus
56
+
57
+ # Allows RSpec to persist some state between runs in order to support
58
+ # the `--only-failures` and `--next-failure` CLI options. We recommend
59
+ # you configure your source control system to ignore this file.
60
+ config.example_status_persistence_file_path = "spec/examples.txt"
61
+
62
+ # Limits the available syntax to the non-monkey patched syntax that is
63
+ # recommended. For more details, see:
64
+ # https://relishapp.com/rspec/rspec-core/docs/configuration/zero-monkey-patching-mode
65
+ config.disable_monkey_patching!
66
+
67
+ # Many RSpec users commonly either run the entire suite or an individual
68
+ # file, and it's useful to allow more verbose output when running an
69
+ # individual spec file.
70
+ if config.files_to_run.one?
71
+ # Use the documentation formatter for detailed output,
72
+ # unless a formatter has already been configured
73
+ # (e.g. via a command-line flag).
74
+ config.default_formatter = "doc"
75
+ end
76
+
77
+ # Print the 10 slowest examples and example groups at the
78
+ # end of the spec run, to help surface which specs are running
79
+ # particularly slow.
80
+ config.profile_examples = 10
81
+
82
+ # Run specs in random order to surface order dependencies. If you find an
83
+ # order dependency and want to debug it, you can fix the order by providing
84
+ # the seed, which is printed after each run.
85
+ # --seed 1234
86
+ config.order = :random
87
+
88
+ # Seed global randomization in this process using the `--seed` CLI option.
89
+ # Setting this allows you to use `--seed` to deterministically reproduce
90
+ # test failures related to randomization by passing the same `--seed` value
91
+ # as the one that triggered the failure.
92
+ Kernel.srand config.seed
93
+ =end
94
+ end
@@ -0,0 +1,3 @@
1
+ RSpec.configure do |config|
2
+ config.include FactoryBot::Syntax::Methods
3
+ end
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
@@ -33,26 +33,40 @@ module DatabaseCleaner
33
33
  def initialize(
34
34
  only: [],
35
35
  except: [],
36
+ batch_deletion: false,
36
37
  cache_tables: true
37
38
  )
38
39
  @only = only
39
40
  @except = except
41
+ @batch_deletion = batch_deletion
40
42
  @cache_tables = cache_tables
41
43
 
42
44
  @deletable_tables = {}
43
45
  end
44
46
 
45
47
  def clean
46
- sorted_table_groups.each do |group|
47
- clean_table_group(group)
48
+ if @batch_deletion
49
+ clean_as_batch
50
+ else
51
+ clean_each
48
52
  end
49
53
  end
50
54
 
51
55
  private
52
56
 
53
- def clean_table_group(group)
54
- group.each do |table|
55
- client.delete(table) if deletable?(table)
57
+ def clean_as_batch
58
+ client.transaction do |tx|
59
+ tx.batch_update do |b|
60
+ each_deletable_table do |table|
61
+ b.batch_update("DELETE #{table} WHERE TRUE")
62
+ end
63
+ end
64
+ end
65
+ end
66
+
67
+ def clean_each
68
+ each_deletable_table do |table|
69
+ client.delete(table)
56
70
  end
57
71
  end
58
72
 
@@ -69,6 +83,18 @@ module DatabaseCleaner
69
83
  end
70
84
  end
71
85
 
86
+ def each_deletable_table(&block)
87
+ each_group do |group|
88
+ group.each do |table|
89
+ block.call(table) if deletable?(table)
90
+ end
91
+ end
92
+ end
93
+
94
+ def each_group(&block)
95
+ sorted_table_groups.each(&block)
96
+ end
97
+
72
98
  def sorted_table_groups
73
99
  return @sorted_table_groups if @cache_tables && @sorted_table_groups
74
100
 
@@ -2,6 +2,6 @@
2
2
 
3
3
  module DatabaseCleaner
4
4
  module Spanner
5
- VERSION = "0.1.0"
5
+ VERSION = "0.1.1"
6
6
  end
7
7
  end