org.apache.flink.api.common.state.KeyGroupAssigner<K> to int