client-flink/flink-1.14/src/main/java/org/apache/celeborn/plugin/flink/RemoteShuffleServiceFactory.java [24:54]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class RemoteShuffleServiceFactory extends AbstractRemoteShuffleServiceFactory
    implements ShuffleServiceFactory<
        RemoteShuffleDescriptor, ResultPartitionWriter, IndexedInputGate> {

  @Override
  public ShuffleMaster<RemoteShuffleDescriptor> createShuffleMaster(
      ShuffleMasterContext shuffleMasterContext) {
    return new RemoteShuffleMaster(shuffleMasterContext, new SimpleResultPartitionAdapter());
  }

  @Override
  public ShuffleEnvironment<ResultPartitionWriter, IndexedInputGate> createShuffleEnvironment(
      ShuffleEnvironmentContext shuffleEnvironmentContext) {
    AbstractRemoteShuffleServiceParameters parameters =
        initializePreCreateShuffleEnvironment(shuffleEnvironmentContext);
    RemoteShuffleResultPartitionFactory resultPartitionFactory =
        new RemoteShuffleResultPartitionFactory(
            parameters.celebornConf,
            parameters.resultPartitionManager,
            parameters.networkBufferPool,
            parameters.bufferSize);
    RemoteShuffleInputGateFactory inputGateFactory =
        new RemoteShuffleInputGateFactory(
            parameters.celebornConf, parameters.networkBufferPool, parameters.bufferSize);

    return new RemoteShuffleEnvironment(
        parameters.networkBufferPool,
        parameters.resultPartitionManager,
        resultPartitionFactory,
        inputGateFactory,
        parameters.celebornConf);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



client-flink/flink-1.17/src/main/java/org/apache/celeborn/plugin/flink/RemoteShuffleServiceFactory.java [28:58]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class RemoteShuffleServiceFactory extends AbstractRemoteShuffleServiceFactory
    implements ShuffleServiceFactory<
        RemoteShuffleDescriptor, ResultPartitionWriter, IndexedInputGate> {

  @Override
  public ShuffleMaster<RemoteShuffleDescriptor> createShuffleMaster(
      ShuffleMasterContext shuffleMasterContext) {
    return new RemoteShuffleMaster(shuffleMasterContext, new SimpleResultPartitionAdapter());
  }

  @Override
  public ShuffleEnvironment<ResultPartitionWriter, IndexedInputGate> createShuffleEnvironment(
      ShuffleEnvironmentContext shuffleEnvironmentContext) {
    AbstractRemoteShuffleServiceParameters parameters =
        initializePreCreateShuffleEnvironment(shuffleEnvironmentContext);
    RemoteShuffleResultPartitionFactory resultPartitionFactory =
        new RemoteShuffleResultPartitionFactory(
            parameters.celebornConf,
            parameters.resultPartitionManager,
            parameters.networkBufferPool,
            parameters.bufferSize);
    RemoteShuffleInputGateFactory inputGateFactory =
        new RemoteShuffleInputGateFactory(
            parameters.celebornConf, parameters.networkBufferPool, parameters.bufferSize);

    return new RemoteShuffleEnvironment(
        parameters.networkBufferPool,
        parameters.resultPartitionManager,
        resultPartitionFactory,
        inputGateFactory,
        parameters.celebornConf);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -