I'm never that happy with returning null. Why not return the key since that will be used to partition on.