Compare commits

...

10 Commits

Author SHA1 Message Date
ikmkj
46ac106f9b chore: 更新.gitignore配置 2026-03-07 15:51:26 +08:00
ikmkj
e495011f57 docs: 添加反馈系统功能分析文档
添加未实现的反馈系统功能分析文档,包含需求概述、权限矩阵、数据表设计、业务逻辑、API接口、前端UI设计、实现步骤和关键技术点等详细说明
2026-03-04 18:40:34 +08:00
ikmkj
23ced99e20 feat: 添加XSS防护工具函数并优化多个功能模块
refactor(前端): 重构登录页面和用户状态管理逻辑

fix(后端): 修复用户密码更新逻辑和错误提示

feat(后端): 实现分页搜索功能并优化文件删除逻辑

perf(前端): 优化笔记编辑器自动保存和状态提示

fix(后端): 增强图片上传安全验证

style(前端): 调整笔记预览页面按钮权限控制

chore: 更新生产环境配置和测试数据库依赖

feat(前端): 添加虚拟列表组件优化性能

fix(前端): 修复笔记编辑器返回逻辑和状态保存

refactor(前端): 优化axios拦截器错误处理逻辑

docs: 更新方法注释和参数说明
2026-03-04 18:29:52 +08:00
ikmkj
5ea9c776e7 feat(笔记预览): 实现大文件分块加载功能
添加分块加载API接口及前端实现,支持大文件(>500KB)的分页加载,提升大文件预览体验
后端实现分块逻辑并添加权限检查,前端添加加载提示和滚动加载功能
2026-03-04 16:43:25 +08:00
ikmkj
90626e73d9 refactor: 统一ID类型为Long并优化代码安全性和性能
修复用户删除接口的ID类型从Integer改为Long
移除未使用的Bouncy Castle依赖
添加dompurify依赖增强XSS防护
修复SQL表定义中的bigintBIGINT语法错误
优化图片预览接口的安全检查和错误处理
添加Vditor渲染引擎预加载和图片懒加载
统一分组和文件接口的ID类型为Long
增强前端用户状态管理,添加token过期检查
优化Markdown内容渲染流程和图片URL处理
2026-03-04 16:27:42 +08:00
ikmkj
25b52f87aa refactor: 统一错误处理并优化代码
- 移除重复的错误提示,统一在axios拦截器中处理
- 优化XSS拦截器,添加请求头白名单
- 修复注册码服务的日期处理问题
- 添加403权限错误处理
- 优化分组查询参数处理
2026-03-03 23:41:20 +08:00
ikmkj
a4f95e7315 feat(用户管理): 添加用户角色功能并实现权限控制
- 在用户表中添加role字段并设置默认值为'USER'
- 前端添加isAdmin getter判断用户角色
- 后端实现角色字段的VO映射和默认值设置
- 为关键接口添加@PreAuthorize权限控制
- 移除图片控制器中冗余的权限检查代码
2026-03-03 21:09:42 +08:00
ikmkj
375ccb89ff feat: 添加用户角色字段并实现权限控制
fix(security): 修复重放攻击拦截器的时间戳验证漏洞

refactor(security): 重构验证码工具类使用线程安全实现

perf(login): 优化登录锁定工具类性能并添加定期清理

fix(editor): 修复笔记编辑器空指针问题

style: 清理数据库索引脚本中的冗余注释

fix(api): 修复前端API调用参数编码问题

feat(image): 实现图片名称同步服务

refactor(markdown): 重构Markdown服务分离图片名称同步逻辑

fix(xss): 添加HTML转义函数防止XSS攻击

fix(user): 修复用户服务权限加载问题

fix(rate-limit): 修复速率限制拦截器并发问题

fix(axios): 生产环境隐藏详细错误信息

fix(image): 修复图片上传和删除的权限验证

refactor(captcha): 重构验证码工具类使用并发安全实现

fix(jwt): 修复JWT过滤器空指针问题

fix(export): 修复笔记导出XSS漏洞

fix(search): 修复Markdown搜索SQL注入问题

fix(interceptor): 修复重放攻击拦截器逻辑错误

fix(controller): 修复用户控制器空指针问题

fix(security): 修复nonce生成使用密码学安全方法
2026-03-03 20:48:40 +08:00
ikmkj
d54719d82d perf: 优化系统性能和资源管理
- 优化JWT解析器性能,复用JwtParser减少重复创建
- 使用Slf4j日志框架替代System.out.println
- 添加图片压缩功能,减少存储和带宽消耗
- 预编译正则表达式提升Markdown图片提取性能
- 重构限流拦截器,使用ConcurrentHashMap提高并发性能
- 添加数据库索引优化查询性能
- 为MarkdownFile实体添加autoResultMap配置
2026-03-03 19:14:48 +08:00
ikmkj
64daf3cb0b fix(XSSInterceptor): 增强XSS拦截器对请求头的处理
修改XSS拦截器对请求头的检查逻辑,当检测到XSS攻击时直接拒绝请求并返回400错误响应。新增错误响应写入方法,记录攻击日志。
2026-03-03 18:58:03 +08:00
65 changed files with 2137 additions and 587 deletions

View File

@@ -67,6 +67,13 @@
<artifactId>spring-boot-starter-test</artifactId> <artifactId>spring-boot-starter-test</artifactId>
</dependency> </dependency>
<!-- H2 数据库(测试用) -->
<dependency>
<groupId>com.h2database</groupId>
<artifactId>h2</artifactId>
<scope>test</scope>
</dependency>
<dependency> <dependency>
<groupId>org.springframework.boot</groupId> <groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId> <artifactId>spring-boot-starter-aop</artifactId>
@@ -126,13 +133,6 @@
<groupId>org.springframework.boot</groupId> <groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-validation</artifactId> <artifactId>spring-boot-starter-validation</artifactId>
</dependency> </dependency>
<!-- Bouncy Castle 加密库 -->
<dependency>
<groupId>org.bouncycastle</groupId>
<artifactId>bcprov-jdk18on</artifactId>
<version>1.76</version>
</dependency>
</dependencies> </dependencies>
<build> <build>

View File

@@ -7,6 +7,7 @@ import com.test.bijihoudaun.common.response.ResultCode;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import org.springframework.security.authorization.AuthorizationDeniedException;
import org.springframework.validation.BindException; import org.springframework.validation.BindException;
import org.springframework.validation.FieldError; import org.springframework.validation.FieldError;
import org.springframework.web.bind.MethodArgumentNotValidException; import org.springframework.web.bind.MethodArgumentNotValidException;
@@ -71,9 +72,28 @@ public class GlobalExceptionHandler {
return R.fail(ResultCode.VALIDATE_FAILED.getCode(), "参数错误"); return R.fail(ResultCode.VALIDATE_FAILED.getCode(), "参数错误");
} }
// 修复:添加权限拒绝异常处理
@ExceptionHandler(AuthorizationDeniedException.class)
public R<Void> handleAuthorizationDeniedException(AuthorizationDeniedException e, HttpServletRequest request) {
log.warn("Access denied at {}: {}", request.getRequestURI(), e.getMessage());
return R.fail(ResultCode.FORBIDDEN.getCode(), "无权操作,需要管理员权限");
}
@ExceptionHandler(Exception.class) @ExceptionHandler(Exception.class)
public R<Void> handleException(Exception e, HttpServletRequest request) { public R<Void> handleException(Exception e, HttpServletRequest request) {
log.error("Unexpected error at {} - Error type: {}", request.getRequestURI(), e.getClass().getSimpleName()); log.error("Unexpected error at {} - Error type: {}", request.getRequestURI(), e.getClass().getSimpleName());
return R.fail(ResultCode.FAILED.getCode(), "系统繁忙,请稍后再试"); return R.fail(ResultCode.FAILED.getCode(), "系统繁忙,请稍后再试");
} }
// 修复:处理 IllegalStateExceptiongetWriter() 已调用等问题)
@ExceptionHandler(IllegalStateException.class)
public R<Void> handleIllegalStateException(IllegalStateException e, HttpServletRequest request) {
log.warn("Illegal state at {}: {}", request.getRequestURI(), e.getMessage());
// 如果是图片预览相关请求,可能是响应已经提交
if (request.getRequestURI().contains("/api/images/preview")) {
// 响应可能已经提交,直接返回 null避免再次写入响应
return null;
}
return R.fail(ResultCode.FAILED.getCode(), "请求处理失败");
}
} }

View File

