freedom-admin/src/main/java/com/freedom/admin/controller/TodoController.java

package com.freedom.admin.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.freedom.admin.base.BaseController;
import com.freedom.admin.common.Result;
import com.freedom.admin.entity.Todo;
import com.freedom.admin.service.TodoService;
import lombok.extern.log4j.Log4j2;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Date;

/**
 * 控制层-待办清单
 *
 * @author zhangjiaxun
 * @description 多用户待办，数据按当前登录用户严格隔离（user_id 由 token 解析，禁止前端传入）
 */
@RestController
@RequestMapping("todo")
@Log4j2
public class TodoController extends BaseController {

    @Autowired
    private TodoService todoService;

    /**
     * 分页查询当前用户的待办列表
     */
    @RequestMapping("findPageList")
    public Result<Todo> findPageList(long pageNum, long pageSize, Todo todo) {
        Long userId = getUserId();
        QueryWrapper<Todo> queryWrapper = new QueryWrapper<>();
        // 数据隔离：只查当前登录用户的数据
        queryWrapper.eq("user_id", userId);
        queryWrapper.eq("del_flag", 0);
        queryWrapper.like(StringUtils.isNotBlank(todo.getTitle()), "title", todo.getTitle());
        queryWrapper.eq(todo.getStatus() != null, "status", todo.getStatus());
        queryWrapper.eq(todo.getPriority() != null, "priority", todo.getPriority());
        // 未完成的排在前面，同状态按创建时间倒序
        queryWrapper.orderByAsc("status").orderByDesc("create_date");
        Page<Todo> page = todoService.page(new Page<>(pageNum, pageSize), queryWrapper);
        return Result.ok().setData(page);
    }

    /**
     * 新增待办
     */
    @RequestMapping("add")
    public Result add(Todo todo) {
        if (StringUtils.isBlank(todo.getTitle())) {
            return Result.error().message("标题不能为空");
        }
        try {
            Long userId = getUserId();
            todo.setId(null);
            // 归属人一律取当前登录用户，忽略前端可能传入的 userId
            todo.setUserId(userId);
            if (todo.getStatus() == null) {
                todo.setStatus(0);
            }
            if (todo.getPriority() == null) {
                todo.setPriority(2);
            }
            todo.setCreateBy(getUsername());
            todo.setCreateDate(new Date());
            todo.setUpdateBy(getUsername());
            todo.setUpdateDate(new Date());
            boolean save = todoService.save(todo);
            return Result.ok().setData(save);
        } catch (Exception e) {
            log.error("新增待办失败", e);
            return Result.error().message("新增失败");
        }
    }

    /**
     * 更新待办
     */
    @RequestMapping("update")
    public Result update(Todo todo) {
        if (todo.getId() == null) {
            return Result.error().message("参数缺失");
        }
        if (StringUtils.isBlank(todo.getTitle())) {
            return Result.error().message("标题不能为空");
        }
        try {
            Long userId = getUserId();
            todo.setUserId(null);
            todo.setUpdateBy(getUsername());
            todo.setUpdateDate(new Date());
            UpdateWrapper<Todo> updateWrapper = new UpdateWrapper<>();
            // 数据隔离：只允许更新自己的数据
            updateWrapper.eq("id", todo.getId()).eq("user_id", userId).eq("del_flag", 0);
            boolean update = todoService.update(todo, updateWrapper);
            return Result.ok().setData(update);
        } catch (Exception e) {
            log.error("更新待办失败", e);
            return Result.error().message("更新失败");
        }
    }

    /**
     * 勾选/取消完成
     */
    @RequestMapping("updateStatus")
    public Result updateStatus(Todo todo) {
        if (todo.getId() == null || todo.getStatus() == null) {
            return Result.error().message("参数缺失");
        }
        try {
            Long userId = getUserId();
            UpdateWrapper<Todo> updateWrapper = new UpdateWrapper<>();
            updateWrapper.eq("id", todo.getId()).eq("user_id", userId).eq("del_flag", 0)
                    .set("status", todo.getStatus())
                    .set("update_by", getUsername())
                    .set("update_date", new Date());
            boolean update = todoService.update(updateWrapper);
            return Result.ok().setData(update);
        } catch (Exception e) {
            log.error("更新待办状态失败", e);
            return Result.error().message("状态更新失败");
        }
    }

    /**
     * 删除待办（逻辑删除，仅能删除自己的数据）
     */
    @RequestMapping("remove")
    public Result remove(Todo todo) {
        if (todo.getId() == null) {
            return Result.error().message("参数缺失");
        }
        try {
            Long userId = getUserId();
            QueryWrapper<Todo> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("id", todo.getId()).eq("user_id", userId);
            boolean remove = todoService.remove(queryWrapper);
            return Result.ok().setData(remove);
        } catch (Exception e) {
            log.error("删除待办失败", e);
            return Result.error().message("删除失败");
        }
    }

}