gitlab-org--gitlab-foss/app/models/ci/build_metadata.rb

module Ci
  # The purpose of this class is to store Build related data that can be disposed.
  # Data that should be persisted forever, should be stored with Ci::Build model.
  class BuildMetadata < ActiveRecord::Base
    extend Gitlab::Ci::Model
    include Presentable
    include ChronicDurationAttribute

    self.table_name = 'ci_builds_metadata'

    belongs_to :build, class_name: 'Ci::Build'
    belongs_to :project

    chronic_duration_attr_reader :timeout_human_readable, :timeout

    after_initialize :set_project_id

    enum timeout_source: {
        unknown_timeout_source: 1,
        project_timeout_source: 2,
        runner_timeout_source: 3
    }

    def save_timeout_state!
      return unless build.runner.present?

      project_timeout = build.project&.build_timeout
      timeout = [project_timeout, build.runner.maximum_timeout].compact.min
      timeout_source = timeout < project_timeout ? :runner_timeout_source : :project_timeout_source

      update_attributes(timeout: timeout, timeout_source: timeout_source)
    end

    private

    def set_project_id
      return unless self.project_id.nil?

      self.project_id = build&.project&.id
    end
  end
end