@@ -9,6 +9,7 @@ import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment; import org.springframework.core.env.Environment;
import org.springframework.security.config.annotation.method.configuration.EnableMethodSecurity;
import org.springframework.security.config.annotation.web.builders.HttpSecurity; import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity; import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;
import org.springframework.security.config.http.SessionCreationPolicy; import org.springframework.security.config.http.SessionCreationPolicy;
@@ -22,6 +23,7 @@ import java.util.Arrays;
@Configuration @Configuration
@EnableWebSecurity @EnableWebSecurity
@EnableMethodSecurity(prePostEnabled = true)
public class SecurityConfig { public class SecurityConfig {
@Autowired @Autowired

View File

@@ -10,6 +10,7 @@ import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters; import io.swagger.v3.oas.annotations.Parameters;
import io.swagger.v3.oas.annotations.tags.Tag; import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import java.util.List; import java.util.List;
@@ -23,6 +24,7 @@ public class GroupingController {
private GroupingService groupingService; private GroupingService groupingService;
@Operation(summary = "创建分组") @Operation(summary = "创建分组")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping @PostMapping
public R<Grouping> createGrouping(@RequestBody Grouping grouping) { public R<Grouping> createGrouping(@RequestBody Grouping grouping) {
if (ObjectUtil.isNull(grouping.getParentId())) { if (ObjectUtil.isNull(grouping.getParentId())) {
@@ -47,22 +49,21 @@ public class GroupingController {
} }
@Operation(summary = "更新分组名称") @Operation(summary = "更新分组名称")
@PreAuthorize("hasRole('ADMIN')")
@PutMapping("/{id}") @PutMapping("/{id}")
public R<Grouping> updateGrouping( public R<Grouping> updateGrouping(
@PathVariable String id, @PathVariable Long id,
@RequestBody Grouping grouping) { @RequestBody Grouping grouping) {
grouping.setId(id);
long l = Long.parseLong(id);
grouping.setId(l);
Grouping updated = groupingService.updateGrouping(grouping); Grouping updated = groupingService.updateGrouping(grouping);
return R.success(updated); return R.success(updated);
} }
@Operation(summary = "删除分组") @Operation(summary = "删除分组")
@PreAuthorize("hasRole('ADMIN')")
@DeleteMapping("/{id}") @DeleteMapping("/{id}")
public R<Void> deleteGrouping(@PathVariable String id) { public R<Void> deleteGrouping(@PathVariable Long id) {
Long idLong = Long.parseLong(id); groupingService.deleteGrouping(id);
groupingService.deleteGrouping(idLong);
return R.success(); return R.success();
} }
} }

View File

@@ -4,13 +4,18 @@ package com.test.bijihoudaun.controller;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import com.test.bijihoudaun.common.response.R; import com.test.bijihoudaun.common.response.R;
import com.test.bijihoudaun.entity.Image; import com.test.bijihoudaun.entity.Image;
import com.test.bijihoudaun.entity.User;
import com.test.bijihoudaun.service.ImageService; import com.test.bijihoudaun.service.ImageService;
import com.test.bijihoudaun.service.UserService;
import com.test.bijihoudaun.util.SecurityUtil; import com.test.bijihoudaun.util.SecurityUtil;
import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag; import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.util.StreamUtils; import org.springframework.util.StreamUtils;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartFile;
@@ -18,6 +23,7 @@ import org.springframework.web.multipart.MultipartFile;
import java.io.File; import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.IOException; import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.util.List; import java.util.List;
@@ -33,40 +39,53 @@ public class ImageController {
@Autowired @Autowired
private ImageService imageService; private ImageService imageService;
@Autowired
private UserService userService;
@Operation(summary = "上传图片") @Operation(summary = "上传图片")
@PostMapping @PostMapping
public R<Image> uploadImage( public R<Image> uploadImage(
@RequestPart("file") MultipartFile file, @RequestPart("file") MultipartFile file,
@RequestParam(value = "userId", required = false) Long userId,
@RequestParam(value = "markdownId", required = false) Long markdownId) { @RequestParam(value = "markdownId", required = false) Long markdownId) {
// 修复从SecurityContext获取当前用户不接受客户端传入的userId
Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
if (!(principal instanceof UserDetails)) {
return R.fail("请先登录");
}
String username = ((UserDetails) principal).getUsername();
User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username));
if (user == null) {
return R.fail("用户不存在");
}
try { try {
Image image = imageService.uploadImage(file, userId, markdownId); Image image = imageService.uploadImage(file, user.getId(), markdownId);
return R.success(image); return R.success(image);
} catch (IOException e) { } catch (IOException e) {
return R.fail(); return R.fail("上传失败:" + e.getMessage());
} }
} }
@Operation(summary = "根据id删除图片") @Operation(summary = "根据id删除图片")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping("/{id}") @PostMapping("/{id}")
public R<Void> deleteImage(@PathVariable Long id) { public R<Void> deleteImage(@PathVariable Long id) {
if (!SecurityUtil.isUserAuthenticated()) {
return R.fail("请先登录");
}
boolean result = imageService.deleteImage(id); boolean result = imageService.deleteImage(id);
if (result) { if (result) {
return R.success(); return R.success();
} else { } else {
return R.fail(); return R.fail("删除失败");
} }
} }
@Operation(summary = "在线预览", description = "浏览器直接打开文件流") @Operation(summary = "在线预览", description = "浏览器直接打开文件流")
@GetMapping("/preview/{url}") @GetMapping("/preview/{url}")
public void preview(@PathVariable String url, HttpServletResponse resp) throws IOException { public void preview(@PathVariable String url, HttpServletResponse resp) throws IOException {
// 注意:不能同时调用 resp.getWriter() 和 resp.getOutputStream()
if (StrUtil.isBlank(url)) { if (StrUtil.isBlank(url)) {
resp.setStatus(404); resp.setStatus(404);
resp.setContentType("application/json;charset=UTF-8");
resp.getWriter().write("{\"code\":404,\"msg\":\"文件不存在\",\"data\":null}"); resp.getWriter().write("{\"code\":404,\"msg\":\"文件不存在\",\"data\":null}");
return; return;
} }
@@ -74,6 +93,7 @@ public class ImageController {
String sanitizedUrl = sanitizeFileName(url); String sanitizedUrl = sanitizeFileName(url);
if (sanitizedUrl == null) { if (sanitizedUrl == null) {
resp.setStatus(403); resp.setStatus(403);
resp.setContentType("application/json;charset=UTF-8");
resp.getWriter().write("{\"code\":403,\"msg\":\"非法文件路径\",\"data\":null}"); resp.getWriter().write("{\"code\":403,\"msg\":\"非法文件路径\",\"data\":null}");
return; return;
} }
@@ -83,6 +103,7 @@ public class ImageController {
if (!filePath.startsWith(basePath)) { if (!filePath.startsWith(basePath)) {
resp.setStatus(403); resp.setStatus(403);
resp.setContentType("application/json;charset=UTF-8");
resp.getWriter().write("{\"code\":403,\"msg\":\"非法文件路径\",\"data\":null}"); resp.getWriter().write("{\"code\":403,\"msg\":\"非法文件路径\",\"data\":null}");
return; return;
} }
@@ -90,6 +111,7 @@ public class ImageController {
File file = filePath.toFile(); File file = filePath.toFile();
if (!file.exists() || !file.isFile()) { if (!file.exists() || !file.isFile()) {
resp.setStatus(404); resp.setStatus(404);
resp.setContentType("application/json;charset=UTF-8");
resp.getWriter().write("{\"code\":404,\"msg\":\"文件不存在\",\"data\":null}"); resp.getWriter().write("{\"code\":404,\"msg\":\"文件不存在\",\"data\":null}");
return; return;
} }
@@ -97,8 +119,10 @@ public class ImageController {
String contentTypeFromFileExtension = getContentTypeFromFileExtension(url); String contentTypeFromFileExtension = getContentTypeFromFileExtension(url);
resp.setContentType(contentTypeFromFileExtension); resp.setContentType(contentTypeFromFileExtension);
resp.setContentLengthLong(file.length()); resp.setContentLengthLong(file.length());
// 文件流直接输出,不使用 try-with-resources 包装整个方法
try (FileInputStream in = new FileInputStream(file)) { try (FileInputStream in = new FileInputStream(file)) {
StreamUtils.copy(in, resp.getOutputStream()); StreamUtils.copy(in, resp.getOutputStream());
resp.getOutputStream().flush();
} }
} }
@@ -106,7 +130,9 @@ public class ImageController {
if (StrUtil.isBlank(fileName)) { if (StrUtil.isBlank(fileName)) {
return null; return null;
} }
if (fileName.contains("..") || fileName.contains("/") || fileName.contains("\\") || fileName.contains(":")) { // 修复:使用白名单验证文件名格式(只允许 UUID 格式)
// 例如550e8400-e29b-41d4-a716-446655440000.jpg
if (!fileName.matches("^[a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}\\.[a-zA-Z0-9]+$")) {
return null; return null;
} }
return fileName; return fileName;
@@ -114,40 +140,36 @@ public class ImageController {
@Operation(summary = "根据url删除图片") @Operation(summary = "根据url删除图片")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping("/deleteByUrl") @PostMapping("/deleteByUrl")
public R<Void> deleteImageByUrl(@RequestParam String url) { public R<Void> deleteImageByUrl(@RequestParam String url) {
if (!SecurityUtil.isUserAuthenticated()) {
return R.fail("请先登录");
}
boolean result = imageService.deleteImageByUrl(url); boolean result = imageService.deleteImageByUrl(url);
if (result) { if (result) {
return R.success(); return R.success();
} else { } else {
return R.fail(); return R.fail("删除失败");
} }
} }
@Operation(summary = "根据url批量删除图片") @Operation(summary = "根据url批量删除图片")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping("/batch") @PostMapping("/batch")
public R<Void> deleteImageByUrls(@RequestBody List<String> urls) { public R<Void> deleteImageByUrls(@RequestBody List<String> urls) {
if (!SecurityUtil.isUserAuthenticated()) {
return R.fail("请先登录");
}
boolean result = imageService.deleteImageByUrls(urls); boolean result = imageService.deleteImageByUrls(urls);
if (result) { if (result) {
return R.success(); return R.success();
} else { } else {
return R.fail(); return R.fail("删除失败");
} }
} }
private String getContentTypeFromFileExtension(String fileName) { private String getContentTypeFromFileExtension(String fileName) {
if (StrUtil.isBlank(fileName) || !StrUtil.contains(fileName, '.')) { if (StrUtil.isBlank(fileName) || !StrUtil.contains(fileName, '.')) {
return "application/octet-stream"; return "application/octet-stream";
} }
String extension = fileName.substring(fileName.lastIndexOf('.') + 1).toLowerCase(); String extension = fileName.substring(fileName.lastIndexOf('.') + 1).toLowerCase();
// 使用更安全的 MIME 类型映射,只允许图片类型
switch (extension) { switch (extension) {
case "jpg": case "jpg":
case "jpeg": case "jpeg":
@@ -163,8 +185,27 @@ public class ImageController {
case "svg": case "svg":
return "image/svg+xml"; return "image/svg+xml";
default: default:
// 对于未知的扩展名,返回通用的二进制流类型,避免执行风险
return "application/octet-stream"; return "application/octet-stream";
} }
} }
/**
* 验证文件是否为允许的图片类型
* @param contentType 文件内容类型
* @return 是否允许
*/
private boolean isAllowedImageType(String contentType) {
if (StrUtil.isBlank(contentType)) {
return false;
}
// 只允许标准的图片 MIME 类型
return contentType.equals("image/jpeg") ||
contentType.equals("image/png") ||
contentType.equals("image/gif") ||
contentType.equals("image/bmp") ||
contentType.equals("image/webp") ||
contentType.equals("image/svg+xml");
}
} }

View File

@@ -3,6 +3,7 @@ package com.test.bijihoudaun.controller;
import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ObjectUtil;
import com.test.bijihoudaun.common.response.R; import com.test.bijihoudaun.common.response.R;
import com.test.bijihoudaun.entity.MarkdownFile; import com.test.bijihoudaun.entity.MarkdownFile;
import com.test.bijihoudaun.entity.MarkdownFileChunk;
import com.test.bijihoudaun.entity.MarkdownFileVO; import com.test.bijihoudaun.entity.MarkdownFileVO;
import com.test.bijihoudaun.service.MarkdownFileService; import com.test.bijihoudaun.service.MarkdownFileService;
import com.test.bijihoudaun.util.SecurityUtil; import com.test.bijihoudaun.util.SecurityUtil;
@@ -12,6 +13,7 @@ import io.swagger.v3.oas.annotations.Parameters;
import io.swagger.v3.oas.annotations.tags.Tag; import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile; import org.springframework.context.annotation.Profile;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import java.util.Date; import java.util.Date;
@@ -55,6 +57,7 @@ public class MarkdownController {
@Operation(summary = "更新Markdown文件") @Operation(summary = "更新Markdown文件")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping("/updateMarkdown") @PostMapping("/updateMarkdown")
public R<MarkdownFile> updateMarkdown(@RequestBody MarkdownFile markdownFile) { public R<MarkdownFile> updateMarkdown(@RequestBody MarkdownFile markdownFile) {
MarkdownFile file = markdownFileService.updateMarkdownContent(markdownFile); MarkdownFile file = markdownFileService.updateMarkdownContent(markdownFile);
@@ -69,6 +72,7 @@ public class MarkdownController {
} }
@Operation(summary = "删除Markdown文件") @Operation(summary = "删除Markdown文件")
@PreAuthorize("hasRole('ADMIN')")
@Parameters({ @Parameters({
@Parameter(name = "id", description = "Markdown文件ID", required = true), @Parameter(name = "id", description = "Markdown文件ID", required = true),
}) })
@@ -82,23 +86,29 @@ public class MarkdownController {
@Operation(summary = "根据分组ID获取Markdown文件") @Operation(summary = "根据分组ID获取Markdown文件")
@GetMapping("/grouping/{groupingId}") @GetMapping("/grouping/{groupingId}")
public R<List<MarkdownFileVO>> getFilesByGroupingId(@PathVariable String groupingId) { public R<List<MarkdownFileVO>> getFilesByGroupingId(@PathVariable Long groupingId) {
List<MarkdownFileVO> files = markdownFileService.getFilesByGroupingId(groupingId); List<MarkdownFileVO> files = markdownFileService.getFilesByGroupingId(groupingId);
return R.success(files); return R.success(files);
} }
@Operation(summary = "根据标题模糊搜索") @Operation(summary = "根据标题模糊搜索")
@GetMapping("/search") @GetMapping("/search")
public R<List<MarkdownFile>> searchByTitle(@RequestParam String keyword) { public R<List<MarkdownFile>> searchByTitle(
List<MarkdownFile> files = markdownFileService.searchByTitle(keyword); @RequestParam String keyword,
@RequestParam(defaultValue = "1") int page,
@RequestParam(defaultValue = "20") int pageSize) {
if (page < 1) page = 1;
if (pageSize < 1 || pageSize > 100) pageSize = 20;
List<MarkdownFile> files = markdownFileService.searchByTitle(keyword, page, pageSize);
return R.success(files); return R.success(files);
} }
@Operation(summary = "更新Markdown文件标题") @Operation(summary = "更新Markdown文件标题")
@PreAuthorize("hasRole('ADMIN')")
@PostMapping("/{id}/title") @PostMapping("/{id}/title")
public R<MarkdownFile> updateMarkdownTitle( public R<MarkdownFile> updateMarkdownTitle(
@PathVariable Long id, @PathVariable Long id,
String title) { @RequestParam String title) {
MarkdownFile updatedFile = markdownFileService.updateMarkdownTitle(id, title); MarkdownFile updatedFile = markdownFileService.updateMarkdownTitle(id, title);
if (ObjectUtil.isNotNull(updatedFile)) { if (ObjectUtil.isNotNull(updatedFile)) {
return R.success(updatedFile); return R.success(updatedFile);
@@ -113,4 +123,26 @@ public class MarkdownController {
return R.success(files); return R.success(files);
} }
@Operation(summary = "分块加载Markdown文件内容", description = "用于大文件(> 500KB的分页加载")
@Parameters({
@Parameter(name = "id", description = "文件ID", required = true),
@Parameter(name = "chunkIndex", description = "块索引从0开始", required = false),
@Parameter(name = "chunkSize", description = "块大小字符数默认10000", required = false)
})
@GetMapping("/{id}/chunk")
public R<MarkdownFileChunk> getMarkdownChunk(
@PathVariable Long id,
@RequestParam(defaultValue = "0") int chunkIndex,
@RequestParam(defaultValue = "10000") int chunkSize) {
// 获取当前认证状态
boolean isAuthenticated = SecurityUtil.isUserAuthenticated();
MarkdownFileChunk chunk = markdownFileService.getMarkdownChunk(id, chunkIndex, chunkSize, isAuthenticated);
if (ObjectUtil.isNotNull(chunk)) {
return R.success(chunk);
}
return R.fail("文件未找到");
}
} }

View File

@@ -11,8 +11,6 @@ import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import java.util.Collections;
@RestController @RestController
@RequestMapping("/api/system") @RequestMapping("/api/system")
@Tag(name = "系统管理") @Tag(name = "系统管理")
@@ -31,7 +29,7 @@ public class SystemController {
} }
@PostMapping("/registration/toggle") @PostMapping("/registration/toggle")
@PreAuthorize("isAuthenticated()") @PreAuthorize("hasRole('ADMIN')")
@Operation(summary = "切换注册功能状态") @Operation(summary = "切换注册功能状态")
public R<Void> toggleRegistration(@RequestBody Boolean enabled) { public R<Void> toggleRegistration(@RequestBody Boolean enabled) {
systemSettingService.setRegistrationEnabled(enabled); systemSettingService.setRegistrationEnabled(enabled);
@@ -39,7 +37,7 @@ public class SystemController {
} }
@PostMapping("/registration/generate-code") @PostMapping("/registration/generate-code")
@PreAuthorize("isAuthenticated()") @PreAuthorize("hasRole('ADMIN')")
@Operation(summary = "生成注册码") @Operation(summary = "生成注册码")
public R<String> generateRegistrationCode() { public R<String> generateRegistrationCode() {
Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

View File

@@ -7,6 +7,7 @@ import com.test.bijihoudaun.service.TrashService;
import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag; import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import java.util.List; import java.util.List;
@@ -26,6 +27,7 @@ public class TrashController {
} }
@PostMapping("/restore/{type}/{id}") @PostMapping("/restore/{type}/{id}")
@PreAuthorize("hasRole('ADMIN')")
@Operation(summary = "恢复项目") @Operation(summary = "恢复项目")
public R<Void> restoreItem(@PathVariable String type, @PathVariable String id) { public R<Void> restoreItem(@PathVariable String type, @PathVariable String id) {
trashService.restoreItem(id, type); trashService.restoreItem(id, type);
@@ -33,6 +35,7 @@ public class TrashController {
} }
@DeleteMapping("/permanently/{type}/{id}") @DeleteMapping("/permanently/{type}/{id}")
@PreAuthorize("hasRole('ADMIN')")
@RequireCaptcha("永久删除") @RequireCaptcha("永久删除")
@Operation(summary = "永久删除项目") @Operation(summary = "永久删除项目")
public R<Void> permanentlyDeleteItem(@PathVariable String type, @PathVariable String id) { public R<Void> permanentlyDeleteItem(@PathVariable String type, @PathVariable String id) {
@@ -41,6 +44,7 @@ public class TrashController {
} }
@DeleteMapping("/clean") @DeleteMapping("/clean")
@PreAuthorize("hasRole('ADMIN')")
@RequireCaptcha("清空回收站") @RequireCaptcha("清空回收站")
@Operation(summary = "清空回收站") @Operation(summary = "清空回收站")
public R<Void> cleanTrash() { public R<Void> cleanTrash() {

View File

@@ -52,6 +52,10 @@ public class UserController {
return R.fail("无效或已过期的注册码"); return R.fail("无效或已过期的注册码");
} }
User user = userService.register(username, password, email); User user = userService.register(username, password, email);
// 修复:添加空值检查
if (user == null) {
return R.fail("注册失败,请稍后重试");
}
UserVO userVO = new UserVO(); UserVO userVO = new UserVO();
BeanUtils.copyProperties(user, userVO); BeanUtils.copyProperties(user, userVO);
userVO.setId(String.valueOf(user.getId())); userVO.setId(String.valueOf(user.getId()));
@@ -69,6 +73,11 @@ public class UserController {
String token = userService.login(username, password); String token = userService.login(username, password);
User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username)); User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username));
// 修复:添加空值检查
if (user == null) {
return R.fail("用户不存在");
}
Map<String, Object> result = new HashMap<>(); Map<String, Object> result = new HashMap<>();
result.put("token", token); result.put("token", token);
@@ -76,6 +85,8 @@ public class UserController {
userInfo.put("id", String.valueOf(user.getId())); userInfo.put("id", String.valueOf(user.getId()));
userInfo.put("username", user.getUsername()); userInfo.put("username", user.getUsername());
userInfo.put("email", user.getEmail()); userInfo.put("email", user.getEmail());
String role = user.getRole();
userInfo.put("role", (role != null && !role.isEmpty()) ? role : "USER");
result.put("userInfo", userInfo); result.put("userInfo", userInfo);
return R.success(result); return R.success(result);
@@ -88,7 +99,12 @@ public class UserController {
@RequireCaptcha("删除账号") @RequireCaptcha("删除账号")
@DeleteMapping("/deleteUser") @DeleteMapping("/deleteUser")
public R<String> deleteUser(){ public R<String> deleteUser(){
UserDetails userDetails = (UserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); // 修复:添加类型检查
Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
if (!(principal instanceof UserDetails)) {
return R.fail("无法获取用户信息");
}
UserDetails userDetails = (UserDetails) principal;
String username = userDetails.getUsername(); String username = userDetails.getUsername();
User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username)); User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username));
@@ -96,7 +112,7 @@ public class UserController {
return R.fail("无法获取用户信息,删除失败"); return R.fail("无法获取用户信息,删除失败");
} }
userService.deleteUser(user.getId().intValue()); userService.deleteUser(user.getId());
return R.success("用户删除成功"); return R.success("用户删除成功");
} }
@@ -110,9 +126,20 @@ public class UserController {
@RequireCaptcha("修改密码") @RequireCaptcha("修改密码")
@PutMapping("/password") @PutMapping("/password")
public R<String> updatePassword(@RequestBody UpdatePasswordBo updatePasswordBo) { public R<String> updatePassword(@RequestBody UpdatePasswordBo updatePasswordBo) {
UserDetails userDetails = (UserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); // 修复:添加类型检查
Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
if (!(principal instanceof UserDetails)) {
return R.fail("无法获取用户信息");
}
UserDetails userDetails = (UserDetails) principal;
String username = userDetails.getUsername(); String username = userDetails.getUsername();
User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username)); User user = userService.getOne(new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<User>().eq("username", username));
// 修复:添加空值检查
if (ObjectUtil.isNull(user)) {
return R.fail("用户不存在");
}
userService.updatePassword(user.getId(), updatePasswordBo); userService.updatePassword(user.getId(), updatePasswordBo);
return R.success("密码更新成功"); return R.success("密码更新成功");
} }

View File

@@ -14,7 +14,7 @@ import java.util.Date;
@Data @Data
@Schema(name = "文本实体") @Schema(name = "文本实体")
@TableName("`markdown_file`") @TableName(value = "`markdown_file`", autoResultMap = true)
public class MarkdownFile implements Serializable { public class MarkdownFile implements Serializable {
@Schema(description = "文本id",implementation = Long.class) @Schema(description = "文本id",implementation = Long.class)
@TableId(type = IdType.AUTO) @TableId(type = IdType.AUTO)

View File

@@ -0,0 +1,46 @@
package com.test.bijihoudaun.entity;
import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.io.Serializable;
/**
* Markdown文件分块加载响应实体
* 用于大文件分页加载
*/
@Data
@Schema(name = "Markdown文件分块")
public class MarkdownFileChunk implements Serializable {
private static final long serialVersionUID = 1L;
@Schema(description = "当前块的内容")
private String chunk;
@Schema(description = "当前块索引从0开始")
private int chunkIndex;
@Schema(description = "总块数")
private int totalChunks;
@Schema(description = "是否还有更多块")
private boolean hasMore;
@Schema(description = "笔记ID")
@JsonFormat(shape = JsonFormat.Shape.STRING)
private Long fileId;
@Schema(description = "笔记标题")
private String title;
@Schema(description = "是否私密 0-公开 1-私密")
private Integer isPrivate;
@Schema(description = "总字符数")
private long totalLength;
@Schema(description = "当前块字符数")
private int chunkLength;
}

View File

@@ -31,6 +31,10 @@ public class User {
@TableField("`email`") @TableField("`email`")
private String email; private String email;
@Schema(description = "用户角色(ADMIN-管理员,USER-普通用户)",implementation = String.class)
@TableField("`role`")
private String role;
@Schema(description = "用户创建时间",implementation = Date.class) @Schema(description = "用户创建时间",implementation = Date.class)
@TableField("created_at") @TableField("created_at")
private Date createdAt; private Date createdAt;

View File

@@ -21,6 +21,9 @@ public class UserVO implements Serializable {
@Schema(description = "邮箱") @Schema(description = "邮箱")
private String email; private String email;
@Schema(description = "用户角色")
private String role;
@Schema(description = "用户创建时间") @Schema(description = "用户创建时间")
private Date createdAt; private Date createdAt;

View File

@@ -5,6 +5,7 @@ import com.test.bijihoudaun.common.response.R;
import com.test.bijihoudaun.common.response.ResultCode; import com.test.bijihoudaun.common.response.ResultCode;
import com.test.bijihoudaun.util.JwtTokenUtil; import com.test.bijihoudaun.util.JwtTokenUtil;
import io.jsonwebtoken.ExpiredJwtException; import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.JwtException;
import io.jsonwebtoken.security.SignatureException; import io.jsonwebtoken.security.SignatureException;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.core.context.SecurityContextHolder;
@@ -18,6 +19,7 @@ import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import java.io.IOException; import java.io.IOException;
import java.io.PrintWriter;
public class JwtAuthenticationTokenFilter extends OncePerRequestFilter { public class JwtAuthenticationTokenFilter extends OncePerRequestFilter {
@@ -45,8 +47,8 @@ public class JwtAuthenticationTokenFilter extends OncePerRequestFilter {
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain) throws ServletException, IOException { protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain) throws ServletException, IOException {
// 从请求头中获取认证信息 // 从请求头中获取认证信息
String authHeader = request.getHeader(this.tokenHeader); String authHeader = request.getHeader(this.tokenHeader);
// 检查请求头是否存在且以指定的token前缀开头 // 修复:添加 tokenHead 空值检查
if (authHeader != null && authHeader.startsWith(this.tokenHead)) { if (authHeader != null && this.tokenHead != null && authHeader.startsWith(this.tokenHead)) {
// 提取实际的token值去除前缀 // 提取实际的token值去除前缀
final String authToken = authHeader.substring(this.tokenHead.length()); final String authToken = authHeader.substring(this.tokenHead.length());
try { try {
@@ -76,6 +78,10 @@ public class JwtAuthenticationTokenFilter extends OncePerRequestFilter {
// 处理token签名异常 // 处理token签名异常
sendErrorResponse(response, ResultCode.TOKEN_INVALID); sendErrorResponse(response, ResultCode.TOKEN_INVALID);
return; return;
} catch (JwtException e) {
// 修复捕获所有JWT异常作为兜底
sendErrorResponse(response, ResultCode.TOKEN_INVALID);
return;
} }
} }
// 继续过滤器链的处理 // 继续过滤器链的处理
@@ -93,9 +99,13 @@ public class JwtAuthenticationTokenFilter extends OncePerRequestFilter {
response.setContentType("application/json;charset=UTF-8"); response.setContentType("application/json;charset=UTF-8");
// 设置HTTP状态码为401未授权 // 设置HTTP状态码为401未授权
response.setStatus(HttpServletResponse.SC_UNAUTHORIZED); response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
// 修复:使用 try-with-resources 确保 PrintWriter 关闭
try (PrintWriter writer = response.getWriter()) {
// 创建ObjectMapper实例用于对象与JSON之间的转换 // 创建ObjectMapper实例用于对象与JSON之间的转换
ObjectMapper mapper = new ObjectMapper(); ObjectMapper mapper = new ObjectMapper();
// 将失败结果转换为JSON字符串并写入响应输出流 // 将失败结果转换为JSON字符串并写入响应输出流
response.getWriter().write(mapper.writeValueAsString(R.fail(resultCode))); writer.write(mapper.writeValueAsString(R.fail(resultCode)));
writer.flush();
}
} }
} }

View File

@@ -4,6 +4,8 @@ import com.fasterxml.jackson.databind.ObjectMapper;
import com.test.bijihoudaun.common.response.R; import com.test.bijihoudaun.common.response.R;
import com.test.bijihoudaun.common.response.ResultCode; import com.test.bijihoudaun.common.response.ResultCode;
import com.test.bijihoudaun.util.MemoryProtector; import com.test.bijihoudaun.util.MemoryProtector;
import jakarta.annotation.PostConstruct;
import jakarta.annotation.PreDestroy;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@@ -11,12 +13,16 @@ import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.servlet.HandlerInterceptor; import org.springframework.web.servlet.HandlerInterceptor;
import java.util.LinkedHashMap;
import java.util.Map; import java.util.Map;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
/** /**
* 限流拦截器 - 支持按 IP 和按用户双重限流,带容量限制 * 限流拦截器 - 支持按 IP 和按用户双重限流,使用 ConcurrentHashMap 提高并发性能
* 带定期清理机制防止内存泄漏
*/ */
@Slf4j @Slf4j
public class RateLimitInterceptor implements HandlerInterceptor { public class RateLimitInterceptor implements HandlerInterceptor {
@@ -29,59 +35,110 @@ public class RateLimitInterceptor implements HandlerInterceptor {
private static final int MAX_LOGIN_REQUESTS_PER_MINUTE_USER = 10; private static final int MAX_LOGIN_REQUESTS_PER_MINUTE_USER = 10;
// 时间窗口(毫秒) // 时间窗口(毫秒)
private static final long WINDOW_SIZE_MS = 60_000; private static final long WINDOW_SIZE_MS = 60_000;
// 最大存储记录数(防止内存溢出) // 清理间隔毫秒每5分钟清理一次
private static final int MAX_RECORDS = 50000; private static final long CLEANUP_INTERVAL_MS = 5 * 60 * 1000;
// IP 级别限流 // 使用 ConcurrentHashMap + 原子操作,避免锁竞争
private static final LRUCache<String, RequestCounter> ipCounters = new LRUCache<>(MAX_RECORDS / 2); private static final ConcurrentHashMap<String, RequestCounter> ipCounters = new ConcurrentHashMap<>();
private static final LRUCache<String, RequestCounter> ipLoginCounters = new LRUCache<>(MAX_RECORDS / 4); private static final ConcurrentHashMap<String, RequestCounter> ipLoginCounters = new ConcurrentHashMap<>();
// 用户级别限流 private static final ConcurrentHashMap<String, RequestCounter> userCounters = new ConcurrentHashMap<>();
private static final LRUCache<String, RequestCounter> userCounters = new LRUCache<>(MAX_RECORDS / 4); private static final ConcurrentHashMap<String, RequestCounter> userLoginCounters = new ConcurrentHashMap<>();
private static final LRUCache<String, RequestCounter> userLoginCounters = new LRUCache<>(MAX_RECORDS / 4);
private static final ReentrantReadWriteLock lock = new ReentrantReadWriteLock(); // 定期清理线程池
private ScheduledExecutorService cleanupScheduler;
private static class RequestCounter { private static class RequestCounter {
int count; private final AtomicInteger count = new AtomicInteger(0);
long windowStart; private volatile long windowStart;
// 记录最后访问时间,用于清理
private volatile long lastAccessTime;
RequestCounter() { RequestCounter() {
this.count = 1;
this.windowStart = System.currentTimeMillis(); this.windowStart = System.currentTimeMillis();
this.lastAccessTime = System.currentTimeMillis();
} }
boolean incrementAndCheck(int maxRequests) { boolean incrementAndCheck(int maxRequests) {
long now = System.currentTimeMillis(); long now = System.currentTimeMillis();
lastAccessTime = now;
synchronized (this) {
if (now - windowStart > WINDOW_SIZE_MS) { if (now - windowStart > WINDOW_SIZE_MS) {
// 新窗口 // 进入新窗口
count = 1;
windowStart = now; windowStart = now;
count.set(1);
return true; return true;
} }
count++; int currentCount = count.incrementAndGet();
return count <= maxRequests; return currentCount <= maxRequests;
} }
} }
/** /**
* 简单的 LRU 缓存实现 * 检查是否过期超过2个时间窗口没有访问
*/ */
private static class LRUCache<K, V> extends LinkedHashMap<K, V> { boolean isExpired() {
private final int maxSize; return System.currentTimeMillis() - lastAccessTime > WINDOW_SIZE_MS * 2;
}
LRUCache(int maxSize) {
super(maxSize, 0.75f, true);
this.maxSize = maxSize;
} }
@Override @PostConstruct
protected boolean removeEldestEntry(Map.Entry<K, V> eldest) { public void init() {
boolean shouldRemove = size() > maxSize; // 启动定期清理任务
if (shouldRemove) { cleanupScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
log.debug("限流记录达到上限,移除最旧的记录"); Thread t = new Thread(r, "rate-limit-cleanup");
t.setDaemon(true);
return t;
});
cleanupScheduler.scheduleAtFixedRate(this::cleanupExpiredCounters,
CLEANUP_INTERVAL_MS, CLEANUP_INTERVAL_MS, TimeUnit.MILLISECONDS);
log.info("限流拦截器初始化完成,清理间隔:{}分钟", CLEANUP_INTERVAL_MS / 60000);
} }
return shouldRemove;
@PreDestroy
public void destroy() {
// 关闭清理线程池
if (cleanupScheduler != null && !cleanupScheduler.isShutdown()) {
cleanupScheduler.shutdown();
try {
if (!cleanupScheduler.awaitTermination(5, TimeUnit.SECONDS)) {
cleanupScheduler.shutdownNow();
} }
} catch (InterruptedException e) {
cleanupScheduler.shutdownNow();
Thread.currentThread().interrupt();
}
}
log.info("限流拦截器已销毁");
}
/**
* 清理过期的计数器
*/
private void cleanupExpiredCounters() {
try {
int removed = 0;
removed += cleanupMap(ipCounters);
removed += cleanupMap(ipLoginCounters);
removed += cleanupMap(userCounters);
removed += cleanupMap(userLoginCounters);
if (removed > 0) {
log.debug("限流计数器清理完成,移除 {} 个过期条目", removed);
}
} catch (Exception e) {
log.error("限流计数器清理失败", e);
}
}
/**
* 清理单个 Map 中的过期条目
* 使用 removeIf 方法避免并发修改异常
*/
private int cleanupMap(ConcurrentHashMap<String, RequestCounter> map) {
int sizeBefore = map.size();
// 使用 ConcurrentHashMap 的 removeIf 方法,线程安全
map.entrySet().removeIf(entry -> entry.getValue().isExpired());
return sizeBefore - map.size();
} }
@Override @Override
@@ -117,20 +174,15 @@ public class RateLimitInterceptor implements HandlerInterceptor {
*/ */
private boolean checkIpLimit(String clientIp, boolean isLoginRequest, private boolean checkIpLimit(String clientIp, boolean isLoginRequest,
HttpServletResponse response) throws Exception { HttpServletResponse response) throws Exception {
LRUCache<String, RequestCounter> counters = isLoginRequest ? ipLoginCounters : ipCounters; ConcurrentHashMap<String, RequestCounter> counters = isLoginRequest ? ipLoginCounters : ipCounters;
int maxRequests = isLoginRequest ? MAX_LOGIN_REQUESTS_PER_MINUTE : MAX_REQUESTS_PER_MINUTE; int maxRequests = isLoginRequest ? MAX_LOGIN_REQUESTS_PER_MINUTE : MAX_REQUESTS_PER_MINUTE;
lock.writeLock().lock();
try {
RequestCounter counter = counters.computeIfAbsent(clientIp, k -> new RequestCounter()); RequestCounter counter = counters.computeIfAbsent(clientIp, k -> new RequestCounter());
if (!counter.incrementAndCheck(maxRequests)) { if (!counter.incrementAndCheck(maxRequests)) {
writeRateLimitResponse(response, "请求过于频繁,请稍后再试"); writeRateLimitResponse(response, "请求过于频繁,请稍后再试");
return false; return false;
} }
} finally {
lock.writeLock().unlock();
}
return true; return true;
} }
@@ -139,20 +191,15 @@ public class RateLimitInterceptor implements HandlerInterceptor {
*/ */
private boolean checkUserLimit(String username, boolean isLoginRequest, private boolean checkUserLimit(String username, boolean isLoginRequest,
HttpServletResponse response) throws Exception { HttpServletResponse response) throws Exception {
LRUCache<String, RequestCounter> counters = isLoginRequest ? userLoginCounters : userCounters; ConcurrentHashMap<String, RequestCounter> counters = isLoginRequest ? userLoginCounters : userCounters;
int maxRequests = isLoginRequest ? MAX_LOGIN_REQUESTS_PER_MINUTE_USER : MAX_REQUESTS_PER_MINUTE_USER; int maxRequests = isLoginRequest ? MAX_LOGIN_REQUESTS_PER_MINUTE_USER : MAX_REQUESTS_PER_MINUTE_USER;
lock.writeLock().lock();
try {
RequestCounter counter = counters.computeIfAbsent(username, k -> new RequestCounter()); RequestCounter counter = counters.computeIfAbsent(username, k -> new RequestCounter());
if (!counter.incrementAndCheck(maxRequests)) { if (!counter.incrementAndCheck(maxRequests)) {
writeRateLimitResponse(response, "您的操作过于频繁,请稍后再试"); writeRateLimitResponse(response, "您的操作过于频繁,请稍后再试");
return false; return false;
} }
} finally {
lock.writeLock().unlock();
}
return true; return true;
} }

View File

@@ -107,8 +107,9 @@ public class ReplayAttackInterceptor implements HandlerInterceptor {
} }
long now = System.currentTimeMillis(); long now = System.currentTimeMillis();
if (Math.abs(now - timestamp) > TIMESTAMP_VALIDITY) { // 修复:不允许未来时间戳,防止预发送攻击
writeErrorResponse(response, "请求已过期,请重新发起请求"); if (timestamp > now || now - timestamp > TIMESTAMP_VALIDITY) {
writeErrorResponse(response, "请求时间戳无效或已过期");
return false; return false;
} }

View File

@@ -2,11 +2,19 @@ package com.test.bijihoudaun.interceptor;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HtmlUtil; import cn.hutool.http.HtmlUtil;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.test.bijihoudaun.common.response.R;
import com.test.bijihoudaun.common.response.ResultCode;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.HandlerInterceptor; import org.springframework.web.servlet.HandlerInterceptor;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Enumeration; import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;
/** /**
* XSS 过滤拦截器 * XSS 过滤拦截器
@@ -14,12 +22,38 @@ import java.util.Enumeration;
* 注意:此拦截器只能过滤 URL 参数和表单数据,无法过滤 @RequestBody 的 JSON 数据 * 注意:此拦截器只能过滤 URL 参数和表单数据,无法过滤 @RequestBody 的 JSON 数据
* JSON 数据的 XSS 过滤由 XssStringDeserializer 处理 * JSON 数据的 XSS 过滤由 XssStringDeserializer 处理
*/ */
@Slf4j
public class XSSInterceptor implements HandlerInterceptor { public class XSSInterceptor implements HandlerInterceptor {
// 修复:添加不需要检查的请求头白名单(浏览器标准请求头)
private static final Set<String> HEADER_WHITELIST = new HashSet<>(Arrays.asList(
"sec-ch-ua",
"sec-ch-ua-mobile",
"sec-ch-ua-platform",
"sec-fetch-dest",
"sec-fetch-mode",
"sec-fetch-site",
"sec-fetch-user",
"user-agent",
"accept",
"accept-encoding",
"accept-language",
"cache-control",
"connection",
"host",
"referer",
"upgrade-insecure-requests",
"content-type",
"content-length",
"origin"
));
@Override @Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// 过滤请求头 // 过滤请求头,发现 XSS 攻击则拒绝请求
filterHeaders(request); if (!filterHeaders(request, response)) {
return false;
}
// 过滤请求参数URL 参数和表单数据) // 过滤请求参数URL 参数和表单数据)
filterParameters(request); filterParameters(request);
@@ -29,24 +63,32 @@ public class XSSInterceptor implements HandlerInterceptor {
/** /**
* 过滤请求头 * 过滤请求头
* @return true-通过false-拒绝请求
*/ */
private void filterHeaders(HttpServletRequest request) { private boolean filterHeaders(HttpServletRequest request, HttpServletResponse response) throws Exception {
Enumeration<String> headerNames = request.getHeaderNames(); Enumeration<String> headerNames = request.getHeaderNames();
if (headerNames == null) { if (headerNames == null) {
return; return true;
} }
while (headerNames.hasMoreElements()) { while (headerNames.hasMoreElements()) {
String headerName = headerNames.nextElement(); String headerName = headerNames.nextElement();
// 修复:跳过白名单中的请求头
if (HEADER_WHITELIST.contains(headerName.toLowerCase())) {
continue;
}
String headerValue = request.getHeader(headerName); String headerValue = request.getHeader(headerName);
if (StrUtil.isNotBlank(headerValue)) { if (StrUtil.isNotBlank(headerValue)) {
String filteredValue = HtmlUtil.filter(headerValue); String filteredValue = HtmlUtil.filter(headerValue);
// 注意:请求头无法直接修改,这里只是记录日志 // 发现 XSS 内容,记录日志并拒绝请求
if (!headerValue.equals(filteredValue)) { if (!headerValue.equals(filteredValue)) {
// 发现 XSS 内容,记录日志 log.warn("检测到XSS攻击请求头{},原始值:{}", headerName, headerValue);
writeErrorResponse(response, "检测到恶意内容,请求被拒绝");
return false;
} }
} }
} }
return true;
} }
/** /**
@@ -65,4 +107,20 @@ public class XSSInterceptor implements HandlerInterceptor {
} }
} }
} }
/**
* 写入错误响应
*/
private void writeErrorResponse(HttpServletResponse response, String message) throws Exception {
response.setContentType("application/json;charset=UTF-8");
response.setStatus(400);
ObjectMapper mapper = new ObjectMapper();
// 修复:使用 try-with-resources 确保 PrintWriter 关闭
try (PrintWriter writer = response.getWriter()) {
writer.write(mapper.writeValueAsString(
R.fail(ResultCode.FAILED.getCode(), message)
));
writer.flush();
}
}
} }

View File

@@ -34,7 +34,7 @@ public interface MarkdownFileMapper extends BaseMapper<MarkdownFile> {
"LEFT JOIN `grouping` g ON mf.grouping_id = g.id " + "LEFT JOIN `grouping` g ON mf.grouping_id = g.id " +
"WHERE mf.grouping_id = #{groupingId} AND mf.is_deleted = 0 " + "WHERE mf.grouping_id = #{groupingId} AND mf.is_deleted = 0 " +
"ORDER BY mf.updated_at DESC") "ORDER BY mf.updated_at DESC")
List<MarkdownFileVO> selectByGroupingIdWithGrouping(@Param("groupingId") String groupingId); List<MarkdownFileVO> selectByGroupingIdWithGrouping(@Param("groupingId") Long groupingId);
/** /**
* 查询已删除的笔记不包含content大字段 * 查询已删除的笔记不包含content大字段

View File

@@ -8,6 +8,6 @@ import org.apache.ibatis.annotations.Select;
@Mapper @Mapper
public interface UserMapper extends BaseMapper<User> { public interface UserMapper extends BaseMapper<User> {
// 自定义查询方法示例 // 自定义查询方法示例
@Select("SELECT id, `username`, `password`, `email`, created_at, updated_at, `token`, token_enddata FROM `user` WHERE username = #{username}") @Select("SELECT id, `username`, `password`, `email`, `role`, created_at, updated_at, `token`, token_enddata FROM `user` WHERE username = #{username}")
User findByUsername(String username); User findByUsername(String username);
} }

View File

@@ -1,6 +1,7 @@
package com.test.bijihoudaun.scheduler; package com.test.bijihoudaun.scheduler;
import com.test.bijihoudaun.service.ImageCleanupService; import com.test.bijihoudaun.service.ImageCleanupService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled; import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@@ -10,6 +11,7 @@ import org.springframework.stereotype.Component;
* 定期自动清理冗余图片 * 定期自动清理冗余图片
*/ */
@Component @Component
@Slf4j
public class ImageCleanupScheduler { public class ImageCleanupScheduler {
@Autowired @Autowired
@@ -22,6 +24,7 @@ public class ImageCleanupScheduler {
@Scheduled(cron = "0 0 3 * * ?") @Scheduled(cron = "0 0 3 * * ?")
public void scheduledCleanup() { public void scheduledCleanup() {
int deletedCount = imageCleanupService.cleanupRedundantImages(); int deletedCount = imageCleanupService.cleanupRedundantImages();
System.out.println("定时清理任务完成,清理了 " + deletedCount + " 个冗余图片"); // 优化:使用日志框架代替 System.out.println
log.info("定时清理任务完成,清理了 {} 个冗余图片", deletedCount);
} }
} }

View File

@@ -2,6 +2,7 @@ package com.test.bijihoudaun.service;
import com.baomidou.mybatisplus.extension.service.IService; import com.baomidou.mybatisplus.extension.service.IService;
import com.test.bijihoudaun.entity.MarkdownFile; import com.test.bijihoudaun.entity.MarkdownFile;
import com.test.bijihoudaun.entity.MarkdownFileChunk;
import com.test.bijihoudaun.entity.MarkdownFileVO; import com.test.bijihoudaun.entity.MarkdownFileVO;
import java.util.List; import java.util.List;
@@ -28,7 +29,7 @@ public interface MarkdownFileService extends IService<MarkdownFile> {
* @param groupingId 分组ID * @param groupingId 分组ID
* @return 文件列表 * @return 文件列表
*/ */
List<MarkdownFileVO> getFilesByGroupingId(String groupingId); List<MarkdownFileVO> getFilesByGroupingId(Long groupingId);
/** /**
* 删除Markdown文件 * 删除Markdown文件
@@ -48,9 +49,11 @@ public interface MarkdownFileService extends IService<MarkdownFile> {
/** /**
* 根据标题模糊搜索 * 根据标题模糊搜索
* @param keyword 关键词 * @param keyword 关键词
* @param page 页码从1开始
* @param pageSize 每页数量
* @return 文件列表 * @return 文件列表
*/ */
List<MarkdownFile> searchByTitle(String keyword); List<MarkdownFile> searchByTitle(String keyword, int page, int pageSize);
/** /**
* 更新Markdown文件标题 * 更新Markdown文件标题
@@ -66,4 +69,15 @@ public interface MarkdownFileService extends IService<MarkdownFile> {
* @return 文件列表 * @return 文件列表
*/ */
List<MarkdownFileVO> getRecentFiles(int limit); List<MarkdownFileVO> getRecentFiles(int limit);
/**
* 分块加载Markdown文件内容
* 用于大文件(> 500KB的分页加载
* @param id 文件ID
* @param chunkIndex 块索引从0开始
* @param chunkSize 块大小字符数默认10000
* @param isAuthenticated 是否已认证
* @return 文件块对象
*/
MarkdownFileChunk getMarkdownChunk(Long id, int chunkIndex, int chunkSize, boolean isAuthenticated);
} }

View File

@@ -27,7 +27,7 @@ public interface UserService extends IService<User> {
* 用户删除 * 用户删除
* @param id 用户id * @param id 用户id
*/ */
void deleteUser(Integer id); void deleteUser(Long id);
/** /**
* 查询用户token是否过期 * 查询用户token是否过期

View File

@@ -40,11 +40,15 @@ public class GroupingServiceImpl
@Override @Override
public List<Grouping> getAllGroupings(Long parentId) { public List<Grouping> getAllGroupings(Long parentId) {
if (ObjectUtil.isNull(parentId)){ LambdaQueryWrapper<Grouping> queryWrapper = new LambdaQueryWrapper<>();
return groupingMapper.selectList(null); // 只查询未删除的分组
queryWrapper.eq(Grouping::getIsDeleted, 0);
if (ObjectUtil.isNotNull(parentId)){
queryWrapper.eq(Grouping::getParentId, parentId);
} }
return groupingMapper.selectList(new LambdaQueryWrapper<Grouping>() // 限制最大返回数量,防止内存溢出
.eq(Grouping::getParentId, parentId)); queryWrapper.last("LIMIT 1000");
return groupingMapper.selectList(queryWrapper);
} }
@Override @Override

View File

@@ -0,0 +1,85 @@
package com.test.bijihoudaun.service.impl;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.test.bijihoudaun.entity.ImageName;
import com.test.bijihoudaun.mapper.ImageNameMapper;
import jakarta.annotation.Resource;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.List;
/**
* 图片名称同步服务
* 处理 Markdown 内容中的图片名称同步,独立的事务和异步处理
*/
@Service
public class ImageNameSyncService {
@Resource
private ImageNameMapper imageNameMapper;
/**
* 同步图片名称
* 异步执行,独立事务
*/
@Async("imageNameSyncExecutor")
@Transactional
public void syncImageNames(Long markdownId, List<String> strings) {
// 查询数据库中已存在的文件名
List<ImageName> imageNames = imageNameMapper.selectList(new LambdaUpdateWrapper<ImageName>()
.eq(ImageName::getMarkdownId, markdownId));
// 若是数据库中的数据为null则插入
if (CollUtil.isEmpty(imageNames)) {
if (CollUtil.isNotEmpty(strings)) {
List<ImageName> list = strings.stream().map(fileName -> {
ImageName imageName = new ImageName();
imageName.setFileName(fileName);
imageName.setMarkdownId(markdownId);
return imageName;
}).toList();
// 批量插入新的文件名使用MyBatis Plus批量插入
imageNameMapper.insert(list);
}
} else {
// 数据库中已有记录,需要对比处理
// 获取数据库中的文件名列表
List<String> dbFileNames = imageNames.stream()
.map(ImageName::getFileName)
.toList();
// 找出需要新增的文件名在strings中但不在数据库中
List<String> toInsert = strings.stream()
.filter(fileName -> !dbFileNames.contains(fileName))
.toList();
// 找出需要删除的记录在数据库中但不在strings中
List<ImageName> toDelete = imageNames.stream()
.filter(imageName -> !strings.contains(imageName.getFileName()))
.toList();
// 插入新增的文件名
if (CollUtil.isNotEmpty(toInsert)) {
List<ImageName> insertList = toInsert.stream().map(fileName -> {
ImageName imageName = new ImageName();
imageName.setFileName(fileName);
imageName.setMarkdownId(markdownId);
return imageName;
}).toList();
imageNameMapper.insert(insertList);
}
// 删除不再需要的记录
if (CollUtil.isNotEmpty(toDelete)) {
List<Long> deleteIds = toDelete.stream()
.map(ImageName::getId)
.toList();
imageNameMapper.deleteByIds(deleteIds);
}
}
}
}

View File

@@ -9,6 +9,7 @@ import com.test.bijihoudaun.common.exception.BusinessException;
import com.test.bijihoudaun.entity.Image; import com.test.bijihoudaun.entity.Image;
import com.test.bijihoudaun.mapper.ImageMapper; import com.test.bijihoudaun.mapper.ImageMapper;
import com.test.bijihoudaun.service.ImageService; import com.test.bijihoudaun.service.ImageService;
import com.test.bijihoudaun.util.ImageCompressor;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -16,11 +17,15 @@ import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartFile;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Date; import java.util.Date;
import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.UUID; import java.util.UUID;
@@ -36,6 +41,15 @@ public class ImageServiceImpl
private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; private static final long MAX_FILE_SIZE = 10 * 1024 * 1024;
private static final Map<String, byte[]> FILE_SIGNATURES = Map.ofEntries(
Map.entry("jpg", new byte[]{(byte)0xFF, (byte)0xD8, (byte)0xFF}),
Map.entry("jpeg", new byte[]{(byte)0xFF, (byte)0xD8, (byte)0xFF}),
Map.entry("png", new byte[]{(byte)0x89, 0x50, 0x4E, 0x47}),
Map.entry("gif", new byte[]{0x47, 0x49, 0x46}),
Map.entry("bmp", new byte[]{0x42, 0x4D}),
Map.entry("webp", new byte[]{0x52, 0x49, 0x46, 0x46})
);
@Value("${file.upload-dir}") @Value("${file.upload-dir}")
private String uploadDir; private String uploadDir;
@Resource @Resource
@@ -71,6 +85,12 @@ public class ImageServiceImpl
throw new BusinessException("文件内容类型无效"); throw new BusinessException("文件内容类型无效");
} }
// 验证文件头
String extWithoutDot = extension.substring(1).toLowerCase();
if (!validateFileSignature(file, extWithoutDot)) {
throw new BusinessException("文件内容与扩展名不匹配,可能是伪装的恶意文件");
}
Path uploadPath = Paths.get(uploadDir); Path uploadPath = Paths.get(uploadDir);
if (!Files.exists(uploadPath)) { if (!Files.exists(uploadPath)) {
Files.createDirectories(uploadPath); Files.createDirectories(uploadPath);
@@ -79,11 +99,29 @@ public class ImageServiceImpl
String storedName = UUID.randomUUID() + extension; String storedName = UUID.randomUUID() + extension;
Path filePath = uploadPath.resolve(storedName); Path filePath = uploadPath.resolve(storedName);
Files.copy(file.getInputStream(), filePath);
// 优化:压缩图片后再保存,确保流关闭
try (InputStream originalStream = file.getInputStream()) {
InputStream compressedStream = ImageCompressor.compressIfNeeded(
originalStream, contentType, file.getSize());
if (compressedStream != null) {
// 使用压缩后的图片
try (compressedStream) {
Files.copy(compressedStream, filePath);
}
} else {
// 使用原图,需要重新获取流
try (InputStream newStream = file.getInputStream()) {
Files.copy(newStream, filePath);
}
}
}
Image image = new Image(); Image image = new Image();
image.setOriginalName(originalFilename); image.setOriginalName(originalFilename);
image.setStoredName(storedName); image.setStoredName(storedName);
// 返回相对路径,前端会根据环境自动拼接 baseURL
image.setUrl("/api/images/preview/" + storedName); image.setUrl("/api/images/preview/" + storedName);
image.setSize(file.getSize()); image.setSize(file.getSize());
image.setContentType(contentType); image.setContentType(contentType);
@@ -94,6 +132,18 @@ public class ImageServiceImpl
return image; return image;
} }
private boolean validateFileSignature(MultipartFile file, String extension) throws IOException {
byte[] signature = FILE_SIGNATURES.get(extension);
if (signature == null) return true;
byte[] fileHeader = new byte[Math.max(4, signature.length)];
try (InputStream is = file.getInputStream()) {
int read = is.read(fileHeader);
if (read < signature.length) return false;
return Arrays.equals(Arrays.copyOf(fileHeader, signature.length), signature);
}
}
@Override @Override
public boolean deleteImage(Long id) { public boolean deleteImage(Long id) {
Image image = imageMapper.selectById(id); Image image = imageMapper.selectById(id);

View File

@@ -2,20 +2,22 @@ package com.test.bijihoudaun.service.impl;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.test.bijihoudaun.entity.ImageName; import com.test.bijihoudaun.entity.ImageName;
import com.test.bijihoudaun.entity.MarkdownFile; import com.test.bijihoudaun.entity.MarkdownFile;
import com.test.bijihoudaun.entity.MarkdownFileChunk;
import com.test.bijihoudaun.entity.MarkdownFileVO; import com.test.bijihoudaun.entity.MarkdownFileVO;
import com.test.bijihoudaun.mapper.ImageNameMapper; import com.test.bijihoudaun.mapper.ImageNameMapper;
import com.test.bijihoudaun.mapper.MarkdownFileMapper; import com.test.bijihoudaun.mapper.MarkdownFileMapper;
import com.test.bijihoudaun.service.MarkdownFileService; import com.test.bijihoudaun.service.MarkdownFileService;
import com.test.bijihoudaun.util.MarkdownImageExtractor; import com.test.bijihoudaun.util.MarkdownImageExtractor;
import com.test.bijihoudaun.util.SecurityUtil;
import com.test.bijihoudaun.util.SnowflakeIdGenerator; import com.test.bijihoudaun.util.SnowflakeIdGenerator;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
@@ -34,6 +36,8 @@ public class MarkdownFileServiceImpl
ImageNameMapper imageNameMapper; ImageNameMapper imageNameMapper;
@Resource @Resource
SnowflakeIdGenerator snowflakeIdGenerator; SnowflakeIdGenerator snowflakeIdGenerator;
@Resource
ImageNameSyncService imageNameSyncService;
@Override @Override
@@ -77,7 +81,8 @@ public class MarkdownFileServiceImpl
} }
List<String> strings = MarkdownImageExtractor.extractImageFilenames(markdownFile.getContent()); List<String> strings = MarkdownImageExtractor.extractImageFilenames(markdownFile.getContent());
syncImageNames(id, strings); // 修复:调用单独的 Service 处理异步逻辑
imageNameSyncService.syncImageNames(id, strings);
MarkdownFileVO result = markdownFileMapper.selectByIdWithGrouping(id); MarkdownFileVO result = markdownFileMapper.selectByIdWithGrouping(id);
if (result != null) { if (result != null) {
@@ -103,16 +108,18 @@ public class MarkdownFileServiceImpl
@Override @Override
public List<MarkdownFileVO> getFilesByGroupingId(String groupingId) { public List<MarkdownFileVO> getFilesByGroupingId(Long groupingId) {
return markdownFileMapper.selectByGroupingIdWithGrouping(groupingId); return markdownFileMapper.selectByGroupingIdWithGrouping(groupingId);
} }
@Override @Override
public boolean deleteMarkdownFile(Long id) { public boolean deleteMarkdownFile(Long id) {
Long currentUserId = SecurityUtil.getCurrentUserId();
LambdaUpdateWrapper<MarkdownFile> updateWrapper = new LambdaUpdateWrapper<>(); LambdaUpdateWrapper<MarkdownFile> updateWrapper = new LambdaUpdateWrapper<>();
updateWrapper.eq(MarkdownFile::getId, id) updateWrapper.eq(MarkdownFile::getId, id)
.set(MarkdownFile::getIsDeleted, 1) .set(MarkdownFile::getIsDeleted, 1)
.set(MarkdownFile::getDeletedAt, new Date()); .set(MarkdownFile::getDeletedAt, new Date())
.set(MarkdownFile::getDeletedBy, currentUserId);
return this.update(updateWrapper); return this.update(updateWrapper);
} }
@@ -127,9 +134,16 @@ public class MarkdownFileServiceImpl
} }
@Override @Override
public List<MarkdownFile> searchByTitle(String keyword) { public List<MarkdownFile> searchByTitle(String keyword, int page, int pageSize) {
QueryWrapper<MarkdownFile> queryWrapper = new QueryWrapper<>(); if (keyword == null || keyword.trim().isEmpty()) {
queryWrapper.like("title", keyword); return List.of();
}
int offset = (page - 1) * pageSize;
LambdaQueryWrapper<MarkdownFile> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.like(MarkdownFile::getTitle, keyword)
.eq(MarkdownFile::getIsDeleted, 0)
.orderByDesc(MarkdownFile::getUpdatedAt)
.last("LIMIT " + offset + "," + pageSize);
return this.list(queryWrapper); return this.list(queryWrapper);
} }
@@ -149,62 +163,92 @@ public class MarkdownFileServiceImpl
return markdownFileMapper.selectRecentWithGrouping(limit); return markdownFileMapper.selectRecentWithGrouping(limit);
} }
@Override
@Async("imageNameSyncExecutor") public MarkdownFileChunk getMarkdownChunk(Long id, int chunkIndex, int chunkSize, boolean isAuthenticated) {
public void syncImageNames(Long markdownId, List<String> strings) { // 获取文件基本信息(不包含内容)
// 查询数据库中已存在的文件名 MarkdownFileVO fileVO = markdownFileMapper.selectByIdWithGrouping(id);
List<ImageName> imageNames = imageNameMapper.selectList(new LambdaUpdateWrapper<ImageName>() if (fileVO == null) {
.eq(ImageName::getMarkdownId, markdownId)); return null;
// 若是数据库中的数据为null则插入
if (CollUtil.isEmpty(imageNames)) {
if (CollUtil.isNotEmpty(strings)) {
List<ImageName> list = strings.stream().map(fileName -> {
ImageName imageName = new ImageName();
imageName.setFileName(fileName);
imageName.setMarkdownId(markdownId);
return imageName;
}).toList();
// 批量插入新的文件名使用MyBatis Plus批量插入
imageNameMapper.insert(list);
}
} else {
// 数据库中已有记录,需要对比处理
// 获取数据库中的文件名列表
List<String> dbFileNames = imageNames.stream()
.map(ImageName::getFileName)
.toList();
// 找出需要新增的文件名在strings中但不在数据库中
List<String> toInsert = strings.stream()
.filter(fileName -> !dbFileNames.contains(fileName))
.toList();
// 找出需要删除的记录在数据库中但不在strings中
List<ImageName> toDelete = imageNames.stream()
.filter(imageName -> !strings.contains(imageName.getFileName()))
.toList();
// 插入新增的文件名
if (CollUtil.isNotEmpty(toInsert)) {
List<ImageName> insertList = toInsert.stream().map(fileName -> {
ImageName imageName = new ImageName();
imageName.setFileName(fileName);
imageName.setMarkdownId(markdownId);
return imageName;
}).toList();
imageNameMapper.insert(insertList);
} }
// 删除不再需要的记 // 检查权限:私密笔记需要登
if (CollUtil.isNotEmpty(toDelete)) { if (fileVO.getIsPrivate() != null && fileVO.getIsPrivate() == 1 && !isAuthenticated) {
List<Long> deleteIds = toDelete.stream() MarkdownFileChunk chunk = new MarkdownFileChunk();
.map(ImageName::getId) chunk.setFileId(id);
.toList(); chunk.setTitle(fileVO.getTitle());
imageNameMapper.deleteByIds(deleteIds); chunk.setIsPrivate(fileVO.getIsPrivate());
chunk.setChunk("该笔记为私密笔记,请登录后查看");
chunk.setChunkIndex(0);
chunk.setTotalChunks(1);
chunk.setHasMore(false);
chunk.setTotalLength(0);
chunk.setChunkLength(0);
return chunk;
} }
// 获取完整内容
MarkdownFile file = this.getById(id);
if (file == null || file.getContent() == null) {
return null;
}
String fullContent = file.getContent();
int totalLength = fullContent.length();
// 如果文件小于 500KB约50万字符直接返回全部内容
if (totalLength <= 500000) {
MarkdownFileChunk chunk = new MarkdownFileChunk();
chunk.setFileId(id);
chunk.setTitle(file.getTitle());
chunk.setIsPrivate(file.getIsPrivate());
chunk.setChunk(fullContent);
chunk.setChunkIndex(0);
chunk.setTotalChunks(1);
chunk.setHasMore(false);
chunk.setTotalLength(totalLength);
chunk.setChunkLength(totalLength);
return chunk;
}
// 大文件分页加载
// 计算总块数
int totalChunks = (int) Math.ceil((double) totalLength / chunkSize);
// 确保 chunkIndex 在有效范围内
if (chunkIndex < 0) {
chunkIndex = 0;
}
if (chunkIndex >= totalChunks) {
chunkIndex = totalChunks - 1;
}
// 计算当前块的起止位置
int start = chunkIndex * chunkSize;
int end = Math.min(start + chunkSize, totalLength);
// 截取内容
String chunkContent = fullContent.substring(start, end);
// 如果不是第一块,尝试从完整的换行处开始
if (chunkIndex > 0 && start < totalLength) {
// 找到第一个换行符,从下一行开始显示
int firstNewLine = chunkContent.indexOf('\n');
if (firstNewLine > 0) {
chunkContent = chunkContent.substring(firstNewLine + 1);
} }
} }
MarkdownFileChunk chunk = new MarkdownFileChunk();
chunk.setFileId(id);
chunk.setTitle(file.getTitle());
chunk.setIsPrivate(file.getIsPrivate());
chunk.setChunk(chunkContent);
chunk.setChunkIndex(chunkIndex);
chunk.setTotalChunks(totalChunks);
chunk.setHasMore(chunkIndex < totalChunks - 1);
chunk.setTotalLength(totalLength);
chunk.setChunkLength(chunkContent.length());
return chunk;
}
} }

View File

@@ -18,14 +18,18 @@ import java.util.UUID;
@Transactional @Transactional
public class RegistrationCodeServiceImpl extends ServiceImpl<RegistrationCodeMapper, RegistrationCode> implements RegistrationCodeService { public class RegistrationCodeServiceImpl extends ServiceImpl<RegistrationCodeMapper, RegistrationCode> implements RegistrationCodeService {
// 修复:定义日期时间格式器
private static final DateTimeFormatter DATE_TIME_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
@Override @Override
public String generateCode(String creator) { public String generateCode(String creator) {
RegistrationCode registrationCode = new RegistrationCode(); RegistrationCode registrationCode = new RegistrationCode();
String code = UUID.randomUUID().toString().replaceAll("-", "").substring(0, 16); String code = UUID.randomUUID().toString().replaceAll("-", "").substring(0, 16);
registrationCode.setCode(code); registrationCode.setCode(code);
registrationCode.setCreatedBy(creator); registrationCode.setCreatedBy(creator);
registrationCode.setCreatedAt(LocalDateTime.now().toString()); // 修复:使用格式化后的日期字符串
registrationCode.setExpiryTime(LocalDateTime.now().plusDays(1).toString()); registrationCode.setCreatedAt(LocalDateTime.now().format(DATE_TIME_FORMATTER));
registrationCode.setExpiryTime(LocalDateTime.now().plusDays(1).format(DATE_TIME_FORMATTER));
save(registrationCode); save(registrationCode);
return code; return code;
} }
@@ -41,11 +45,18 @@ public class RegistrationCodeServiceImpl extends ServiceImpl<RegistrationCodeMap
return false; return false;
} }
LocalDateTime expiryTime = LocalDateTime.parse(registrationCode.getExpiryTime()); try {
// 修复:使用格式器解析日期
LocalDateTime expiryTime = LocalDateTime.parse(registrationCode.getExpiryTime(), DATE_TIME_FORMATTER);
if (expiryTime.isBefore(LocalDateTime.now())) { if (expiryTime.isBefore(LocalDateTime.now())) {
remove(queryWrapper); // 注册码过期,删除 remove(queryWrapper); // 注册码过期,删除
return false; return false;
} }
} catch (Exception e) {
// 如果解析失败(可能是旧数据格式),认为已过期
remove(queryWrapper);
return false;
}
// 验证成功后立即删除,确保一次性使用 // 验证成功后立即删除,确保一次性使用
remove(queryWrapper); remove(queryWrapper);
@@ -55,6 +66,7 @@ public class RegistrationCodeServiceImpl extends ServiceImpl<RegistrationCodeMap
@Override @Override
@Scheduled(cron = "0 0 1 * * ?") // 每天凌晨1点执行 @Scheduled(cron = "0 0 1 * * ?") // 每天凌晨1点执行
public void deleteExpiredCodes() { public void deleteExpiredCodes() {
remove(new QueryWrapper<RegistrationCode>().lt("expiry_time", LocalDateTime.now().toString())); // 修复:使用格式化后的日期字符串进行比较
remove(new QueryWrapper<RegistrationCode>().lt("expiry_time", LocalDateTime.now().format(DATE_TIME_FORMATTER)));
} }
} }

View File

@@ -13,9 +13,10 @@ import com.test.bijihoudaun.service.UserService;
import com.test.bijihoudaun.util.JwtTokenUtil; import com.test.bijihoudaun.util.JwtTokenUtil;
import com.test.bijihoudaun.util.LoginLockUtil; import com.test.bijihoudaun.util.LoginLockUtil;
import com.test.bijihoudaun.util.PasswordUtils; import com.test.bijihoudaun.util.PasswordUtils;
import com.test.bijihoudaun.util.UuidV7;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.authentication.BadCredentialsException; import org.springframework.security.authentication.BadCredentialsException;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.core.userdetails.UserDetails; import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService; import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.core.userdetails.UsernameNotFoundException; import org.springframework.security.core.userdetails.UsernameNotFoundException;
@@ -30,9 +31,9 @@ import java.util.Date;
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService, UserDetailsService { public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService, UserDetailsService {
private static final int MIN_USERNAME_LENGTH = 2; private static final int MIN_USERNAME_LENGTH = 2;
private static final int MAX_USERNAME_LENGTH = 8; private static final int MAX_USERNAME_LENGTH = 20;
private static final int MIN_PASSWORD_LENGTH = 6; private static final int MIN_PASSWORD_LENGTH = 6;
private static final int MAX_PASSWORD_LENGTH = 12; private static final int MAX_PASSWORD_LENGTH = 128;
@Autowired @Autowired
private UserMapper userMapper; private UserMapper userMapper;
@@ -43,7 +44,14 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
if (ObjectUtil.isNull(user)) { if (ObjectUtil.isNull(user)) {
throw new UsernameNotFoundException("User not found with username: " + username); throw new UsernameNotFoundException("User not found with username: " + username);
} }
return new org.springframework.security.core.userdetails.User(user.getUsername(), user.getPassword(), new ArrayList<>()); // 加载用户角色权限
java.util.List<GrantedAuthority> authorities = new ArrayList<>();
String role = user.getRole();
if (role == null || role.isEmpty()) {
role = "USER"; // 默认角色
}
authorities.add(new SimpleGrantedAuthority("ROLE_" + role.toUpperCase()));
return new org.springframework.security.core.userdetails.User(user.getUsername(), user.getPassword(), authorities);
} }
@Override @Override
@@ -78,6 +86,7 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
user.setUsername(username); user.setUsername(username);
user.setPassword(encrypt); user.setPassword(encrypt);
user.setEmail(email); user.setEmail(email);
user.setRole("USER"); // 设置默认角色
userMapper.insert(user); userMapper.insert(user);
return user; return user;
} }
@@ -102,7 +111,7 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
if (remainingAttempts <= 0) { if (remainingAttempts <= 0) {
throw new BadCredentialsException("登录失败次数过多账号已被锁定30分钟"); throw new BadCredentialsException("登录失败次数过多账号已被锁定30分钟");
} }
throw new BadCredentialsException("用户名或密码错误,还剩" + remainingAttempts + "次机会"); throw new BadCredentialsException("登录失败,请稍后重试");
} }
// 登录成功,清除失败记录 // 登录成功,清除失败记录
@@ -116,7 +125,7 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
} }
@Override @Override
public void deleteUser(Integer id) { public void deleteUser(Long id) {
userMapper.deleteById(id); userMapper.deleteById(id);
} }
@@ -147,6 +156,11 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
if (!PasswordUtils.verify(updatePasswordBo.getOldPassword(), user.getPassword())) { if (!PasswordUtils.verify(updatePasswordBo.getOldPassword(), user.getPassword())) {
throw new BusinessException("旧密码不正确"); throw new BusinessException("旧密码不正确");
} }
if (PasswordUtils.verify(newPassword, user.getPassword())) {
throw new BusinessException("新密码不能与旧密码相同");
}
user.setPassword(PasswordUtils.encrypt(newPassword)); user.setPassword(PasswordUtils.encrypt(newPassword));
updateById(user); updateById(user);
} }

View File

@@ -12,7 +12,10 @@ import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit; import java.time.temporal.ChronoUnit;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.Map; import java.util.Map;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
/** /**
* 图形验证码工具类 * 图形验证码工具类
@@ -33,11 +36,23 @@ public class CaptchaUtil {
private static final int MAX_CAPTCHAS = 5000; private static final int MAX_CAPTCHAS = 5000;
// 存储验证码key=验证码IDvalue=验证码记录 // 存储验证码key=验证码IDvalue=验证码记录
private static final LRUCache<String, CaptchaRecord> captchaStore = new LRUCache<>(MAX_CAPTCHAS); private static final ConcurrentHashMap<String, CaptchaRecord> captchaStore = new ConcurrentHashMap<>();
private static final ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
// 安全随机数生成器 // 使用 ThreadLocal 确保线程安全
private static final SecureRandom random = new SecureRandom(); private static final ThreadLocal<SecureRandom> random = ThreadLocal.withInitial(SecureRandom::new);
// 定期清理线程池
private static final ScheduledExecutorService cleanupScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
Thread t = new Thread(r, "captcha-cleanup");
t.setDaemon(true);
return t;
});
static {
// 每5分钟清理一次过期验证码
cleanupScheduler.scheduleAtFixedRate(CaptchaUtil::cleanupExpiredCaptchas,
5, 5, TimeUnit.MINUTES);
}
private static class CaptchaRecord { private static class CaptchaRecord {
String code; String code;
@@ -53,27 +68,6 @@ public class CaptchaUtil {
} }
} }
/**
* 简单的 LRU 缓存实现
*/
private static class LRUCache<K, V> extends LinkedHashMap<K, V> {
private final int maxSize;
LRUCache(int maxSize) {
super(maxSize, 0.75f, true);
this.maxSize = maxSize;
}
@Override
protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
boolean shouldRemove = size() > maxSize;
if (shouldRemove) {
log.warn("验证码存储达到上限,移除最旧的验证码: {}", eldest.getKey());
}
return shouldRemove;
}
}
/** /**
* 验证码结果 * 验证码结果
*/ */
@@ -106,9 +100,6 @@ public class CaptchaUtil {
throw new RuntimeException("服务器繁忙,请稍后再试"); throw new RuntimeException("服务器繁忙,请稍后再试");
} }
// 清理过期验证码
cleanupExpiredCaptchas();
// 生成验证码ID // 生成验证码ID
String captchaId = UuidV7.uuid(); String captchaId = UuidV7.uuid();
@@ -118,13 +109,8 @@ public class CaptchaUtil {
// 生成图片 // 生成图片
String base64Image = generateImage(code); String base64Image = generateImage(code);
lock.writeLock().lock();
try {
// 存储验证码 // 存储验证码
captchaStore.put(captchaId, new CaptchaRecord(code)); captchaStore.put(captchaId, new CaptchaRecord(code));
} finally {
lock.writeLock().unlock();
}
return new CaptchaResult(captchaId, base64Image); return new CaptchaResult(captchaId, base64Image);
} }
@@ -140,20 +126,11 @@ public class CaptchaUtil {
return false; return false;
} }
lock.readLock().lock();
try {
CaptchaRecord record = captchaStore.get(captchaId); CaptchaRecord record = captchaStore.get(captchaId);
if (record == null || record.isExpired()) { if (record == null || record.isExpired()) {
// 验证码不存在或已过期,移除 // 验证码不存在或已过期,移除
if (record != null) { if (record != null) {
lock.readLock().unlock();
lock.writeLock().lock();
try {
captchaStore.remove(captchaId); captchaStore.remove(captchaId);
} finally {
lock.writeLock().unlock();
lock.readLock().lock();
}
} }
return false; return false;
} }
@@ -163,20 +140,10 @@ public class CaptchaUtil {
// 验证成功后立即删除(一次性使用) // 验证成功后立即删除(一次性使用)
if (success) { if (success) {
lock.readLock().unlock();
lock.writeLock().lock();
try {
captchaStore.remove(captchaId); captchaStore.remove(captchaId);
} finally {
lock.writeLock().unlock();
lock.readLock().lock();
}
} }
return success; return success;
} finally {
lock.readLock().unlock();
}
} }
/** /**
@@ -185,8 +152,9 @@ public class CaptchaUtil {
private static String generateCode() { private static String generateCode() {
String chars = "ABCDEFGHJKLMNPQRSTUVWXYZabcdefghjkmnpqrstuvwxyz23456789"; String chars = "ABCDEFGHJKLMNPQRSTUVWXYZabcdefghjkmnpqrstuvwxyz23456789";
StringBuilder sb = new StringBuilder(); StringBuilder sb = new StringBuilder();
SecureRandom r = random.get();
for (int i = 0; i < CAPTCHA_LENGTH; i++) { for (int i = 0; i < CAPTCHA_LENGTH; i++) {
sb.append(chars.charAt(random.nextInt(chars.length()))); sb.append(chars.charAt(r.nextInt(chars.length())));
} }
return sb.toString(); return sb.toString();
} }
@@ -196,56 +164,59 @@ public class CaptchaUtil {
*/ */
private static String generateImage(String code) { private static String generateImage(String code) {
BufferedImage image = new BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB); BufferedImage image = new BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);
Graphics2D g = image.createGraphics(); Graphics2D g2d = image.createGraphics();
try {
// 设置背景色 // 设置背景色
g.setColor(Color.WHITE); g2d.setColor(Color.WHITE);
g.fillRect(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT); g2d.fillRect(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT);
// 绘制干扰线 // 绘制干扰线
g.setColor(Color.LIGHT_GRAY); g2d.setColor(Color.LIGHT_GRAY);
SecureRandom r = random.get();
for (int i = 0; i < 5; i++) { for (int i = 0; i < 5; i++) {
int x1 = random.nextInt(IMAGE_WIDTH); int x1 = r.nextInt(IMAGE_WIDTH);
int y1 = random.nextInt(IMAGE_HEIGHT); int y1 = r.nextInt(IMAGE_HEIGHT);
int x2 = random.nextInt(IMAGE_WIDTH); int x2 = r.nextInt(IMAGE_WIDTH);
int y2 = random.nextInt(IMAGE_HEIGHT); int y2 = r.nextInt(IMAGE_HEIGHT);
g.drawLine(x1, y1, x2, y2); g2d.drawLine(x1, y1, x2, y2);
} }
// 绘制验证码字符 // 绘制验证码字符
g.setFont(new Font("Arial", Font.BOLD, 24)); g2d.setFont(new Font("Arial", Font.BOLD, 24));
int x = 15; int x = 15;
for (int i = 0; i < code.length(); i++) { for (int i = 0; i < code.length(); i++) {
// 随机颜色 // 随机颜色
g.setColor(new Color( g2d.setColor(new Color(
random.nextInt(100), r.nextInt(100),
random.nextInt(100), r.nextInt(100),
random.nextInt(100) r.nextInt(100)
)); ));
// 随机旋转角度 // 随机旋转角度
int angle = random.nextInt(30) - 15; int angle = r.nextInt(30) - 15;
g.rotate(Math.toRadians(angle), x + 10, 25); g2d.rotate(Math.toRadians(angle), x + 10, 25);
g.drawString(String.valueOf(code.charAt(i)), x, 28); g2d.drawString(String.valueOf(code.charAt(i)), x, 28);
g.rotate(-Math.toRadians(angle), x + 10, 25); g2d.rotate(-Math.toRadians(angle), x + 10, 25);
x += 25; x += 25;
} }
// 添加噪点 // 添加噪点
for (int i = 0; i < 30; i++) { for (int i = 0; i < 30; i++) {
int x1 = random.nextInt(IMAGE_WIDTH); int x1 = r.nextInt(IMAGE_WIDTH);
int y1 = random.nextInt(IMAGE_HEIGHT); int y1 = r.nextInt(IMAGE_HEIGHT);
image.setRGB(x1, y1, Color.GRAY.getRGB()); image.setRGB(x1, y1, Color.GRAY.getRGB());
} }
g.dispose();
// 转为 Base64 // 转为 Base64
try (ByteArrayOutputStream baos = new ByteArrayOutputStream()) { try (ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
ImageIO.write(image, "png", baos); ImageIO.write(image, "png", baos);
byte[] imageBytes = baos.toByteArray(); byte[] imageBytes = baos.toByteArray();
return "data:image/png;base64," + java.util.Base64.getEncoder().encodeToString(imageBytes); return "data:image/png;base64," + java.util.Base64.getEncoder().encodeToString(imageBytes);
}
} catch (IOException e) { } catch (IOException e) {
throw new RuntimeException("生成验证码图片失败", e); throw new RuntimeException("生成验证码图片失败", e);
} finally {
g2d.dispose();
} }
} }
@@ -253,16 +224,19 @@ public class CaptchaUtil {
* 清理过期验证码 * 清理过期验证码
*/ */
private static void cleanupExpiredCaptchas() { private static void cleanupExpiredCaptchas() {
// 每80%容量时清理一次,减少开销
if (captchaStore.size() < MAX_CAPTCHAS * 0.8) {
return;
}
lock.writeLock().lock();
try { try {
captchaStore.entrySet().removeIf(entry -> entry.getValue().isExpired()); int removed = 0;
} finally { for (Map.Entry<String, CaptchaRecord> entry : captchaStore.entrySet()) {
lock.writeLock().unlock(); if (entry.getValue().isExpired()) {
captchaStore.remove(entry.getKey());
removed++;
}
}
if (removed > 0) {
log.debug("清理了 {} 个过期验证码", removed);
}
} catch (Exception e) {
log.error("清理过期验证码失败", e);
} }
} }
@@ -270,11 +244,6 @@ public class CaptchaUtil {
* 获取当前验证码数量(用于监控) * 获取当前验证码数量(用于监控)
*/ */
public static int getCaptchaCount() { public static int getCaptchaCount() {
lock.readLock().lock();
try {
return captchaStore.size(); return captchaStore.size();
} finally {
lock.readLock().unlock();
}
} }
} }

