mirror of
https://github.com/mentos1386/zdravko.git
synced 2024-11-29 10:41:19 +00:00
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package workflows
|
|
|
|
import (
|
|
"sort"
|
|
"time"
|
|
|
|
"code.tjo.space/mentos1386/zdravko/database/models"
|
|
"code.tjo.space/mentos1386/zdravko/internal/activities"
|
|
"go.temporal.io/sdk/workflow"
|
|
)
|
|
|
|
type MonitorWorkflowParam struct {
|
|
Script string
|
|
MonitorId string
|
|
WorkerGroupIds []string
|
|
}
|
|
|
|
func (w *Workflows) MonitorWorkflowDefinition(ctx workflow.Context, param MonitorWorkflowParam) (models.MonitorStatus, error) {
|
|
workerGroupIds := param.WorkerGroupIds
|
|
sort.Strings(workerGroupIds)
|
|
|
|
for _, workerGroupId := range workerGroupIds {
|
|
ctx = workflow.WithActivityOptions(ctx, workflow.ActivityOptions{
|
|
StartToCloseTimeout: 60 * time.Second,
|
|
TaskQueue: workerGroupId,
|
|
})
|
|
|
|
heatlcheckParam := activities.HealtcheckParam{
|
|
Script: param.Script,
|
|
}
|
|
|
|
var monitorResult *activities.MonitorResult
|
|
err := workflow.ExecuteActivity(ctx, w.activities.Monitor, heatlcheckParam).Get(ctx, &monitorResult)
|
|
if err != nil {
|
|
return models.MonitorUnknown, err
|
|
}
|
|
|
|
status := models.MonitorFailure
|
|
if monitorResult.Success {
|
|
status = models.MonitorSuccess
|
|
}
|
|
|
|
historyParam := activities.HealtcheckAddToHistoryParam{
|
|
MonitorId: param.MonitorId,
|
|
Status: status,
|
|
Note: monitorResult.Note,
|
|
WorkerGroupId: workerGroupId,
|
|
}
|
|
|
|
var historyResult *activities.MonitorAddToHistoryResult
|
|
err = workflow.ExecuteActivity(ctx, w.activities.MonitorAddToHistory, historyParam).Get(ctx, &historyResult)
|
|
if err != nil {
|
|
return models.MonitorUnknown, err
|
|
}
|
|
}
|
|
|
|
return models.MonitorSuccess, nil
|
|
}
|