zoujiajian 1 год назад
Родитель
Сommit
41fdcb53a1

+ 3 - 0
netflix-service/src/main/java/com/cyksj/service/midjourney/MidjourneyAccountService.java

@@ -4,6 +4,8 @@ import com.baomidou.mybatisplus.extension.service.IService;
 import com.cyksj.model.entity.MidjourneyAccount;
 import com.cyksj.model.entity.MidjourneyUser;
 
+import java.util.Date;
+
 /**
  * @author zwhui
  * @date 2024/4/12 15:54
@@ -21,4 +23,5 @@ public interface MidjourneyAccountService extends IService<MidjourneyAccount> {
 
     MidjourneyUser getMidjourneyUserToken(Long userId, Long relationId);
 
+    void updateUserExpiryTime(Long relationId, Date expiryTime);
 }

+ 8 - 4
netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidJourneyAccountServiceImpl.java

@@ -27,10 +27,7 @@ import javax.imageio.stream.FileImageOutputStream;
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Random;
+import java.util.*;
 
 /**
  * @author zwhui
@@ -226,6 +223,13 @@ public class MidJourneyAccountServiceImpl extends ServiceImpl<MidjourneyAccountM
         }
     }
 
+    @Override
+    public void updateUserExpiryTime(Long relationId, Date expiryTime) {
+        midjourneyUserMapper.update(null, Wrappers.lambdaUpdate(MidjourneyUser.class)
+                .set(MidjourneyUser::getExpireTime, expiryTime)
+                .eq(MidjourneyUser::getRelationId, relationId));
+    }
+
     /**
      * 获取车位信息
      *

+ 1 - 0
netflix-service/src/main/java/com/cyksj/service/relation/impl/GroupRelationFrontServiceImpl.java

@@ -1501,6 +1501,7 @@ public class GroupRelationFrontServiceImpl implements GroupRelationFrontService
 					ticket.setMjUserFastNum(search.getMjUserFastNum());
 					ticket.setMjUserRelaxNum(search.getMjUserRelaxNum());
 				}
+				midjourneyAccountService.updateUserExpiryTime(ticket.getRelationId(), ticket.getExpiryTime());
 			} catch (Exception e) {
 			}
 		}