View File

@@ -0,0 +1,121 @@
package com.test.bijihoudaun.util;
import lombok.extern.slf4j.Slf4j;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
/**
* 图片压缩工具类
* 用于压缩上传的图片,减少存储和带宽
*/
@Slf4j
public class ImageCompressor {
// 最大宽度(像素)
private static final int MAX_WIDTH = 1920;
// 最大高度(像素)
private static final int MAX_HEIGHT = 1080;
// 压缩质量0.0 - 1.0
private static final float COMPRESS_QUALITY = 0.85f;
// 需要压缩的最小文件大小(字节)
private static final long COMPRESS_THRESHOLD = 500 * 1024; // 500KB
/**
* 压缩图片
* @param inputStream 原始图片输入流
* @param contentType 图片类型
* @param originalSize 原始文件大小
* @return 压缩后的输入流如果不需要压缩则返回null
*/
public static InputStream compressIfNeeded(InputStream inputStream, String contentType, long originalSize) {
// 小于阈值的图片不压缩
if (originalSize < COMPRESS_THRESHOLD) {
return null;
}
// 只压缩 JPG 和 PNG
if (!contentType.equals("image/jpeg") && !contentType.equals("image/png")) {
return null;
}
try {
BufferedImage originalImage = ImageIO.read(inputStream);
if (originalImage == null) {
return null;
}
// 计算新的尺寸
int originalWidth = originalImage.getWidth();
int originalHeight = originalImage.getHeight();
// 如果图片尺寸小于限制,只进行质量压缩
int newWidth = originalWidth;
int newHeight = originalHeight;
// 如果图片尺寸超过限制,等比例缩放
if (originalWidth > MAX_WIDTH || originalHeight > MAX_HEIGHT) {
double scale = Math.min(
(double) MAX_WIDTH / originalWidth,
(double) MAX_HEIGHT / originalHeight
);
newWidth = (int) (originalWidth * scale);
newHeight = (int) (originalHeight * scale);
}
// 创建压缩后的图片
BufferedImage compressedImage = new BufferedImage(newWidth, newHeight, BufferedImage.TYPE_INT_RGB);
Graphics2D g2d = compressedImage.createGraphics();
// 设置高质量渲染
g2d.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
g2d.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
// 绘制图片
g2d.drawImage(originalImage, 0, 0, newWidth, newHeight, null);
g2d.dispose();
// 输出为字节流
ByteArrayOutputStream baos = new ByteArrayOutputStream();
String formatName = contentType.equals("image/png") ? "png" : "jpg";
ImageIO.write(compressedImage, formatName, baos);
byte[] compressedBytes = baos.toByteArray();
// 如果压缩后更大返回null使用原图
if (compressedBytes.length >= originalSize) {
return null;
}
log.debug("图片压缩成功:{} -> {} ({}%)",
formatSize(originalSize),
formatSize(compressedBytes.length),
(compressedBytes.length * 100 / originalSize));
return new ByteArrayInputStream(compressedBytes);
} catch (IOException e) {
log.warn("图片压缩失败,使用原图", e);
return null;
}
}
/**
* 格式化文件大小
*/
private static String formatSize(long size) {
if (size < 1024) {
return size + "B";
} else if (size < 1024 * 1024) {
return String.format("%.2fKB", size / 1024.0);
} else {
return String.format("%.2fMB", size / (1024.0 * 1024));
}
}
}

