KAFKA-14595 Move ReassignPartitionsCommand to java

tools/src/main/java/org/apache/kafka/tools/reassign/ActiveMoveState.java

@@ -0,0 +1,81 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+import java.util.Objects;
+
+/**
+ * A replica log directory move state where the move is in progress.
+ */
+public final class ActiveMoveState implements LogDirMoveState {
+    private final String currentLogDir;
+
+    private final String targetLogDir;
+
+    private final String futureLogDir;
+
+    /**
+     * @param currentLogDir       The current log directory.
+     * @param futureLogDir        The log directory that the replica is moving to.
+     * @param targetLogDir        The log directory that we wanted the replica to move to.
+     */
+    public ActiveMoveState(String currentLogDir, String targetLogDir, String futureLogDir) {
+        this.currentLogDir = currentLogDir;
+        this.targetLogDir = targetLogDir;
+        this.futureLogDir = futureLogDir;
+    }
+
+    public String currentLogDir() {
+        return currentLogDir;
+    }
+
+    public String targetLogDir() {
+        return targetLogDir;
+    }
+
+    public String futureLogDir() {
+        return futureLogDir;
+    }
+
+    @Override
+    public boolean done() {
+        return false;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        ActiveMoveState that = (ActiveMoveState) o;
+        return Objects.equals(currentLogDir, that.currentLogDir) && Objects.equals(targetLogDir, that.targetLogDir) && Objects.equals(futureLogDir, that.futureLogDir);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(currentLogDir, targetLogDir, futureLogDir);
+    }
+
+    @Override
+    public String toString() {
+        return "ActiveMoveState{" +
+            "currentLogDir='" + currentLogDir + '\'' +
+            ", targetLogDir='" + targetLogDir + '\'' +
+            ", futureLogDir='" + futureLogDir + '\'' +
+            '}';
+    }
+}

tools/src/main/java/org/apache/kafka/tools/reassign/CancelledMoveState.java

@@ -0,0 +1,73 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+import java.util.Objects;
+
+/**
+ * A replica log directory move state where there is no move in progress, but we did not
+ * reach the target log directory.
+ */
+public final class CancelledMoveState implements LogDirMoveState {
+    private final String currentLogDir;
+
+    private final String targetLogDir;
+
+    /**
+     * @param currentLogDir       The current log directory.
+     * @param targetLogDir        The log directory that we wanted the replica to move to.
+     */
+    public CancelledMoveState(String currentLogDir, String targetLogDir) {
+        this.currentLogDir = currentLogDir;
+        this.targetLogDir = targetLogDir;
+    }
+
+    public String currentLogDir() {
+        return currentLogDir;
+    }
+
+    public String targetLogDir() {
+        return targetLogDir;
+    }
+
+    @Override
+    public boolean done() {
+        return true;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        CancelledMoveState that = (CancelledMoveState) o;
+        return Objects.equals(currentLogDir, that.currentLogDir) && Objects.equals(targetLogDir, that.targetLogDir);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(currentLogDir, targetLogDir);
+    }
+
+    @Override
+    public String toString() {
+        return "CancelledMoveState{" +
+            "currentLogDir='" + currentLogDir + '\'' +
+            ", targetLogDir='" + targetLogDir + '\'' +
+            '}';
+    }
+}

tools/src/main/java/org/apache/kafka/tools/reassign/CompletedMoveState.java

@@ -0,0 +1,63 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+import java.util.Objects;
+
+/**
+ * The completed replica log directory move state.
+ */
+public final class CompletedMoveState implements LogDirMoveState {
+    private final String targetLogDir;
+
+    /**
+     * @param targetLogDir        The log directory that we wanted the replica to move to.
+     */
+    public CompletedMoveState(String targetLogDir) {
+        this.targetLogDir = targetLogDir;
+    }
+
+    public String targetLogDir() {
+        return targetLogDir;
+    }
+
+    @Override
+    public boolean done() {
+        return true;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        CompletedMoveState that = (CompletedMoveState) o;
+        return Objects.equals(targetLogDir, that.targetLogDir);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(targetLogDir);
+    }
+
+    @Override
+    public String toString() {
+        return "CompletedMoveState{" +
+            "targetLogDir='" + targetLogDir + '\'' +
+            '}';
+    }
+}

tools/src/main/java/org/apache/kafka/tools/reassign/LogDirMoveState.java

@@ -0,0 +1,28 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+/**
+ * The state of a replica log directory movement.
+ */
+public interface LogDirMoveState {
+    /**
+     * True if the move is done without errors.
+     */
+    boolean done();
+}

tools/src/main/java/org/apache/kafka/tools/reassign/MissingLogDirMoveState.java

@@ -0,0 +1,63 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+import java.util.Objects;
+
+/**
+ * A replica log directory move state where the source replica is missing.
+ */
+public final class MissingLogDirMoveState implements LogDirMoveState {
+    private final String targetLogDir;
+
+    /**
+     * @param targetLogDir        The log directory that we wanted the replica to move to.
+     */
+    public MissingLogDirMoveState(String targetLogDir) {
+        this.targetLogDir = targetLogDir;
+    }
+
+    public String targetLogDir() {
+        return targetLogDir;
+    }
+
+    @Override
+    public boolean done() {
+        return false;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        MissingLogDirMoveState that = (MissingLogDirMoveState) o;
+        return Objects.equals(targetLogDir, that.targetLogDir);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(targetLogDir);
+    }
+
+    @Override
+    public String toString() {
+        return "MissingLogDirMoveState{" +
+            "targetLogDir='" + targetLogDir + '\'' +
+            '}';
+    }
+}

tools/src/main/java/org/apache/kafka/tools/reassign/MissingReplicaMoveState.java

@@ -0,0 +1,63 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.kafka.tools.reassign;
+
+import java.util.Objects;
+
+/**
+ * A replica log directory move state where the source log directory is missing.
+ */
+public final class MissingReplicaMoveState implements LogDirMoveState {
+    private final String targetLogDir;
+
+    /**
+     * @param targetLogDir        The log directory that we wanted the replica to move to.
+     */
+    public MissingReplicaMoveState(String targetLogDir) {
+        this.targetLogDir = targetLogDir;
+    }
+
+    public String targetLogDir() {
+        return targetLogDir;
+    }
+
+    @Override
+    public boolean done() {
+        return false;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        MissingReplicaMoveState that = (MissingReplicaMoveState) o;
+        return Objects.equals(targetLogDir, that.targetLogDir);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(targetLogDir);
+    }
+
+    @Override
+    public String toString() {
+        return "MissingReplicaMoveState{" +
+            "targetLogDir='" + targetLogDir + '\'' +
+            '}';
+    }
+}