package com.casic.config.task;

import com.casic.config.TimeConfig;
import com.casic.service.DayDataService;
import com.casic.service.HourDataService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.SchedulingConfigurer;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.config.TriggerTask;
import org.springframework.scheduling.support.CronTrigger;

@Configuration
@EnableScheduling
public class TaskConfigurer implements SchedulingConfigurer {

    @Autowired
    private HourDataService hourDataProvider;
    @Autowired
    private DayDataService dayDataProvider;
    @Autowired
    private TimeConfig timeConfig;

    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
        TriggerTask triggrtTask = new TriggerTask(nextDayDevice(),
                triggerContext -> {
                    return new CronTrigger(timeConfig.getDayCronTime()).nextExecutionTime(triggerContext);
                });
        taskRegistrar.addTriggerTask(triggrtTask);
        triggrtTask = new TriggerTask(nextHourDevice(),
                triggerContext -> {
                    return new CronTrigger(timeConfig.getHourCronTime()).nextExecutionTime(triggerContext);
                });
        taskRegistrar.addTriggerTask(triggrtTask);
    }

    private Runnable nextDayDevice() {
        return new Runnable() {
            @Override
            public void run() {
                dayDataProvider.nextDayData();
            }
        };
    }

    private Runnable nextHourDevice() {
        return new Runnable() {
            @Override
            public void run() {
                hourDataProvider.nextHourData();
            }
        };
    }

}