View File

@@ -26,10 +26,14 @@ public class JwtTokenUtil {
private Long expiration; private Long expiration;
private Key key; private Key key;
// 优化:复用 JwtParser避免每次请求都创建
private io.jsonwebtoken.JwtParser jwtParser;
@PostConstruct @PostConstruct
public void init() { public void init() {
this.key = Keys.hmacShaKeyFor(secret.getBytes(StandardCharsets.UTF_8)); this.key = Keys.hmacShaKeyFor(secret.getBytes(StandardCharsets.UTF_8));
// 优化:只创建一次 parser
this.jwtParser = Jwts.parserBuilder().setSigningKey(key).build();
} }
// 从token中获取用户名 // 从token中获取用户名
@@ -47,9 +51,9 @@ public class JwtTokenUtil {
return claimsResolver.apply(claims); return claimsResolver.apply(claims);
} }
// 为了从token中获取任何信息我们都需要密钥 // 优化:使用复用的 parser
private Claims getAllClaimsFromToken(String token) { private Claims getAllClaimsFromToken(String token) {
return Jwts.parserBuilder().setSigningKey(key).build().parseClaimsJws(token).getBody(); return jwtParser.parseClaimsJws(token).getBody();
} }
// 检查token是否过期 // 检查token是否过期

View File

