I am very skeptical about this change. Each shard has approx 10000 columns. And we are reading 128 rows without pagination. This could hold a single connection for way too long. Either we should paginate or revert back to the older way of getting states for one shard at a time.