package com.peanut.modules.job.task;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.peanut.modules.common.entity.MyUserEntity;
import com.peanut.modules.common.entity.UserVip;
import com.peanut.modules.common.service.MyUserService;
import com.peanut.modules.master.service.UserVipService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.List;

@Component("userVipTask")
public class UserVipTask implements ITask{

    @Autowired
    private UserVipService vipService;
    @Autowired
    private MyUserService userService;

    @Override
    public void run(String params) {
        LambdaQueryWrapper<UserVip> wrapper = new LambdaQueryWrapper();
        wrapper.eq(UserVip::getState,0);
        List<UserVip> list = vipService.list(wrapper);
        if (list.size() > 0) {
            for (UserVip userVip : list) {
                if (userVip.getEndTime().getTime()<new Date().getTime()){
                    userVip.setState(1);
                    vipService.saveOrUpdate(userVip);
                    MyUserEntity user = userService.getById(userVip.getUserId());
                    user.setVip("0");
                    userService.saveOrUpdate(user);
                }
            }
        }
    }

}