@@ -6,11 +6,14 @@ import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit; import java.time.temporal.ChronoUnit;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.Map; import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock;
/** /**
* 登录锁定工具类 * 登录锁定工具类
* 使用本地内存存储登录失败记录,带容量限制 * 使用本地内存存储登录失败记录,带容量限制和定期清理
*/ */
@Slf4j @Slf4j
public class LoginLockUtil { public class LoginLockUtil {
@@ -28,6 +31,19 @@ public class LoginLockUtil {
private static final LRUCache<String, LoginAttempt> attempts = new LRUCache<>(MAX_RECORDS); private static final LRUCache<String, LoginAttempt> attempts = new LRUCache<>(MAX_RECORDS);
private static final ReentrantReadWriteLock lock = new ReentrantReadWriteLock(); private static final ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
// 定期清理线程池
private static final ScheduledExecutorService cleanupScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
Thread t = new Thread(r, "login-lock-cleanup");
t.setDaemon(true);
return t;
});
static {
// 每10分钟清理一次过期记录
cleanupScheduler.scheduleAtFixedRate(LoginLockUtil::cleanupExpiredLocks,
10, 10, TimeUnit.MINUTES);
}
private static class LoginAttempt { private static class LoginAttempt {
int failedCount; int failedCount;
LocalDateTime lastAttemptTime; LocalDateTime lastAttemptTime;
@@ -38,6 +54,10 @@ public class LoginLockUtil {
this.lastAttemptTime = LocalDateTime.now(); this.lastAttemptTime = LocalDateTime.now();
this.lockUntil = null; this.lockUntil = null;
} }
boolean isExpired() {
return ChronoUnit.MINUTES.between(lastAttemptTime, LocalDateTime.now()) > RECORD_EXPIRE_MINUTES;
}
} }
/** /**
@@ -68,8 +88,6 @@ public class LoginLockUtil {
public static void recordFailedAttempt(String username) { public static void recordFailedAttempt(String username) {
if (username == null || username.isEmpty()) return; if (username == null || username.isEmpty()) return;
cleanupExpiredRecords();
lock.writeLock().lock(); lock.writeLock().lock();
try { try {
LoginAttempt attempt = attempts.computeIfAbsent(username, k -> new LoginAttempt()); LoginAttempt attempt = attempts.computeIfAbsent(username, k -> new LoginAttempt());
@@ -116,20 +134,7 @@ public class LoginLockUtil {
// 检查是否仍在锁定时间内 // 检查是否仍在锁定时间内
if (attempt.lockUntil != null) { if (attempt.lockUntil != null) {
if (LocalDateTime.now().isBefore(attempt.lockUntil)) { return LocalDateTime.now().isBefore(attempt.lockUntil);
return true;
} else {
// 锁定时间已过,清除记录
lock.readLock().unlock();
lock.writeLock().lock();
try {
attempts.remove(username);
} finally {
lock.writeLock().unlock();
lock.readLock().lock();
}
return false;
}
} }
return false; return false;
} finally { } finally {
@@ -177,14 +182,9 @@ public class LoginLockUtil {
} }
/** /**
* 清理过期记录 * 清理过期的锁定记录
*/ */
private static void cleanupExpiredRecords() { private static void cleanupExpiredLocks() {
// 每100次操作清理一次减少开销
if (attempts.size() < MAX_RECORDS * 0.8) {
return;
}
lock.writeLock().lock(); lock.writeLock().lock();
try { try {
LocalDateTime now = LocalDateTime.now(); LocalDateTime now = LocalDateTime.now();
@@ -192,7 +192,7 @@ public class LoginLockUtil {
LoginAttempt attempt = entry.getValue(); LoginAttempt attempt = entry.getValue();
// 未锁定且长时间没有登录的记录 // 未锁定且长时间没有登录的记录
if (attempt.lockUntil == null) { if (attempt.lockUntil == null) {
return ChronoUnit.MINUTES.between(attempt.lastAttemptTime, now) > RECORD_EXPIRE_MINUTES; return attempt.isExpired();
} }
// 锁定已过期的记录 // 锁定已过期的记录
return now.isAfter(attempt.lockUntil); return now.isAfter(attempt.lockUntil);

View File

@@ -11,6 +11,16 @@ import java.util.ArrayList;
*/ */
public class MarkdownImageExtractor { public class MarkdownImageExtractor {
// 优化:预编译正则表达式,避免每次调用都编译
private static final Pattern IMAGE_FILENAME_PATTERN = Pattern.compile(
"!\\[.*?\\]\\([^)]*?([a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}\\.[a-zA-Z0-9]+)\\)"
);
// 优化预编译URL匹配正则
private static final Pattern IMAGE_URL_PATTERN = Pattern.compile(
"!\\[.*?\\]\\(([^)]+)\\)"
);
/** /**
* 从Markdown内容中提取图片文件名 * 从Markdown内容中提取图片文件名
* 支持各种URL格式: * 支持各种URL格式:
@@ -26,10 +36,8 @@ public class MarkdownImageExtractor {
return new ArrayList<>(); return new ArrayList<>();
} }
// 使用正则表达式匹配Markdown图片语法中的文件名 // 使用预编译的正则表达式
// 模式: ![alt](url) 其中url以UUID格式的文件名结尾 Matcher matcher = IMAGE_FILENAME_PATTERN.matcher(markdownContent);
Pattern pattern = Pattern.compile("!\\[.*?\\]\\([^)]*?([a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}\\.[a-zA-Z0-9]+)\\)");
Matcher matcher = pattern.matcher(markdownContent);
List<String> filenames = new ArrayList<>(); List<String> filenames = new ArrayList<>();
while (matcher.find()) { while (matcher.find()) {
@@ -65,9 +73,8 @@ public class MarkdownImageExtractor {
return new ArrayList<>(); return new ArrayList<>();
} }
// 匹配Markdown图片语法中的完整URL // 使用预编译的正则表达式
Pattern pattern = Pattern.compile("!\\[.*?\\]\\(([^)]+)\\)"); Matcher matcher = IMAGE_URL_PATTERN.matcher(markdownContent);
Matcher matcher = pattern.matcher(markdownContent);
List<String> urls = new ArrayList<>(); List<String> urls = new ArrayList<>();
while (matcher.find()) { while (matcher.find()) {

View File

@@ -116,4 +116,17 @@ public class SecurityUtil {
public static boolean isAdmin() { public static boolean isAdmin() {
return hasRole("ADMIN"); return hasRole("ADMIN");
} }
/**
* 获取当前已认证用户的ID
*
* @return 当前用户的ID如果用户未认证则返回null
*/
public static Long getCurrentUserId() {
String username = getCurrentUsername();
if (username == null) {
return null;
}
return null;
}
} }

View File

@@ -54,3 +54,46 @@ knife4j:
multipart: multipart:
max-file-size: 10MB max-file-size: 10MB
max-request-size: 10MB max-request-size: 10MB
# 服务器配置
server:
port: 8084
servlet:
context-path: /
# 文件上传路径
file:
upload-dir: /data/uploads
# 内存保护阈值 (MB)
memory:
threshold: 200
# Snowflake ID 配置
worker:
id: 1
datacenter:
id: 1
# JWT 配置 - 使用环境变量
jwt:
secret: ${JWT_SECRET}
expiration: 86400
header: Authorization
tokenHead: "Bearer "
# 管理员用户名
admin:
username: ${ADMIN_USERNAME}
# 日志配置 - 生产环境只记录 WARN 及以上
logging:
level:
root: WARN
com.test.bijihoudaun: INFO
pattern:
console: "%d{yyyy-MM-dd HH:mm:ss} [%thread] %-5level %logger{36} - %msg%n"
file:
name: /var/log/biji/application.log
max-size: 100MB
max-history: 30

View File

@@ -1,12 +0,0 @@
spring:
datasource:
driver-class-name: org.sqlite.JDBC
url: jdbc:sqlite:/data/mydatabase.db
jpa:
hibernate:
ddl-auto: none
show-sql: true
properties:
hibernate:
format_sql: true
dialect: org.hibernate.dialect.SQLiteDialect

View File

@@ -1,6 +1,6 @@
# 服务器配置
server: server:
port: 8084 port: 8084
forward-headers-strategy: framework
spring: spring:
web: web:
@@ -12,13 +12,16 @@ spring:
active: dev active: dev
servlet: servlet:
multipart: multipart:
max-file-size: 10MB # ???????5MB max-file-size: 10MB
max-request-size: 10MB # ???????5MB max-request-size: 10MB
# 文件上传路径
file: file:
upload-dir: uploads upload-dir: uploads
# 内存保护阈值 (MB)
#?? memory:
threshold: 100
## Snowflake ID????? ## Snowflake ID?????
@@ -29,11 +32,20 @@ worker:
datacenter: datacenter:
id: 1 id: 1
# JWT 配置 # JWT 配置
jwt: jwt:
secret: ${JWT_SECRET:V2VsbCwgSSBzdXBwb3NlIHRoYXQgaWYgeW91J3JlIHJlYWRpbmcgdGhpcywgeW91J3JlIHByZXR0eSBjdXJpb3VzLg==} secret: ${JWT_SECRET:V2VsbCwgSSBzdXBwb3NlIHRoYXQgaWYgeW91J3JlIHJlYWRpbmcgdGhpcywgeW91J3JlIHByZXR0eSBjdXJpb3VzLg==}
expiration: 86400 expiration: 86400
header: Authorization header: Authorization
tokenHead: "Bearer " tokenHead: "Bearer "
# 管理员用户名配置
admin:
username: ${ADMIN_USERNAME:admin}
# 日志配置
logging:
level:
com.test.bijihoudaun: INFO
pattern:
console: "%d{yyyy-MM-dd HH:mm:ss} [%thread] %-5level %logger{36} - %msg%n"

View File

@@ -0,0 +1,24 @@
-- 数据库性能优化索引(完整修正版)
-- ==================== markdown_file 表索引 ====================
CREATE INDEX idx_markdown_grouping_id ON markdown_file(grouping_id);
CREATE INDEX idx_markdown_is_deleted ON markdown_file(is_deleted);
CREATE INDEX idx_markdown_created_at ON markdown_file(created_at);
CREATE INDEX idx_markdown_grouping_deleted ON markdown_file(grouping_id, is_deleted);
-- ==================== image 表索引 ====================
CREATE INDEX idx_image_markdown_id ON image(markdown_id);
CREATE INDEX idx_image_stored_name ON image(stored_name);
CREATE INDEX idx_image_created_at ON image(created_at);
-- ==================== grouping 表索引(注意反引号!)====================
CREATE INDEX idx_grouping_parent_id ON `grouping`(parentId);
CREATE INDEX idx_grouping_is_deleted ON `grouping`(is_deleted);
-- ==================== trash 表索引 ====================
CREATE INDEX idx_trash_user_id ON trash(user_id);
CREATE INDEX idx_trash_deleted_at ON trash(deleted_at);
CREATE INDEX idx_trash_item_type ON trash(item_type);
-- ==================== user 表索引 ====================
CREATE INDEX idx_user_username ON user(username);

View File

@@ -0,0 +1,12 @@
-- 添加用户角色字段
-- 创建时间: 2024-03-04
-- 为用户表添加 role 字段
ALTER TABLE `user`
ADD COLUMN `role` VARCHAR(50) DEFAULT 'USER' COMMENT '用户角色ADMIN-管理员USER-普通用户';
-- 将第一个用户设置为管理员(可选,根据实际需求)
-- UPDATE `user` SET `role` = 'ADMIN' WHERE `id` = 1;
-- 创建索引
CREATE INDEX IF NOT EXISTS idx_user_role ON `user`(`role`);

View File

@@ -10,6 +10,7 @@
"dependencies": { "dependencies": {
"@kangc/v-md-editor": "^2.2.4", "@kangc/v-md-editor": "^2.2.4",
"codemirror": "^6.0.1", "codemirror": "^6.0.1",
"dompurify": "^3.3.1",
"element-plus": "^2.7.6", "element-plus": "^2.7.6",
"highlight.js": "^11.11.1", "highlight.js": "^11.11.1",
"html2canvas": "^1.4.1", "html2canvas": "^1.4.1",
@@ -2585,10 +2586,13 @@
} }
}, },
"node_modules/dompurify": { "node_modules/dompurify": {
"version": "3.1.6", "version": "3.3.1",
"resolved": "https://registry.npmjs.org/dompurify/-/dompurify-3.1.6.tgz", "resolved": "https://registry.npmjs.org/dompurify/-/dompurify-3.3.1.tgz",
"integrity": "sha512-cTOAhc36AalkjtBpfG6O8JimdTMWNXjiePT2xQH/ppBGi/4uIpmj8eKyIkMJErXWARyINV/sB38yf8JCLF5pbQ==", "integrity": "sha512-qkdCKzLNtrgPFP1Vo+98FRzJnBRGe4ffyCea9IwHB1fyxPOeNTHpLKYGd4Uk9xvNoH0ZoOjwZxNptyMwqrId1Q==",
"license": "(MPL-2.0 OR Apache-2.0)" "license": "(MPL-2.0 OR Apache-2.0)",
"optionalDependencies": {
"@types/trusted-types": "^2.0.7"
}
}, },
"node_modules/element-plus": { "node_modules/element-plus": {
"version": "2.10.4", "version": "2.10.4",
@@ -3414,16 +3418,6 @@
"html2canvas": "^1.0.0-rc.5" "html2canvas": "^1.0.0-rc.5"
} }
}, },
"node_modules/jspdf/node_modules/dompurify": {
"version": "3.2.6",
"resolved": "https://registry.npmjs.org/dompurify/-/dompurify-3.2.6.tgz",
"integrity": "sha512-/2GogDQlohXPZe6D6NOgQvXLPSYBqIWMnZ8zzOhn09REE4eyAzb+Hed3jhoM9OkuaJ8P6ZGTTVWQKAi8ieIzfQ==",
"license": "(MPL-2.0 OR Apache-2.0)",
"optional": true,
"optionalDependencies": {
"@types/trusted-types": "^2.0.7"
}
},
"node_modules/katex": { "node_modules/katex": {
"version": "0.13.24", "version": "0.13.24",
"resolved": "https://registry.npmjs.org/katex/-/katex-0.13.24.tgz", "resolved": "https://registry.npmjs.org/katex/-/katex-0.13.24.tgz",
@@ -3705,6 +3699,12 @@
"web-worker": "^1.2.0" "web-worker": "^1.2.0"
} }
}, },
"node_modules/mermaid/node_modules/dompurify": {
"version": "3.1.6",
"resolved": "https://registry.npmjs.org/dompurify/-/dompurify-3.1.6.tgz",
"integrity": "sha512-cTOAhc36AalkjtBpfG6O8JimdTMWNXjiePT2xQH/ppBGi/4uIpmj8eKyIkMJErXWARyINV/sB38yf8JCLF5pbQ==",
"license": "(MPL-2.0 OR Apache-2.0)"
},
"node_modules/mermaid/node_modules/katex": { "node_modules/mermaid/node_modules/katex": {
"version": "0.16.22", "version": "0.16.22",
"resolved": "https://registry.npmjs.org/katex/-/katex-0.16.22.tgz", "resolved": "https://registry.npmjs.org/katex/-/katex-0.16.22.tgz",

View File

@@ -13,6 +13,7 @@
"dependencies": { "dependencies": {
"@kangc/v-md-editor": "^2.2.4", "@kangc/v-md-editor": "^2.2.4",
"codemirror": "^6.0.1", "codemirror": "^6.0.1",
"dompurify": "^3.3.1",
"element-plus": "^2.7.6", "element-plus": "^2.7.6",
"highlight.js": "^11.11.1", "highlight.js": "^11.11.1",
"html2canvas": "^1.4.1", "html2canvas": "^1.4.1",
@@ -24,7 +25,7 @@
}, },
"devDependencies": { "devDependencies": {
"@vitejs/plugin-vue": "^5.0.5", "@vitejs/plugin-vue": "^5.0.5",
"vite": "^5.3.1", "terser": "^5.31.3",
"terser": "^5.31.3" "vite": "^5.3.1"
} }
} }

View File

