|
|
@@ -61,9 +61,19 @@ public class MidjourneyController {
|
|
|
}
|
|
|
}else {
|
|
|
Object fastNum = redisService.get(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId());
|
|
|
- midjourneyUser.setMjFastNum(Integer.parseInt(fastNum.toString()));
|
|
|
+ if (fastNum == null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjFastNum(), RedisService.key.MIDJOURNEY_FAST_LIMIT.getTimeout());
|
|
|
+ }else {
|
|
|
+ midjourneyUser.setMjFastNum(Integer.parseInt(fastNum.toString()));
|
|
|
+ }
|
|
|
Object relaxNum = redisService.get(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId());
|
|
|
- midjourneyUser.setMjRelaxNum(Integer.parseInt(relaxNum.toString()));
|
|
|
+ if (relaxNum == null){
|
|
|
+ if (midjourneyUser.getMjRelaxNum() != null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjRelaxNum(), RedisService.key.MIDJOURNEY_RELAX_LIMIT.getTimeout());
|
|
|
+ }
|
|
|
+ }else {
|
|
|
+ midjourneyUser.setMjRelaxNum(Integer.parseInt(relaxNum.toString()));
|
|
|
+ }
|
|
|
}
|
|
|
return midjourneyUser;
|
|
|
}
|