MINOR: Refine `PartitionAssignor` server-side interface (#13524)

This patch updates the `PartitionAssignor` server-side interface used in the new group coordinator for the new consumer group protocol as follow:
- It switches subscription from topic names to topic ids in order to be closer to the server side implementation.
- It switches assignment from Set to Map<Integer, Set> to be closer to the server side implementation.
- It adds getters for all attributes.
- It makes all attributes final private.

Reviewers: Jeff Kim <jeff.kim@confluent.io>, Alexandre Dupriez <alexandre.dupriez@gmail.com>, David Jacot <djacot@confluent.io>

group-coordinator/src/main/java/org/apache/kafka/coordinator/group/assignor/AssignmentMemberSpec.java

@@ -16,11 +16,13 @@
  */
 package org.apache.kafka.coordinator.group.assignor;
 
-import org.apache.kafka.common.TopicPartition;
+import org.apache.kafka.common.Uuid;
 
 import java.util.Collection;
+import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;
+import java.util.Set;
 
 /**
  * The assignment specification for a consumer group member.
@@ -29,58 +31,84 @@ public class AssignmentMemberSpec {
     /**
      * The instance ID if provided.
      */
-    final Optional<String> instanceId;
+    private final Optional<String> instanceId;
 
     /**
      * The rack ID if provided.
      */
-    final Optional<String> rackId;
+    private final Optional<String> rackId;
 
     /**
-     * The topics that the member is subscribed to.
+     * Topics Ids that the member is subscribed to.
      */
-    final Collection<String> subscribedTopics;
+    private final Collection<Uuid> subscribedTopicIds;
 
     /**
-     * The current target partitions of the member.
+     * Partitions assigned keyed by topicId.
      */
-    final Collection<TopicPartition> targetPartitions;
+    private final Map<Uuid, Set<Integer>> assignedPartitions;
+
+    /**
+     * @return The instance ID as an Optional.
+     */
+    public Optional<String> instanceId() {
+        return instanceId;
+    }
+
+    /**
+     * @return The rack ID as an Optional.
+     */
+    public Optional<String> rackId() {
+        return rackId;
+    }
+
+    /**
+     * @return Collection of subscribed topic Ids.
+     */
+    public Collection<Uuid> subscribedTopicIds() {
+        return subscribedTopicIds;
+    }
+
+    /**
+     * @return Assigned partitions keyed by topic Ids.
+     */
+    public Map<Uuid, Set<Integer>> assignedPartitions() {
+        return assignedPartitions;
+    }
 
     public AssignmentMemberSpec(
         Optional<String> instanceId,
         Optional<String> rackId,
-        Collection<String> subscribedTopics,
-        Collection<TopicPartition> targetPartitions
+        Collection<Uuid> subscribedTopicIds,
+        Map<Uuid, Set<Integer>> assignedPartitions
     ) {
         Objects.requireNonNull(instanceId);
         Objects.requireNonNull(rackId);
-        Objects.requireNonNull(subscribedTopics);
-        Objects.requireNonNull(targetPartitions);
+        Objects.requireNonNull(subscribedTopicIds);
+        Objects.requireNonNull(assignedPartitions);
         this.instanceId = instanceId;
         this.rackId = rackId;
-        this.subscribedTopics = subscribedTopics;
-        this.targetPartitions = targetPartitions;
+        this.subscribedTopicIds = subscribedTopicIds;
+        this.assignedPartitions = assignedPartitions;
     }
 
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
-
         AssignmentMemberSpec that = (AssignmentMemberSpec) o;
-
         if (!instanceId.equals(that.instanceId)) return false;
         if (!rackId.equals(that.rackId)) return false;
-        if (!subscribedTopics.equals(that.subscribedTopics)) return false;
-        return targetPartitions.equals(that.targetPartitions);
+        if (!subscribedTopicIds.equals(that.subscribedTopicIds)) return false;
+        return assignedPartitions.equals(that.assignedPartitions);
     }
 
     @Override
     public int hashCode() {
         int result = instanceId.hashCode();
         result = 31 * result + rackId.hashCode();
-        result = 31 * result + subscribedTopics.hashCode();
-        result = 31 * result + targetPartitions.hashCode();
+        result = 31 * result + subscribedTopicIds.hashCode();
+        result = 31 * result + assignedPartitions.hashCode();
         return result;
     }
 
@@ -88,8 +116,8 @@ public class AssignmentMemberSpec {
     public String toString() {
         return "AssignmentMemberSpec(instanceId=" + instanceId +
             ", rackId=" + rackId +
-            ", subscribedTopics=" + subscribedTopics +
-            ", targetPartitions=" + targetPartitions +
+            ", subscribedTopicIds=" + subscribedTopicIds +
+            ", assignedPartitions=" + assignedPartitions +
             ')';
     }
 }

group-coordinator/src/main/java/org/apache/kafka/coordinator/group/assignor/AssignmentSpec.java