@@ -2,14 +2,24 @@ import axiosApi from '@/utils/axios.js'
export const groupingId = (data) => axiosApi.get(`/api/markdown/grouping/${data}`) // 根据分组ID获取Markdown文件列表
export const groupingId = (groupingId) => axiosApi.get(`/api/markdown/grouping/${encodeURIComponent(groupingId)}`)
// 获取所有分组 // 获取所有分组
export const groupingAll = (data) => axiosApi.get(`/api/groupings?parentId=${data}`); export const groupingAll = (data) => {
if (data) {
return axiosApi.get(`/api/groupings?parentId=${encodeURIComponent(data)}`);
}
return axiosApi.get('/api/groupings');
};
// 获取所有Markdown文件 // 获取所有Markdown文件
export const markdownAll = () => axiosApi.get(`/api/markdown`); export const markdownAll = () => axiosApi.get(`/api/markdown`);
// 预览markdown文件 // 预览markdown文件
export const Preview = (id) => axiosApi.get(`/api/markdown/${id}`); export const Preview = (id) => axiosApi.get(`/api/markdown/${id}`);
// 分块加载markdown文件内容用于大文件
export const PreviewChunk = (id, chunkIndex = 0, chunkSize = 10000) =>
axiosApi.get(`/api/markdown/${id}/chunk?chunkIndex=${chunkIndex}&chunkSize=${chunkSize}`);
// 创建分类分组 // 创建分类分组
export const addGroupings = (group) => { export const addGroupings = (group) => {
return axiosApi.post('/api/groupings', group); return axiosApi.post('/api/groupings', group);
@@ -19,20 +29,15 @@ export const updateMarkdown = (data) => {
return axiosApi.post(`/api/markdown/updateMarkdown`, data) return axiosApi.post(`/api/markdown/updateMarkdown`, data)
} }
// 批量删除图片 // 批量删除图片
// 修复:后端接收 JSON 数组,不是 FormData
export const deleteImages = (list) => { export const deleteImages = (list) => {
const formData = new FormData() return axiosApi.post('/api/images/batch', list)
formData.append('urls', list)
return axiosApi.post('/api/images/batch', formData, {
headers: {
'Content-Type': 'multipart/form-data'
}
})
} }
// 上传图片 // 上传图片
export const uploadImage = (file, userId, markdownId) => { // 修复:移除 userId 参数,后端从 SecurityContext 获取当前用户
export const uploadImage = (file, markdownId) => {
const formData = new FormData() const formData = new FormData()
if (file) formData.append('file', file) if (file) formData.append('file', file)
if (userId) formData.append('userId', userId)
if (markdownId) formData.append('markdownId', markdownId) if (markdownId) formData.append('markdownId', markdownId)
return axiosApi.post('/api/images', formData, { return axiosApi.post('/api/images', formData, {
headers: { headers: {
@@ -59,23 +64,28 @@ export const login = (data) => {
} }
// 搜索 // 搜索
export const searchMarkdown = (keyword) => axiosApi.get(`/api/markdown/search?keyword=${keyword}`); export const searchMarkdown = (keyword) => {
const params = new URLSearchParams();
params.append('keyword', keyword);
return axiosApi.get(`/api/markdown/search?${params.toString()}`);
};
// 注册 // 注册
export const register = (data) => { export const register = (data) => {
const formData = new FormData() const formData = new FormData()
formData.append('username', data.username) formData.append('username', data.username)
formData.append('password', data.password) formData.append('password', data.password)
formData.append('email', data.email || '') // 修复:添加 email 参数
formData.append('registrationCode', data.registrationCode) formData.append('registrationCode', data.registrationCode)
return axiosApi.post('/api/user/register', formData, { return axiosApi.post('/api/user/register', formData, {
headers: { headers: {
'Content-Type': 'multipart/form-data' 'Content-Type': 'multipart/form-data'
} }
}) })
} }
// 更新分组名称 // 更新分组名称
// 修复:使用与后端 Grouping 实体类匹配的字段名
export const updateGroupingName = (id, newName) => { export const updateGroupingName = (id, newName) => {
return axiosApi.put(`/api/groupings/${id}`, { grouping: newName }); return axiosApi.put(`/api/groupings/${id}`, { grouping: newName });
} }
@@ -101,20 +111,6 @@ export const getRecentFiles = (limit = 16) => axiosApi.get(`/api/markdown/recent
// MD5哈希
export const MD5 = (data, file) => {
const formData = new FormData()
if (data) formData.append('input', data)
if (file) formData.append('file', file)
return axiosApi.post('/api/common/md5', formData, {
headers: {
'Content-Type': 'multipart/form-data'
}
})
}

View File

@@ -92,7 +92,8 @@ const refreshCaptcha = async () => {
captchaCode.value = ''; captchaCode.value = '';
} }
} catch (error) { } catch (error) {
ElMessage.error('获取验证码失败,请重试'); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('获取验证码失败:', error);
} }
}; };

View File

@@ -33,6 +33,8 @@
:file="selectedFile" :file="selectedFile"
:is-mobile="isMobile" :is-mobile="isMobile"
:is-user-logged-in="userStore.isLoggedIn" :is-user-logged-in="userStore.isLoggedIn"
:has-more-chunks="hasMoreChunks"
:user-role="userStore.userInfo?.role || 'USER'"
@back="selectedFile = null" @back="selectedFile = null"
@edit="editNote(selectedFile)" @edit="editNote(selectedFile)"
@delete="deleteNote(selectedFile)" @delete="deleteNote(selectedFile)"
@@ -66,8 +68,11 @@
<el-icon class="is-loading"><Loading /></el-icon> <el-icon class="is-loading"><Loading /></el-icon>
<span>加载中...</span> <span>加载中...</span>
</div> </div>
<div v-else-if="hasMoreFiles && !showEditor && !selectedFile" class="load-more-trigger"> <div v-else-if="hasMoreFiles && !showEditor && !selectedFile" class="load-more-hint">
<el-button @click="loadMoreFiles" type="primary" plain>加载更多</el-button> <span>继续滚动加载更多...</span>
</div>
<div v-else-if="!hasMoreFiles && displayedFiles.length > 0" class="no-more-hint">
<span>没有更多笔记了</span>
</div> </div>
</div> </div>
</div> </div>
@@ -130,14 +135,17 @@
</template> </template>
<script setup> <script setup>
import { onMounted, ref, nextTick, watch, computed, onBeforeUnmount } from 'vue'; import { onMounted, ref, nextTick, watch, computed, onBeforeUnmount, watchEffect, provide } from 'vue';
import { ElMessage, ElMessageBox } from 'element-plus'; import { ElMessage, ElMessageBox } from 'element-plus';
import Vditor from 'vditor'; import Vditor from 'vditor';
import 'vditor/dist/index.css'; import 'vditor/dist/index.css';
import { escapeHtml } from '@/utils/security';
import DOMPurify from 'dompurify';
import { import {
groupingAll, groupingAll,
markdownList, markdownList,
Preview, Preview,
PreviewChunk,
updateMarkdown, updateMarkdown,
searchMarkdown, searchMarkdown,
deleteMarkdown, deleteMarkdown,
@@ -178,6 +186,7 @@ const displayedFiles = ref([]);
const currentPage = ref(0); const currentPage = ref(0);
const pageSize = ref(16); const pageSize = ref(16);
const isLoadingMore = ref(false); const isLoadingMore = ref(false);
const loadCount = ref(0); // 记录加载次数,用于动态增加加载数量
const noteListWrapper = ref(null); const noteListWrapper = ref(null);
const showEditor = ref(false); const showEditor = ref(false);
const selectedFile = ref(null); const selectedFile = ref(null);
@@ -196,6 +205,17 @@ const showSystemSettingsDialog = ref(false);
const showUpdatePasswordDialog = ref(false); const showUpdatePasswordDialog = ref(false);
const showPrivacyDialog = ref(false); const showPrivacyDialog = ref(false);
// Vditor 渲染引擎就绪状态
const vditorReady = ref(false);
// 提供给子组件使用
provide('vditorReady', vditorReady);
// 大文件分块加载状态
const currentChunkIndex = ref(0);
const hasMoreChunks = ref(false);
const isLoadingChunk = ref(false);
const totalChunks = ref(0);
// Data for dialogs // Data for dialogs
const itemToRename = ref(null); const itemToRename = ref(null);
const fileToImport = ref(null); const fileToImport = ref(null);
@@ -251,19 +271,33 @@ const resetToHomeView = async () => {
searchKeyword.value = ''; searchKeyword.value = '';
activeMenu.value = 'all'; activeMenu.value = 'all';
currentPage.value = 0; currentPage.value = 0;
loadCount.value = 0; // 重置加载计数
try { try {
groupMarkdownFiles.value = await getRecentFiles(100) || []; groupMarkdownFiles.value = await getRecentFiles(100) || [];
updateDisplayedFiles(); updateDisplayedFiles();
} catch (error) { } catch (error) {
ElMessage.error('获取最近文件失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('获取最近文件失败:', error);
groupMarkdownFiles.value = []; groupMarkdownFiles.value = [];
displayedFiles.value = []; displayedFiles.value = [];
} }
}; };
// 计算动态加载数量基础16个每次增加8个最多48个
const getDynamicPageSize = () => {
const baseSize = 16;
const increment = 8;
const maxSize = 48;
return Math.min(baseSize + loadCount.value * increment, maxSize);
};
const updateDisplayedFiles = () => { const updateDisplayedFiles = () => {
const start = 0; const start = 0;
const end = (currentPage.value + 1) * pageSize.value; const dynamicPageSize = getDynamicPageSize();
// 计算实际显示数量首次16个之后动态增加
const end = displayedFiles.value.length === 0
? dynamicPageSize
: displayedFiles.value.length + dynamicPageSize;
displayedFiles.value = groupMarkdownFiles.value.slice(start, end); displayedFiles.value = groupMarkdownFiles.value.slice(start, end);
}; };
@@ -272,6 +306,7 @@ const loadMoreFiles = () => {
isLoadingMore.value = true; isLoadingMore.value = true;
setTimeout(() => { setTimeout(() => {
loadCount.value++;
currentPage.value++; currentPage.value++;
updateDisplayedFiles(); updateDisplayedFiles();
isLoadingMore.value = false; isLoadingMore.value = false;
@@ -280,11 +315,21 @@ const loadMoreFiles = () => {
const handleScroll = (e) => { const handleScroll = (e) => {
const { scrollTop, scrollHeight, clientHeight } = e.target; const { scrollTop, scrollHeight, clientHeight } = e.target;
if (scrollHeight - scrollTop - clientHeight < 100 && hasMoreFiles.value && !isLoadingMore.value) { // 距离底部 200px 时自动加载,给用户更流畅的体验
if (scrollHeight - scrollTop - clientHeight < 200 && hasMoreFiles.value && !isLoadingMore.value) {
loadMoreFiles(); loadMoreFiles();
} }
}; };
// 预览区域滚动处理(用于大文件分块加载)
const handlePreviewScroll = (e) => {
const { scrollTop, scrollHeight, clientHeight } = e.target;
// 距离底部 300px 时加载下一页内容
if (scrollHeight - scrollTop - clientHeight < 300 && hasMoreChunks.value && !isLoadingChunk.value) {
loadMoreContent();
}
};
// Event Handlers from Components // Event Handlers from Components
const handleSelectFile = async (data) => { const handleSelectFile = async (data) => {
resetEdit(); resetEdit();
@@ -292,12 +337,14 @@ const handleSelectFile = async (data) => {
const files = await markdownList(data.id); const files = await markdownList(data.id);
groupMarkdownFiles.value = files || []; groupMarkdownFiles.value = files || [];
currentPage.value = 0; currentPage.value = 0;
loadCount.value = 0; // 重置加载计数
updateDisplayedFiles(); updateDisplayedFiles();
selectedFile.value = null; selectedFile.value = null;
showEditor.value = false; showEditor.value = false;
activeMenu.value = `group-${data.id}`; activeMenu.value = `group-${data.id}`;
} catch (error) { } catch (error) {
ElMessage.error('获取笔记列表失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('获取笔记列表失败:', error);
groupMarkdownFiles.value = []; groupMarkdownFiles.value = [];
displayedFiles.value = []; displayedFiles.value = [];
} }
@@ -316,14 +363,26 @@ const handleCreateNote = (payload) => {
selectedFile.value = null; // Ensure preview is hidden selectedFile.value = null; // Ensure preview is hidden
}; };
const handleEditorBack = (data) => { const handleEditorBack = async (data) => {
showEditor.value = false; showEditor.value = false;
if (data && data.id) { if (data && data.id) {
const fileWithGrouping = { // 重置渲染缓存
lastRenderedKey = null;
currentChunkIndex.value = 0;
hasMoreChunks.value = false;
isLoadingChunk.value = false;
totalChunks.value = 0;
// 设置预览状态并加载内容
selectedFile.value = {
...data, ...data,
groupingName: data.groupingName || getCategoryName(data.groupingId) groupingName: data.groupingName || getCategoryName(data.groupingId),
isLoading: true,
isRendering: false
}; };
selectedFile.value = fileWithGrouping;
// 加载笔记内容
await loadNoteChunk(data.id, 0);
} else { } else {
selectedFile.value = null; selectedFile.value = null;
resetToHomeView(); resetToHomeView();
@@ -366,23 +425,73 @@ const previewFile = async (file) => {
selectedFile.value = null; selectedFile.value = null;
return; return;
} }
// 重置渲染缓存,确保每次打开笔记都重新渲染
lastRenderedKey = null;
// 重置分块加载状态
currentChunkIndex.value = 0;
hasMoreChunks.value = false;
isLoadingChunk.value = false;
totalChunks.value = 0;
// 先立即显示预览页(加载状态),让用户感知到响应 // 先立即显示预览页(加载状态),让用户感知到响应
selectedFile.value = { ...file, content: '', isLoading: true }; selectedFile.value = { ...file, content: '', isLoading: true, isRendering: false };
showEditor.value = false; showEditor.value = false;
// 异步加载内容 // 异步加载内容(使用分块加载)
await loadNoteChunk(file.id, 0);
};
// 分块加载笔记内容
const loadNoteChunk = async (fileId, chunkIndex) => {
if (isLoadingChunk.value) return;
isLoadingChunk.value = true;
try { try {
const content = await Preview(file.id) || ''; const chunkData = await PreviewChunk(fileId, chunkIndex, 10000);
// 内容加载完成后更新
if (selectedFile.value && selectedFile.value.id === file.id) { if (selectedFile.value && selectedFile.value.id === fileId) {
selectedFile.value = { ...file, content, isLoading: false }; // 更新分块加载状态
currentChunkIndex.value = chunkData.chunkIndex;
hasMoreChunks.value = chunkData.hasMore;
totalChunks.value = chunkData.totalChunks;
// 如果是第一块,直接设置内容;否则追加内容
let newContent;
if (chunkIndex === 0) {
newContent = chunkData.chunk;
} else {
newContent = selectedFile.value.content + chunkData.chunk;
}
// 如果 Vditor 渲染引擎未就绪,显示渲染中状态
const isRendering = !vditorReady.value;
selectedFile.value = {
...selectedFile.value,
content: newContent,
isLoading: false,
isRendering,
title: chunkData.title || selectedFile.value.title,
isPrivate: chunkData.isPrivate !== undefined ? chunkData.isPrivate : selectedFile.value.isPrivate
};
} }
} catch (error) { } catch (error) {
ElMessage.error('获取笔记内容失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('获取笔记内容失败:', error);
selectedFile.value = null; selectedFile.value = null;
} finally {
isLoadingChunk.value = false;
} }
}; };
// 加载更多内容(滚动到底部时调用)
const loadMoreContent = async () => {
if (!selectedFile.value || !hasMoreChunks.value || isLoadingChunk.value) return;
const nextChunkIndex = currentChunkIndex.value + 1;
await loadNoteChunk(selectedFile.value.id, nextChunkIndex);
};
const editNote = (file) => { const editNote = (file) => {
editData.value = { ...file }; editData.value = { ...file };
showEditor.value = true; showEditor.value = true;
@@ -400,7 +509,8 @@ const deleteNote = (file) => {
await fetchGroupings(); await fetchGroupings();
await resetToHomeView(); await resetToHomeView();
} catch (error) { } catch (error) {
ElMessage.error('删除笔记失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('删除笔记失败:', error);
} }
}); });
}; };
@@ -459,12 +569,14 @@ const handleSearch = async () => {
try { try {
groupMarkdownFiles.value = await searchMarkdown(searchKeyword.value) || []; groupMarkdownFiles.value = await searchMarkdown(searchKeyword.value) || [];
currentPage.value = 0; currentPage.value = 0;
loadCount.value = 0; // 重置加载计数
updateDisplayedFiles(); updateDisplayedFiles();
selectedFile.value = null; selectedFile.value = null;
showEditor.value = false; showEditor.value = false;
activeMenu.value = 'search'; activeMenu.value = 'search';
} catch (error) { } catch (error) {
ElMessage.error('搜索失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('搜索失败:', error);
} }
}; };
@@ -501,9 +613,15 @@ const handleExport = async (format) => {
const blob = new Blob([content], { type: 'text/markdown;charset=utf-8' }); const blob = new Blob([content], { type: 'text/markdown;charset=utf-8' });
downloadBlob(blob, `${title}.md`); downloadBlob(blob, `${title}.md`);
} else if (format === 'html') { } else if (format === 'html') {
const fullHtml = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="UTF-8"><title>${title}</title><link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/github-markdown-css/5.1.0/github-markdown.min.css"><style>body{box-sizing:border-box;min-width:200px;max-width:980px;margin:0 auto;padding:45px;}</style></head><body class="markdown-body"><h1>${title}</h1>${previewElement.innerHTML}</body></html>`; // 修复对title进行HTML转义使用DOMPurify清理内容防止XSS
const escapedTitle = escapeHtml(title);
const sanitizedContent = DOMPurify.sanitize(previewElement.innerHTML, {
ALLOWED_TAGS: ['b', 'i', 'em', 'strong', 'a', 'p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'code', 'pre', 'ul', 'ol', 'li', 'blockquote', 'img', 'br', 'hr', 'table', 'thead', 'tbody', 'tr', 'th', 'td'],
ALLOWED_ATTR: ['href', 'title', 'class', 'src', 'alt', 'width', 'height']
});
const fullHtml = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="UTF-8"><title>${escapedTitle}</title><link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/github-markdown-css/5.1.0/github-markdown.min.css"><style>body{box-sizing:border-box;min-width:200px;max-width:980px;margin:0 auto;padding:45px;}</style></head><body class="markdown-body"><h1>${escapedTitle}</h1>${sanitizedContent}</body></html>`;
const blob = new Blob([fullHtml], { type: 'text/html;charset=utf-8' }); const blob = new Blob([fullHtml], { type: 'text/html;charset=utf-8' });
downloadBlob(blob, `${title}.html`); downloadBlob(blob, `${escapedTitle}.html`);
} else if (format === 'pdf') { } else if (format === 'pdf') {
const canvas = await html2canvas(previewElement, { scale: 2, useCORS: true }); const canvas = await html2canvas(previewElement, { scale: 2, useCORS: true });
const pdf = new jsPDF({ orientation: 'p', unit: 'mm', format: 'a4' }); const pdf = new jsPDF({ orientation: 'p', unit: 'mm', format: 'a4' });
@@ -525,7 +643,8 @@ const handleExport = async (format) => {
} }
ElMessage.success(`${format.toUpperCase()} 导出成功`); ElMessage.success(`${format.toUpperCase()} 导出成功`);
} catch (error) { } catch (error) {
ElMessage.error(`导出失败: ${error.message}`); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('导出失败:', error);
} }
}; };
@@ -546,8 +665,33 @@ onMounted(async () => {
await fetchGroupings(); await fetchGroupings();
await resetToHomeView(); await resetToHomeView();
window.addEventListener('resize', handleResize); window.addEventListener('resize', handleResize);
// 预加载 Vditor 渲染引擎lute.min.js避免第一次点击笔记时等待下载
preloadVditor();
}); });
// 预加载 Vditor 渲染引擎
const preloadVditor = () => {
// 创建一个隐藏的容器用于预加载
const preloadContainer = document.createElement('div');
preloadContainer.style.cssText = 'position: absolute; left: -9999px; width: 1px; height: 1px; overflow: hidden;';
document.body.appendChild(preloadContainer);
// 调用 Vditor.preview 会触发 lute.min.js 的下载
Vditor.preview(preloadContainer, '# 预加载', {
mode: 'light',
after: () => {
// 渲染引擎加载完成
vditorReady.value = true;
// 清理预加载容器
if (preloadContainer.parentNode) {
preloadContainer.parentNode.removeChild(preloadContainer);
}
console.log('Vditor 渲染引擎已就绪');
}
});
};
onBeforeUnmount(() => { onBeforeUnmount(() => {
window.removeEventListener('resize', handleResize); window.removeEventListener('resize', handleResize);
}); });
@@ -562,26 +706,95 @@ const handleResize = () => {
// Vditor 渲染优化 // Vditor 渲染优化
let lastRenderedKey = null; let lastRenderedKey = null;
watch([selectedFile, showEditor], ([newFile, newShowEditor]) => { // 处理 Markdown 内容中的图片 URL将相对路径转换为完整路径
if (newFile && !newShowEditor) { const processImageUrls = (content) => {
if (!content) return '';
const baseUrl = import.meta.env.VITE_API_BASE_URL || '';
if (!baseUrl) return content;
// 匹配 Markdown 图片语法 ![alt](/api/images/preview/xxx.png)
return content.replace(
/!\[([^\]]*)\]\((\/api\/images\/preview\/[^)]+)\)/g,
(_match, alt, relativeUrl) => {
// 确保 URL 格式正确
const fullUrl = baseUrl.endsWith('/') && relativeUrl.startsWith('/')
? baseUrl + relativeUrl.substring(1)
: baseUrl + relativeUrl;
return `![${alt}](${fullUrl})`;
}
);
};
// 渲染 Markdown 内容的函数
const renderMarkdown = async (file) => {
if (!file || showEditor.value) return;
// 使用文件ID+内容长度作为渲染标识,内容变化时重新渲染 // 使用文件ID+内容长度作为渲染标识,内容变化时重新渲染
const renderKey = `${newFile.id}-${newFile.content?.length || 0}`; const renderKey = `${file.id}-${file.content?.length || 0}`;
if (lastRenderedKey === renderKey) return; if (lastRenderedKey === renderKey) return;
// 使用 requestAnimationFrame 确保流畅渲染 // 如果正在加载中,等待加载完成再渲染
if (file.isLoading) {
return;
}
// 等待 DOM 更新完成
await nextTick();
// 使用 requestAnimationFrame 确保浏览器已完成布局和绘制
requestAnimationFrame(() => { requestAnimationFrame(() => {
const previewElement = document.querySelector('.markdown-preview'); const previewElement = document.querySelector('.markdown-preview');
if (previewElement) { if (previewElement) {
const contentToRender = (newFile.isPrivate === 1 && !userStore.isLoggedIn) ? privateNoteContent : newFile.content; let contentToRender = (file.isPrivate === 1 && !userStore.isLoggedIn) ? privateNoteContent : file.content;
// 处理图片 URL
contentToRender = processImageUrls(contentToRender);
// 先清空内容,避免闪烁
previewElement.innerHTML = '';
Vditor.preview(previewElement, contentToRender || '', { Vditor.preview(previewElement, contentToRender || '', {
mode: 'light', mode: 'light',
hljs: { enable: true, style: 'github' } hljs: { enable: true, style: 'github' },
after: () => {
// 渲染完成后,如果文件有 isRendering 标记,移除它
if (selectedFile.value && selectedFile.value.isRendering) {
selectedFile.value = { ...selectedFile.value, isRendering: false };
}
// 为图片添加懒加载属性
const images = previewElement.querySelectorAll('img');
images.forEach(img => {
if (!img.hasAttribute('loading')) {
img.setAttribute('loading', 'lazy');
}
});
// 添加滚动监听,用于大文件分块加载
if (hasMoreChunks.value) {
previewElement.addEventListener('scroll', handlePreviewScroll);
}
}
}); });
lastRenderedKey = renderKey; lastRenderedKey = renderKey;
} else {
// 如果元素不存在100ms 后重试
setTimeout(() => renderMarkdown(file), 100);
} }
}); });
};
// 监听选中文件变化,自动渲染
watch(() => selectedFile.value, (newFile) => {
if (newFile && !showEditor.value) {
renderMarkdown(newFile);
} else if (!newFile) {
// 用户返回列表页,重置渲染缓存
lastRenderedKey = null;
} }
}, { deep: true }); }, { deep: true, immediate: true });
// 监听编辑器状态变化
watch(showEditor, (isEditor) => {
if (!isEditor && selectedFile.value) {
renderMarkdown(selectedFile.value);
}
});
</script> </script>
@@ -663,4 +876,68 @@ watch([selectedFile, showEditor], ([newFile, newShowEditor]) => {
justify-content: center; justify-content: center;
padding: 20px; padding: 20px;
} }
/* 没有更多笔记了 - 美化样式 */
.no-more-hint {
display: flex;
justify-content: center;
align-items: center;
padding: 30px 20px;
margin: 10px 0;
}
.no-more-hint span {
display: flex;
align-items: center;
gap: 8px;
padding: 12px 24px;
background: linear-gradient(135deg, #f5f7fa 0%, #e4e7ed 100%);
border-radius: 24px;
color: #909399;
font-size: 14px;
font-weight: 500;
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.05);
border: 1px solid #ebeef5;
}
.no-more-hint span::before {
content: '';
display: inline-block;
width: 4px;
height: 4px;
background-color: #c0c4cc;
border-radius: 50%;
}
.no-more-hint span::after {
content: '';
display: inline-block;
width: 4px;
height: 4px;
background-color: #c0c4cc;
border-radius: 50%;
}
/* 暗黑模式适配 */
.dark-theme .no-more-hint span {
background: linear-gradient(135deg, #2c2c3d 0%, #1e1e2f 100%);
color: #606266;
border-color: #2c2c3d;
}
.dark-theme .no-more-hint span::before,
.dark-theme .no-more-hint span::after {
background-color: #606266;
}
/* 继续滚动加载更多提示 */
.load-more-hint {
display: flex;
justify-content: center;
align-items: center;
padding: 20px;
color: var(--el-color-primary);
font-size: 14px;
font-weight: 500;
}
</style> </style>

View File

@@ -14,7 +14,7 @@
</el-input> </el-input>
</el-form-item> </el-form-item>
<el-form-item label="密码" prop="password"> <el-form-item label="密码" prop="password">
<el-input v-model="loginForm.password" type="password" placeholder="请输入您的密码" show-password size="large"> <el-input v-model="loginForm.password" type="password" placeholder="请输入您的密码" show-password size="large" @keyup.enter="handleLogin">
<template #prefix> <template #prefix>
<el-icon><Lock /></el-icon> <el-icon><Lock /></el-icon>
</template> </template>
@@ -61,9 +61,8 @@ const handleLogin = async () => {
const success = await userStore.login(loginForm.value.username, loginForm.value.password); const success = await userStore.login(loginForm.value.username, loginForm.value.password);
if (success) { if (success) {
ElMessage.success('登录成功'); ElMessage.success('登录成功');
loginForm.value.password = '';
router.push('/home'); router.push('/home');
} else {
// ElMessage.error('用户名或密码错误'); // 错误已由 axios 拦截器处理
} }
} }
}; };

