Class ShuffleUtils

java.lang.Object
org.apache.flink.runtime.shuffle.ShuffleUtils

public class ShuffleUtils extends Object
Common utility methods for shuffle service.
  • Method Details

    • applyWithShuffleTypeCheck

      public static <T, SD extends ShuffleDescriptor> T applyWithShuffleTypeCheck(Class<SD> shuffleDescriptorClass, ShuffleDescriptor shuffleDescriptor, Function<UnknownShuffleDescriptor,T> functionOfUnknownDescriptor, Function<SD,T> functionOfKnownDescriptor)
      Apply different functions to known and unknown ShuffleDescriptors.

      Also casts known ShuffleDescriptor.

      Type Parameters:
      T - return type of called functions
      SD - concrete type of shuffleDescriptor to check
      Parameters:
      shuffleDescriptorClass - concrete class of shuffleDescriptor
      shuffleDescriptor - concrete shuffle descriptor to check
      functionOfUnknownDescriptor - supplier to call in case shuffleDescriptor is unknown
      functionOfKnownDescriptor - function to call in case shuffleDescriptor is known
      Returns:
      result of either function call