Skip to content

Commit 17640a8

Browse files
committed
[FLINK-33653][runtime] Introduce a benchmark for balanced tasks scheduling.
Broaden the access scope for classes associated with the balanced task scheduling benchmark.
1 parent 8dc699a commit 17640a8

File tree

5 files changed

+8
-8
lines changed

5 files changed

+8
-8
lines changed

flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PendingRequest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
import java.util.Set;
3333
import java.util.concurrent.CompletableFuture;
3434

35-
final class PendingRequest implements WeightLoadable {
35+
public final class PendingRequest implements WeightLoadable {
3636

3737
private final SlotRequestId slotRequestId;
3838

@@ -72,7 +72,7 @@ static PendingRequest createBatchRequest(
7272
slotRequestId, resourceProfile, loadingWeight, preferredAllocations, true);
7373
}
7474

75-
static PendingRequest createNormalRequest(
75+
public static PendingRequest createNormalRequest(
7676
SlotRequestId slotRequestId,
7777
ResourceProfile resourceProfile,
7878
LoadingWeight loadingWeight,

flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/slotpool/PhysicalSlotRequest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ public boolean willSlotBeOccupiedIndefinitely() {
6565
return slotWillBeOccupiedIndefinitely;
6666
}
6767

68-
PendingRequest toPendingRequest() {
68+
public PendingRequest toPendingRequest() {
6969
return slotWillBeOccupiedIndefinitely
7070
? PendingRequest.createNormalRequest(
7171
slotRequestId,

flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/LocalInputPreferredSlotSharingStrategy.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@
4747
* belong to the same SlotSharingGroup, tend to be put in the same ExecutionSlotSharingGroup.
4848
* Co-location constraints will be respected.
4949
*/
50-
class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy {
50+
public class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy {
5151

5252
LocalInputPreferredSlotSharingStrategy(
5353
final SchedulingTopology topology,
@@ -56,7 +56,7 @@ class LocalInputPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy
5656
super(topology, logicalSlotSharingGroups, coLocationGroups);
5757
}
5858

59-
static class Factory implements SlotSharingStrategy.Factory {
59+
public static class Factory implements SlotSharingStrategy.Factory {
6060

6161
public LocalInputPreferredSlotSharingStrategy create(
6262
final SchedulingTopology topology,

flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/SlotSharingStrategy.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
import java.util.Set;
2727

2828
/** Strategy which determines {@link ExecutionSlotSharingGroup} for each execution vertex. */
29-
interface SlotSharingStrategy {
29+
public interface SlotSharingStrategy {
3030

3131
ExecutionSlotSharingGroup getExecutionSlotSharingGroup(ExecutionVertexID executionVertexId);
3232

flink-runtime/src/main/java/org/apache/flink/runtime/scheduler/TaskBalancedPreferredSlotSharingStrategy.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
* the same SlotSharingGroup, tend to be put evenly in each ExecutionSlotSharingGroup. Co-location
3636
* constraints will be respected.
3737
*/
38-
class TaskBalancedPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy {
38+
public class TaskBalancedPreferredSlotSharingStrategy extends AbstractSlotSharingStrategy {
3939

4040
public static final Logger LOG =
4141
LoggerFactory.getLogger(TaskBalancedPreferredSlotSharingStrategy.class);
@@ -57,7 +57,7 @@ protected Map<ExecutionVertexID, ExecutionSlotSharingGroup> computeExecutionSlot
5757
.build();
5858
}
5959

60-
static class Factory implements SlotSharingStrategy.Factory {
60+
public static class Factory implements SlotSharingStrategy.Factory {
6161

6262
public TaskBalancedPreferredSlotSharingStrategy create(
6363
final SchedulingTopology topology,

0 commit comments

Comments
 (0)