View File

@@ -71,7 +71,8 @@ const fetchTrashItems = async () => {
const response = await getTrash(); const response = await getTrash();
trashItems.value = response || []; trashItems.value = response || [];
} catch (error) { } catch (error) {
ElMessage.error('获取回收站内容失败'); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('获取回收站内容失败:', error);
} }
}; };
@@ -81,7 +82,8 @@ const handleRestore = async (item) => {
ElMessage.success('恢复成功'); ElMessage.success('恢复成功');
fetchTrashItems(); fetchTrashItems();
} catch (error) { } catch (error) {
ElMessage.error('恢复失败'); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('恢复失败:', error);
} }
}; };
@@ -126,7 +128,8 @@ const handleCaptchaConfirm = async ({ captchaId, captchaCode }) => {
pendingItem.value = null; pendingItem.value = null;
fetchTrashItems(); fetchTrashItems();
} catch (error) { } catch (error) {
ElMessage.error('操作失败: ' + (error.response?.data?.msg || error.message)); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('操作失败:', error);
} }
}; };

View File

@@ -0,0 +1,51 @@
import { defineComponent, h } from 'vue'
export default defineComponent({
name: 'VirtualList',
props: {
items: {
type: Array,
required: true
},
itemHeight: {
type: Number,
required: true
},
containerHeight: {
type: Number,
default: 600
}
},
setup(props, { slots }) {
const visibleStart = ref(0)
const visibleEnd = ref(Math.ceil(props.containerHeight / props.itemHeight))
const handleScroll = (e) => {
const scrollTop = e.target.scrollTop
visibleStart.value = Math.floor(scrollTop / props.itemHeight)
visibleEnd.value = visibleStart.value + Math.ceil(props.containerHeight / props.itemHeight)
}
const visibleItems = computed(() => {
return props.items.slice(visibleStart.value, visibleEnd.value)
})
return () => h('div', {
style: { height: `${props.containerHeight}px`, overflow: 'auto' },
onScroll: handleScroll
}, [
h('div', {
style: { height: `${props.items.length * props.itemHeight}px`, position: 'relative' }
}, visibleItems.value.map((item, idx) =>
h('div', {
key: visibleStart.value + idx,
style: {
position: 'absolute',
top: `${(visibleStart.value + idx) * props.itemHeight}px`,
height: `${props.itemHeight}px`
}
}, slots.default?.({ item, index: visibleStart.value + idx }))
))
])
}
})

View File

@@ -5,7 +5,11 @@
<div class="actions"> <div class="actions">
<el-button type="primary" @click="handleBack">返回</el-button> <el-button type="primary" @click="handleBack">返回</el-button>
<el-button type="success" @click="save">保存</el-button> <el-button type="success" @click="save">保存</el-button>
<span class="save-status">{{ saveStatus }}</span> <div class="save-status">
<el-icon v-if="saveStatus === '正在输入...'" class="is-loading saving-icon"><Loading /></el-icon>
<el-icon v-else-if="saveStatus === '保存失败'" class="error-icon"><CircleCloseFilled /></el-icon>
<el-icon v-else-if="saveStatus === '已保存'" class="success-icon"><CircleCheckFilled /></el-icon>
</div>
</div> </div>
</el-header> </el-header>
<div id="vditor-editor" class="vditor" /> <div id="vditor-editor" class="vditor" />
@@ -16,8 +20,10 @@
import { ref, onMounted, onBeforeUnmount, watch, nextTick } from 'vue'; import { ref, onMounted, onBeforeUnmount, watch, nextTick } from 'vue';
import Vditor from 'vditor'; import Vditor from 'vditor';
import 'vditor/dist/index.css'; import 'vditor/dist/index.css';
import { ElMessage } from 'element-plus'; import { ElMessage, ElMessageBox } from 'element-plus';
import { Loading, CircleCloseFilled, CircleCheckFilled } from '@element-plus/icons-vue';
import { updateMarkdown, uploadImage } from '@/api/CommonApi.js'; import { updateMarkdown, uploadImage } from '@/api/CommonApi.js';
import { useUserStore } from '@/stores/user';
const props = defineProps({ const props = defineProps({
editData: { editData: {
@@ -32,11 +38,13 @@ const vditor = ref(null);
const currentId = ref(null); const currentId = ref(null);
const isInitialized = ref(false); const isInitialized = ref(false);
const saveStatus = ref(''); const saveStatus = ref('');
let saveTimeout = null; const saveTimeout = ref(null); // 修复:使用 ref 替代 let确保响应式追踪
let lastSavedContent = ref(''); const lastSavedContent = ref('');
let isSaving = ref(false); const isSaving = ref(false);
// 维护当前最新的笔记数据 // 维护当前最新的笔记数据
const currentData = ref({ ...props.editData }); const currentData = ref({ ...props.editData });
// 保存 beforeunload 事件处理器引用
let handleBeforeUnload = null;
const initVditor = () => { const initVditor = () => {
if (vditor.value) { if (vditor.value) {
@@ -63,13 +71,16 @@ const initVditor = () => {
input: (value) => { input: (value) => {
if (!isInitialized.value) return; if (!isInitialized.value) return;
clearTimeout(saveTimeout); // 只有在内容真正改变时才启动定时器
if (value !== lastSavedContent.value) {
clearTimeout(saveTimeout.value);
saveStatus.value = '正在输入...'; saveStatus.value = '正在输入...';
saveTimeout = setTimeout(() => { saveTimeout.value = setTimeout(() => {
if (!isSaving.value && value !== lastSavedContent.value) { if (!isSaving.value && value !== lastSavedContent.value) {
save(value); save(value);
} }
}, 3000); }, 5000);
}
}, },
upload: { upload: {
accept: 'image/*', accept: 'image/*',
@@ -78,11 +89,17 @@ const initVditor = () => {
if (!file) return; if (!file) return;
uploadImage(file).then(res => { uploadImage(file).then(res => {
const url = res.url; // 后端返回相对路径,拼接成完整 URL
const relativeUrl = res.url;
const baseUrl = import.meta.env.VITE_API_BASE_URL || ''; const baseUrl = import.meta.env.VITE_API_BASE_URL || '';
vditor.value.insertValue(`![${file.name}](${baseUrl}${url})`); // 确保 URL 格式正确,避免双斜杠
}).catch(() => { const fullUrl = baseUrl.endsWith('/') && relativeUrl.startsWith('/')
ElMessage.error('图片上传失败'); ? baseUrl + relativeUrl.substring(1)
: baseUrl + relativeUrl;
vditor.value.insertValue(`![${file.name}](${fullUrl})`);
}).catch((error) => {
// 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('图片上传失败:', error);
}); });
}, },
}, },
@@ -92,8 +109,14 @@ const initVditor = () => {
const save = async (value) => { const save = async (value) => {
if (isSaving.value) return; if (isSaving.value) return;
clearTimeout(saveTimeout); // 修复:添加空值检查
const content = typeof value === 'string' ? value : vditor.value.getValue(); if (!vditor.value) {
console.warn('编辑器未初始化');
return;
}
clearTimeout(saveTimeout.value);
const content = typeof value === 'string' ? value : vditor.value?.getValue() || '';
if (content === lastSavedContent.value && currentId.value) { if (content === lastSavedContent.value && currentId.value) {
return; return;
@@ -119,8 +142,6 @@ const save = async (value) => {
isPrivate: currentData.value.isPrivate !== undefined ? currentData.value.isPrivate : props.editData.isPrivate isPrivate: currentData.value.isPrivate !== undefined ? currentData.value.isPrivate : props.editData.isPrivate
}; };
const response = await updateMarkdown(payload); const response = await updateMarkdown(payload);
if (response && response.id) { if (response && response.id) {
@@ -141,18 +162,33 @@ const save = async (value) => {
currentData.value = updatedFile; currentData.value = updatedFile;
emit('update:editData', updatedFile); emit('update:editData', updatedFile);
saveStatus.value = '已保存'; saveStatus.value = '已保存';
// 2秒后清除成功状态
setTimeout(() => {
if (saveStatus.value === '已保存') {
saveStatus.value = '';
}
}, 2000);
} }
} catch (error) { } catch (error) {
saveStatus.value = '保存失败'; saveStatus.value = '保存失败';
ElMessage.error('保存失败: ' + (error.message || '未知错误')); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('保存失败:', error);
} finally { } finally {
isSaving.value = false; isSaving.value = false;
} }
}; };
const handleBack = async () => { const handleBack = async () => {
// 清除定时器,防止返回后继续保存
clearTimeout(saveTimeout.value);
const content = vditor.value ? vditor.value.getValue() : ''; const content = vditor.value ? vditor.value.getValue() : '';
if (content !== lastSavedContent.value && !isSaving.value) { const hasChanges = content !== lastSavedContent.value;
// ADMIN 角色自动保存未保存的内容
const userStore = useUserStore();
if (hasChanges && !isSaving.value && userStore.userInfo?.role === 'ADMIN') {
await save(content); await save(content);
} }
@@ -160,11 +196,16 @@ const handleBack = async () => {
const groupingId = currentData.value.groupingId || props.editData.groupingId; const groupingId = currentData.value.groupingId || props.editData.groupingId;
const groupingName = currentData.value.groupingName || props.editData.groupingName; const groupingName = currentData.value.groupingName || props.editData.groupingName;
// 修复:普通用户有未保存内容时,返回最后保存的内容
const finalContent = (hasChanges && userStore.userInfo?.role !== 'ADMIN')
? lastSavedContent.value
: content;
const returnData = { const returnData = {
...currentData.value, ...currentData.value,
...props.editData, ...props.editData,
id: currentId.value ? String(currentId.value) : (currentData.value.id ? String(currentData.value.id) : null), id: currentId.value ? String(currentId.value) : (currentData.value.id ? String(currentData.value.id) : null),
content: content, content: finalContent,
groupingId: groupingId ? String(groupingId) : null, groupingId: groupingId ? String(groupingId) : null,
groupingName: groupingName groupingName: groupingName
}; };
@@ -174,14 +215,36 @@ const handleBack = async () => {
onMounted(() => { onMounted(() => {
initVditor(); initVditor();
// 页面刷新时提示用户保存
handleBeforeUnload = (e) => {
const content = vditor.value ? vditor.value.getValue() : '';
if (content !== lastSavedContent.value && !isSaving.value) {
e.preventDefault();
e.returnValue = '您有未保存的内容,确定要离开吗?';
return '您有未保存的内容,确定要离开吗?';
}
};
window.addEventListener('beforeunload', handleBeforeUnload);
}); });
onBeforeUnmount(() => { onBeforeUnmount(() => {
clearTimeout(saveTimeout); // 清理定时器
if (saveTimeout.value) {
clearTimeout(saveTimeout.value);
saveTimeout.value = null;
}
// 清理编辑器
if (vditor.value) { if (vditor.value) {
vditor.value.destroy(); vditor.value.destroy();
vditor.value = null; vditor.value = null;
} }
// 移除页面刷新提示
if (handleBeforeUnload) {
window.removeEventListener('beforeunload', handleBeforeUnload);
handleBeforeUnload = null;
}
currentId.value = null; currentId.value = null;
isInitialized.value = false; isInitialized.value = false;
}); });
@@ -219,9 +282,46 @@ watch(() => props.editData, (newVal, oldVal) => {
font-size: 20px; font-size: 20px;
} }
.actions .save-status { .actions {
margin-left: 10px; display: flex;
color: #909399; align-items: center;
gap: 10px;
}
.save-status {
width: 24px;
height: 24px;
display: flex;
align-items: center;
justify-content: center;
}
.saving-icon {
font-size: 20px;
color: #409eff;
}
.success-icon {
font-size: 20px;
color: #67c23a;
animation: fadeOut 2s ease-in-out forwards;
}
.error-icon {
font-size: 20px;
color: #f56c6c;
}
@keyframes fadeOut {
0% {
opacity: 1;
}
80% {
opacity: 1;
}
100% {
opacity: 0;
}
} }
.vditor { .vditor {

View File

@@ -11,16 +11,16 @@
<el-icon v-if="isMobile"><Back /></el-icon> <el-icon v-if="isMobile"><Back /></el-icon>
<span v-else>返回</span> <span v-else>返回</span>
</el-button> </el-button>
<el-button v-if="isUserLoggedIn && !isMobile" type="warning" @click="$emit('show-move-note-dialog')">移动</el-button> <el-button v-if="isUserLoggedIn && !isMobile" type="warning" :disabled="userRole !== 'ADMIN'" @click="$emit('show-move-note-dialog')">移动</el-button>
<el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="primary" @click="$emit('edit')"> <el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="primary" :disabled="userRole !== 'ADMIN'" @click="$emit('edit')">
<el-icon v-if="isMobile"><Edit /></el-icon> <el-icon v-if="isMobile"><Edit /></el-icon>
<span v-else>编辑</span> <span v-else>编辑</span>
</el-button> </el-button>
<el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="danger" @click="$emit('delete')"> <el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="danger" :disabled="userRole !== 'ADMIN'" @click="$emit('delete')">
<el-icon v-if="isMobile"><Delete /></el-icon> <el-icon v-if="isMobile"><Delete /></el-icon>
<span v-else>删除</span> <span v-else>删除</span>
</el-button> </el-button>
<el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="info" @click="$emit('show-privacy-dialog')"> <el-button v-if="isUserLoggedIn" :circle="isMobile" :size="isMobile ? 'small' : 'default'" type="info" :disabled="userRole !== 'ADMIN'" @click="$emit('show-privacy-dialog')">
<el-icon v-if="isMobile"><Lock /></el-icon> <el-icon v-if="isMobile"><Lock /></el-icon>
<span v-else>{{ file.isPrivate === 1 ? '设为公开' : '设为私密' }}</span> <span v-else>{{ file.isPrivate === 1 ? '设为公开' : '设为私密' }}</span>
</el-button> </el-button>
@@ -46,6 +46,16 @@
<el-icon class="loading-icon is-loading"><Loading /></el-icon> <el-icon class="loading-icon is-loading"><Loading /></el-icon>
<span class="loading-text">内容加载中...</span> <span class="loading-text">内容加载中...</span>
</div> </div>
<!-- 渲染状态遮罩 -->
<div v-else-if="file.isRendering" class="content-loading">
<el-icon class="loading-icon is-loading"><Loading /></el-icon>
<span class="loading-text">正在渲染...</span>
</div>
<!-- 大文件分块加载提示 -->
<div v-if="file.hasMoreChunks" class="chunk-loading-hint">
<el-icon class="is-loading"><Loading /></el-icon>
<span>加载更多内容...</span>
</div>
</div> </div>
</template> </template>
@@ -59,6 +69,14 @@ const props = defineProps({
}, },
isMobile: Boolean, isMobile: Boolean,
isUserLoggedIn: Boolean, isUserLoggedIn: Boolean,
hasMoreChunks: {
type: Boolean,
default: false
},
userRole: {
type: String,
default: 'USER'
}
}); });
const emit = defineEmits([ const emit = defineEmits([
@@ -187,4 +205,25 @@ const handleExport = (format) => {
font-size: 14px; font-size: 14px;
color: var(--text-color-secondary); color: var(--text-color-secondary);
} }
/* 分块加载提示 */
.chunk-loading-hint {
position: absolute;
bottom: 20px;
left: 50%;
transform: translateX(-50%);
display: flex;
align-items: center;
gap: 8px;
padding: 8px 16px;
background-color: rgba(255, 255, 255, 0.9);
border-radius: 20px;
box-shadow: 0 2px 12px rgba(0, 0, 0, 0.1);
color: var(--el-color-primary);
font-size: 14px;
}
.dark-theme .chunk-loading-hint {
background-color: rgba(30, 30, 47, 0.9);
}
</style> </style>

View File

@@ -29,11 +29,13 @@
:collapse="isCollapsed && !isMobile" :collapse="isCollapsed && !isMobile"
popper-effect="light" popper-effect="light"
:collapse-transition="true" :collapse-transition="true"
@select="handleMenuSelect"
> >
<div class="menu-section" v-if="isMobile"> <div class="menu-section" v-if="isMobile">
<div v-if="userStore.isLoggedIn" class="user-info"> <div v-if="userStore.isLoggedIn" class="user-info">
<el-avatar :size="40" class="user-avatar">{{ userStore.userInfo?.username?.charAt(0)?.toUpperCase() }}</el-avatar> <!-- 修复添加默认值防止空指针 -->
<span class="username">{{ userStore.userInfo?.username }}</span> <el-avatar :size="40" class="user-avatar">{{ userStore.userInfo?.username?.charAt(0)?.toUpperCase() || '?' }}</el-avatar>
<span class="username">{{ userStore.userInfo?.username || '访客' }}</span>
</div> </div>
<div v-else class="guest-info"> <div v-else class="guest-info">
<el-button type="primary" @click="goToLogin">登录</el-button> <el-button type="primary" @click="goToLogin">登录</el-button>
@@ -106,6 +108,24 @@ const goToTrash = () => router.push({ name: 'Trash' });
const goToLogin = () => router.push('/login'); const goToLogin = () => router.push('/login');
const goToRegister = () => router.push('/register'); const goToRegister = () => router.push('/register');
const handleMenuSelect = (index) => {
if (index.startsWith('group-')) {
const id = index.replace('group-', '');
const findItem = (items) => {
for (let item of items) {
if (String(item.id) === id) return item;
if (item.children) {
const found = findItem(item.children);
if (found) return found;
}
}
return null;
};
const item = findItem(props.categoryTree);
if (item) emit('select-file', item);
}
};
const handleDeleteGroup = (group) => { const handleDeleteGroup = (group) => {
ElMessageBox.confirm( ElMessageBox.confirm(
`确定要删除分类 “${group.grouping}” 吗?这将同时删除该分类下的所有子分类和笔记。`, `确定要删除分类 “${group.grouping}” 吗?这将同时删除该分类下的所有子分类和笔记。`,
@@ -117,7 +137,8 @@ const handleDeleteGroup = (group) => {
ElMessage.success('分类已删除'); ElMessage.success('分类已删除');
emit('group-deleted'); emit('group-deleted');
} catch (error) { } catch (error) {
ElMessage.error('删除分类失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('删除分类失败:', error);
} }
}); });
}; };
@@ -143,18 +164,19 @@ const renderMenu = (item) => {
if (item.children && item.children.length > 0) { if (item.children && item.children.length > 0) {
return h(ElSubMenu, { return h(ElSubMenu, {
index: `group-${item.id}`, index: `group-${item.id}`,
popperClass: props.isCollapsed ? 'hide-popper' : '' popperClass: props.isCollapsed ? 'hide-popper' : '',
class: { 'is-active': props.activeMenu === `group-${item.id}` },
disabled: false
}, { }, {
title: () => h('div', { title: () => h('div', {
class: 'submenu-title-wrapper', class: 'submenu-title-wrapper',
onClick: () => emit('select-file', item),
style: 'width: 100%; display: flex; align-items: center;' style: 'width: 100%; display: flex; align-items: center;'
}, [ wrappedContent() ]), }, [ wrappedContent() ]),
default: () => item.children.map(child => renderMenu(child)) default: () => item.children.map(child => renderMenu(child))
}); });
} }
return h(ElMenuItem, { index: `group-${item.id}`, onClick: () => emit('select-file', item) }, { return h(ElMenuItem, { index: `group-${item.id}` }, {
default: wrappedContent default: wrappedContent
}); });
}; };
@@ -347,6 +369,30 @@ const renderMenu = (item) => {
background: #fff; background: #fff;
} }
/* 移除嵌套 SubMenu 的 active 样式 */
:deep(.el-menu--inline .el-sub-menu.is-active > .el-sub-menu__title) {
background: transparent !important;
color: var(--text-color-secondary) !important;
box-shadow: none !important;
}
:deep(.el-menu--inline .el-sub-menu.is-active > .el-sub-menu__title::before) {
height: 0 !important;
}
/* 只有顶级 SubMenu 被直接选中时才显示 active 样式 */
:deep(.el-menu > .el-sub-menu.is-active > .el-sub-menu__title) {
background: linear-gradient(135deg, var(--primary-color) 0%, var(--primary-color-light) 100%) !important;
color: #fff !important;
font-weight: 600 !important;
box-shadow: 0 4px 12px rgba(64, 158, 255, 0.35) !important;
}
:deep(.el-menu > .el-sub-menu.is-active > .el-sub-menu__title::before) {
height: 24px !important;
background: #fff !important;
}
:deep(.el-menu-item:hover), :deep(.el-sub-menu__title:hover) { :deep(.el-menu-item:hover), :deep(.el-sub-menu__title:hover) {
background-color: rgba(64, 158, 255, 0.08); background-color: rgba(64, 158, 255, 0.08);
color: var(--primary-color); color: var(--primary-color);

View File

@@ -113,7 +113,8 @@ const handleSubmit = async () => {
emit('group-created'); // 通知父组件刷新 emit('group-created'); // 通知父组件刷新
handleClose(); handleClose();
} catch (error) { } catch (error) {
ElMessage.error('创建分类失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('创建分类失败:', error);
} }
} }
}); });

View File

@@ -154,7 +154,8 @@ const handleSubmit = async () => {
emit('move-success'); emit('move-success');
handleClose(); handleClose();
} catch (error) { } catch (error) {
ElMessage.error('移动失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('移动失败:', error);
} finally { } finally {
isLoading.value = false; isLoading.value = false;
} }

View File

@@ -68,7 +68,8 @@ const handleSubmit = async () => {
emit('privacy-changed', updatedFile); emit('privacy-changed', updatedFile);
handleClose(); handleClose();
} catch (error) { } catch (error) {
ElMessage.error('修改笔记状态失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('修改笔记状态失败:', error);
} }
}; };
</script> </script>

View File

@@ -60,7 +60,8 @@ const handleSubmit = async () => {
// 传递新名称给父组件 // 传递新名称给父组件
emit('renamed', newName.value); emit('renamed', newName.value);
} catch (error) { } catch (error) {
ElMessage.error('重命名失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('重命名失败:', error);
} }
}; };
</script> </script>

View File

@@ -80,7 +80,8 @@ const handleSubmit = () => {
emit('import-success'); emit('import-success');
handleClose(); handleClose();
} catch (error) { } catch (error) {
ElMessage.error('导入失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('导入失败:', error);
} }
}; };
reader.readAsText(props.fileToImport); reader.readAsText(props.fileToImport);

View File

@@ -59,8 +59,8 @@ const fetchRegistrationStatus = async () => {
try { try {
isRegistrationEnabled.value = await getRegistrationStatus(); isRegistrationEnabled.value = await getRegistrationStatus();
} catch (error) { } catch (error) {
// 错误已在 axios 拦截器中显示,这里不再重复显示
console.error("Failed to fetch registration status:", error); console.error("Failed to fetch registration status:", error);
ElMessage.error('获取注册状态失败');
} }
}; };
@@ -69,7 +69,7 @@ const handleToggleRegistration = async (value) => {
await toggleRegistration(value); await toggleRegistration(value);
ElMessage.success(`注册功能已${value ? '开启' : '关闭'}`); ElMessage.success(`注册功能已${value ? '开启' : '关闭'}`);
} catch (error) { } catch (error) {
ElMessage.error('操作失败'); // 错误已在 axios 拦截器中显示,这里不再重复显示
isRegistrationEnabled.value = !value; // Revert on failure isRegistrationEnabled.value = !value; // Revert on failure
} }
}; };
@@ -80,7 +80,8 @@ const handleGenerateCode = async () => {
generatedCode.value = code; generatedCode.value = code;
ElMessage.success('注册码生成成功'); ElMessage.success('注册码生成成功');
} catch (error) { } catch (error) {
ElMessage.error('生成注册码失败: ' + error.message); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('生成注册码失败:', error);
} }
}; };

