Class: Webhookdb::Tasks::Regress

Inherits:
Rake::TaskLib
  • Object
show all
Defined in:
lib/webhookdb/tasks/regress.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Regress 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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
64
65
66
 
# File 'lib/webhookdb/tasks/regress.rb', line 11

def initialize
  super()
  namespace :regress do
    desc "Creates databases for all orgs that do not have them."
    task :prepare do
      Webhookdb.load_app
      Webhookdb::Organization.where(readonly_connection_url_raw: nil).each do |org|
        org.prepare_database_connections
        org.migrate_replication_tables
      end
    end

    desc "Prints out all service integrations that have untrimmed logged webhooks."
    task :list_available do
      Webhookdb.load_app
      opaque_ids = Webhookdb::LoggedWebhook.where(truncated_at: nil).
        distinct(:service_integration_opaque_id).
        select_map(:service_integration_opaque_id)
      sints = Webhookdb::ServiceIntegration.where(opaque_id: opaque_ids).all
      self.print_service_integrations(sints)
    end

    desc "Replay the last :count webhooks going to the service integration with the given opaque id. " \
         "Use -1 for all webhooks."
    task :replay, [:opaque_id, :count] do |_, args|
      opaque_id = args.fetch(:opaque_id)
      count = args.fetch(:count).to_i
      Webhookdb.load_app
      ds = Webhookdb::LoggedWebhook.where(truncated_at: nil).
        where(service_integration_opaque_id: opaque_id).
        order(Sequel.asc(:inserted_at))
      ds = ds.limit(count) if count >= 0
      ds.paged_each do |lw|
        good = lw.retry_one
        puts "#{lw.pk} failed" unless good
      end
    end

    desc "Prints out all service integrations that have backfill info available."
    task :list_backfill do
      Webhookdb.load_app
      sints = Webhookdb::ServiceIntegration.exclude(backfill_key: nil).all
      self.print_service_integrations(sints)
    end

    desc "Runs a backfill for the service integration with the given opaque id. " \
         "Regression backfills are limited to one page."
    task :backfill, [:opaque_id] do |_, args|
      opaque_id = args.fetch(:opaque_id)
      Webhookdb.load_app
      sint = Webhookdb::ServiceIntegration[opaque_id:] or raise "No service integration for #{opaque_id}"
      bfjob = Webhookdb::BackfillJob.create_recursive(service_integration: sint, incremental: false)
      sint.replicator.backfill(bfjob)
    end
  end
end

Instance Method Details



68
69
70
71
72
73
74
75
 
# File 'lib/webhookdb/tasks/regress.rb', line 68

def print_service_integrations(iter)
  rows = iter.map do |sint|
    name = "#{sint.organization.name} (#{sint.organization_id})"
    [name, sint.service_name, sint.opaque_id, sint.table_name]
  end
  rows.sort!
  self.table_print rows
end

#table_print(rows) ⇒ Object 



77
78
79
80
81
82
83
84
85
 
# File 'lib/webhookdb/tasks/regress.rb', line 77

def table_print(rows)
  max_lengths = rows.each_with_object({}) do |row, m|
    row.each_with_index { |value, idx| m[idx] = [m.fetch(idx, 0), value.to_s.length].max }
  end
  row_strings = rows.map do |row|
    row.each_with_index.map { |value, idx| "%-#{max_lengths[idx]}s" % value }.join("\t")
  end
  puts row_strings
end