linwe
2023-11-09 feb3872fdd6c7bd59b8f3969a85978fefc7130b3
src/main/java/org/springblade/modules/task/service/impl/TaskServiceImpl.java
@@ -17,7 +17,9 @@
package org.springblade.modules.task.service.impl;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.modules.task.entity.TaskEntity;
import org.springblade.modules.task.mapper.TaskMapper;
import org.springblade.modules.task.service.ITaskService;
@@ -25,6 +27,8 @@
import org.springframework.stereotype.Service;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
/**
 * 任务表 服务实现类
@@ -34,6 +38,7 @@
 */
@Service
public class TaskServiceImpl extends BaseServiceImpl<TaskMapper, TaskEntity> implements ITaskService {
   @Override
   public IPage<TaskVO> selectTaskPage(IPage<TaskVO> page, TaskVO task) {
@@ -51,7 +56,7 @@
    * @return
    */
   @Override
   public Long saveTask(Integer type, String name, Integer frequency, String remark, Long createUser) {
   public Long saveTask(Integer type, String name, Integer frequency, String remark, Long createUser, String houseCode) {
      TaskEntity taskEntity = new TaskEntity();
      taskEntity.setType(type);
      taskEntity.setName(name);
@@ -59,6 +64,7 @@
      taskEntity.setRemark(remark);
      taskEntity.setCreateTime(new Date());
      taskEntity.setCreateUser(createUser);
      taskEntity.setHouseCode(houseCode);
      return baseMapper.insert(taskEntity) > 0 ? taskEntity.getId() : 0;
   }
@@ -75,4 +81,18 @@
      taskEntity.setStatus(status);
      return baseMapper.updateById(taskEntity) > 0 ? 1L : 0;
   }
   @Override
   public Object countNumber(String houseCode, Integer status) {
      Map<String, Object> objectObjectHashMap = new HashMap<>();
      // 总数
      Long all = baseMapper.selectCount(Wrappers.<TaskEntity>lambdaQuery().eq(TaskEntity::getCreateUser, AuthUtil.getUserId() ));
      // 已处理
      Long processed = baseMapper.selectCount(Wrappers.<TaskEntity>lambdaQuery().eq(TaskEntity::getCreateUser, AuthUtil.getUserId())
         .eq(TaskEntity::getStatus, 2));
      objectObjectHashMap.put("all", all);
      objectObjectHashMap.put("processed", processed);
      return objectObjectHashMap;
   }
}