feat: 初始化收支管理系统

2026-08-30 14:36:22 +08:00
commit 733ed4bb86
79 changed files with 5808 additions and 0 deletions
@@ -0,0 +1,160 @@
+package com.example.finance.controller;
+
+import com.alibaba.excel.EasyExcel;
+import com.example.finance.common.PageResult;
+import com.example.finance.common.Result;
+import com.example.finance.dto.TransactionQuery;
+import com.example.finance.dto.TransactionRequest;
+import com.example.finance.service.TransactionService;
+import com.example.finance.vo.TransactionExportVO;
+import com.example.finance.vo.TransactionVO;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import jakarta.validation.Valid;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.PutMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 交易记录控制器
+ *
+ * <p>提供交易记录的分页查询、新增、更新、删除接口</p>
+ */
+@RestController
+@RequestMapping("/api/transaction")
+@Tag(name = "交易记录管理", description = "收支交易记录的增删改查相关接口")
+public class TransactionController {
+
+    private final TransactionService transactionService;
+
+    public TransactionController(TransactionService transactionService) {
+        this.transactionService = transactionService;
+    }
+
+    /**
+     * 分页查询交易记录
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param query 分页查询参数
+     * @return 分页交易记录列表
+     */
+    @GetMapping("/page")
+    @Operation(summary = "分页查询交易记录", description = "根据查询条件分页获取当前用户的交易记录")
+    public Result<PageResult<TransactionVO>> page(HttpServletRequest request,
+                                                  @Parameter(description = "分页查询参数") TransactionQuery query) {
+        Long userId = (Long) request.getAttribute("userId");
+        return Result.success(transactionService.page(userId, query));
+    }
+
+    /**
+     * 新增交易记录
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param transactionRequest 交易记录请求参数
+     * @return 操作结果
+     */
+    @PostMapping
+    @Operation(summary = "新增交易记录", description = "为当前用户新增一条交易记录")
+    public Result<Void> create(HttpServletRequest request, @RequestBody @Valid TransactionRequest transactionRequest) {
+        Long userId = (Long) request.getAttribute("userId");
+        transactionService.create(userId, transactionRequest);
+        return Result.success();
+    }
+
+    /**
+     * 更新交易记录
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param id 交易记录ID
+     * @param transactionRequest 交易记录请求参数
+     * @return 操作结果
+     */
+    @PutMapping("/{id}")
+    @Operation(summary = "更新交易记录", description = "根据交易记录ID更新交易信息")
+    public Result<Void> update(HttpServletRequest request,
+                               @Parameter(description = "交易记录ID") @PathVariable Long id,
+                               @RequestBody @Valid TransactionRequest transactionRequest) {
+        Long userId = (Long) request.getAttribute("userId");
+        transactionService.update(userId, id, transactionRequest);
+        return Result.success();
+    }
+
+    /**
+     * 删除交易记录
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param id 交易记录ID
+     * @return 操作结果
+     */
+    @DeleteMapping("/{id}")
+    @Operation(summary = "删除交易记录", description = "根据交易记录ID删除指定交易记录")
+    public Result<Void> delete(HttpServletRequest request,
+                               @Parameter(description = "交易记录ID") @PathVariable Long id) {
+        Long userId = (Long) request.getAttribute("userId");
+        transactionService.delete(userId, id);
+        return Result.success();
+    }
+
+    /**
+     * 切换交易记录的"计入统计"标记
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param id 交易记录ID
+     * @param body 请求体，包含 includeInStats 字段
+     * @return 操作结果
+     */
+    @PutMapping("/{id}/stats")
+    @Operation(summary = "切换计入统计标记", description = "切换交易记录是否计入统计（用于表格中开关）")
+    public Result<Void> toggleStats(HttpServletRequest request,
+                                    @Parameter(description = "交易记录ID") @PathVariable Long id,
+                                    @RequestBody Map<String, Boolean> body) {
+        Long userId = (Long) request.getAttribute("userId");
+        Boolean includeInStats = body != null ? body.get("includeInStats") : null;
+        transactionService.toggleStats(userId, id, includeInStats);
+        return Result.success();
+    }
+
+    /**
+     * 导出交易记录为 Excel 文件
+     *
+     * @param request HTTP请求，从中获取用户ID
+     * @param response HTTP响应，用于输出文件流
+     * @param query 查询条件
+     * @throws IOException 文件输出异常
+     */
+    @GetMapping("/export")
+    @Operation(summary = "导出交易记录", description = "根据查询条件导出交易记录为 Excel 文件")
+    public void export(HttpServletRequest request,
+                       HttpServletResponse response,
+                       @Parameter(description = "查询条件") TransactionQuery query) throws IOException {
+        Long userId = (Long) request.getAttribute("userId");
+
+        // 设置响应头
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
+        response.setCharacterEncoding("utf-8");
+        String fileName = URLEncoder.encode("交易记录_" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd_HHmmss")), StandardCharsets.UTF_8).replaceAll("\\+", "%20");
+        response.setHeader("Content-Disposition", "attachment;filename=" + fileName + ".xlsx");
+
+        // 查询数据并写入 Excel
+        List<TransactionExportVO> data = transactionService.export(userId, query);
+        EasyExcel.write(response.getOutputStream(), TransactionExportVO.class)
+                .sheet("交易记录")
+                .doWrite(data);
+    }
+}