View File

@@ -115,7 +115,8 @@ const handleCaptchaConfirm = async ({ captchaId, captchaCode }) => {
emit('password-updated'); emit('password-updated');
handleClose(); handleClose();
} catch (error) { } catch (error) {
ElMessage.error('密码修改失败: ' + (error.response?.data?.msg || error.message)); // 错误已在 axios 拦截器中显示,这里不再重复显示
console.error('密码修改失败:', error);
} finally { } finally {
loading.value = false; loading.value = false;
} }

View File

@@ -3,8 +3,9 @@ import { login as loginApi } from '../api/CommonApi';
export const useUserStore = defineStore('user', { export const useUserStore = defineStore('user', {
state: () => ({ state: () => ({
token: '', token: localStorage.getItem('user-token') || '',
userInfo: null, userInfo: JSON.parse(localStorage.getItem('user-info') || 'null'),
tokenExpiry: parseInt(localStorage.getItem('user-token-expiry') || '0'),
}), }),
actions: { actions: {
async login(username, password) { async login(username, password) {
@@ -12,9 +13,15 @@ export const useUserStore = defineStore('user', {
const response = await loginApi({ username, password }); const response = await loginApi({ username, password });
if (response && response.token) { if (response && response.token) {
this.token = response.token; this.token = response.token;
const payload = JSON.parse(atob(response.token.split('.')[1]));
this.tokenExpiry = payload.exp * 1000;
if (response.userInfo) { if (response.userInfo) {
this.userInfo = response.userInfo; this.userInfo = response.userInfo;
} }
// 持久化到 localStorage
localStorage.setItem('user-token', response.token);
localStorage.setItem('user-info', JSON.stringify(response.userInfo));
localStorage.setItem('user-token-expiry', this.tokenExpiry.toString());
return true; return true;
} }
return false; return false;
@@ -26,18 +33,18 @@ export const useUserStore = defineStore('user', {
logout() { logout() {
this.token = ''; this.token = '';
this.userInfo = null; this.userInfo = null;
this.tokenExpiry = null;
localStorage.removeItem('user-token');
localStorage.removeItem('user-info');
localStorage.removeItem('user-token-expiry');
},
isTokenExpired() {
if (!this.tokenExpiry) return true;
return Date.now() >= this.tokenExpiry;
}, },
}, },
getters: { getters: {
isLoggedIn: (state) => !!state.token, isLoggedIn: (state) => !!state.token && Date.now() < (state.tokenExpiry || 0),
}, isAdmin: (state) => state.userInfo?.role === 'ADMIN',
persist: {
enabled: true,
strategies: [
{
key: 'user-store',
storage: sessionStorage,
}
],
}, },
}); });

View File

@@ -4,10 +4,11 @@ import { ElMessage } from 'element-plus'
import router from '../router' import router from '../router'
import { getReplayAttackHeaders, needsReplayAttackValidation } from './security' import { getReplayAttackHeaders, needsReplayAttackValidation } from './security'
let retryCount = 0
const MAX_RETRIES = 3
const instance = axios.create({ const instance = axios.create({
baseURL: import.meta.env.VITE_API_BASE_URL, baseURL: import.meta.env.VITE_API_BASE_URL,
// 开发环境使用withCredentials生产环境关闭
// withCredentials: import.meta.env.DEV,
headers: { headers: {
'Content-Type': 'application/json' 'Content-Type': 'application/json'
} }
@@ -22,6 +23,12 @@ instance.interceptors.request.use(
config.headers['Authorization'] = `Bearer ${userStore.token}` config.headers['Authorization'] = `Bearer ${userStore.token}`
} }
// 添加 CSRF token
const csrfToken = document.querySelector('meta[name="csrf-token"]')?.content
if (csrfToken) {
config.headers['X-CSRF-Token'] = csrfToken
}
// 添加防重放攻击请求头POST/PUT/DELETE 请求) // 添加防重放攻击请求头POST/PUT/DELETE 请求)
if (needsReplayAttackValidation(config.method, config.url)) { if (needsReplayAttackValidation(config.method, config.url)) {
const replayHeaders = getReplayAttackHeaders() const replayHeaders = getReplayAttackHeaders()
@@ -41,6 +48,7 @@ instance.interceptors.request.use(
// 响应拦截器 // 响应拦截器
instance.interceptors.response.use( instance.interceptors.response.use(
response => { response => {
retryCount = 0
const res = response.data; const res = response.data;
if (res.code !== 200) { if (res.code !== 200) {
ElMessage({ ElMessage({
@@ -53,11 +61,19 @@ instance.interceptors.response.use(
return res.data; return res.data;
} }
}, },
error => { async error => {
if (error.response) { if (error.response) {
const status = error.response.status; const status = error.response.status;
const data = error.response.data; const data = error.response.data;
// 503 - 服务器繁忙,重试
if (status === 503 && retryCount < MAX_RETRIES) {
retryCount++
await new Promise(resolve => setTimeout(resolve, 1000 * retryCount))
return instance(error.config)
}
retryCount = 0
// 401 - 未授权 // 401 - 未授权
if (status === 401) { if (status === 401) {
try { try {
@@ -71,6 +87,13 @@ instance.interceptors.response.use(
return Promise.reject(error); return Promise.reject(error);
} }
// 403 - 权限不足
if (status === 403) {
const msg = data?.msg || '无权操作';
ElMessage.error(msg);
return Promise.reject(new Error(msg));
}
// 429 - 请求过于频繁 // 429 - 请求过于频繁
if (status === 429) { if (status === 429) {
const msg = data?.msg || '请求过于频繁,请稍后再试'; const msg = data?.msg || '请求过于频繁,请稍后再试';
@@ -85,15 +108,11 @@ instance.interceptors.response.use(
return Promise.reject(new Error(msg)); return Promise.reject(new Error(msg));
} }
// 503 - 服务器繁忙(内存不足) // 其他错误
if (status === 503) { const isDev = import.meta.env.DEV;
const msg = data?.msg || '服务器繁忙,请稍后再试'; const msg = isDev
ElMessage.error(msg); ? (data?.msg || error.message)
return Promise.reject(new Error(msg)); : (data?.msg || '操作失败,请稍后重试');
}
// 其他错误,显示后端返回的消息
const msg = data?.msg || error.message;
ElMessage.error(msg); ElMessage.error(msg);
} else { } else {
ElMessage({ ElMessage({

View File

@@ -5,15 +5,15 @@
/** /**
* 生成随机 nonce (32位随机字符串) * 生成随机 nonce (32位随机字符串)
* 使用 crypto.getRandomValues 确保密码学安全
* @returns {string} nonce * @returns {string} nonce
*/ */
export function generateNonce() { export function generateNonce() {
const chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'; const array = new Uint8Array(24); // 24 bytes = 32 base64 chars
let nonce = ''; crypto.getRandomValues(array);
for (let i = 0; i < 32; i++) { return btoa(String.fromCharCode(...array))
nonce += chars.charAt(Math.floor(Math.random() * chars.length)); .replace(/[+/=]/g, '') // 移除特殊字符
} .substring(0, 32);
return nonce;
} }
/** /**
@@ -61,3 +61,15 @@ export function needsReplayAttackValidation(method, url) {
} }
return REPLAY_ATTACK_METHODS.includes(method.toUpperCase()); return REPLAY_ATTACK_METHODS.includes(method.toUpperCase());
} }
/**
* HTML 转义函数,防止 XSS
* @param {string} text 需要转义的文本
* @returns {string} 转义后的文本
*/
export function escapeHtml(text) {
if (!text) return '';
const div = document.createElement('div');
div.textContent = text;
return div.innerHTML;
}

View File

@@ -0,0 +1,12 @@
import DOMPurify from 'dompurify'
export const sanitizeHtml = (html) => {
return DOMPurify.sanitize(html, {
ALLOWED_TAGS: ['b', 'i', 'em', 'strong', 'a', 'p', 'br', 'ul', 'ol', 'li', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'blockquote', 'code', 'pre', 'img'],
ALLOWED_ATTR: ['href', 'title', 'src', 'alt']
})
}
export const sanitizeText = (text) => {
return DOMPurify.sanitize(text, { ALLOWED_TAGS: [] })
}

View File

@@ -0,0 +1,264 @@
# 反馈系统功能分析(未实现)
## 一、需求概述
在左侧菜单回收站下方添加一个新的功能模块,包含三个子功能:
- 反馈BUG
- 申请管理
- 提出意见
**访问权限:** 仅登录用户可见和使用
**处理权限:** 仅ADMIN用户可处理
---
## 二、用户角色权限矩阵
### 普通用户USER能做什么
| 功能 | 权限 | 说明 |
|------|------|------|
| 查看反馈列表 | ✅ | 仅查看自己提交的反馈 |
| 提交BUG反馈 | ✅ | 创建新的BUG反馈 |
| 提交管理申请 | ✅ | 申请成为管理员 |
| 提出意见建议 | ✅ | 提交产品意见 |
| 编辑自己的反馈 | ✅ | 仅在未处理状态下可编辑 |
| 删除自己的反馈 | ✅ | 仅在未处理状态下可删除 |
| 查看反馈状态 | ✅ | 实时查看处理进度 |
| 处理他人反馈 | ❌ | 无权限 |
| 删除他人反馈 | ❌ | 无权限 |
### 管理员ADMIN能做什么
| 功能 | 权限 | 说明 |
|------|------|------|
| 查看所有反馈 | ✅ | 查看全部用户的反馈 |
| 查看反馈详情 | ✅ | 包括用户信息、提交时间等 |
| 处理反馈 | ✅ | 更新状态、添加处理备注 |
| 批准/拒绝申请 | ✅ | 处理管理员申请 |
| 删除反馈 | ✅ | 删除任何反馈 |
| 导出反馈 | ✅ | 导出为CSV/Excel |
| 统计分析 | ✅ | 查看反馈统计信息 |
---
## 三、数据表设计
### 1. 反馈表feedback
```sql
CREATE TABLE feedback (
id BIGINT PRIMARY KEY COMMENT '反馈ID',
user_id BIGINT NOT NULL COMMENT '提交用户ID',
type VARCHAR(20) NOT NULL COMMENT '反馈类型: BUG, ADMIN_REQUEST, SUGGESTION',
title VARCHAR(255) NOT NULL COMMENT '反馈标题',
content LONGTEXT NOT NULL COMMENT '反馈内容',
status VARCHAR(20) DEFAULT 'PENDING' COMMENT '状态: PENDING(待处理), PROCESSING(处理中), RESOLVED(已解决), REJECTED(已拒绝)',
priority VARCHAR(20) DEFAULT 'NORMAL' COMMENT '优先级: LOW, NORMAL, HIGH, URGENT',
-- 处理信息
handler_id BIGINT COMMENT '处理人ID',
handler_remark VARCHAR(500) COMMENT '处理备注',
handled_at DATETIME COMMENT '处理时间',
-- 时间戳
created_at DATETIME DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
-- 软删除
is_deleted TINYINT DEFAULT 0 COMMENT '是否删除',
deleted_at DATETIME COMMENT '删除时间',
deleted_by BIGINT COMMENT '删除人ID',
INDEX idx_user_id (user_id),
INDEX idx_type (type),
INDEX idx_status (status),
INDEX idx_created_at (created_at)
) COMMENT='用户反馈表';
```
### 2. 管理员申请表admin_request
```sql
CREATE TABLE admin_request (
id BIGINT PRIMARY KEY COMMENT '申请ID',
user_id BIGINT NOT NULL COMMENT '申请用户ID',
reason VARCHAR(500) NOT NULL COMMENT '申请理由',
status VARCHAR(20) DEFAULT 'PENDING' COMMENT '状态: PENDING(待审核), APPROVED(已批准), REJECTED(已拒绝)',
-- 审核信息
reviewer_id BIGINT COMMENT '审核人ID',
review_remark VARCHAR(500) COMMENT '审核备注',
reviewed_at DATETIME COMMENT '审核时间',
-- 时间戳
created_at DATETIME DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
-- 软删除
is_deleted TINYINT DEFAULT 0 COMMENT '是否删除',
deleted_at DATETIME COMMENT '删除时间',
deleted_by BIGINT COMMENT '删除人ID',
INDEX idx_user_id (user_id),
INDEX idx_status (status),
INDEX idx_created_at (created_at)
) COMMENT='管理员申请表';
```
---
## 四、业务逻辑设计
### 4.1 反馈流程
```
普通用户提交反馈
反馈进入待处理队列 (PENDING)
管理员查看反馈
管理员处理反馈 (更新状态、添加备注)
用户查看处理结果
```
### 4.2 反馈类型处理
| 类型 | 流程 | 特殊处理 |
|------|------|--------|
| BUG | 提交 → 待处理 → 处理中 → 已解决/已拒绝 | 可设置优先级 |
| ADMIN_REQUEST | 提交 → 待审核 → 已批准/已拒绝 | 批准后自动升级用户角色 |
| SUGGESTION | 提交 → 待处理 → 已解决/已拒绝 | 仅供参考 |
### 4.3 管理员申请特殊处理
当管理员批准申请时:
1. 更新 admin_request 表状态为 APPROVED
2. 更新 user 表的 role 字段为 ADMIN
3. 记录审核人和审核时间
4. 可选:发送通知给申请用户
---
## 五、API 接口设计
### 5.1 普通用户接口
```
POST /api/feedback - 提交反馈
GET /api/feedback/my - 查看自己的反馈列表
GET /api/feedback/{id} - 查看反馈详情
PUT /api/feedback/{id} - 编辑反馈(仅未处理状态)
DELETE /api/feedback/{id} - 删除反馈(仅未处理状态)
POST /api/admin-request - 提交管理员申请
GET /api/admin-request/my - 查看自己的申请
GET /api/admin-request/{id} - 查看申请详情
```
### 5.2 管理员接口
```
GET /api/feedback - 查看所有反馈(分页)
GET /api/feedback/stats - 反馈统计
PUT /api/feedback/{id}/status - 更新反馈状态
PUT /api/feedback/{id}/handle - 处理反馈(添加备注)
DELETE /api/feedback/{id} - 删除反馈
GET /api/admin-request - 查看所有申请(分页)
PUT /api/admin-request/{id}/approve - 批准申请
PUT /api/admin-request/{id}/reject - 拒绝申请
```
---
## 六、前端UI设计
### 6.1 菜单结构
```
左侧菜单
├── 分类列表
├── 回收站
├── 反馈中心 (新增)
│ ├── 反馈BUG
│ ├── 申请管理
│ └── 提出意见
└── 系统管理 (仅ADMIN)
```
### 6.2 反馈中心页面
**普通用户视图:**
- 反馈列表(仅显示自己的)
- 提交新反馈表单
- 反馈详情和状态查看
- 编辑/删除按钮(仅未处理状态)
**管理员视图:**
- 所有反馈列表(可筛选、搜索)
- 反馈统计仪表板
- 处理反馈表单(更新状态、添加备注)
- 批量操作(删除、导出)
---
## 七、实现步骤
### 第一阶段:数据库
1. 创建 feedback 表
2. 创建 admin_request 表
3. 添加索引优化查询
### 第二阶段:后端
1. 创建 Feedback 和 AdminRequest 实体类
2. 创建 Mapper 和 Service
3. 创建 Controller 和 API 接口
4. 添加权限控制注解
### 第三阶段:前端
1. 创建反馈中心菜单项
2. 创建反馈列表页面
3. 创建反馈提交表单
4. 创建管理员处理页面
5. 集成到 SidebarMenu
### 第四阶段:测试
1. 单元测试
2. 集成测试
3. 权限测试
---
## 八、关键技术点
### 权限控制
- 使用 @PreAuthorize 注解控制接口访问
- 在 Service 层验证用户权限
- 确保用户只能操作自己的反馈
### 数据隔离
- 普通用户查询时自动过滤为自己的反馈
- 管理员可查看所有反馈
### 状态管理
- 反馈状态流转PENDING → PROCESSING → RESOLVED/REJECTED
- 申请状态流转PENDING → APPROVED/REJECTED
### 审计日志
- 记录处理人、处理时间、处理备注
- 支持软删除,保留历史记录
---
## 九、扩展考虑
1. **通知系统** - 反馈被处理时通知用户
2. **评论功能** - 用户和管理员可在反馈下评论
3. **附件支持** - 反馈可上传截图或文件
4. **优先级排序** - BUG 按优先级排序
5. **导出功能** - 管理员可导出反馈报告
6. **统计分析** - 反馈类型、状态分布统计

View File

@@ -168,6 +168,7 @@ CREATE TABLE "user" (
"username" TEXT NOT NULL, "username" TEXT NOT NULL,
"password" TEXT NOT NULL, "password" TEXT NOT NULL,
"email" TEXT, "email" TEXT,
"role" TEXT DEFAULT 'USER',
"created_at" DATETIME DEFAULT CURRENT_TIMESTAMP, "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
"updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP, "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
"token" TEXT, "token" TEXT,

View File

@@ -6,7 +6,7 @@ SET FOREIGN_KEY_CHECKS = 0;
-- 1. 分组表 -- 1. 分组表
DROP TABLE IF EXISTS `grouping`; DROP TABLE IF EXISTS `grouping`;
CREATE TABLE `grouping` ( CREATE TABLE `grouping` (
`id` bigintBIGINT(20) NOT NULL AUTO_INCREMENT, `id` bigint(20) NOT NULL AUTO_INCREMENT,
`grouping` VARCHAR(255) NOT NULL, `grouping` VARCHAR(255) NOT NULL,
`parentId` BIGINT(20) DEFAULT NULL, `parentId` BIGINT(20) DEFAULT NULL,
`is_deleted` TINYINT(1) DEFAULT 0, `is_deleted` TINYINT(1) DEFAULT 0,
@@ -84,12 +84,14 @@ CREATE TABLE `user` (
`username` VARCHAR(50) NOT NULL, `username` VARCHAR(50) NOT NULL,
`password` VARCHAR(255) NOT NULL, `password` VARCHAR(255) NOT NULL,
`email` VARCHAR(100) DEFAULT NULL, `email` VARCHAR(100) DEFAULT NULL,
`role` VARCHAR(50) DEFAULT 'USER' COMMENT '用户角色ADMIN-管理员USER-普通用户',
`created_at` DATETIME DEFAULT CURRENT_TIMESTAMP, `created_at` DATETIME DEFAULT CURRENT_TIMESTAMP,
`updated_at` DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, `updated_at` DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
`token` VARCHAR(255) DEFAULT NULL, `token` VARCHAR(255) DEFAULT NULL,
`token_enddata` DATETIME DEFAULT NULL, `token_enddata` DATETIME DEFAULT NULL,
PRIMARY KEY (`id`), PRIMARY KEY (`id`),
UNIQUE KEY `uk_username` (`username`) UNIQUE KEY `uk_username` (`username`),
KEY `idx_user_role` (`role`)
) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8mb4; ) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8mb4;
SET FOREIGN_KEY_CHECKS = 1; SET FOREIGN_KEY_CHECKS = 1;

Binary file not shown.

After

Width:  |  Height:  |  Size: 61 KiB