package db

import "gitlink.org.cn/cloudream/jcs-pub/client/types"

type SpaceSyncTaskDB struct {
	*DB
}

func (db *DB) SpaceSyncTask() *SpaceSyncTaskDB {
	return &SpaceSyncTaskDB{db}
}

func (db *SpaceSyncTaskDB) Create(ctx SQLContext, task *types.SpaceSyncTask) error {
	return ctx.Create(task).Error
}

func (db *SpaceSyncTaskDB) GetAll(ctx SQLContext) ([]types.SpaceSyncTask, error) {
	var tasks []types.SpaceSyncTask
	err := ctx.Find(&tasks).Order("TaskID ASC").Error
	return tasks, err
}

func (*SpaceSyncTaskDB) Delete(ctx SQLContext, taskID types.SpaceSyncTaskID) error {
	return ctx.Delete(&types.SpaceSyncTask{}, taskID).Error
}

func (*SpaceSyncTaskDB) BatchDelete(ctx SQLContext, taskIDs []types.SpaceSyncTaskID) error {
	if len(taskIDs) == 0 {
		return nil
	}

	return ctx.Where("TaskID IN (?)", taskIDs).Delete(&types.SpaceSyncTask{}).Error
}