code: public synchronized int getPartitionStoreCount() {
          int count = 0;
          for (String store : storeToPartitionIds.keySet()) {
              count += storeToPartitionIds.get(store).size();
          }
          return count;
      }
docs: Returns the total count of partitions across all stores.
  
  @return returns the total count of partitions across all stores.