cost_time.go

// Copyright (c) 2021 Terminus, Inc.
//
// This program is free software: you can use, redistribute, and/or modify
// it under the terms of the GNU Affero General Public License, version 3
// or later ("AGPL"), as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package costtimeutil

import (
	"time"

	"github.com/erda-project/erda/modules/pipeline/spec"
)

func CalculateTaskCostTimeSec(task *spec.PipelineTask) (cost int64) {
	if task.CostTimeSec >= 0 {
		return task.CostTimeSec
	}
	if task.TimeBegin.IsZero() {
		return -1
	}
	if task.TimeEnd.IsZero() && task.Status.IsEndStatus() { // 终态，但 timeEnd 异常为空
		task.TimeEnd = task.TimeUpdated
	}
	if task.TimeEnd.IsZero() { // 正在运行中
		return int64(time.Now().Sub(task.TimeBegin).Seconds())
	}
	return int64(task.TimeEnd.Sub(task.TimeBegin).Seconds())
}

func CalculateTaskQueueTimeSec(task *spec.PipelineTask) (cost int64) {
	defer func() {
		if task.Status.IsEndStatus() && cost < 0 {
			cost = 0
		}
	}()
	if task.QueueTimeSec >= 0 {
		return task.QueueTimeSec
	}
	if task.Extra.TimeBeginQueue.IsZero() {
		return -1
	}
	if task.Extra.TimeEndQueue.IsZero() { // 正在运行中
		return int64(time.Now().Sub(task.Extra.TimeBeginQueue).Seconds())
	}
	return int64(task.Extra.TimeEndQueue.Sub(task.Extra.TimeBeginQueue).Seconds())
}

func CalculatePipelineCostTimeSec(p *spec.Pipeline) (cost int64) {
	defer func() {
		if p.Status.IsEndStatus() && cost < 0 {
			cost = 0
		}
	}()
	if p.CostTimeSec >= 0 {
		return p.CostTimeSec
	}
	// 还没开始
	if p.TimeBegin == nil || p.TimeBegin.IsZero() {
		return -1
	}
	// 正在运行中
	if p.TimeEnd == nil || p.TimeEnd.IsZero() {
		return int64(time.Now().Sub(*p.TimeBegin).Seconds())
	}
	return int64(p.TimeEnd.Sub(*p.TimeBegin).Seconds())
}