This class will manage the keys during a reduce-by-key worklet
execution.