package service

import (
	"context"
	"fmt"
	"strconv"
	"sync"

	"gitee.com/linqwen/momo/sys/job/model"

	"gitee.com/linqwen/momo/app"
)

type scheduleService struct {
	app.BaseService[model.ScheduleEntity]
}

func NewScheduleService() *scheduleService {
	return &scheduleService{BaseService: *app.NewBaseService(JobDb, model.ScheduleEntity{})}
}

var (
	taskCancelFuncs = make(map[int64]context.CancelFunc)
	taskLock        sync.Mutex
)

func (bs scheduleService) RunTask(id string) error {
	// var schedule model.ScheduleEntity
	// if err := JobDb.Preload("Task").First(&schedule, id).Error; err != nil {
	// 	return err
	// }
	// ctx, cancel := context.WithCancel(context.Background())
	// taskLock.Lock()
	// taskCancelFuncs[schedule.Id] = cancel
	// taskLock.Unlock()
	// if err := schedule.RunTask(ctx); err != nil {
	// 	return err
	// }
	return nil
}

func (bs scheduleService) EndTask(id string) error {
	taskId, err := strconv.ParseInt(id, 16, 64)
	if err != nil {
		fmt.Printf("转换出错: %v\n", err)
		return err
	}
	taskLock.Lock()
	if cancel, ok := taskCancelFuncs[taskId]; ok {
		cancel()
		delete(taskCancelFuncs, taskId)
		taskLock.Unlock()
		return nil
	} else {
		return fmt.Errorf("endtask error")
	}

}