fix alpha 限速

midjourney/src/main/java/com/yhlxj/service/midjourney/MidjourneyService.java

@@ -52,4 +52,5 @@ public interface MidjourneyService {
 
     void saveConversationRecord(String userToken, AlphaSubmitRequestDTO conversationRequest,String carid,String status);
 
+    void userSubmitLimit(MidjourneyUser user,String action,Object customId,Boolean again);
 }

midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -650,6 +650,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
     /**
      * 用户生图次数限制
      */
+    @Override
     public void userSubmitLimit(MidjourneyUser user,String action,Object customId,Boolean again) {
         if (again) {
             return;
@@ -1450,13 +1451,13 @@ public class MidjourneyServiceImpl implements MidjourneyService {
     @Override
     public void saveConversationRecord(String userToken, AlphaSubmitRequestDTO conversationRequest,String carid,String status) {
         LambdaQueryWrapper<MidjourneySession> wrapper = Wrappers.lambdaQuery(MidjourneySession.class).eq(MidjourneySession::getCarId, carid);
-        MidjourneySession chatgptSession = midjourneySessionMapper.selectOne(wrapper);
+        MidjourneySession midjourneySession = midjourneySessionMapper.selectOne(wrapper);
         MidjourneyUserConversationRecord midjourneyUserConversationRecord = midjourneyUserConversationRecordMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversationRecord.class).eq(MidjourneyUserConversationRecord::getConversationId, conversationRequest.getId()));
         if (midjourneyUserConversationRecord == null) {
             midjourneyUserConversationRecord = new MidjourneyUserConversationRecord();
             midjourneyUserConversationRecord.setUserToken(userToken);
-            midjourneyUserConversationRecord.setCarId(chatgptSession.getCarId());
-            midjourneyUserConversationRecord.setCarName(chatgptSession.getCarName());
+            midjourneyUserConversationRecord.setCarId(midjourneySession.getCarId());
+            midjourneyUserConversationRecord.setCarName(midjourneySession.getCarName());
             midjourneyUserConversationRecord.setConversationId(conversationRequest.getId());
             midjourneyUserConversationRecord.setMode(conversationRequest.getFlags().getMode());
             midjourneyUserConversationRecord.setAction(conversationRequest.getEventType());
@@ -1469,8 +1470,8 @@ public class MidjourneyServiceImpl implements MidjourneyService {
             return;
         }
 
-        if (StringUtils.isNotBlank(chatgptSession.getCarId())) {
-            updateExperienceAndScore(chatgptSession.getCarId(), System.currentTimeMillis());
+        if (StringUtils.isNotBlank(midjourneySession.getCarId())) {
+            updateExperienceAndScore(midjourneySession.getCarId(), System.currentTimeMillis());
         }
 
     }

midjourney/src/main/java/com/yhlxj/web/mirror/AlphaMidjourneyController.java

@@ -65,6 +65,11 @@ public class AlphaMidjourneyController {
         if (StringUtils.isBlank(mode)) {
             return badRequestResponse("模式错误.请选择fast 或者 relax模式");
         }
+        try {
+            midjourneyService.userSubmitLimit(user,"imagine", "1234",false);
+        }catch (Exception e){
+            return badRequestResponse(e.getMessage());
+        }
 
         if (MODE_TURBO.equals(mode)) {
             return badRequestResponse("暂不支持turbo 模式.请选择fast 或者 relax模式");