flykhan
/
kos
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更改 bot 创建规则：限制 bot 创建数量为 10 个

This commit is contained in:
flykhan 2023-03-10 23:56:42 +08:00
parent cb8d9f4e3e
commit 2e9ab75bdf
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backendcloud/backend/src/main/java/com/kob/backend/service/impl/user/bot/AddServiceImpl.java
View File

@ -1,5 +1,6 @@
package com.kob.backend.service.impl.user.bot; package com.kob.backend.service.impl.user.bot;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.kob.backend.mapper.BotMapper; import com.kob.backend.mapper.BotMapper;
import com.kob.backend.pojo.Bot; import com.kob.backend.pojo.Bot;
import com.kob.backend.pojo.User; import com.kob.backend.pojo.User;
@ -49,7 +50,7 @@ public class AddServiceImpl implements AddService {
description = "这个用户很懒，什么也没留下~"; description = "这个用户很懒，什么也没留下~";
} }
if (description != null && description.length() > 300) { if (description.length() > 300) {
map.put("error_message", "描述信息长度不能大于300"); map.put("error_message", "描述信息长度不能大于300");
return map; return map;
} }
@ -64,6 +65,13 @@ public class AddServiceImpl implements AddService {
return map; return map;
} }
QueryWrapper<Bot> queryWrapper = new QueryWrapper<>();
queryWrapper.eq("user_id", user.getId());
if (botMapper.selectCount(queryWrapper) >= 10) {
map.put("error_message", "每个用户最多可以创建 10 个 Bot");
return map;
}
// 定义当前时间 // 定义当前时间
Date now = new Date(); Date now = new Date();
// 定义一个新 Bot ;创建时间和修改时间开始应该一样 // 定义一个新 Bot ;创建时间和修改时间开始应该一样