package main

import (
	"context"
	"database/sql"
	"fmt"
	"gitee.com/aywc_1/aywc_judge/src/appinit"
	"gitee.com/aywc_1/aywc_judge/src/configuration"
	"gitee.com/aywc_1/aywc_judge/src/models"
	"github.com/jmoiron/sqlx"
	"github.com/robfig/cron/v3"
	log "github.com/sirupsen/logrus"
	"os"
)

var MyCron *cron.Cron
var uploadPath = ""
var processPath = ""
var configFilePath = ""

const DeviceWarmProcessSql = "select id,upload_file_name,create_time from device_warm  where status=0 and TIMESTAMPDIFF(SECOND,create_time,now())>? limit 100   "

func initCron() error {
	expired_check := configuration.GetYamlValue([]string{"config", "business", "expired_picture_process", "cron"}, 0, "0/5 * * * * *").(string)
	MyCron = cron.New(cron.WithSeconds()) //支持秒级定时器
	_, err := MyCron.AddFunc(expired_check, DeleteExpirePicture)
	if err != nil {
		log.Error(err)
		return err
	}

	return nil
}

var islock = false

func DeleteExpirePicture() {
	fmt.Println(islock)
	if islock {
		log.Info("locked.return.......")
		return
	}
	tx, err := appinit.GetDB().BeginTxx(context.Background(), nil)
	if err != nil {
		log.Error("事务失败:", err)
		return
	}
	islock = true     //加锁
	defer clearTx(tx) //清理事务
	expired_picture := configuration.GetYamlValue([]string{"config", "business", "expired_picture_process", "expired"}, 0, 86400).(int)
	uploadPath := configuration.GetYamlValue([]string{"upload_path"}, 0, "/upload").(string)
	rows, err := tx.Queryx(DeviceWarmProcessSql, expired_picture)
	if err != nil {
		tx.Rollback()
		return
	}
	defer rows.Close()
	tms := []models.DeviceWarmModel{}
	_ = sqlx.StructScan(rows, &tms)
	for _, tm := range tms {
		sourcePath := uploadPath + "/" + tm.Id + "_" + tm.UploadFileName
		_, err = tx.Exec("delete from  device_warm  where id=?  ", tm.Id)
		if err != nil {
			tx.Rollback()
		}
		err := os.Remove(sourcePath)
		if err != nil {
			log.Errorf("Failed to remove file,err:%s\n", err.Error())
			continue
		}

	}

}

func clearTx(tx *sqlx.Tx) {
	err := tx.Commit()
	if err != nil && err != sql.ErrTxDone {
		log.Println("tx error:", err)
	}
	islock = false
}

func main() {
	configuration.InitLog("check")
	log.Info("Starting check cron application...")
	c := make(chan error)
	go func() {
		err := appinit.DBInit()
		if err != nil {
			c <- err
		}

	}()
	go func() {
		err := initCron()
		if err != nil {
			c <- err
		}
		MyCron.Start() //开启定时任务
	}()
	err := <-c
	log.Fatal(err)
}