package com.whyc.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.mapper.TaskMapper;
import com.whyc.pojo.db_wms.Task;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class TaskService {

    @Resource
    private TaskMapper mapper;

    public void add(Task task) {
        mapper.insert(task);
    }

    public void update(Task task) {
        UpdateWrapper<Task> update = Wrappers.update();
        update.eq("task_no",task.getTaskNo());
        mapper.update(task,update);
    }

    public List<Task> getFinishedList() {
        QueryWrapper<Task> query = Wrappers.query();
        query.isNotNull("end_time");
        return mapper.selectList(query);
    }

    public List<Task> getOngoingList() {
        QueryWrapper<Task> query = Wrappers.query();
        query.isNull("end_time");
        return mapper.selectList(query);
    }
}