@@ -28,12 +28,12 @@ public class AssignmentSpec {
     /**
      * The members keyed by member id.
      */
-    final Map<String, AssignmentMemberSpec> members;
+    private final Map<String, AssignmentMemberSpec> members;
 
     /**
-     * The topics' metadata keyed by topic id
+     * The topics' metadata keyed by topic id.
      */
-    final Map<Uuid, AssignmentTopicMetadata> topics;
+    private final Map<Uuid, AssignmentTopicMetadata> topics;
 
     public AssignmentSpec(
         Map<String, AssignmentMemberSpec> members,
@@ -45,13 +45,25 @@ public class AssignmentSpec {
         this.topics = topics;
     }
 
+    /**
+     * @return Member metadata keyed by member Ids.
+     */
+    public Map<String, AssignmentMemberSpec> members() {
+        return members;
+    }
+
+    /**
+     * @return Topic metadata keyed by topic Ids.
+     */
+    public Map<Uuid, AssignmentTopicMetadata> topics() {
+        return topics;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
-
         AssignmentSpec that = (AssignmentSpec) o;
-
         if (!members.equals(that.members)) return false;
         return topics.equals(that.topics);
     }

group-coordinator/src/main/java/org/apache/kafka/coordinator/group/assignor/AssignmentTopicMetadata.java

@@ -16,53 +16,44 @@
  */
 package org.apache.kafka.coordinator.group.assignor;
 
-import java.util.Objects;
-
 /**
  * Metadata of a topic.
  */
 public class AssignmentTopicMetadata {
-    /**
-     * The topic name.
-     */
-    final String topicName;
 
     /**
      * The number of partitions.
      */
-    final int numPartitions;
+    private final int numPartitions;
 
     public AssignmentTopicMetadata(
-        String topicName,
         int numPartitions
     ) {
-        Objects.requireNonNull(topicName);
-        this.topicName = topicName;
         this.numPartitions = numPartitions;
     }
 
+    /**
+     * @return The number of partitions present for the topic.
+     */
+    public int numPartitions() {
+        return numPartitions;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
-
         AssignmentTopicMetadata that = (AssignmentTopicMetadata) o;
-
-        if (numPartitions != that.numPartitions) return false;
-        return topicName.equals(that.topicName);
+        return numPartitions == that.numPartitions;
     }
 
     @Override
     public int hashCode() {
-        int result = topicName.hashCode();
-        result = 31 * result + numPartitions;
-        return result;
+        return numPartitions;
     }
 
     @Override
     public String toString() {
-        return "AssignmentTopicMetadata(topicName=" + topicName +
-            ", numPartitions=" + numPartitions +
-            ')';
+        return "AssignmentTopicMetadata(numPartitions=" + numPartitions + ')';
     }
 }

group-coordinator/src/main/java/org/apache/kafka/coordinator/group/assignor/GroupAssignment.java

@@ -26,7 +26,7 @@ public class GroupAssignment {
     /**
      * The member assignments keyed by member id.
      */
-    final Map<String, MemberAssignment> members;
+    private final Map<String, MemberAssignment> members;
 
     public GroupAssignment(
         Map<String, MemberAssignment> members
@@ -35,13 +35,18 @@ public class GroupAssignment {
         this.members = members;
     }
 
+    /**
+     * @return Member assignments keyed by member Ids.
+     */
+    public Map<String, MemberAssignment> members() {
+        return members;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
-
         GroupAssignment that = (GroupAssignment) o;
-
         return members.equals(that.members);
     }
 

group-coordinator/src/main/java/org/apache/kafka/coordinator/group/assignor/MemberAssignment.java

@@ -16,34 +16,38 @@
  */
 package org.apache.kafka.coordinator.group.assignor;
 
-import org.apache.kafka.common.TopicPartition;
+import org.apache.kafka.common.Uuid;
 
-import java.util.Collection;
+import java.util.Map;
 import java.util.Objects;
+import java.util.Set;
 
 /**
  * The partition assignment for a consumer group member.
  */
 public class MemberAssignment {
     /**
-     * The target partitions assigned to this member.
+     * The target partitions assigned to this member keyed by topicId.
      */
-    final Collection<TopicPartition> targetPartitions;
+    private final Map<Uuid, Set<Integer>> targetPartitions;
 
-    public MemberAssignment(
-        Collection<TopicPartition> targetPartitions
-    ) {
+    public MemberAssignment(Map<Uuid, Set<Integer>> targetPartitions) {
         Objects.requireNonNull(targetPartitions);
         this.targetPartitions = targetPartitions;
     }
 
+    /**
+     * @return Target partitions keyed by topic Ids.
+     */
+    public Map<Uuid, Set<Integer>> targetPartitions() {
+        return this.targetPartitions;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
-
         MemberAssignment that = (MemberAssignment) o;
-
         return targetPartitions.equals(that.targetPartitions);
     }