loadSpaceStatistics - explicit EXTENSION

Signed-off-by: qGYdXbY2 <[email protected]>
heremaps · Dec 6, 2024 · 1519fc1 · 1519fc1
1 parent 219cbf3
commit 1519fc1
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/xyz-jobs/xyz-job-service/src/main/java/com/here/xyz/jobs/steps/compiler/SpaceCopy.java b/xyz-jobs/xyz-job-service/src/main/java/com/here/xyz/jobs/steps/compiler/SpaceCopy.java
@@ -35,6 +35,7 @@
 import com.here.xyz.jobs.steps.impl.transport.CopySpacePre;
 import com.here.xyz.jobs.steps.impl.transport.CopySpacePost;
 import com.here.xyz.models.hub.Ref;
+import com.here.xyz.models.hub.Space;
 import com.here.xyz.responses.StatisticsResponse;
 import com.here.xyz.util.web.HubWebClient;
 import com.here.xyz.util.web.XyzWebClient.WebClientException;
@@ -64,15 +65,23 @@ private static int threadCountCalc( long sourceFeatureCount, long targetFeatureC
     return PARALLELIZTATION_THREAD_MAX; 
   }
 
+
+  private static StatisticsResponse _loadSpaceStatistics(String spaceId) throws WebClientException
+  {
+   Space sourceSpace = HubWebClient.getInstance(Config.instance.HUB_ENDPOINT).loadSpace(spaceId);
+   boolean isExtended = sourceSpace.getExtension() != null;
+   return HubWebClient.getInstance(Config.instance.HUB_ENDPOINT).loadSpaceStatistics(spaceId, isExtended ? SpaceContext.EXTENSION : null);  
+  }
+
   public static CompilationStepGraph compileSteps(String sourceId, String targetId, String jobId, Filters filters, Ref versionRef) 
   {
     final String sourceSpaceId = sourceId,
                  targetSpaceId = targetId;
 
     StatisticsResponse sourceStatistics = null, targetStatistics = null;
     try {
-      sourceStatistics = HubWebClient.getInstance(Config.instance.HUB_ENDPOINT).loadSpaceStatistics(sourceSpaceId,SpaceContext.EXTENSION);
-      targetStatistics = HubWebClient.getInstance(Config.instance.HUB_ENDPOINT).loadSpaceStatistics(targetSpaceId,SpaceContext.EXTENSION);
+      sourceStatistics = _loadSpaceStatistics(sourceSpaceId);
+      targetStatistics = _loadSpaceStatistics(targetSpaceId);
     } catch (WebClientException e) {
       String errMsg = String.format("Unable to get Staistics for %s", sourceStatistics == null ? sourceSpaceId : targetSpaceId );
       throw new JobCompiler.CompilationError(errMsg);