def __init__(self, subnet, name):
super(AzureNIC, self).__init__()
self.subnet = subnet
self.name = name
self.resource_group = azure_network.GetResourceGroup()
self.location = self.subnet.vnet.location
self.args = ['--nic-name', self.name]
def __init__(self, disk_spec, vm, lun, is_image=False):
super(AzureDisk, self).__init__(disk_spec)
self.host_caching = FLAGS.azure_host_caching
self.vm = vm
self.vm_name = vm.name
self.name = self.vm_name + str(lun)
self.resource_group = azure_network.GetResourceGroup()
self.storage_account = vm.storage_account
# lun is Azure's abbreviation for "logical unit number"
self.lun = lun
self.is_image = is_image
self._deleted = False
self.machine_type = vm.machine_type
if self.disk_type == PREMIUM_STORAGE:
self.metadata.update({
disk.MEDIA: disk.SSD,
disk.REPLICATION: disk.ZONE,
HOST_CACHING: self.host_caching,
})
elif self.disk_type == STANDARD_DISK:
self.metadata.update({
disk.MEDIA:
disk.HDD,
disk.REPLICATION:
AZURE_REPLICATION_MAP[FLAGS.azure_storage_type],
HOST_CACHING:
self.host_caching,
})
elif self.disk_type == disk.LOCAL:
media = disk.SSD if LocalDiskIsSSD(self.machine_type) else disk.HDD
self.metadata.update({
disk.MEDIA: media,
disk.REPLICATION: disk.NONE,
})
def __init__(self, managed_relational_db_spec):
super(AzureManagedRelationalDb,
self).__init__(managed_relational_db_spec)
self.spec = managed_relational_db_spec
self.instance_id = 'pkb-db-instance-' + FLAGS.run_uri
self.region = self.spec.vm_spec.zone
self.resource_group = azure_network.GetResourceGroup(self.region)
def __init__(self, location, availability_zone, name):
super(AzurePublicIPAddress, self).__init__()
self.location = location
self.availability_zone = availability_zone
self.name = name
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
def __init__(self, spec): """Initializes the cluster.""" super(AksCluster, self).__init__(spec) self.resource_group = azure_network.GetResourceGroup(self.zone) self.name = 'pkbcluster%s' % FLAGS.run_uri self.service_principal = service_principal.ServicePrincipal.GetInstance() self._deleted = False
def __init__(self, spec):
super(AzureRedisCache, self).__init__(spec)
self.redis_region = FLAGS.redis_region
self.resource_group = azure_network.GetResourceGroup(self.redis_region)
self.azure_tier = FLAGS.azure_tier
self.azure_redis_size = FLAGS.azure_redis_size
self.failover_style = FLAGS.redis_failover_style
def __init__(self, vm_spec):
"""Initialize an Azure virtual machine.
Args:
vm_spec: virtual_machine.BaseVmSpec object of the vm.
"""
super(AzureVirtualMachine, self).__init__(vm_spec)
self.network = azure_network.AzureNetwork.GetNetwork(self)
self.firewall = azure_network.AzureFirewall.GetFirewall()
self.max_local_disks = NUM_LOCAL_VOLUMES.get(self.machine_type) or 1
self._lun_counter = itertools.count()
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
self.public_ip = AzurePublicIPAddress(self.zone,
self.name + '-public-ip')
self.nic = AzureNIC(self.network.subnet, self.name + '-nic',
self.public_ip.name,
vm_spec.accelerated_networking)
self.storage_account = self.network.storage_account
self.image = vm_spec.image or self.IMAGE_URN
disk_spec = disk.BaseDiskSpec('azure_os_disk')
self.os_disk = azure_disk.AzureDisk(disk_spec,
self.name,
self.machine_type,
self.storage_account,
None,
is_image=True)
def __init__(self, relational_db_spec):
super(AzureRelationalDb, self).__init__(relational_db_spec)
self.instance_id = 'pkb-db-instance-' + FLAGS.run_uri
self.region = self.spec.vm_spec.zone
self.resource_group = azure_network.GetResourceGroup(self.region)
self.unmanaged_db_exists = None if self.is_managed_db else False
def __init__(self,
disk_spec,
vm_name,
machine_type,
storage_account,
lun,
is_image=False):
super(AzureDisk, self).__init__(disk_spec)
self.host_caching = FLAGS.azure_host_caching
self.name = None
self.vm_name = vm_name
self.resource_group = azure_network.GetResourceGroup()
self.storage_account = storage_account
# lun is Azure's abbreviation for "logical unit number"
self.lun = lun
self.is_image = is_image
self._deleted = False
if self.disk_type == PREMIUM_STORAGE:
self.metadata = PREMIUM_STORAGE_METADATA
elif self.disk_type == STANDARD_DISK:
self.metadata = {
disk.MEDIA: disk.HDD,
disk.REPLICATION:
AZURE_REPLICATION_MAP[FLAGS.azure_storage_type],
disk.LEGACY_DISK_TYPE: disk.STANDARD
}
elif self.disk_type == disk.LOCAL:
media = disk.SSD if LocalDiskIsSSD(machine_type) else disk.HDD
self.metadata = {
disk.MEDIA: media,
disk.REPLICATION: disk.NONE,
disk.LEGACY_DISK_TYPE: disk.LOCAL
}
def __init__(self, spec):
super(AzureRedisCache, self).__init__(spec)
self.name = 'pkb-%s' % FLAGS.run_uri
self.redis_region = FLAGS.redis_region
self.resource_group = azure_network.GetResourceGroup(self.redis_region)
self.azure_tier = FLAGS.azure_tier
self.azure_redis_size = FLAGS.azure_redis_size
def __init__(self, vm_spec):
"""Initialize an Azure virtual machine.
Args:
vm_spec: virtual_machine.BaseVirtualMachineSpec object of the vm.
"""
super(AzureVirtualMachine, self).__init__(vm_spec)
self.network = azure_network.AzureNetwork.GetNetwork(self)
self.firewall = azure_network.AzureFirewall.GetFirewall()
self.max_local_disks = 1
self.resource_group = azure_network.GetResourceGroup()
self.public_ip = AzurePublicIPAddress(self.zone,
self.name + '-public-ip')
self.nic = AzureNIC(self.network.subnet, self.name + '-nic')
self.storage_account = self.network.storage_account
self.image = vm_spec.image or self.IMAGE_URN
disk_spec = disk.BaseDiskSpec('azure_os_disk')
self.os_disk = azure_disk.AzureDisk(disk_spec,
self.name,
self.machine_type,
self.storage_account,
None,
is_image=True)
def __init__(self, vm_spec):
"""Initialize an Azure virtual machine.
Args:
vm_spec: virtual_machine.BaseVmSpec object of the vm.
"""
super(AzureVirtualMachine, self).__init__(vm_spec)
# PKB zone can be either a region or a region with an availability zone.
# Format for Azure availability zone support is "region-availability_zone"
# Example: eastus2-1 is Azure region eastus2 with availability zone 1.
self.region = util.GetRegionFromZone(self.zone)
self.availability_zone = util.GetAvailabilityZoneFromZone(self.zone)
self.use_dedicated_host = vm_spec.use_dedicated_host
self.num_vms_per_host = vm_spec.num_vms_per_host
self.network = azure_network.AzureNetwork.GetNetwork(self)
self.firewall = azure_network.AzureFirewall.GetFirewall()
self.max_local_disks = NUM_LOCAL_VOLUMES.get(self.machine_type) or 1
self._lun_counter = itertools.count()
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
self.public_ip = AzurePublicIPAddress(self.region,
self.availability_zone,
self.name + '-public-ip')
self.nic = AzureNIC(self.network.subnet, self.name + '-nic',
self.public_ip.name,
vm_spec.accelerated_networking, self.network.nsg)
self.storage_account = self.network.storage_account
if vm_spec.image:
self.image = vm_spec.image
elif self.machine_type in _MACHINE_TYPES_ONLY_SUPPORT_GEN2_IMAGES:
if hasattr(type(self), 'GEN2_IMAGE_URN'):
self.image = type(self).GEN2_IMAGE_URN
else:
raise errors.Benchmarks.UnsupportedConfigError(
'No Azure gen2 image.')
else:
self.image = type(self).IMAGE_URN
self.host = None
if self.use_dedicated_host:
self.host_series_sku = _GetSkuType(self.machine_type)
self.host_list = None
self.low_priority = vm_spec.low_priority
self.low_priority_status_code = None
self.spot_early_termination = False
self.ultra_ssd_enabled = False
disk_spec = disk.BaseDiskSpec('azure_os_disk')
disk_spec.disk_type = (vm_spec.boot_disk_type
or self.storage_account.storage_type)
if vm_spec.boot_disk_size:
disk_spec.disk_size = vm_spec.boot_disk_size
self.os_disk = azure_disk.AzureDisk(disk_spec,
self,
None,
is_image=True)
def __init__(self, registry_spec): super(AzureContainerRegistry, self).__init__(registry_spec) self.resource_group = azure_network.GetResourceGroup(self.zone) self.login_server = None self.sku = 'Basic' self._deleted = False self.acr_id = None self.service_principal = service_principal.ServicePrincipal.GetInstance()
def __init__(self, subnet, name, public_ip):
super(AzureNIC, self).__init__()
self.subnet = subnet
self.name = name
self.public_ip = public_ip
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
self.location = self.subnet.vnet.location
self.args = ['--nics', self.name]
def __init__(self, relational_db_spec):
super(AzureRelationalDb, self).__init__(relational_db_spec)
if util.IsZone(self.spec.db_spec.zone):
raise errors.Config.InvalidValue(
'Availability zones are currently not supported by Azure DBs')
self.region = util.GetRegionFromZone(self.spec.db_spec.zone)
self.resource_group = azure_network.GetResourceGroup(self.region)
self.unmanaged_db_exists = None if self.is_managed_db else False
def __init__(self, disk_spec, zone):
super(AzureSmbService, self).__init__(disk_spec, zone)
self.name = 'azure-smb-fs-%s' % FLAGS.run_uri
self.region = util.GetRegionFromZone(self.zone)
self.resource_group = azure_network.GetResourceGroup(self.region)
# set during _Create()
self.connection_args: List[str] = None
self.storage_account_key: str = None
self.storage_account_name: str = None
def __init__(self, spec):
super(AzureRedisCache, self).__init__(spec)
self.redis_region = FLAGS.redis_region
self.resource_group = azure_network.GetResourceGroup(self.redis_region)
self.azure_redis_size = FLAGS.azure_redis_size
self.failover_style = FLAGS.redis_failover_style
if self.failover_style == managed_memory_store.Failover.FAILOVER_SAME_REGION:
self.azure_tier = 'Premium'
else:
self.azure_tier = 'Basic'
def __init__(self, spec):
"""Initializes the cluster."""
super(AksCluster, self).__init__(spec)
self.region = util.GetRegionFromZone(self.zone)
self.resource_group = azure_network.GetResourceGroup(self.region)
self.name = 'pkbcluster%s' % FLAGS.run_uri
# TODO(pclay): replace with built in service principal once I figure out how
# to make it work with ACR
self.service_principal = service_principal.ServicePrincipal.GetInstance(
)
self.cluster_version = FLAGS.container_cluster_version
self._deleted = False
def __init__(self, spec):
"""Initializes the cluster."""
super(AksCluster, self).__init__(spec)
if util.IsZone(spec.vm_spec.zone):
raise errors.Config.InvalidValue(
'Availability zones are currently not supported by Aks Cluster')
self.location = util.GetLocationFromZone(self.zone)
self.resource_group = azure_network.GetResourceGroup(self.location)
self.name = 'pkbcluster%s' % FLAGS.run_uri
# TODO(pclay): replace with built in service principal once I figure out how
# to make it work with ACR
self.service_principal = service_principal.ServicePrincipal.GetInstance()
self._deleted = False
def __init__(self, vm_spec):
"""Initialize an Azure virtual machine.
Args:
vm_spec: virtual_machine.BaseVmSpec object of the vm.
"""
super(AzureVirtualMachine, self).__init__(vm_spec)
# PKB zone can be either a location or a location with an availability zone.
# Format for Azure availability zone support is "location-availability_zone"
# Example: eastus2-1 is Azure location eastus2 with availability zone 1.
self.location = util.GetLocationFromZone(self.zone)
self.availability_zone = util.GetAvailabilityZoneFromZone(self.zone)
self.use_dedicated_host = vm_spec.use_dedicated_host
# TODO(buggay): implement num_vms_per_host functionality
self.num_vms_per_host = vm_spec.num_vms_per_host
if self.num_vms_per_host:
raise NotImplementedError(
'Num vms per host for Azure is not supported.')
self.network = azure_network.AzureNetwork.GetNetwork(self)
self.firewall = azure_network.AzureFirewall.GetFirewall()
self.max_local_disks = NUM_LOCAL_VOLUMES.get(self.machine_type) or 1
self._lun_counter = itertools.count()
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
self.public_ip = AzurePublicIPAddress(self.location,
self.availability_zone,
self.name + '-public-ip')
self.nic = AzureNIC(self.network.subnet, self.name + '-nic',
self.public_ip.name,
vm_spec.accelerated_networking)
self.storage_account = self.network.storage_account
self.image = vm_spec.image or self.IMAGE_URN
self.host = None
if self.use_dedicated_host:
self.host_series_sku = _GetSkuType(self.machine_type)
self.host_list = None
disk_spec = disk.BaseDiskSpec('azure_os_disk')
disk_spec.disk_type = (vm_spec.boot_disk_type
or self.storage_account.storage_type)
if vm_spec.boot_disk_size:
disk_spec.disk_size = vm_spec.boot_disk_size
self.os_disk = azure_disk.AzureDisk(disk_spec,
self.name,
self.machine_type,
self.storage_account,
None,
is_image=True)
def __init__(self,
subnet,
name,
public_ip,
accelerated_networking,
private_ip=None):
super(AzureNIC, self).__init__()
self.subnet = subnet
self.name = name
self.public_ip = public_ip
self.private_ip = private_ip
self._deleted = False
self.resource_group = azure_network.GetResourceGroup()
self.region = self.subnet.vnet.region
self.args = ['--nics', self.name]
self.accelerated_networking = accelerated_networking
def PrepareService(self, location):
# abs is "Azure Blob Storage"
prefix = 'pkb%sabs' % FLAGS.run_uri
self.resource_group = azure_network.GetResourceGroup()
# We use a different Azure storage account than the VM account
# because a) we need to be able to set the storage class
# separately, including using a blob-specific storage account and
# b) this account might be in a different location than any
# VM-related account.
self.storage_account = azure_network.AzureStorageAccount(
FLAGS.azure_storage_type,
location or DEFAULT_AZURE_REGION,
prefix + 'storage',
kind=FLAGS.azure_blob_account_kind)
self.storage_account.Create()
def __init__(self, registry_spec):
super(AzureContainerRegistry, self).__init__(registry_spec)
self.resource_group = azure_network.GetResourceGroup(self.zone)
self.login_server = None
self.sku = 'Basic'
self._deleted = False
def __init__(self, disk_spec, zone): super(AzureSmbService, self).__init__(disk_spec, zone) self.name = 'azure-smb-fs-%s' % FLAGS.run_uri self.location = util.GetLocationFromZone(self.zone) self.resource_group = azure_network.GetResourceGroup(self.location)
def __init__(self):
super().__init__()
self.topic_name = 'pkb-topic-{0}'.format(FLAGS.run_uri)
self.subscription_name = 'pkb-subscription-{0}'.format(FLAGS.run_uri)
self.namespace_name = 'pkb-namespace-{0}'.format(FLAGS.run_uri)
self.resource_group = azure_network.GetResourceGroup()
def __init__(self, location, name):
super(AzurePublicIPAddress, self).__init__()
self.location = location
self.name = name
self.resource_group = azure_network.GetResourceGroup()
def __init__(self, spec):
"""Initializes the cluster."""
super(AcsKubernetesCluster, self).__init__(spec)
self.resource_group = azure_network.GetResourceGroup(self.zone)
self.name = 'pkbcluster%s' % FLAGS.run_uri
self._deleted = False
def __init__(self, cluster_spec):
super(AciCluster, self).__init__(cluster_spec)
self.location = util.GetLocationFromZone(self.zone)
self.resource_group = azure_network.GetResourceGroup(self.location)
def __init__(self, cluster_spec): super(AciCluster, self).__init__(cluster_spec) self.resource_group = azure_network.GetResourceGroup(self.zone)
