JCS-pub/client/internal/spacesyncer/space_syncer.go

package spacesyncer

import (
	"context"
	"fmt"
	"sync"

	"gitlink.org.cn/cloudream/common/pkgs/async"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/jcs-pub/client/internal/cluster"
	"gitlink.org.cn/cloudream/jcs-pub/client/internal/db"
	"gitlink.org.cn/cloudream/jcs-pub/client/internal/metacache"
	stgpool "gitlink.org.cn/cloudream/jcs-pub/common/pkgs/storage/pool"
	jcstypes "gitlink.org.cn/cloudream/jcs-pub/common/types"
)

const (
	logMod = "SpaceSyncer"
)

type SpaceSyncerEvent interface {
	IsSpaceSyncerEvent() bool
}

type SpaceSyncer struct {
	db        *db.DB
	stgPool   *stgpool.Pool
	spaceMeta *metacache.UserSpaceMeta
	cluster   *cluster.Cluster
	lock      sync.Mutex
	tasks     map[jcstypes.SpaceSyncTaskID]*task
}

func New(db *db.DB, stgPool *stgpool.Pool, spaceMeta *metacache.UserSpaceMeta, cluster *cluster.Cluster) *SpaceSyncer {
	return &SpaceSyncer{
		db:        db,
		stgPool:   stgPool,
		spaceMeta: spaceMeta,
		cluster:   cluster,
		tasks:     make(map[jcstypes.SpaceSyncTaskID]*task),
	}
}

func (s *SpaceSyncer) Start() *async.UnboundChannel[SpaceSyncerEvent] {
	s.lock.Lock()
	defer s.lock.Unlock()

	log := logger.WithField("Mod", logMod)

	ch := async.NewUnboundChannel[SpaceSyncerEvent]()

	allTask, err := db.DoTx01(s.db, s.db.SpaceSyncTask().GetAll)
	if err != nil {
		log.Warnf("load task from db: %v", err)
	} else {
		var rms []jcstypes.SpaceSyncTaskID
		for _, t := range allTask {
			ctx, cancel := context.WithCancel(context.Background())
			tsk := task{
				Task:     t,
				Context:  ctx,
				CancelFn: cancel,
			}

			switch tr := t.Trigger.(type) {
			case *jcstypes.SpaceSyncTriggerOnce:
				// Once类型的任务没有执行完也不执行了
				rms = append(rms, t.TaskID)

			case *jcstypes.SpaceSyncTriggerInterval:
				triggerInterval(s, &tsk, tr)

			case *jcstypes.SpaceSyncTriggerAt:
				triggerAt(s, &tsk, tr)
			}

			log.Infof("load task %v from db", t.TaskID)
		}

		if len(rms) > 0 {
			err := s.db.SpaceSyncTask().BatchDelete(s.db.DefCtx(), rms)
			if err != nil {
				log.Warnf("batch delete task: %v", err)
			} else {
				log.Infof("%v once task deleted", len(rms))
			}
		}
	}

	return ch
}

func (s *SpaceSyncer) Stop() {
	s.lock.Lock()
	defer s.lock.Unlock()

	if !s.cluster.IsMaster() {
		return
	}

	for _, t := range s.tasks {
		t.CancelFn()
	}

	s.tasks = make(map[jcstypes.SpaceSyncTaskID]*task)
}

func (s *SpaceSyncer) CreateTask(t jcstypes.SpaceSyncTask) (*TaskInfo, error) {
	log := logger.WithField("Mod", logMod)

	if !s.cluster.IsMaster() {
		return nil, fmt.Errorf("not master, create task aborted")
	}

	d := s.db
	err := d.DoTx(func(tx db.SQLContext) error {
		err := d.SpaceSyncTask().Create(tx, &t)
		if err != nil {
			return err
		}

		return nil
	})
	if err != nil {
		return nil, fmt.Errorf("creating space sync task: %w", err)
	}

	ctx, cancel := context.WithCancel(context.Background())
	tsk := task{
		Task:     t,
		Context:  ctx,
		CancelFn: cancel,
	}

	s.lock.Lock()
	s.tasks[t.TaskID] = &tsk
	s.lock.Unlock()

	switch tr := t.Trigger.(type) {
	case *jcstypes.SpaceSyncTriggerOnce:
		triggerOnce(s, &tsk)

	case *jcstypes.SpaceSyncTriggerInterval:
		triggerInterval(s, &tsk, tr)

	case *jcstypes.SpaceSyncTriggerAt:
		triggerAt(s, &tsk, tr)
	}

	log.Infof("task %v created", t.TaskID)

	return &TaskInfo{
		Task: t,
	}, nil
}

func (s *SpaceSyncer) CancelTask(taskID jcstypes.SpaceSyncTaskID) {
	log := logger.WithField("Mod", logMod)

	if !s.cluster.IsMaster() {
		return
	}

	s.lock.Lock()
	defer s.lock.Unlock()

	t := s.tasks[taskID]
	if t == nil {
		log.Infof("task %v not found, cancel aborted", taskID)
		return
	}

	t.CancelFn()
	delete(s.tasks, taskID)

	err := s.db.SpaceSyncTask().Delete(s.db.DefCtx(), taskID)
	if err != nil {
		log.Warnf("delete task %v from db: %v", taskID, err)
	}

	log.Infof("task %v canceled", taskID)
}

func (s *SpaceSyncer) GetTask(taskID jcstypes.SpaceSyncTaskID) *jcstypes.SpaceSyncTask {
	s.lock.Lock()
	defer s.lock.Unlock()

	if !s.cluster.IsMaster() {
		return nil
	}

	tsk := s.tasks[taskID]
	if tsk == nil {
		return nil
	}

	// TODO 考虑复制一份状态，防止修改
	t := tsk.Task
	return &t
}

type TaskInfo struct {
	Task jcstypes.SpaceSyncTask
}

type task struct {
	Task     jcstypes.SpaceSyncTask
	Context  context.Context
	CancelFn func()
}