[Refactor] decreased shuffle related transform in spark code