Merge pull request #100 from RodrigoMNardi/feature/github/calc_time_execution

Execution time

Author: RodrigoMNardi

db/migrate/20240924140825_add_ci_job_stage_execution_time.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+#  SPDX-License-Identifier: BSD-2-Clause
+#
+#  20240924140825_add_ci_job_stage_execution_time.rb
+#  Part of NetDEF CI System
+#
+#  Copyright (c) 2024 by
+#  Network Device Education Foundation, Inc. ("NetDEF")
+#
+
+class AddCiJobStageExecutionTime < ActiveRecord::Migration[6.0]
+  def change
+    add_column :ci_jobs, :execution_time, :integer
+    add_column :stages, :execution_time, :integer
+  end
+end

db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[7.2].define(version: 2024_06_17_121935) do
+ActiveRecord::Schema[7.2].define(version: 2024_09_24_140825) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
 
@@ -77,6 +77,7 @@
     t.integer "retry", default: 0
     t.integer "parent_stage_id"
     t.bigint "stage_id"
+    t.integer "execution_time"
     t.index ["check_suite_id"], name: "index_ci_jobs_on_check_suite_id"
     t.index ["stage_id"], name: "index_ci_jobs_on_stage_id"
   end
@@ -160,6 +161,7 @@
     t.datetime "updated_at", null: false
     t.bigint "check_suite_id"
     t.bigint "stage_configuration_id"
+    t.integer "execution_time"
     t.index ["check_suite_id"], name: "index_stages_on_check_suite_id"
     t.index ["stage_configuration_id"], name: "index_stages_on_stage_configuration_id"
   end

lib/github/build/summary.rb

@@ -135,12 +135,14 @@ def finished_stage_update(stage, output)
         if stage.jobs.failure.empty?
           logger(Logger::WARN, "Stage: #{stage.name} finished - success")
           stage.success(@github, output: output, agent: @agent)
+          stage.update_execution_time
 
           return
         end
 
         logger(Logger::WARN, "Stage: #{stage.name} finished - failure")
         stage.failure(@github, output: output, agent: @agent)
+        stage.update_execution_time
       end
 
       def update_summary(stage)

lib/github/update_status.rb

@@ -62,9 +62,11 @@ def update_status
         @job.in_progress(@github_check)
       when 'success'
         @job.success(@github_check)
+        @job.update_execution_time
         slack_notify_success
       else
         failure
+        @job.update_execution_time
         slack_notify_failure
       end
 

lib/models/ci_job.rb

@@ -36,6 +36,15 @@ def finished?
     !%w[queued in_progress].include?(status)
   end
 
+  def update_execution_time
+    started = audit_statuses.find_by(status: :in_progress)
+    finished = audit_statuses.find_by(status: %i[success failure])
+
+    return if started.nil? || finished.nil?
+
+    update(execution_time: finished.created_at - started.created_at)
+  end
+
   def create_check_run(agent: 'Github')
     AuditStatus.create(auditable: self, status: :queued, agent: agent, created_at: Time.now)
     update(status: :queued)

lib/models/stage.rb

@@ -18,6 +18,15 @@ class Stage < ActiveRecord::Base
 
   default_scope -> { order(id: :asc) }, all_queries: true
 
+  def update_execution_time
+    started = audit_statuses.find_by(status: :in_progress)
+    finished = audit_statuses.find_by(status: %i[success failure])
+
+    return if started.nil? || finished.nil?
+
+    update(execution_time: finished.created_at - started.created_at)
+  end
+
   def running?
     jobs.where(status: %i[queued in_progress]).any?
   end

spec/lib/models/ci_job_spec.rb

@@ -80,4 +80,23 @@
       end
     end
   end
+
+  describe '#execution_time' do
+    let(:stage) { create(:ci_job, check_ref: nil) }
+
+    context 'when CiJob started and finished' do
+      it 'must update status' do
+        stage.in_progress(github)
+        stage.success(github)
+        stage.update_execution_time
+      end
+    end
+
+    context 'when CiJob started and not finished' do
+      it 'must update status' do
+        stage.in_progress(github)
+        stage.update_execution_time
+      end
+    end
+  end
 end