zwhui 2 жил өмнө
parent
commit
71c800d29f

+ 6 - 0
netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidJourneyAccountServiceImpl.java

@@ -81,7 +81,12 @@ public class MidJourneyAccountServiceImpl extends ServiceImpl<MidjourneyAccountM
     @Override
     @Transactional(rollbackFor = Throwable.class)
     public void updateAccount(MidjourneyAccount midjourneyAccount) {
+        MidjourneyAccount account = baseMapper.selectById(midjourneyAccount.getId());
+        if (account == null) {
+            throw BusinessRuntimeException.getInstance("账号不存在");
+        }
         baseMapper.updateById(midjourneyAccount);
+        midjourneyAccount.setInstanceId(account.getInstanceId());
         updAccount(midjourneyAccount);
     }
 
@@ -92,6 +97,7 @@ public class MidJourneyAccountServiceImpl extends ServiceImpl<MidjourneyAccountM
             throw BusinessRuntimeException.getInstance("账号不存在");
         }
         midjourneyAccount.setStatus(!midjourneyAccount.getStatus());
+        baseMapper.updateById(midjourneyAccount);
         updAccount(midjourneyAccount);
     }