JCS-pub/internal/event/check_unavailable_cache.go

package event

import (
	"database/sql"
	"fmt"

	"github.com/jmoiron/sqlx"
	"github.com/samber/lo"
	"gitlink.org.cn/cloudream/common/consts"
	"gitlink.org.cn/cloudream/common/utils/logger"
	"gitlink.org.cn/cloudream/db/model"
	mysql "gitlink.org.cn/cloudream/db/sql"
)

type CheckUnavailableCache struct {
	NodeID int
}

func NewCheckUnavailableCache(nodeID int) *CheckUnavailableCache {
	return &CheckUnavailableCache{
		NodeID: nodeID,
	}
}

func (t *CheckUnavailableCache) TryMerge(other Event) bool {
	event, ok := other.(*CheckUnavailableCache)
	if !ok {
		return false
	}
	if event.NodeID != t.NodeID {
		return false
	}

	return true
}

func (t *CheckUnavailableCache) Execute(execCtx ExecuteContext) {
	err := execCtx.Args.DB.DoTx(sql.LevelSerializable, func(tx *sqlx.Tx) error {
		node, err := mysql.Node.GetByID(tx, t.NodeID)
		if err == sql.ErrNoRows {
			return nil
		}
		if err != nil {
			return fmt.Errorf("get node failed, err: %w", err)
		}

		if node.State != consts.NODE_STATE_UNAVAILABLE {
			return nil
		}

		caches, err := mysql.Cache.GetNodeCaches(tx, t.NodeID)
		if err != nil {
			return fmt.Errorf("get node caches failed, err: %w", err)
		}

		err = mysql.Cache.DeleteNodeAll(tx, t.NodeID)
		if err != nil {
			return fmt.Errorf("delete node all caches failed, err: %w", err)
		}

		execCtx.Executor.Post(NewCheckRepCount(lo.Map(caches, func(ch model.Cache, index int) string { return ch.HashValue })))
		return nil
	})

	if err != nil {
		logger.WithField("NodeID", t.NodeID).Warn(err.Error())
	}
}