Cherrypicking partition number not changing fix#1408

Tang8330 wants to merge 6 commits intosegmentio:mainsegmentio/kafka-go:mainfrom

artie-labs:pr-1365artie-labs/kafka-go:pr-1365Copy head branch name to clipboard