diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PendingRequest.java b/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PendingRequest.java index c25ef0daa6196..339db7f575f0a 100644 --- a/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PendingRequest.java +++ b/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PendingRequest.java @@ -32,7 +32,7 @@ import java.util.Set; import java.util.concurrent.CompletableFuture; -final class PendingRequest implements WeightLoadable { +public final class PendingRequest implements WeightLoadable { private final SlotRequestId slotRequestId; @@ -72,7 +72,7 @@ static PendingRequest createBatchRequest( slotRequestId, resourceProfile, loadingWeight, preferredAllocations, true); } - static PendingRequest createNormalRequest( + public static PendingRequest createNormalRequest( SlotRequestId slotRequestId, ResourceProfile resourceProfile, LoadingWeight loadingWeight, diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PhysicalSlotRequest.java b/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PhysicalSlotRequest.java index 89122440a5631..08a9bfe5dbf0c 100644 --- a/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PhysicalSlotRequest.java +++ b/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PhysicalSlotRequest.java @@ -65,7 +65,7 @@ public boolean willSlotBeOccupiedIndefinitely() { return slotWillBeOccupiedIndefinitely; } - PendingRequest toPendingRequest() { + public PendingRequest toPendingRequest() { return slotWillBeOccupiedIndefinitely ? PendingRequest.createNormalRequest( slotRequestId, diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/LocalInputPreferredSlotSharingStrategy.java b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/LocalInputPreferredSlotSharingStrategy.java index 8a57e6b5cdb94..c3778a84677ac 100644 --- a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/LocalInputPreferredSlotSharingStrategy.java +++ b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/LocalInputPreferredSlotSharingStrategy.java @@ -47,7 +47,7 @@ * belong to the same SlotSharingGroup, tend to be put in the same ExecutionSlotSharingGroup. * Co-location constraints will be respected. */ -class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy { +public class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy { LocalInputPreferredSlotSharingStrategy( final SchedulingTopology topology, @@ -56,7 +56,7 @@ class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy super(topology, logicalSlotSharingGroups, coLocationGroups); } - static class Factory implements SlotSharingStrategy.Factory { + public static class Factory implements SlotSharingStrategy.Factory { public LocalInputPreferredSlotSharingStrategy create( final SchedulingTopology topology, diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/SlotSharingStrategy.java b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/SlotSharingStrategy.java index 248961a107c2c..ebb9baf7891bc 100644 --- a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/SlotSharingStrategy.java +++ b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/SlotSharingStrategy.java @@ -26,7 +26,7 @@ import java.util.Set; /** Strategy which determines {@link ExecutionSlotSharingGroup} for each execution vertex. */ -interface SlotSharingStrategy { +public interface SlotSharingStrategy { ExecutionSlotSharingGroup getExecutionSlotSharingGroup(ExecutionVertexID executionVertexId); diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/TaskBalancedPreferredSlotSharingStrategy.java b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/TaskBalancedPreferredSlotSharingStrategy.java index 571ac9d3f573c..4efef9914a94f 100644 --- a/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/TaskBalancedPreferredSlotSharingStrategy.java +++ b/flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/TaskBalancedPreferredSlotSharingStrategy.java @@ -35,7 +35,7 @@ * the same SlotSharingGroup, tend to be put evenly in each ExecutionSlotSharingGroup. Co-location * constraints will be respected. */ -class TaskBalancedPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy { +public class TaskBalancedPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy { public static final Logger LOG = LoggerFactory.getLogger(TaskBalancedPreferredSlotSharingStrategy.class); @@ -57,7 +57,7 @@ protected Map computeExecutionSlot .build(); } - static class Factory implements SlotSharingStrategy.Factory { + public static class Factory implements SlotSharingStrategy.Factory { public TaskBalancedPreferredSlotSharingStrategy create( final SchedulingTopology topology,