offerpai/offerpai_backend
6
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

修改权限控制

Browse Source
This commit is contained in:
zk
2026-03-12 16:40:23 +08:00
parent 24fe596d27
commit bebf9c9c50
6 changed files with 166 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client-api/src/main/java/org/jiayunet/aop/FuncPermissionAspect.java
+6 -5
View File
@@ -12,7 +12,7 @@ import org.springframework.stereotype.Component;
/**
* 功能权限校验切面
* <p>拦截 @FuncPermission 注解，校验权限并扣减库存后放行</p>
* <p>拦截 @FuncPermission 注解，校验权限并扣减库存后放行，业务异常时自动回退</p>
*
* @author zk
*/
@@ -32,14 +32,15 @@ public class FuncPermissionAspect {
log.info("功能权限校验 userId:{} funcCode:{}", userId, funcCode);
// 校验权限 + 扣减库存
funcPermissionServer.checkAndDeduct(userId, funcCode);
// 校验权限 + 扣减库存，返回使用记录ID
Long logId = funcPermissionServer.checkAndDeduct(userId, funcCode);
// 放行，业务异常时回退库存
// 放行，业务异常时回退使用记录和库存
try {
return joinPoint.proceed();
} catch (Exception e) {
funcPermissionServer.rollbackCount(userId, funcCode);
log.warn("业务异常，回退使用记录 logId:{} userId:{} funcCode:{}", logId, userId, funcCode);
funcPermissionServer.rollbackUsage(logId, userId, funcCode);
throw e;
}
}
client-api/src/main/java/org/jiayunet/server/FuncPermissionServer.java
+87 -29
View File
@@ -5,15 +5,21 @@ import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import lombok.extern.slf4j.Slf4j;
import org.jiayunet.exception.BusinessException;
import org.jiayunet.exception.BusinessExpCodeEnum;
import org.jiayunet.mapper.FuncPermissionMapper;
import org.jiayunet.mapper.UserFuncPermissionStockMapper;
import org.jiayunet.mapper.UserFuncUsageLogMapper;
import org.jiayunet.pojo.po.FuncPermission;
import org.jiayunet.pojo.po.UserFuncPermissionStock;
import org.jiayunet.pojo.po.UserFuncUsageLog;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.time.Instant;
import java.time.LocalDate;
import java.time.ZoneId;
/**
* 功能权限服务（校验、扣减、查询、添加库存）
* 功能权限服务（校验、扣减、查询、添加库存、回退）
*
* @author zk
*/
@@ -21,39 +27,72 @@ import java.time.Instant;
@Slf4j
public class FuncPermissionServer {
@Autowired
private FuncPermissionMapper funcPermissionMapper;
@Autowired
private UserFuncPermissionStockMapper userFuncPermissionStockMapper;
@Autowired
private UserFuncUsageLogMapper userFuncUsageLogMapper;
/**
* 校验用户功能权限并扣减库存
* <p>优先使用每日免费额度，免费额度用完后走付费库存</p>
*
* @param userId 用户ID
* @param funcCode 功能权限编码
* @return 使用记录ID（用于异常回退）
*/
public void checkAndDeduct(Long userId, String funcCode) {
// 查询用户功能权限库存
public Long checkAndDeduct(Long userId, String funcCode) {
// 1. 查功能权限定义
FuncPermission funcPermission = funcPermissionMapper.selectOne(
new LambdaQueryWrapper<FuncPermission>()
.eq(FuncPermission::getFuncCode, funcCode)
.eq(FuncPermission::getStatus, 1)
);
if (funcPermission == null) {
throw new BusinessException(BusinessExpCodeEnum.PERMISSION_DENIED, "功能不存在或未启用");
}
// 2. 判断每日免费额度
int dailyFreeCount = funcPermission.getDailyFreeCount() == null ? 0 : funcPermission.getDailyFreeCount();
if (dailyFreeCount > 0) {
Instant todayStart = LocalDate.now().atStartOfDay(ZoneId.systemDefault()).toInstant();
Long todayUsed = userFuncUsageLogMapper.selectCount(
new LambdaQueryWrapper<UserFuncUsageLog>()
.eq(UserFuncUsageLog::getUserId, userId)
.eq(UserFuncUsageLog::getFuncCode, funcCode)
.ge(UserFuncUsageLog::getCreateTime, todayStart)
);
if (todayUsed < dailyFreeCount) {
// 免费额度未用完，插入使用记录，直接放行
return insertUsageLog(userId, funcCode);
}
}
// 3. 免费额度用完或无免费额度，查付费库存
UserFuncPermissionStock stock = userFuncPermissionStockMapper.selectOne(
new LambdaQueryWrapper<UserFuncPermissionStock>()
.eq(UserFuncPermissionStock::getUserId, userId)
.eq(UserFuncPermissionStock::getFuncCode, funcCode)
);
// 无记录，无权限
if (stock == null) {
throw new BusinessException(BusinessExpCodeEnum.PERMISSION_DENIED, "无该功能权限");
}
// 时间维度校验
// 4. 时间维度校验
if (stock.getTimeLimit() == 1 && stock.getExpireTime() != null && stock.getExpireTime().isBefore(Instant.now())) {
throw new BusinessException(BusinessExpCodeEnum.PERMISSION_DENIED, "功能权限已过期");
}
// 不限次，直接放行
// 5. 次数维度校验
if (stock.getCountLimit() == 0) {
return;
// 不限次，直接放行
return insertUsageLog(userId, funcCode);
}
// 限次，扣减库存（乐观锁，SQL原子扣减）
// 限次，SQL原子扣减
int rows = userFuncPermissionStockMapper.update(null,
new LambdaUpdateWrapper<UserFuncPermissionStock>()
.setSql("remain_count = remain_count - 1")
@@ -61,10 +100,48 @@ public class FuncPermissionServer {
.eq(UserFuncPermissionStock::getFuncCode, funcCode)
.gt(UserFuncPermissionStock::getRemainCount, 0)
);
if (rows == 0) {
throw new BusinessException(BusinessExpCodeEnum.PERMISSION_DENIED, "功能使用次数已用完");
}
return insertUsageLog(userId, funcCode);
}
/**
* 插入使用记录
*
* @param userId 用户ID
* @param funcCode 功能编码
* @return 记录ID
*/
private Long insertUsageLog(Long userId, String funcCode) {
UserFuncUsageLog usageLog = new UserFuncUsageLog();
usageLog.setUserId(userId);
usageLog.setFuncCode(funcCode);
userFuncUsageLogMapper.insert(usageLog);
return usageLog.getId();
}
/**
* 回退使用记录（业务异常时调用）
* <p>删除使用记录，并尝试回退付费库存次数（SQL条件自动过滤）</p>
*
* @param logId 使用记录ID
* @param userId 用户ID
* @param funcCode 功能编码
*/
public void rollbackUsage(Long logId, Long userId, String funcCode) {
// 删除使用记录
userFuncUsageLogMapper.deleteById(logId);
// 尝试回退库存次数（count_limit=1才会匹配，其他情况update 0行无影响）
userFuncPermissionStockMapper.update(null,
new LambdaUpdateWrapper<UserFuncPermissionStock>()
.setSql("remain_count = remain_count + 1")
.eq(UserFuncPermissionStock::getUserId, userId)
.eq(UserFuncPermissionStock::getFuncCode, funcCode)
.eq(UserFuncPermissionStock::getCountLimit, 1)
);
}
/**
@@ -94,7 +171,6 @@ public class FuncPermissionServer {
UserFuncPermissionStock existing = query(userId, funcCode);
if (existing == null) {
// 无记录，新增
UserFuncPermissionStock stock = new UserFuncPermissionStock();
stock.setUserId(userId);
stock.setFuncCode(funcCode);
@@ -138,7 +214,6 @@ public class FuncPermissionServer {
UserFuncPermissionStock existing = query(userId, funcCode);
if (existing == null) {
// 无记录，新增
UserFuncPermissionStock stock = new UserFuncPermissionStock();
stock.setUserId(userId);
stock.setFuncCode(funcCode);
@@ -168,21 +243,4 @@ public class FuncPermissionServer {
userFuncPermissionStockMapper.update(null, wrapper);
}
/**
* 回退次数库存（业务异常时调用，仅限次维度有效）
*
* @param userId 用户ID
* @param funcCode 功能权限编码
*/
public void rollbackCount(Long userId, String funcCode) {
userFuncPermissionStockMapper.update(null,
new LambdaUpdateWrapper<UserFuncPermissionStock>()
.setSql("remain_count = remain_count + 1")
.eq(UserFuncPermissionStock::getUserId, userId)
.eq(UserFuncPermissionStock::getFuncCode, funcCode)
.eq(UserFuncPermissionStock::getCountLimit, 1)
);
}
}
manager/src/main/java/org/jiayunet/mapper/UserFuncUsageLogMapper.java
+14
View File
@@ -0,0 +1,14 @@
package org.jiayunet.mapper;
import org.apache.ibatis.annotations.Mapper;
import org.jiayunet.pojo.po.UserFuncUsageLog;
/**
* 用户功能使用记录Mapper
*
* @author zk
*/
@Mapper
public interface UserFuncUsageLogMapper extends CommonMapper<UserFuncUsageLog> {
}
manager/src/main/java/org/jiayunet/pojo/po/FuncPermission.java
+5
View File
@@ -30,6 +30,11 @@ public class FuncPermission {
*/
private String funcName;
/**
* 每日免费使用次数，0表示无免费额度
*/
private Integer dailyFreeCount;
/**
* 状态 1=启用 0=禁用
*/
manager/src/main/java/org/jiayunet/pojo/po/UserFuncUsageLog.java
+36
View File
@@ -0,0 +1,36 @@
package org.jiayunet.pojo.po;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import java.time.Instant;
/**
* 用户功能使用记录表
*
* @author zk
*/
@Data
@TableName(value = "bg_user_func_usage_log")
public class UserFuncUsageLog {
@TableId(type = IdType.ASSIGN_ID)
private Long id;
/**
* 用户ID
*/
private Long userId;
/**
* 功能编码
*/
private String funcCode;
/**
* 使用时间
*/
private Instant createTime;
}
项目结构说明.md
+18 -12
View File
@@ -56,7 +56,8 @@ offerpie/back-end
│ ├─ RouteMenuMapper.java # 路由菜单Mapper
│ ├─ FuncPermissionMapper.java # 功能权限Mapper
│ ├─ UserRouteMenuStockMapper.java # 用户路由菜单库存Mapper
│ └─ UserFuncPermissionStockMapper.java # 用户功能权限库存Mapper
│ ├─ UserFuncPermissionStockMapper.java # 用户功能权限库存Mapper
│ └─ UserFuncUsageLogMapper.java # 用户功能使用记录Mapper
├─ pojo/
│ ├─ po/ # 持久化实体
│ │ ├─ User.java
@@ -64,7 +65,8 @@ offerpie/back-end
│ │ ├─ RouteMenu.java # 路由菜单表（bg_route_menu）
│ │ ├─ FuncPermission.java # 功能权限表（bg_func_permission）
│ │ ├─ UserRouteMenuStock.java # 用户路由菜单库存表（bg_user_route_menu_stock）
│ │ └─ UserFuncPermissionStock.java # 用户功能权限库存表（bg_user_func_permission_stock）
│ │ ├─ UserFuncPermissionStock.java # 用户功能权限库存表（bg_user_func_permission_stock）
│ │ └─ UserFuncUsageLog.java # 用户功能使用记录表（bg_user_func_usage_log）
│ └─ vo/ # ViewObject（OssUrlVo 等）
└─ server/ # 业务 Service（OssServer、SmsServer 等）
```
@@ -75,7 +77,7 @@ offerpie/back-end
|------|----------|-----------|
| **client-api** | - 面向终端用户的 REST API <br> - 启动 Spring Boot 应用 <br> - 短信验证码登录（含自动注册） <br> - **功能权限校验**：注解 + 切面 + 权限服务（校验、扣减、回退） <br> - **路由菜单**：获取用户有效菜单树 | `ClientApplication`、`LoginController`、`RouteMenuController`、`FuncPermission`、`FuncPermissionAspect`、`FuncPermissionServer`、`RouteMenuServer`、`RouteMenuVo` |
| **common** | - **统一配置**：OSS、Redis、Security、WxPay、Sms 等 <br> - **跨层工具**：HTTP、IP、认证、验证码、Redis Server 等 <br> - **全局拦截/切面**：日志、TraceId、黑名单、SQL 打印 <br> - **统一异常/响应**：`GlobalExceptionAdvice`、`UnifiedResponse` <br> - **业务抽象**：邮件发送、微信支付（Native/JS/Transfer） <br> - **公共 POJO**：登录令牌、防重放信息等 | `config/`, `tool/`, `interceptor/`, `aop/`, `exception/`, `email/`, `wxPay/`, `pojo/` |
| **manager** | - **业务实体**（`User`、`OssFile`、`RouteMenu`、`FuncPermission`、`UserRouteMenuStock`、`UserFuncPermissionStock`） <br> - **MyBatis Mapper**（`UserMapper`、`OssFileMapper`、`RouteMenuMapper`、`FuncPermissionMapper`、`UserRouteMenuStockMapper`、`UserFuncPermissionStockMapper`） <br> - **业务 API**：文件上传/下载、健康检查等 <br> - **业务逻辑**：服务层、工具类等 <br> - **既供 B 端 UI（待实现）使用，也供 C 端业务直接调用** | `controller/`, `mapper/`, `pojo/po/`, `pojo/vo/`, `server/`, `constant/` |
| **manager** | - **业务实体**（`User`、`OssFile`、`RouteMenu`、`FuncPermission`、`UserRouteMenuStock`、`UserFuncPermissionStock`、`UserFuncUsageLog`） <br> - **MyBatis Mapper**（`UserMapper`、`OssFileMapper`、`RouteMenuMapper`、`FuncPermissionMapper`、`UserRouteMenuStockMapper`、`UserFuncPermissionStockMapper`、`UserFuncUsageLogMapper`） <br> - **业务 API**：文件上传/下载、健康检查等 <br> - **业务逻辑**：服务层、工具类等 <br> - **既供 B 端 UI（待实现）使用，也供 C 端业务直接调用** | `controller/`, `mapper/`, `pojo/po/`, `pojo/vo/`, `server/`, `constant/` |
## 3️⃣ 关键业务实体
| 实体 | 所属模块 | 作用概述 |
@@ -83,9 +85,10 @@ offerpie/back-end
| `User` | manager | 记录用户基础信息（手机号、邮箱、密码、昵称、微信绑定等），配合 `UserMapper` 完成持久化。 |
| `OssFile` | manager | 描述 OSS（对象存储）中文件的元数据（路径、大小、标签等），通过 `OssFileMapper` 进行增删改查。 |
| `RouteMenu` | manager | 路由菜单表（bg_route_menu），支持多级树形结构，通过 rootId/parentId 表达层级关系，openAccess 标识是否公开免费。 |
| `FuncPermission` | manager | 功能权限表（bg_func_permission），定义功能点编码（func_code，最长12字符，唯一约束）。 |
| `FuncPermission` | manager | 功能权限表（bg_func_permission），定义功能点编码（func_code，最长12字符，唯一约束），daily_free_count 配置每日免费次数。 |
| `UserRouteMenuStock` | manager | 用户路由菜单库存表（bg_user_route_menu_stock），记录用户拥有的菜单权限，支持时间维度。 |
| `UserFuncPermissionStock` | manager | 用户功能权限库存表（bg_user_func_permission_stock），记录用户拥有的功能权限，支持时间/次数/复合维度。 |
| `UserFuncUsageLog` | manager | 用户功能使用记录表（bg_user_func_usage_log），记录每次功能使用，用于免费次数统计和异常回退。 |
| `RouteMenuVo` | client-api | 路由菜单树形VO，包含 children 子菜单列表，供前端渲染动态路由。 |
| `LoginVo` | client-api | 登录成功后返回的用户信息（userId、nick）。 |
| `SmsLoginDto` | client-api | 短信验证码登录的请求参数（mobileNumber、code）。 |
@@ -96,13 +99,14 @@ offerpie/back-end
- **后端**：AOP 切面拦截 `@FuncPermission` 注解，校验权限 + 扣减库存，业务异常自动回退次数
- **权限来源**：商品模块下单成功后写入库存表，权限框架不关心来源
### 数据库表（4张）
### 数据库表（5张）
| 表名 | 说明 |
|------|------|
| `bg_route_menu` | 路由菜单定义（树形结构，open_access 标识公开免费菜单） |
| `bg_user_route_menu_stock` | 用户路由菜单库存（时间维度） |
| `bg_func_permission` | 功能权限定义（func_code 唯一） |
| `bg_func_permission` | 功能权限定义（func_code 唯一，daily_free_count 配置每日免费次数） |
| `bg_user_func_permission_stock` | 用户功能权限库存（时间/次数/复合维度） |
| `bg_user_func_usage_log` | 用户功能使用记录（每次使用插入，用于免费次数统计和异常回退） |
### 库存维度
- `time_limit` + `expire_time`：时间维度，0=不限时，1=限时
@@ -112,12 +116,14 @@ offerpie/back-end
### 切面校验流程
1. 拿注解上的 funcCode + 当前登录用户 userId
2. 查库存表（唯一索引 userId + funcCode）
3. 无记录 → 无权限
4. time_limit=1 且过期 → 已过期
5. count_limit=0 → 直接放行
6. count_limit=1 → SQL 原子扣减 `remain_count = remain_count - 1`（WHERE remain_count > 0）
7. 业务方法异常 → 自动回退次数
2. 查 bg_func_permission（funcCode + status=1），拿到 dailyFreeCount
3. dailyFreeCount > 0 → COUNT 今日使用记录，未超额 → 插入使用记录，放行
4. 免费额度用完或无免费额度 → 查付费库存表（userId + funcCode）
5. 无记录 → 无权限
6. time_limit=1 且过期 → 已过期
7. count_limit=0 → 插入使用记录，放行
8. count_limit=1 → SQL 原子扣减 `remain_count = remain_count - 1`（WHERE remain_count > 0）→ 插入使用记录，放行
9. 业务方法异常 → 删除使用记录 + 尝试回退库存次数（count_limit=1 时 remain_count + 1）
## 5️⃣ 共享技术栈（位于 `common`）
| 类别 | 关键实现 | 位置 |