diff --git a/backendcloud/backend/src/main/java/com/kob/backend/service/impl/user/bot/AddServiceImpl.java b/backendcloud/backend/src/main/java/com/kob/backend/service/impl/user/bot/AddServiceImpl.java index df1afa5..a6f9a11 100644 --- a/backendcloud/backend/src/main/java/com/kob/backend/service/impl/user/bot/AddServiceImpl.java +++ b/backendcloud/backend/src/main/java/com/kob/backend/service/impl/user/bot/AddServiceImpl.java @@ -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 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 ;创建时间和修改时间开始应该一样