Files
cloudpods/pkg/multicloud/azure/dbinstance.go
2021-08-13 17:24:41 +08:00

407 lines
12 KiB
Go

// Copyright 2019 Yunion
//
// 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.
package azure
import (
"fmt"
"net/url"
"strings"
"time"
"yunion.io/x/jsonutils"
"yunion.io/x/pkg/errors"
api "yunion.io/x/onecloud/pkg/apis/compute"
"yunion.io/x/onecloud/pkg/cloudprovider"
"yunion.io/x/onecloud/pkg/multicloud"
)
type SDBInstanceVirtualNetworkRule struct {
ID string `json:"id"`
Name string `json:"name"`
Type string `json:"type"`
Properties SDBInstanceVirtualNetworkRuleProperties `json:"properties"`
}
type SDBInstanceVirtualNetworkRuleProperties struct {
IgnoreMissingVnetServiceEndpoint bool `json:"ignoreMissingVnetServiceEndpoint"`
State string `json:"state"`
VirtualNetworkSubnetID string `json:"virtualNetworkSubnetId"`
}
type SDBInstanceSku struct {
Name string `json:"name"`
Tier string `json:"tier"`
Family string `json:"family"`
Capacity int `json:"capacity"`
}
type SDBInstanceStorageProfile struct {
StorageMB int `json:"storageMB"`
BackupRetentionDays int `json:"backupRetentionDays"`
StorageIops int `json:"storageIops"`
GeoRedundantBackup string `json:"geoRedundantBackup"`
}
type SDBInstanceDelegatedSubnetArguments struct {
SubnetArmResourceId string `json:"subnetArmResourceId"`
}
type SDBInstanceProperties struct {
AdministratorLogin string `json:"administratorLogin"`
StorageProfile SDBInstanceStorageProfile `json:"storageProfile"`
Version string `json:"version"`
SslEnforcement string `json:"sslEnforcement"`
UserVisibleState string `json:"userVisibleState"`
FullyQualifiedDomainName string `json:"fullyQualifiedDomainName"`
EarliestRestoreDate time.Time `json:"earliestRestoreDate"`
ReplicationRole string `json:"replicationRole"`
MasterServerId string `json:"masterServerId"`
ReplicaCapacity int `json:"replicaCapacity"`
DelegatedSubnetArguments SDBInstanceDelegatedSubnetArguments `json:"delegatedSubnetArguments"`
}
type SDBInstance struct {
region *SRegion
multicloud.SDBInstanceBase
multicloud.AzureTags
Sku SDBInstanceSku `json:"sku"`
Properties SDBInstanceProperties `json:"properties"`
Location string `json:"location"`
ID string `json:"id"`
Name string `json:"name"`
Type string `json:"type"`
}
func (self *SRegion) GetIDBInstances() ([]cloudprovider.ICloudDBInstance, error) {
instanceTypes := []string{
"Microsoft.DBForMariaDB/servers",
"Microsoft.DBforMySQL/servers",
"Microsoft.DBforMySQL/flexibleServers",
"Microsoft.DBforPostgreSQL/servers",
"Microsoft.DBforPostgreSQL/flexibleServers",
}
DBInstances := []SDBInstance{}
for i := range instanceTypes {
instances, err := self.ListDBInstance(instanceTypes[i])
if err != nil {
if errors.Cause(err) != cloudprovider.ErrNotFound {
return nil, errors.Wrap(err, "self.ListDBInstance()")
}
}
DBInstances = append(DBInstances, instances...)
}
result := []cloudprovider.ICloudDBInstance{}
for i := range DBInstances {
DBInstances[i].region = self
result = append(result, &DBInstances[i])
}
sqlServers, err := self.ListSQLServer()
if err != nil {
return nil, errors.Wrapf(err, "ListSQLServer")
}
for i := range sqlServers {
sqlServers[i].region = self
result = append(result, &sqlServers[i])
}
managedSQLServers, err := self.ListManagedSQLServer()
if err != nil {
return nil, errors.Wrapf(err, "ListManagedSQLServer")
}
for i := range managedSQLServers {
managedSQLServers[i].region = self
result = append(result, &managedSQLServers[i])
}
return result, nil
}
func (self *SRegion) GetIDBInstanceById(instanceId string) (cloudprovider.ICloudDBInstance, error) {
if strings.Index(strings.ToLower(instanceId), "microsoft.sql/servers") > 0 {
return self.GetSQLServer(instanceId)
}
if strings.Index(strings.ToLower(instanceId), "microsoft.sql/managedinstances") > 0 {
return self.GetManagedSQLServer(instanceId)
}
rds, err := self.GetDBInstanceById(instanceId)
if err != nil {
return nil, errors.Wrapf(err, "self.get(%s, url.Values{}, &newrds)", instanceId)
}
return rds, nil
}
func (self *SRegion) GetDBInstanceById(instanceId string) (*SDBInstance, error) {
rds := SDBInstance{}
err := self.get(instanceId, url.Values{}, &rds)
if err != nil {
return nil, errors.Wrapf(err, "self.get(%s, url.Values{}, &newrds)", instanceId)
}
rds.region = self
return &rds, nil
}
func (self *SRegion) ListDBInstance(instanceType string) ([]SDBInstance, error) {
result := []SDBInstance{}
err := self.list(instanceType, url.Values{}, &result)
if err != nil {
return nil, errors.Wrapf(err, "list(%s)", instanceType)
}
for i := range result {
result[i].region = self
}
return result, nil
}
func (self *SRegion) ListDBInstanceReplica(Id string) ([]SDBInstance, error) {
type replicas struct {
Value []SDBInstance
}
result := replicas{}
err := self.get(fmt.Sprintf("%s/replicas", Id), url.Values{}, &result)
if err != nil {
return nil, errors.Wrapf(err, "get(%s)", Id)
}
for i := range result.Value {
result.Value[i].region = self
}
return result.Value, nil
}
func (self *SRegion) ListDBInstanceVirtualNetworkRule(Id string) ([]SDBInstanceVirtualNetworkRule, error) {
type networksRules struct {
Value []SDBInstanceVirtualNetworkRule
}
result := networksRules{}
err := self.get(fmt.Sprintf("%s/virtualNetworkRules", Id), url.Values{}, &result)
if err != nil {
return nil, errors.Wrapf(err, "get(%s)", Id)
}
return result.Value, nil
}
func (rds *SDBInstance) GetName() string {
return rds.Name
}
func (rds *SDBInstance) GetId() string {
return strings.ToLower(rds.ID)
}
func (rds *SDBInstance) GetGlobalId() string {
return rds.GetId()
}
func (rds *SDBInstance) GetStatus() string {
return api.DBINSTANCE_RUNNING
}
func (self *SDBInstance) GetProjectId() string {
return getResourceGroup(self.ID)
}
func (rds *SDBInstance) GetExpiredAt() time.Time {
return time.Time{}
}
func (rds *SDBInstance) GetStorageType() string {
switch rds.Sku.Tier {
case "Basic":
return api.STORAGE_AZURE_BASIC
case "General Purpose":
return api.STORAGE_AZURE_GENERAL_PURPOSE
case "Memory Optimized":
return api.STORAGE_AZURE_GENERAL_PURPOSE
default:
return api.STORAGE_AZURE_BASIC
}
}
func (rds *SDBInstance) Refresh() error {
newrds := SDBInstance{}
err := rds.region.get(rds.ID, url.Values{}, &newrds)
if err != nil {
return errors.Wrapf(err, "rds.region.get(%s, url.Values{}, &newdb)", rds.ID)
}
err = jsonutils.Update(rds, newrds)
if err != nil {
return err
}
rds.Tags = newrds.Tags
return nil
}
func (rds *SDBInstance) GetMasterInstanceId() string {
if len(rds.Properties.MasterServerId) > 0 {
return strings.ToLower(rds.Properties.MasterServerId)
}
return ""
}
func (rds *SDBInstance) GetPort() int {
switch rds.GetEngine() {
case api.DBINSTANCE_TYPE_POSTGRESQL:
return 5432
case api.DBINSTANCE_TYPE_MYSQL:
return 3306
case api.DBINSTANCE_TYPE_MARIADB:
return 3306
default:
return 0
}
}
func (rds *SDBInstance) GetEngine() string {
databaseType := strings.Split(rds.Type, "/")
switch databaseType[0] {
case "Microsoft.DBforPostgreSQL":
return api.DBINSTANCE_TYPE_POSTGRESQL
case "Microsoft.DBforMySQL":
return api.DBINSTANCE_TYPE_MYSQL
case "Microsoft.DBforMariaDB":
return api.DBINSTANCE_TYPE_MARIADB
default:
return ""
}
}
func (rds *SDBInstance) GetEngineVersion() string {
return rds.Properties.Version
}
func (rds *SDBInstance) GetInstanceType() string {
return rds.Sku.Name
}
func (rds *SDBInstance) GetVcpuCount() int {
return rds.Sku.Capacity
}
func (rds *SDBInstance) GetVmemSizeMB() int {
if strings.Contains(rds.Type, "server") {
switch rds.Sku.Tier {
case "Basic":
return rds.Sku.Capacity * 2 * 1024
case "General Purpose":
return rds.Sku.Capacity * 5 * 1024
case "GeneralPurpose":
return int(float32(rds.Sku.Capacity) * 5.2 * 1024)
case "Memory Optimized":
return rds.Sku.Capacity * 10 * 1024
default:
return 0
}
}
return 0
}
func (rds *SDBInstance) GetDiskSizeGB() int {
return rds.Properties.StorageProfile.StorageMB / 1024
}
func (rds *SDBInstance) GetCategory() string {
return rds.Sku.Tier
}
func (rds *SDBInstance) GetMaintainTime() string {
return ""
}
func (rds *SDBInstance) GetConnectionStr() string {
return rds.Properties.FullyQualifiedDomainName
}
// func (rds *SDBInstance) GetInternalConnectionStr() string
func (rds *SDBInstance) GetIVpcId() string {
splited := strings.Split(rds.Properties.DelegatedSubnetArguments.SubnetArmResourceId, "/subnets")
return splited[0]
}
func (rds *SDBInstance) GetDBNetworks() ([]cloudprovider.SDBInstanceNetwork, error) {
result := []cloudprovider.SDBInstanceNetwork{}
delegateNet := cloudprovider.SDBInstanceNetwork{NetworkId: rds.Properties.DelegatedSubnetArguments.SubnetArmResourceId}
result = append(result, delegateNet)
return result, nil
}
func (rds *SDBInstance) GetZone1Id() string {
return ""
}
func (rds *SDBInstance) GetZone2Id() string {
return ""
}
func (rds *SDBInstance) GetZone3Id() string {
return ""
}
func (rds *SDBInstance) GetIDBInstanceParameters() ([]cloudprovider.ICloudDBInstanceParameter, error) {
configs, err := rds.region.ListDBInstanceConfiguration(rds.ID)
if err != nil {
return nil, errors.Wrapf(err, "rds.region.ListDBInstanceConfiguration(%s)", rds.ID)
}
result := []cloudprovider.ICloudDBInstanceParameter{}
for i := range configs {
result = append(result, &configs[i])
}
return result, nil
}
func (rds *SDBInstance) GetIDBInstanceDatabases() ([]cloudprovider.ICloudDBInstanceDatabase, error) {
db, err := rds.region.ListSDBInstanceDatabase(rds.ID)
if err != nil {
return nil, errors.Wrapf(err, "rds.region.ListSDBInstanceDatabase(%s)", rds.ID)
}
result := []cloudprovider.ICloudDBInstanceDatabase{}
for i := range db {
result = append(result, &db[i])
}
return result, nil
}
func (rds *SDBInstance) GetIDBInstanceAccounts() ([]cloudprovider.ICloudDBInstanceAccount, error) {
accounts := []cloudprovider.ICloudDBInstanceAccount{}
if len(rds.Properties.AdministratorLogin) > 0 {
account := &SDBInstanceAccount{instance: rds, AccountName: rds.Properties.AdministratorLogin}
accounts = append(accounts, account)
}
return accounts, nil
}
func (rds *SDBInstance) GetIDBInstanceBackups() ([]cloudprovider.ICloudDBInstanceBackup, error) {
return []cloudprovider.ICloudDBInstanceBackup{}, nil
}
func (rds *SDBInstance) Delete() error {
return cloudprovider.ErrNotImplemented
}
func (self *SDBInstance) SetTags(tags map[string]string, replace bool) error {
if !replace {
for k, v := range self.Tags {
if _, ok := tags[k]; !ok {
tags[k] = v
}
}
}
_, err := self.region.client.SetTags(self.ID, tags)
if err != nil {
return errors.Wrapf(err, "self.region.client.SetTags(%s,%s)", self.ID, jsonutils.Marshal(tags).String())
}
return nil
}