IGNITE-26748 Add rolling upgrade commands for control.sh

modules/control-utility/src/test/java/org/apache/ignite/internal/commandline/CommandHandlerParsingTest.java

@@ -76,6 +76,7 @@
 import org.apache.ignite.internal.management.metric.MetricCommand;
 import org.apache.ignite.internal.management.performancestatistics.PerformanceStatisticsCommand;
 import org.apache.ignite.internal.management.property.PropertyCommand;
+import org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand;
 import org.apache.ignite.internal.management.snapshot.SnapshotCommand;
 import org.apache.ignite.internal.management.snapshot.SnapshotRestoreCommand;
 import org.apache.ignite.internal.management.tx.TxCommand;
@@ -1391,6 +1392,7 @@ private boolean requireArgs(Class<?> cmd) {
             cmd == MetaCommand.class ||
             cmd == WarmUpCommand.class ||
             cmd == PropertyCommand.class ||
+            cmd == RollingUpgradeCommand.class ||
             cmd == SystemViewCommand.class ||
             cmd == MetricCommand.class ||
             cmd == DefragmentationCommand.class ||

modules/control-utility/src/test/java/org/apache/ignite/testsuites/IgniteControlUtilityTestSuite2.java

@@ -37,6 +37,7 @@
 import org.apache.ignite.util.IdleVerifyDumpTest;
 import org.apache.ignite.util.MetricCommandTest;
 import org.apache.ignite.util.PerformanceStatisticsCommandTest;
+import org.apache.ignite.util.RollingUpgradeCommandTest;
 import org.apache.ignite.util.SystemViewCommandTest;
 import org.junit.runner.RunWith;
 import org.junit.runners.Suite;
@@ -63,6 +64,7 @@
     GridCommandHandlerConsistencySensitiveTest.class,
     GridCommandHandlerConsistencyRepairCorrectnessAtomicTest.class,
 
+    RollingUpgradeCommandTest.class,
     SystemViewCommandTest.class,
     MetricCommandTest.class,
     PerformanceStatisticsCommandTest.class,

modules/control-utility/src/test/java/org/apache/ignite/util/RollingUpgradeCommandTest.java

@@ -0,0 +1,75 @@
+/*
+ * 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.ignite.util;
+
+import org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand;
+import org.apache.ignite.internal.util.lang.IgnitePair;
+import org.apache.ignite.lang.IgniteProductVersion;
+import org.junit.Test;
+
+import static org.apache.ignite.internal.IgniteNodeAttributes.ATTR_BUILD_VER;
+import static org.apache.ignite.internal.commandline.CommandHandler.EXIT_CODE_OK;
+
+/** Tests {@link RollingUpgradeCommand} command. */
+public class RollingUpgradeCommandTest extends GridCommandHandlerClusterByClassAbstractTest {
+    /** */
+    public static final String ENABLE = "enable";
+
+    /** */
+    public static final String DISABLE = "disable";
+
+    /** */
+    public static final String ROLLING_UPGRADE = "--rolling-upgrade";
+
+    /** */
+    @Test
+    public void testCommands() {
+        int res = execute(ROLLING_UPGRADE, DISABLE);
+
+        assertEquals(EXIT_CODE_OK, res);
+        assertEquals("Rolling upgrade disabled", lastOperationResult);
+
+        IgniteProductVersion curVer = IgniteProductVersion.fromString(crd.localNode().attribute(ATTR_BUILD_VER));
+
+        String targetVerStr = curVer.major() + "." + (curVer.minor() + 1) + ".0";
+
+        res = execute(ROLLING_UPGRADE, ENABLE, targetVerStr);
+        assertEquals(EXIT_CODE_OK, res);
+
+        assertTrue(crd.context().rollingUpgrade().enabled());
+
+        IgniteProductVersion targetVer = IgniteProductVersion.fromString(targetVerStr);
+
+        IgnitePair<IgniteProductVersion> versions = crd.context().rollingUpgrade().versions();
+
+        assertNotNull(versions);
+
+        assertEquals(curVer, versions.get1());
+        assertEquals(targetVer, versions.get2());
+
+        assertEquals(
+            "Rolling upgrade enabled [currentVersion=" + curVer + ", targetVersion=" + targetVer + ']',
+            lastOperationResult
+        );
+
+        res = execute(ROLLING_UPGRADE, DISABLE);
+
+        assertEquals(EXIT_CODE_OK, res);
+        assertEquals("Rolling upgrade disabled", lastOperationResult);
+    }
+}

modules/core/src/main/java/org/apache/ignite/internal/management/IgniteCommandRegistry.java

@@ -34,6 +34,7 @@
 import org.apache.ignite.internal.management.performancestatistics.PerformanceStatisticsCommand;
 import org.apache.ignite.internal.management.persistence.PersistenceCommand;
 import org.apache.ignite.internal.management.property.PropertyCommand;
+import org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand;
 import org.apache.ignite.internal.management.snapshot.SnapshotCommand;
 import org.apache.ignite.internal.management.tracing.TracingConfigurationCommand;
 import org.apache.ignite.internal.management.tx.TxCommand;
@@ -67,6 +68,7 @@ public IgniteCommandRegistry() {
             new TracingConfigurationCommand(),
             new WarmUpCommand(),
             new PropertyCommand(),
+            new RollingUpgradeCommand(),
             new SystemViewCommand(),
             new MetricCommand(),
             new PersistenceCommand(),

modules/core/src/main/java/org/apache/ignite/internal/management/rollingupgrade/RollingUpgradeCommand.java

@@ -0,0 +1,91 @@
+/*
+ * 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.ignite.internal.management.rollingupgrade;
+
+import java.io.IOException;
+import java.io.ObjectInput;
+import java.io.ObjectOutput;
+import org.apache.ignite.internal.dto.IgniteDataTransferObject;
+import org.apache.ignite.internal.management.api.Argument;
+import org.apache.ignite.internal.management.api.CommandRegistryImpl;
+import org.apache.ignite.internal.management.api.Positional;
+import org.apache.ignite.internal.util.typedef.internal.U;
+
+/** Rolling upgrade commands. */
+public class RollingUpgradeCommand extends CommandRegistryImpl {
+    /** */
+    public RollingUpgradeCommand() {
+        super(
+            new RollingUpgradeEnableCommand(),
+            new RollingUpgradeDisableCommand()
+        );
+    }
+
+    /** Base rolling upgrade command argument. */
+    public static class RollingUpgradeCommandArg extends IgniteDataTransferObject {
+        /** */
+        private static final long serialVersionUID = 0;
+
+        /** {@inheritDoc} */
+        @Override protected void writeExternalData(ObjectOutput out) throws IOException {
+            // No-op.
+        }
+
+        /** {@inheritDoc} */
+        @Override protected void readExternalData(ObjectInput in) throws IOException, ClassNotFoundException {
+            // No-op.
+        }
+    }
+
+    /** Rolling upgrade enable command argument. */
+    public static class RollingUpgradeEnableCommandArg extends RollingUpgradeCommandArg {
+        /** */
+        private static final long serialVersionUID = 0;
+
+        /** Target version. */
+        @Argument(description = "Target Ignite version", example = "2.18.0")
+        @Positional
+        private String targetVersion;
+
+        /** */
+        public String targetVersion() {
+            return targetVersion;
+        }
+
+        /** */
+        public void targetVersion(String targetVersion) {
+            this.targetVersion = targetVersion;
+        }
+
+        /** {@inheritDoc} */
+        @Override protected void writeExternalData(ObjectOutput out) throws IOException {
+            U.writeString(out, targetVersion);
+        }
+
+        /** {@inheritDoc} */
+        @Override protected void readExternalData(ObjectInput in) throws IOException, ClassNotFoundException {
+            targetVersion = U.readString(in);
+        }
+    }
+
+    /** Rolling upgrade disable command argument. */
+    public static class RollingUpgradeDisableCommandArg extends RollingUpgradeCommandArg {
+        /** */
+        private static final long serialVersionUID = 0;
+    }
+}

modules/core/src/main/java/org/apache/ignite/internal/management/rollingupgrade/RollingUpgradeDisableCommand.java

@@ -0,0 +1,49 @@
+/*
+ * 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.ignite.internal.management.rollingupgrade;
+
+import java.util.Collection;
+import org.apache.ignite.cluster.ClusterNode;
+import org.apache.ignite.internal.management.api.ComputeCommand;
+import org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand.RollingUpgradeDisableCommandArg;
+
+import static org.apache.ignite.internal.management.api.CommandUtils.coordinatorOrNull;
+import static org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand.RollingUpgradeCommandArg;
+
+/** Command to disable rolling upgrade mode. */
+public class RollingUpgradeDisableCommand implements ComputeCommand<RollingUpgradeCommandArg, String> {
+    /** {@inheritDoc} */
+    @Override public String description() {
+        return "Disable rolling upgrade";
+    }
+
+    /** {@inheritDoc} */
+    @Override public Class<RollingUpgradeDisableCommandArg> argClass() {
+        return RollingUpgradeDisableCommandArg.class;
+    }
+
+    /** {@inheritDoc} */
+    @Override public Class<RollingUpgradeTask> taskClass() {
+        return RollingUpgradeTask.class;
+    }
+
+    /** {@inheritDoc} */
+    @Override public Collection<ClusterNode> nodes(Collection<ClusterNode> nodes, RollingUpgradeCommandArg arg) {
+        return coordinatorOrNull(nodes);
+    }
+}

modules/core/src/main/java/org/apache/ignite/internal/management/rollingupgrade/RollingUpgradeEnableCommand.java

@@ -0,0 +1,49 @@
+/*
+ * 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.ignite.internal.management.rollingupgrade;
+
+import java.util.Collection;
+import org.apache.ignite.cluster.ClusterNode;
+import org.apache.ignite.internal.management.api.ComputeCommand;
+import org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand.RollingUpgradeEnableCommandArg;
+
+import static org.apache.ignite.internal.management.api.CommandUtils.coordinatorOrNull;
+import static org.apache.ignite.internal.management.rollingupgrade.RollingUpgradeCommand.RollingUpgradeCommandArg;
+
+/** Command to enable rolling upgrade. */
+public class RollingUpgradeEnableCommand implements ComputeCommand<RollingUpgradeCommandArg, String> {
+    /** {@inheritDoc} */
+    @Override public String description() {
+        return "Enable rolling upgrade";
+    }
+
+    /** {@inheritDoc} */
+    @Override public Class<RollingUpgradeEnableCommandArg> argClass() {
+        return RollingUpgradeEnableCommandArg.class;
+    }
+
+    /** {@inheritDoc} */
+    @Override public Class<RollingUpgradeTask> taskClass() {
+        return RollingUpgradeTask.class;
+    }
+
+    /** {@inheritDoc} */
+    @Override public Collection<ClusterNode> nodes(Collection<ClusterNode> nodes, RollingUpgradeCommandArg arg) {
+        return coordinatorOrNull(nodes);
+    }
+}