// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
//go:build plus

package tasks

import (
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
	"github.com/TeaOSLab/EdgeAPI/internal/goman"
	"github.com/iwind/TeaGo/dbs"
	"time"
)

func init() {
	dbs.OnReadyDone(func() {
		goman.New(func() {
			NewNodeScheduleTask(1 * time.Minute).Start()
		})
	})
}

// NodeScheduleTask 节点调度任务
type NodeScheduleTask struct {
	BaseTask

	ticker *time.Ticker
}

func NewNodeScheduleTask(duration time.Duration) *NodeScheduleTask {
	return &NodeScheduleTask{
		ticker: time.NewTicker(duration),
	}
}

func (this *NodeScheduleTask) Start() {
	for range this.ticker.C {
		err := this.Loop()
		if err != nil {
			this.logErr("NodeScheduleTask", err.Error())
		}
	}
}

func (this *NodeScheduleTask) Loop() error {
	if !this.IsPrimaryNode() {
		return nil
	}

	var tx *dbs.Tx

	// 恢复动作过期的节点
	err := this.recoverNodes(tx)
	if err != nil {
		return err
	}

	// 下线租期到期的节点
	err = models.SharedNodeDAO.NotifyNodesWithOfflineDay(tx)
	if err != nil {
		return err
	}

	return nil
}

func (this *NodeScheduleTask) recoverNodes(tx *dbs.Tx) error {
	nodeIds, err := models.SharedNodeDAO.FindNodeIdsWithExpiredActions(tx)
	if err != nil {
		return err
	}
	for _, nodeId := range nodeIds {
		err = models.SharedNodeDAO.ResetNodeActionStatus(tx, nodeId, true)
		if err != nil {
			return err
		}
	}
	return nil
}