mlx

api-common/src/main/java/api/common/util/HttpUtil.java

@@ -150,6 +150,7 @@ public class HttpUtil {
         }
     }
 
+
 //
 //    /**
 //     * 通过 get 请求下载文件

simulation-oauth-server/src/main/java/com/css/simulation/oauth/server/cofiguration/oauth/OauthAuthorizationServerConfiguration.java

@@ -55,7 +55,7 @@ public class OauthAuthorizationServerConfiguration extends AuthorizationServerCo
                         "simulation-resource-scheduler",
                         "simulation-resource-sever"
                 )  // 客户端拥有的资源列表
-                .authorizedGrantTypes("password", "refresh_token") // 允许的授权方式
+                .authorizedGrantTypes("client_credentials", "password", "refresh_token") // 允许的授权方式
                 .scopes("all")    // 授权范围划分
                 .autoApprove(true)
                 .and()

simulation-resource-scheduler/src/main/java/com/css/simulation/resource/scheduler/service/TaskService.java

@@ -14,9 +14,13 @@ import com.css.simulation.resource.scheduler.pojo.po.TaskIndexPO;
 import com.css.simulation.resource.scheduler.pojo.po.TaskPO;
 import com.css.simulation.resource.scheduler.pojo.to.ScoreTO;
 import com.css.simulation.resource.scheduler.util.MinioUtil;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import io.minio.MinioClient;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.http.client.config.RequestConfig;
+import org.apache.http.impl.client.CloseableHttpClient;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.data.redis.core.StringRedisTemplate;
@@ -30,6 +34,12 @@ import java.util.stream.Collectors;
 public class TaskService {
 
     private final String USER_ID = "simulation-resource-scheduler";
+
+
+    @Autowired
+    CloseableHttpClient closeableHttpClient;
+    @Autowired
+    RequestConfig requestConfig;
     @Autowired
     MinioClient minioClient;
     @Value("${minio.bucket-name}")
@@ -62,6 +72,14 @@ public class TaskService {
     String pyPath;
     @Value("${scheduler.linux-temp-path}")
     String linuxTempPath;
+    @Value("${simulation-cloud.client-id}")
+    String clientId;
+    @Value("${simulation-cloud.client-secret}")
+    String clientSecret;
+    @Value("${simulation-cloud.token-uri}")
+    String tokenUri;
+    @Value("${simulation-cloud.evaluation-level-uri}")
+    String evaluationLevelUri;
 
     public void taskTick(String taskId) {
         log.info("------- /tick 接收到任务 " + taskId + "的心跳！");
@@ -200,6 +218,18 @@ public class TaskService {
         totalTaskIndex.setIsDeleted("0");
         taskIndexMapper.insertTotalIndex(totalTaskIndex);
 
+        // 调用 server 的接口，计算评价等级
+        String tokenUrl = tokenUri + "?grant_type=client_credentials"
+                + "&client_id=" + clientId
+                + "client_secret" + clientSecret;
+        String response = HttpUtil.get(closeableHttpClient, requestConfig, tokenUrl);
+        ObjectMapper objectMapper = new ObjectMapper();
+        JsonNode jsonNode = objectMapper.readTree(response);
+        String accessToken = jsonNode.path("access_token").asText();
+        Map<String, String> headers = new HashMap<>();
+        headers.put("Authorization", "Bearer " + accessToken);
+        HttpUtil.post(closeableHttpClient, requestConfig, evaluationLevelUri, headers, null);
+
 
     }