migrated CentroidAssignments to a class to hide default constructor, minor change from long to int

Author: john-wagster

server/src/main/java/org/elasticsearch/index/codec/vectors/CentroidAssignments.java

@@ -9,7 +9,19 @@
 
 package org.elasticsearch.index.codec.vectors;
 
-record CentroidAssignments(int numCentroids, float[][] cachedCentroids, int[] assignments, int[] soarAssignments) {
+final class CentroidAssignments {
+
+    private final int numCentroids;
+    private final float[][] cachedCentroids;
+    private final int[] assignments;
+    private final int[] soarAssignments;
+
+    private CentroidAssignments(int numCentroids, float[][] cachedCentroids, int[] assignments, int[] soarAssignments) {
+        this.numCentroids = numCentroids;
+        this.cachedCentroids = cachedCentroids;
+        this.assignments = assignments;
+        this.soarAssignments = soarAssignments;
+    }
 
     CentroidAssignments(float[][] centroids, int[] assignments, int[] soarAssignments) {
         this(centroids.length, centroids, assignments, soarAssignments);
@@ -18,4 +30,21 @@ record CentroidAssignments(int numCentroids, float[][] cachedCentroids, int[] as
     CentroidAssignments(int numCentroids, int[] assignments, int[] soarAssignments) {
         this(numCentroids, null, assignments, soarAssignments);
     }
+
+    // Getters and setters
+    public int numCentroids() {
+        return numCentroids;
+    }
+
+    public float[][] cachedCentroids() {
+        return cachedCentroids;
+    }
+
+    public int[] assignments() {
+        return assignments;
+    }
+
+    public int[] soarAssignments() {
+        return soarAssignments;
+    }
 }

server/src/main/java/org/elasticsearch/index/codec/vectors/cluster/KMeans.java

@@ -57,7 +57,7 @@ private boolean stepLloyd(FloatVectorValues vectors, float[][] centroids, int[]
         throws IOException {
         boolean changed = false;
         int dim = vectors.dimension();
-        long[] centroidCounts = new long[centroids.length];
+        int[] centroidCounts = new int[centroids.length];
         float[][] nextCentroids = new float[centroids.length][dim];
 
         for (int i = 0; i < sampleSize; i++) {