|
|
@@ -68,7 +68,7 @@ public class MidjourneyController {
|
|
|
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()));
|
|
|
+ midjourneyUser.setMjFastNum(Long.parseLong(fastNum.toString()));
|
|
|
}
|
|
|
Object relaxNum = redisService.get(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId());
|
|
|
if (relaxNum == null){
|
|
|
@@ -76,7 +76,7 @@ public class MidjourneyController {
|
|
|
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()));
|
|
|
+ midjourneyUser.setMjRelaxNum(Long.parseLong(relaxNum.toString()));
|
|
|
}
|
|
|
}
|
|
|
return midjourneyUser;
|
|
|
@@ -115,12 +115,14 @@ public class MidjourneyController {
|
|
|
/**
|
|
|
* 恢复次数
|
|
|
*/
|
|
|
- public void recoverUserLimit(Long id,Integer mode){
|
|
|
+ public void recoverUserLimit(Long id,Integer mode,Long num){
|
|
|
if (mode == 1){
|
|
|
redisService.incr(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + id, 1L);
|
|
|
}
|
|
|
if (mode == 2){
|
|
|
- redisService.incr(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + id, 1L);
|
|
|
+ if (num != null){
|
|
|
+ redisService.incr(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + id, 1L);
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -169,7 +171,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitImagine(user, submitImagineDTO.getPrompt(), submitImagineDTO.getBase64Array());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|
|
|
@@ -191,7 +193,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitDescribe(user, submitDescribeDTO.getBase64());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|
|
|
@@ -213,7 +215,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitBlend(user, submitBlendDTO.getDimensions(), submitBlendDTO.getBase64Array());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|
|
|
@@ -235,7 +237,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitModal(user, submitModalDTO.getTaskId(), submitModalDTO.getPrompt(), submitModalDTO.getMaskBase64());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|
|
|
@@ -257,7 +259,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitShorten(user, submitShortenDTO.getPrompt());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|
|
|
@@ -279,7 +281,7 @@ public class MidjourneyController {
|
|
|
try {
|
|
|
conversation = midjourneyService.submitAction(user, actionDTO.getTaskId(), actionDTO.getCustomId());
|
|
|
} catch (Exception e) {
|
|
|
- recoverUserLimit(user.getId(), user.getMode());
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
throw BusinessRuntimeException.getInstance(e.getMessage());
|
|
|
}
|
|
|
syncUser(user.getId(), user.getMode(), num);
|