From e39e08ea547a9bb21b8427e6469078df79d169ef Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Thu, 10 Apr 2025 11:53:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E7=A7=AF=E5=88=86=E6=89=A3?=
 =?UTF-8?q?=E9=99=A4=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ruoyi/platform/scheduling/ResourceOccupyTask.java | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ResourceOccupyTask.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ResourceOccupyTask.java
index 3fc162f4..8598f596 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ResourceOccupyTask.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ResourceOccupyTask.java
@@ -1,14 +1,15 @@
 package com.ruoyi.platform.scheduling;
 
-import com.ruoyi.system.api.constant.Constant;
 import com.ruoyi.platform.domain.DevEnvironment;
 import com.ruoyi.platform.domain.ServiceVersion;
 import com.ruoyi.platform.mapper.DevEnvironmentDao;
+import com.ruoyi.platform.mapper.ResourceOccupyDao;
 import com.ruoyi.platform.mapper.ServiceDao;
 import com.ruoyi.platform.service.JupyterService;
 import com.ruoyi.platform.service.ResourceOccupyService;
 import com.ruoyi.platform.service.ServiceService;
 import com.ruoyi.platform.vo.PodStatusVo;
+import com.ruoyi.system.api.constant.Constant;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 
@@ -23,6 +24,9 @@ public class ResourceOccupyTask {
     @Resource
     private ResourceOccupyService resourceOccupyService;
 
+    @Resource
+    private ResourceOccupyDao resourceOccupyDao;
+
     @Resource
     private DevEnvironmentDao devEnvironmentDao;
 
@@ -46,7 +50,10 @@ public class ResourceOccupyTask {
                 devEnv.setUrl(podStatusVo.getUrl());
                 this.devEnvironmentDao.update(devEnv);
             }
-            if (Constant.Running.equals(devEnv.getStatus())) {
+            Long userId = resourceOccupyDao.getResourceOccupyByTask(Constant.TaskType_Dev, Long.valueOf(devEnv.getId()), null, null).get(0).getUserId();
+            if (resourceOccupyDao.getUserCredit(userId) <= 0) {
+                jupyterService.stopJupyterService(devEnv.getId());
+            } else if (Constant.Running.equals(devEnv.getStatus())) {
                 resourceOccupyService.deducing(Constant.TaskType_Dev, Long.valueOf(devEnv.getId()), null, null, null);
             }
         }