148 lines
5.7 KiB
Python
148 lines
5.7 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2022 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.
|
|
"""Command to create a database migration private connection."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
from googlecloudsdk.api_lib.database_migration import api_util
|
|
from googlecloudsdk.api_lib.database_migration import private_connections
|
|
from googlecloudsdk.api_lib.database_migration import resource_args
|
|
from googlecloudsdk.calliope import base
|
|
from googlecloudsdk.command_lib.database_migration import flags
|
|
from googlecloudsdk.command_lib.database_migration.private_connections import flags as pc_flags
|
|
from googlecloudsdk.command_lib.util.concepts import concept_parsers
|
|
from googlecloudsdk.command_lib.util.concepts import presentation_specs
|
|
from googlecloudsdk.core import log
|
|
|
|
DESCRIPTION = 'Create a Database Migration private connection'
|
|
EXAMPLES = """\
|
|
To create a private connection with VPC Peering called 'my-private-connection', run:
|
|
|
|
$ {command} my-private-connection --region=us-central1 --display-name=my-private-connection --vpc=vpc-example --subnet=10.0.0.0/29
|
|
|
|
To create a private connection with PSC Interface called 'my-privateConnection', run:
|
|
|
|
$ {command} my-private-connection --location=us-central1 --display-name=my-private-connection --network-attachment=network-attachment-example
|
|
|
|
To use a private connection, all migrations and connection profiles that use this configuration must be in the same region.
|
|
|
|
|
|
"""
|
|
|
|
|
|
@base.ReleaseTracks(base.ReleaseTrack.GA)
|
|
@base.UniverseCompatible
|
|
class Create(base.Command):
|
|
"""Create a Database Migration private connection."""
|
|
detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES}
|
|
|
|
@staticmethod
|
|
def CommonArgs(parser, release_track):
|
|
"""Common arguments for all release tracks.
|
|
|
|
Args:
|
|
parser: An argparse parser that you can use to add arguments that go on
|
|
the command line after this command. Positional arguments are allowed.
|
|
release_track: Some arguments are added based on the command release
|
|
track.
|
|
"""
|
|
resource_args.AddPrivateConnectionResourceArg(parser, 'to create')
|
|
|
|
pc_flags.AddDisplayNameFlag(parser)
|
|
pc_flags.AddNoAsyncFlag(parser)
|
|
pc_flags.AddSkipValidationFlag(parser)
|
|
pc_flags.AddValidateOnlyFlag(parser)
|
|
flags.AddLabelsCreateFlags(parser)
|
|
|
|
config_group = parser.add_group(mutex=True, required=True)
|
|
|
|
vpc_peering_group = config_group.add_group(
|
|
help='Arguments for VPC Peering configuration.'
|
|
)
|
|
vpc_peering_group.add_argument(
|
|
'--subnet',
|
|
help="""A free subnet for peering. (CIDR of /29).""",
|
|
required=True,
|
|
)
|
|
|
|
# Add VPC resource arg inside the VPC Peering group
|
|
vpc_spec = presentation_specs.ResourcePresentationSpec(
|
|
'--vpc',
|
|
resource_args.GetVpcResourceSpec(),
|
|
'Resource ID of the VPC network to peer with.',
|
|
group=vpc_peering_group,
|
|
required=True,
|
|
) # Ensure VPC is required within this group
|
|
concept_parsers.ConceptParser([vpc_spec]).AddToParser(vpc_peering_group)
|
|
|
|
# --- PSC Interface Group ---
|
|
psc_group = config_group.add_group(
|
|
help='Arguments for Private Service Connect Interface configuration.'
|
|
)
|
|
pc_flags.AddNetworkAttachmentFlag(psc_group)
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
"""Args is called by calliope to gather arguments for this command."""
|
|
Create.CommonArgs(parser, base.ReleaseTrack.GA)
|
|
|
|
def Run(self, args):
|
|
"""Create a Database Migration private connection.
|
|
|
|
Args:
|
|
args: argparse.Namespace, The arguments that this command was invoked
|
|
with.
|
|
|
|
Returns:
|
|
A dict object representing the operations resource describing the create
|
|
operation if the create was successful.
|
|
"""
|
|
private_connection_ref = args.CONCEPTS.private_connection.Parse()
|
|
parent_ref = private_connection_ref.Parent().RelativeName()
|
|
|
|
pc_client = private_connections.PrivateConnectionsClient(
|
|
release_track=self.ReleaseTrack())
|
|
result_operation = pc_client.Create(
|
|
parent_ref, private_connection_ref.privateConnectionsId, args)
|
|
|
|
client = api_util.GetClientInstance(self.ReleaseTrack())
|
|
messages = api_util.GetMessagesModule(self.ReleaseTrack())
|
|
resource_parser = api_util.GetResourceParser(self.ReleaseTrack())
|
|
|
|
if args.IsKnownAndSpecified('no_async'):
|
|
log.status.Print(
|
|
'Waiting for private connection [{}] to be created with [{}]'.format(
|
|
private_connection_ref.privateConnectionsId,
|
|
result_operation.name))
|
|
|
|
api_util.HandleLRO(client, result_operation,
|
|
client.projects_locations_privateConnections)
|
|
|
|
log.status.Print('Created private connection {} [{}]'.format(
|
|
private_connection_ref.privateConnectionsId, result_operation.name))
|
|
return
|
|
|
|
operation_ref = resource_parser.Create(
|
|
'datamigration.projects.locations.operations',
|
|
operationsId=result_operation.name,
|
|
projectsId=private_connection_ref.projectsId,
|
|
locationsId=private_connection_ref.locationsId)
|
|
|
|
return client.projects_locations_operations.Get(
|
|
messages.DatamigrationProjectsLocationsOperationsGetRequest(
|
|
name=operation_ref.operationsId))
|