ENG-4894: StreamSnapshot* refactoring and @BalancePartitions