142 lines
4.2 KiB
Python
142 lines
4.2 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2016 Google LLC. All Rights Reserved.
|
|
#
|
|
# Licensed 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.
|
|
"""Rollback node pool command."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
from apitools.base.py import exceptions as apitools_exceptions
|
|
|
|
from googlecloudsdk.api_lib.container import util
|
|
from googlecloudsdk.calliope import base
|
|
from googlecloudsdk.calliope import exceptions
|
|
from googlecloudsdk.command_lib.container import flags
|
|
from googlecloudsdk.core import log
|
|
from googlecloudsdk.core.console import console_io
|
|
|
|
|
|
def _Args(parser):
|
|
"""Register flags for this command.
|
|
|
|
Args:
|
|
parser: an argparse.ArgumentParser-like object. It is mocked out in order to
|
|
capture some information, but behaves like an ArgumentParser.
|
|
"""
|
|
|
|
flags.AddAsyncFlag(parser)
|
|
flags.AddNodePoolNameArg(parser, 'The name of the node pool to rollback.')
|
|
flags.AddNodePoolClusterFlag(
|
|
parser, 'The cluster from which to rollback the node pool.')
|
|
flags.AddRespectPodDisruptionBudgetFlag(parser)
|
|
parser.add_argument(
|
|
'--timeout',
|
|
type=int,
|
|
default=1800, # Seconds
|
|
hidden=True,
|
|
help='THIS ARGUMENT NEEDS HELP TEXT.')
|
|
|
|
|
|
@base.ReleaseTracks(base.ReleaseTrack.GA)
|
|
class Rollback(base.Command):
|
|
"""Rollback a node-pool upgrade."""
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
_Args(parser)
|
|
|
|
def Run(self, args):
|
|
"""This is what gets called when the user runs this command.
|
|
|
|
Args:
|
|
args: an argparse namespace. All the arguments that were provided to this
|
|
command invocation.
|
|
|
|
Returns:
|
|
Some value that we want to have printed later.
|
|
|
|
Raises:
|
|
util.Error, if rollback failed.
|
|
"""
|
|
adapter = self.context['api_adapter']
|
|
location_get = self.context['location_get']
|
|
location = location_get(args)
|
|
|
|
pool_ref = adapter.ParseNodePool(args.name, location)
|
|
|
|
console_io.PromptContinue(
|
|
message='Node Pool: [{node_pool_id}], of Cluster: [{cluster_name}] '
|
|
'will be rolled back to previous configuration. This operation is '
|
|
'long-running and will block other operations on the cluster '
|
|
'(including delete) until it has run to completion.'.format(
|
|
node_pool_id=pool_ref.nodePoolId, cluster_name=pool_ref.clusterId),
|
|
throw_if_unattended=True,
|
|
cancel_on_no=True)
|
|
|
|
try:
|
|
# Make sure it exists (will raise appropriate error if not)
|
|
adapter.GetNodePool(pool_ref)
|
|
|
|
op_ref = adapter.RollbackUpgrade(pool_ref, respect_pdb=args.respect_pdb)
|
|
|
|
if not args.async_:
|
|
adapter.WaitForOperation(
|
|
op_ref,
|
|
'Rolling back {0}'.format(pool_ref.nodePoolId),
|
|
timeout_s=args.timeout)
|
|
|
|
except apitools_exceptions.HttpError as error:
|
|
raise exceptions.HttpException(error, util.HTTP_ERROR_FORMAT)
|
|
|
|
log.UpdatedResource(pool_ref)
|
|
return op_ref
|
|
|
|
|
|
@base.ReleaseTracks(base.ReleaseTrack.BETA)
|
|
class RollbackBeta(Rollback):
|
|
"""Rollback a node-pool upgrade."""
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
_Args(parser)
|
|
|
|
|
|
@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
|
|
class RollbackAlpha(Rollback):
|
|
"""Rollback a node-pool upgrade."""
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
_Args(parser)
|
|
|
|
Rollback.detailed_help = {
|
|
'brief':
|
|
'Rollback a node-pool upgrade.',
|
|
'DESCRIPTION':
|
|
"""
|
|
Rollback a node-pool upgrade.
|
|
|
|
Rollback is a method used after a canceled or failed node-pool upgrade. It
|
|
makes a best-effort attempt to revert the pool back to its original state.
|
|
""",
|
|
'EXAMPLES':
|
|
"""\
|
|
To roll back a canceled or failed upgrade in "node-pool-1" in the
|
|
cluster "sample-cluster", run:
|
|
|
|
$ {command} node-pool-1 --cluster=sample-cluster
|
|
""",
|
|
}
|