更改 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

View File

@ -1,5 +1,6 @@
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.pojo.Bot;
import com.kob.backend.pojo.User;
@ -49,7 +50,7 @@ public class AddServiceImpl implements AddService {
description = "这个用户很懒,什么也没留下~";
}
if (description != null && description.length() > 300) {
if (description.length() > 300) {
map.put("error_message", "描述信息长度不能大于300");
return map;
}
@ -64,6 +65,13 @@ public class AddServiceImpl implements AddService {
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();
// 定义一个新 Bot ;创建时间和修改时间开始应该一样