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_connectionObject

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_runObject

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_selectorsObject

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

#loggerObject

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_limitObject

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

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



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

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
# 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
          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