statuseable.rb 2.4 KB
Newer Older
K
Kamil Trzcinski 已提交
1
module Statuseable
K
Kamil Trzcinski 已提交
2 3
  extend ActiveSupport::Concern

4 5 6
  AVAILABLE_STATUSES = %w(pending running success failed canceled skipped)

  class_methods do
K
Kamil Trzcinski 已提交
7
    def status_sql
8 9
      builds = all.select('count(*)').to_sql
      success = all.success.select('count(*)').to_sql
K
Kamil Trzcinski 已提交
10
      ignored = all.ignored.select('count(*)').to_sql if all.respond_to?(:ignored)
K
Kamil Trzcinski 已提交
11
      ignored ||= '0'
12 13 14
      pending = all.pending.select('count(*)').to_sql
      running = all.running.select('count(*)').to_sql
      canceled = all.canceled.select('count(*)').to_sql
K
Kamil Trzcinski 已提交
15
      skipped = all.skipped.select('count(*)').to_sql
K
Kamil Trzcinski 已提交
16 17

      deduce_status = "(CASE
18
        WHEN (#{builds})=0 THEN NULL
K
Kamil Trzcinski 已提交
19
        WHEN (#{builds})=(#{skipped}) THEN 'skipped'
20 21 22
        WHEN (#{builds})=(#{success})+(#{ignored})+(#{skipped}) THEN 'success'
        WHEN (#{builds})=(#{pending})+(#{skipped}) THEN 'pending'
        WHEN (#{builds})=(#{canceled})+(#{success})+(#{ignored})+(#{skipped}) THEN 'canceled'
K
Kamil Trzcinski 已提交
23 24 25 26 27 28 29
        WHEN (#{running})+(#{pending})>0 THEN 'running'
        ELSE 'failed'
      END)"

      deduce_status
    end

K
Kamil Trzcinski 已提交
30
    def status
31
      all.pluck(self.status_sql).first
K
Kamil Trzcinski 已提交
32 33
    end

34 35 36 37 38
    def started_at
      all.minimum(:started_at)
    end

    def finished_at
K
Kamil Trzcinski 已提交
39
      all.maximum(:finished_at)
40
    end
K
Kamil Trzcinski 已提交
41 42 43
  end

  included do
44
    validates :status, inclusion: { in: AVAILABLE_STATUSES }
K
Kamil Trzcinski 已提交
45 46 47 48 49 50 51

    state_machine :status, initial: :pending do
      state :pending, value: 'pending'
      state :running, value: 'running'
      state :failed, value: 'failed'
      state :success, value: 'success'
      state :canceled, value: 'canceled'
K
Kamil Trzcinski 已提交
52
      state :skipped, value: 'skipped'
K
Kamil Trzcinski 已提交
53 54 55 56 57 58
    end

    scope :running, -> { where(status: 'running') }
    scope :pending, -> { where(status: 'pending') }
    scope :success, -> { where(status: 'success') }
    scope :failed, -> { where(status: 'failed')  }
K
Kamil Trzcinski 已提交
59
    scope :canceled, -> { where(status: 'canceled')  }
K
Kamil Trzcinski 已提交
60
    scope :skipped, -> { where(status: 'skipped')  }
K
Kamil Trzcinski 已提交
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
    scope :running_or_pending, -> { where(status: [:running, :pending]) }
    scope :finished, -> { where(status: [:success, :failed, :canceled]) }
  end

  def started?
    !pending? && !canceled? && started_at
  end

  def active?
    running? || pending?
  end

  def complete?
    canceled? || success? || failed?
  end
76 77 78 79 80 81 82 83 84 85

  private

  def calculate_duration
    if started_at && finished_at
      finished_at - started_at
    elsif started_at
      Time.now - started_at
    end
  end
86
end