Class: PactBroker::DB::CleanTask

Inherits:
Rake::TaskLib
  • Object
show all
Defined in:
lib/pact_broker/tasks/clean_task.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(&block) ⇒ CleanTask

Returns a new instance of CleanTask.



11
12
13
14
15
16
17
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 11

def initialize &block
  require "pact_broker/db/clean_incremental"
  @version_deletion_limit = 1000
  @dry_run = false
  @keep_version_selectors = PactBroker::DB::CleanIncremental::DEFAULT_KEEP_SELECTORS
  rake_task(&block)
end

Instance Attribute Details

#database_connection ⇒ Object

Returns the value of attribute database_connection.



5
6
7
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 5

def database_connection
  @database_connection
end

#dry_run ⇒ Object

Returns the value of attribute dry_run.



9
10
11
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 9

def dry_run
  @dry_run
end

#keep_version_selectors ⇒ Object

Returns the value of attribute keep_version_selectors.



6
7
8
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 6

def keep_version_selectors
  @keep_version_selectors
end

#logger ⇒ Object

Returns the value of attribute logger.



8
9
10
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 8

def logger
  @logger
end

#version_deletion_limit ⇒ Object

Returns the value of attribute version_deletion_limit.



7
8
9
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 7

def version_deletion_limit
  @version_deletion_limit
end

Instance Method Details

#add_defaults_to_keep_selectors ⇒ Object 



69
70
71
72
73
74
75
76
77
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 69

def add_defaults_to_keep_selectors
  if keep_version_selectors.none?(&:currently_deployed?)
    keep_version_selectors << PactBroker::DB::Clean::Selector.new(deployed: true)
  end

  if keep_version_selectors.none?(&:currently_supported?)
    keep_version_selectors <<  PactBroker::DB::Clean::Selector.new(released: true)
  end
end

#output(string, payload = {}) ⇒ Object 



65
66
67
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 65

def output string, payload = {}
  logger ? logger.info(string, payload: payload) : puts("#{string} #{payload.to_json}")
end

#rake_task(&block) ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/pact_broker/tasks/clean_task.rb', line 26

def rake_task &block
  namespace :pact_broker do
    namespace :db do
      desc "Clean unnecessary pacts and verifications from database"
      task :clean do | _t, _args |

        instance_eval(&block)

        require "pact_broker/db/clean_incremental"
        require "pact_broker/error"
        require "yaml"
        require "benchmark"

        raise PactBroker::Error.new("You must specify the version_deletion_limit") unless version_deletion_limit

        prefix = dry_run ? "[DRY RUN] " : ""

        if keep_version_selectors.nil? || keep_version_selectors.empty?
          raise PactBroker::Error.new("You must specify which versions to keep")
        else
          add_defaults_to_keep_selectors
          output "#{prefix}Deleting oldest #{version_deletion_limit} versions, keeping versions that match the configured selectors", keep_version_selectors
        end

        start_time = Time.now
        results = PactBroker::DB::CleanIncremental.call(database_connection,
          keep: keep_version_selectors,
          limit: version_deletion_limit,
          logger: logger,
          dry_run: dry_run
        )
        end_time = Time.now
        elapsed_seconds = (end_time - start_time).to_i
        output "Results (#{elapsed_seconds} seconds)", results
      end
    end
  end
end