zwhui 2 년 전
부모
커밋
71c800d29f
1개의 변경된 파일6개의 추가작업 그리고 0개의 파일을 삭제
  1. 6 0
      netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidJourneyAccountServiceImpl.java

+ 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);
     }