package com.mzl.flower.service.impl;

import cn.hutool.core.util.IdUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.mzl.flower.entity.CalendarDO;
import com.mzl.flower.mapper.CalendarMapper;
import com.mzl.flower.service.calendar.CalendarService;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author @TaoJie
 * @since 2024-09-12
 */
@Service
public class CalendarServiceImpl extends ServiceImpl<CalendarMapper, CalendarDO> implements CalendarService {

    @Override
    public void generateCalendar(int year) {
        List<CalendarDO> CalendarDOList = new ArrayList<>();
        LocalDateTime startDate = LocalDateTime.of(year, 1, 1, 0, 0);
        LocalDateTime endDate = LocalDateTime.of(year, 12, 31, 23, 59);


        //删除年的数据
        QueryWrapper<CalendarDO> queryWrapper=new QueryWrapper();
        queryWrapper.lambda().eq(CalendarDO::getCalYear,year);
        baseMapper.delete(queryWrapper);

        for (LocalDateTime date = startDate; !date.isAfter(endDate); date = date.plusDays(1)) {
            CalendarDO CalendarDO = new CalendarDO();
            CalendarDO.setId(IdUtil.simpleUUID());
            CalendarDO.setCalYear(String.valueOf(date.getYear()));
            CalendarDO.setCalMonth(String.valueOf(date.getMonthValue()));
            CalendarDO.setCalDay(String.valueOf(date.getDayOfMonth()));
            CalendarDO.setCalDate(date);
            CalendarDO.setCalWeek(String.valueOf(date.getDayOfWeek().getValue()));
            CalendarDO.setType(0);
            CalendarDO.setCreateTime(LocalDateTime.now());
            CalendarDO.setUpdateTime(LocalDateTime.now());
            CalendarDOList.add(CalendarDO);

            // 重新生成年的数据
            baseMapper.insert(CalendarDO);
        }

    }
}