Преглед на файлове

feat: 限制company_rank资源上限

许家凯 преди 2 години
родител
ревизия
bdc8fbf0a9
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      src/main/java/com/winhc/max/compute/graph/job/CompanyRankGraphJob.java

+ 2 - 0
src/main/java/com/winhc/max/compute/graph/job/CompanyRankGraphJob.java

@@ -26,6 +26,7 @@ public class CompanyRankGraphJob {
         String maxIter = parameterTool.getOrDefault("maxIter", "30");
         String variance = parameterTool.getOrDefault("variance", "0.00000001");
         String numWorkers = parameterTool.getOrDefault("numWorkers", "70");
+        String workerCPU = parameterTool.getOrDefault("workerCPU", "2");
 
         System.out.println("input args: " + String.join(" ", args));
 
@@ -34,6 +35,7 @@ public class CompanyRankGraphJob {
         job.set("company.rank.variance", variance);
         job.setMaxIteration(-1);
         job.setNumWorkers(Integer.parseInt(numWorkers));
+        job.setWorkerCPU(Integer.parseInt(workerCPU) * 100);
 
 
         job.setGraphLoaderClass(CompanyRankVertexReader.class);