mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-06-23 23:44:12 +08:00
766 lines
24 KiB
Go
766 lines
24 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 models
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
"yunion.io/x/log"
|
|
"yunion.io/x/pkg/errors"
|
|
"yunion.io/x/pkg/util/compare"
|
|
"yunion.io/x/pkg/util/netutils"
|
|
"yunion.io/x/sqlchemy"
|
|
|
|
api "yunion.io/x/onecloud/pkg/apis/compute"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
|
|
"yunion.io/x/onecloud/pkg/cloudprovider"
|
|
"yunion.io/x/onecloud/pkg/httperrors"
|
|
"yunion.io/x/onecloud/pkg/mcclient"
|
|
)
|
|
|
|
type SVpcManager struct {
|
|
db.SEnabledStatusStandaloneResourceBaseManager
|
|
}
|
|
|
|
var VpcManager *SVpcManager
|
|
|
|
func init() {
|
|
VpcManager = &SVpcManager{
|
|
SEnabledStatusStandaloneResourceBaseManager: db.NewEnabledStatusStandaloneResourceBaseManager(
|
|
SVpc{},
|
|
"vpcs_tbl",
|
|
"vpc",
|
|
"vpcs",
|
|
),
|
|
}
|
|
VpcManager.SetVirtualObject(VpcManager)
|
|
}
|
|
|
|
type SVpc struct {
|
|
db.SEnabledStatusStandaloneResourceBase
|
|
db.SExternalizedResourceBase
|
|
|
|
SManagedResourceBase
|
|
|
|
IsDefault bool `default:"false" list:"admin" create:"admin_optional"`
|
|
|
|
CidrBlock string `charset:"ascii" nullable:"true" list:"admin" create:"admin_required"`
|
|
|
|
CloudregionId string `width:"36" charset:"ascii" nullable:"false" list:"admin" create:"admin_required"`
|
|
}
|
|
|
|
func (manager *SVpcManager) GetContextManagers() [][]db.IModelManager {
|
|
return [][]db.IModelManager{
|
|
{CloudregionManager},
|
|
}
|
|
}
|
|
|
|
func (self *SVpcManager) AllowListItems(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowList(userCred, self)
|
|
}
|
|
|
|
func (self *SVpcManager) AllowCreateItem(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowCreate(userCred, self)
|
|
}
|
|
|
|
func (self *SVpc) AllowGetDetails(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowGet(userCred, self)
|
|
}
|
|
|
|
func (self *SVpc) AllowUpdateItem(ctx context.Context, userCred mcclient.TokenCredential) bool {
|
|
return db.IsAdminAllowUpdate(userCred, self)
|
|
}
|
|
|
|
func (self *SVpc) AllowDeleteItem(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowDelete(userCred, self)
|
|
}
|
|
|
|
func (self *SVpc) GetCloudRegionId() string {
|
|
if len(self.CloudregionId) == 0 {
|
|
return api.DEFAULT_REGION_ID
|
|
} else {
|
|
return self.CloudregionId
|
|
}
|
|
}
|
|
|
|
func (self *SVpc) CustomizeCreate(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) error {
|
|
idstr, _ := data.GetString("id")
|
|
if len(idstr) > 0 {
|
|
self.Id = idstr
|
|
}
|
|
return self.SEnabledStatusStandaloneResourceBase.CustomizeCreate(ctx, userCred, ownerId, query, data)
|
|
}
|
|
|
|
func (self *SVpc) getNatgatewayQuery() *sqlchemy.SQuery {
|
|
return NatGatewayManager.Query().Equals("vpc_id", self.Id)
|
|
}
|
|
|
|
func (self *SVpc) GetNatgatewayCount() (int, error) {
|
|
return self.getNatgatewayQuery().CountWithError()
|
|
}
|
|
|
|
func (self *SVpc) GetNatgateways() ([]SNatGateway, error) {
|
|
nats := []SNatGateway{}
|
|
err := db.FetchModelObjects(NatGatewayManager, self.getNatgatewayQuery(), &nats)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return nats, nil
|
|
}
|
|
|
|
func (self *SVpc) ValidateDeleteCondition(ctx context.Context) error {
|
|
cnt, err := self.GetNetworkCount()
|
|
if err != nil {
|
|
return httperrors.NewInternalServerError("GetNetworkCount fail %s", err)
|
|
}
|
|
if cnt > 0 {
|
|
return httperrors.NewNotEmptyError("VPC not empty, please delete network first")
|
|
}
|
|
cnt, err = self.GetNatgatewayCount()
|
|
if err != nil {
|
|
return httperrors.NewInternalServerError("GetNatgatewayCount fail %v", err)
|
|
}
|
|
if cnt > 0 {
|
|
return httperrors.NewNotEmptyError("VPC not empty, please delete nat gateway first")
|
|
}
|
|
if self.Id == api.DEFAULT_VPC_ID {
|
|
return httperrors.NewProtectedResourceError("not allow to delete default vpc")
|
|
}
|
|
return self.SEnabledStatusStandaloneResourceBase.ValidateDeleteCondition(ctx)
|
|
}
|
|
|
|
func (self *SVpc) getWireQuery() *sqlchemy.SQuery {
|
|
wires := WireManager.Query()
|
|
if self.Id == api.DEFAULT_VPC_ID {
|
|
return wires.Filter(sqlchemy.OR(sqlchemy.IsNull(wires.Field("vpc_id")),
|
|
sqlchemy.IsEmpty(wires.Field("vpc_id")),
|
|
sqlchemy.Equals(wires.Field("vpc_id"), self.Id)))
|
|
} else {
|
|
return wires.Equals("vpc_id", self.Id)
|
|
}
|
|
}
|
|
|
|
func (self *SVpc) GetWireCount() (int, error) {
|
|
q := self.getWireQuery()
|
|
return q.CountWithError()
|
|
}
|
|
|
|
func (self *SVpc) GetWires() []SWire {
|
|
wires := make([]SWire, 0)
|
|
q := self.getWireQuery()
|
|
err := db.FetchModelObjects(WireManager, q, &wires)
|
|
if err != nil {
|
|
log.Errorf("getWires fail %s", err)
|
|
return nil
|
|
}
|
|
return wires
|
|
}
|
|
|
|
func (self *SVpc) getNetworkQuery() *sqlchemy.SQuery {
|
|
q := NetworkManager.Query()
|
|
wireQ := self.getWireQuery().SubQuery()
|
|
q = q.In("wire_id", wireQ.Query(wireQ.Field("id")).SubQuery())
|
|
return q
|
|
}
|
|
|
|
func (self *SVpc) GetNetworkCount() (int, error) {
|
|
q := self.getNetworkQuery()
|
|
return q.CountWithError()
|
|
}
|
|
|
|
func (self *SVpc) GetRouteTableQuery() *sqlchemy.SQuery {
|
|
return RouteTableManager.Query().Equals("vpc_id", self.Id)
|
|
}
|
|
|
|
func (self *SVpc) GetRouteTables() []SRouteTable {
|
|
q := self.GetRouteTableQuery()
|
|
routes := []SRouteTable{}
|
|
db.FetchModelObjects(RouteTableManager, q, &routes)
|
|
return routes
|
|
}
|
|
|
|
func (self *SVpc) GetRouteTableCount() (int, error) {
|
|
return self.GetRouteTableQuery().CountWithError()
|
|
}
|
|
|
|
func (self *SVpc) getMoreDetails(extra *jsonutils.JSONDict) *jsonutils.JSONDict {
|
|
cnt, _ := self.GetWireCount()
|
|
extra.Add(jsonutils.NewInt(int64(cnt)), "wire_count")
|
|
cnt, _ = self.GetNetworkCount()
|
|
extra.Add(jsonutils.NewInt(int64(cnt)), "network_count")
|
|
cnt, _ = self.GetRouteTableCount()
|
|
extra.Add(jsonutils.NewInt(int64(cnt)), "routetable_count")
|
|
cnt, _ = self.GetNatgatewayCount()
|
|
extra.Add(jsonutils.NewInt(int64(cnt)), "natgateway_count")
|
|
/* region, err := self.GetRegion()
|
|
if err != nil {
|
|
log.Errorf("failed getting region for vpc %s(%s)", self.Name, self.Id)
|
|
return extra
|
|
}
|
|
extra.Add(jsonutils.NewString(region.GetName()), "region")
|
|
if len(region.GetExternalId()) > 0 {
|
|
extra.Add(jsonutils.NewString(region.GetExternalId()), "region_external_id")
|
|
}*/
|
|
|
|
info := self.getCloudProviderInfo()
|
|
extra.Update(jsonutils.Marshal(&info))
|
|
|
|
return extra
|
|
}
|
|
|
|
func (self *SVpc) getCloudProviderInfo() SCloudProviderInfo {
|
|
region, _ := self.GetRegion()
|
|
provider := self.GetCloudprovider()
|
|
return MakeCloudProviderInfo(region, nil, provider)
|
|
}
|
|
|
|
func (self *SVpc) GetRegion() (*SCloudregion, error) {
|
|
region, err := CloudregionManager.FetchById(self.CloudregionId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return region.(*SCloudregion), nil
|
|
}
|
|
|
|
func (self *SVpc) getZoneByExternalId(externalId string) (*SZone, error) {
|
|
region, err := self.GetRegion()
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "getZoneByExternalId.GetRegion")
|
|
}
|
|
zones := []SZone{}
|
|
q := ZoneManager.Query().Equals("cloudregion_id", region.Id).Equals("external_id", externalId)
|
|
err = db.FetchModelObjects(ZoneManager, q, &zones)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "getZoneByExternalId.FetchModelObjects")
|
|
}
|
|
if len(zones) == 1 {
|
|
return &zones[0], nil
|
|
}
|
|
if len(zones) == 0 {
|
|
return nil, fmt.Errorf("failed to found zone by externalId %s in cloudregion %s(%s)", externalId, region.Name, region.Id)
|
|
}
|
|
return nil, fmt.Errorf("found %d duplicate zones by externalId %s in cloudregion %s(%s)", len(zones), externalId, region.Name, region.Id)
|
|
}
|
|
|
|
func (self *SVpc) GetCustomizeColumns(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) *jsonutils.JSONDict {
|
|
extra := self.SEnabledStatusStandaloneResourceBase.GetCustomizeColumns(ctx, userCred, query)
|
|
return self.getMoreDetails(extra)
|
|
}
|
|
|
|
func (self *SVpc) GetExtraDetails(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) (*jsonutils.JSONDict, error) {
|
|
extra, err := self.SEnabledStatusStandaloneResourceBase.GetExtraDetails(ctx, userCred, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return self.getMoreDetails(extra), nil
|
|
}
|
|
|
|
func (manager *SVpcManager) getVpcsByRegion(region *SCloudregion, provider *SCloudprovider) ([]SVpc, error) {
|
|
vpcs := make([]SVpc, 0)
|
|
q := manager.Query().Equals("cloudregion_id", region.Id)
|
|
if provider != nil {
|
|
q = q.Equals("manager_id", provider.Id)
|
|
}
|
|
err := db.FetchModelObjects(manager, q, &vpcs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return vpcs, nil
|
|
}
|
|
|
|
func (self *SVpc) setDefault(def bool) error {
|
|
var err error
|
|
if self.IsDefault != def {
|
|
_, err = db.Update(self, func() error {
|
|
self.IsDefault = def
|
|
return nil
|
|
})
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (manager *SVpcManager) SyncVPCs(ctx context.Context, userCred mcclient.TokenCredential, provider *SCloudprovider, region *SCloudregion, vpcs []cloudprovider.ICloudVpc) ([]SVpc, []cloudprovider.ICloudVpc, compare.SyncResult) {
|
|
lockman.LockClass(ctx, manager, db.GetLockClassKey(manager, userCred))
|
|
defer lockman.ReleaseClass(ctx, manager, db.GetLockClassKey(manager, userCred))
|
|
|
|
localVPCs := make([]SVpc, 0)
|
|
remoteVPCs := make([]cloudprovider.ICloudVpc, 0)
|
|
syncResult := compare.SyncResult{}
|
|
|
|
dbVPCs, err := manager.getVpcsByRegion(region, provider)
|
|
if err != nil {
|
|
syncResult.Error(err)
|
|
return nil, nil, syncResult
|
|
}
|
|
|
|
for i := range dbVPCs {
|
|
if taskman.TaskManager.IsInTask(&dbVPCs[i]) {
|
|
syncResult.Error(fmt.Errorf("object in task"))
|
|
return nil, nil, syncResult
|
|
}
|
|
}
|
|
|
|
removed := make([]SVpc, 0)
|
|
commondb := make([]SVpc, 0)
|
|
commonext := make([]cloudprovider.ICloudVpc, 0)
|
|
added := make([]cloudprovider.ICloudVpc, 0)
|
|
|
|
err = compare.CompareSets(dbVPCs, vpcs, &removed, &commondb, &commonext, &added)
|
|
if err != nil {
|
|
syncResult.Error(err)
|
|
return nil, nil, syncResult
|
|
}
|
|
|
|
for i := 0; i < len(removed); i += 1 {
|
|
err = removed[i].syncRemoveCloudVpc(ctx, userCred)
|
|
if err != nil {
|
|
syncResult.DeleteError(err)
|
|
} else {
|
|
syncResult.Delete()
|
|
}
|
|
}
|
|
for i := 0; i < len(commondb); i += 1 {
|
|
err = commondb[i].SyncWithCloudVpc(ctx, userCred, commonext[i])
|
|
if err != nil {
|
|
syncResult.UpdateError(err)
|
|
} else {
|
|
syncMetadata(ctx, userCred, &commondb[i], commonext[i])
|
|
localVPCs = append(localVPCs, commondb[i])
|
|
remoteVPCs = append(remoteVPCs, commonext[i])
|
|
syncResult.Update()
|
|
}
|
|
}
|
|
for i := 0; i < len(added); i += 1 {
|
|
new, err := manager.newFromCloudVpc(ctx, userCred, added[i], provider, region)
|
|
if err != nil {
|
|
syncResult.AddError(err)
|
|
} else {
|
|
syncMetadata(ctx, userCred, new, added[i])
|
|
localVPCs = append(localVPCs, *new)
|
|
remoteVPCs = append(remoteVPCs, added[i])
|
|
syncResult.Add()
|
|
}
|
|
}
|
|
|
|
return localVPCs, remoteVPCs, syncResult
|
|
}
|
|
|
|
func (self *SVpc) syncRemoveCloudVpc(ctx context.Context, userCred mcclient.TokenCredential) error {
|
|
lockman.LockObject(ctx, self)
|
|
defer lockman.ReleaseObject(ctx, self)
|
|
|
|
err := self.ValidateDeleteCondition(ctx)
|
|
if err != nil { // cannot delete
|
|
self.markAllNetworksUnknown(userCred)
|
|
_, err = self.PerformDisable(ctx, userCred, nil, nil)
|
|
if err == nil {
|
|
err = self.SetStatus(userCred, api.VPC_STATUS_UNKNOWN, "sync to delete")
|
|
}
|
|
} else {
|
|
err = self.RealDelete(ctx, userCred)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (self *SVpc) SyncWithCloudVpc(ctx context.Context, userCred mcclient.TokenCredential, extVPC cloudprovider.ICloudVpc) error {
|
|
diff, err := db.UpdateWithLock(ctx, self, func() error {
|
|
extVPC.Refresh()
|
|
// self.Name = extVPC.GetName()
|
|
self.Status = extVPC.GetStatus()
|
|
self.CidrBlock = extVPC.GetCidrBlock()
|
|
self.IsDefault = extVPC.GetIsDefault()
|
|
self.ExternalId = extVPC.GetGlobalId()
|
|
|
|
self.IsEmulated = extVPC.IsEmulated()
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
db.OpsLog.LogSyncUpdate(self, diff, userCred)
|
|
return nil
|
|
}
|
|
|
|
func (manager *SVpcManager) newFromCloudVpc(ctx context.Context, userCred mcclient.TokenCredential, extVPC cloudprovider.ICloudVpc, provider *SCloudprovider, region *SCloudregion) (*SVpc, error) {
|
|
vpc := SVpc{}
|
|
vpc.SetModelManager(manager, &vpc)
|
|
|
|
newName, err := db.GenerateName(manager, userCred, extVPC.GetName())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
vpc.Name = newName
|
|
vpc.Status = extVPC.GetStatus()
|
|
vpc.ExternalId = extVPC.GetGlobalId()
|
|
vpc.IsDefault = extVPC.GetIsDefault()
|
|
vpc.CidrBlock = extVPC.GetCidrBlock()
|
|
vpc.CloudregionId = region.Id
|
|
|
|
vpc.ManagerId = provider.Id
|
|
|
|
vpc.IsEmulated = extVPC.IsEmulated()
|
|
|
|
err = manager.TableSpec().Insert(&vpc)
|
|
if err != nil {
|
|
log.Errorf("newFromCloudVpc fail %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
db.OpsLog.LogEvent(&vpc, db.ACT_CREATE, vpc.GetShortDesc(ctx), userCred)
|
|
|
|
return &vpc, nil
|
|
}
|
|
|
|
func (self *SVpc) markAllNetworksUnknown(userCred mcclient.TokenCredential) error {
|
|
wires := self.GetWires()
|
|
if wires == nil || len(wires) == 0 {
|
|
return nil
|
|
}
|
|
for i := 0; i < len(wires); i += 1 {
|
|
wires[i].markNetworkUnknown(userCred)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (manager *SVpcManager) InitializeData() error {
|
|
vpcObj, err := manager.FetchById(api.DEFAULT_VPC_ID)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
defVpc := SVpc{}
|
|
defVpc.SetModelManager(VpcManager, &defVpc)
|
|
|
|
defVpc.Id = api.DEFAULT_VPC_ID
|
|
defVpc.Name = "Default"
|
|
defVpc.CloudregionId = api.DEFAULT_REGION_ID
|
|
defVpc.Description = "Default VPC"
|
|
defVpc.Status = api.VPC_STATUS_AVAILABLE
|
|
defVpc.IsDefault = true
|
|
err = manager.TableSpec().Insert(&defVpc)
|
|
if err != nil {
|
|
log.Errorf("Insert default vpc fail: %s", err)
|
|
}
|
|
return err
|
|
} else {
|
|
return err
|
|
}
|
|
} else {
|
|
vpc := vpcObj.(*SVpc)
|
|
if vpc.Status != api.VPC_STATUS_AVAILABLE {
|
|
_, err = db.Update(vpc, func() error {
|
|
vpc.Status = api.VPC_STATUS_AVAILABLE
|
|
return nil
|
|
})
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (manager *SVpcManager) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data *jsonutils.JSONDict) (*jsonutils.JSONDict, error) {
|
|
regionId := jsonutils.GetAnyString(data, []string{"region", "cloudregion", "cloudregion_id"})
|
|
if len(regionId) == 0 {
|
|
return nil, httperrors.NewMissingParameterError("cloudregion_id")
|
|
}
|
|
regionObj, err := CloudregionManager.FetchByIdOrName(userCred, regionId)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, httperrors.NewResourceNotFoundError2(CloudregionManager.Keyword(), regionId)
|
|
} else {
|
|
return nil, httperrors.NewGeneralError(err)
|
|
}
|
|
}
|
|
region := regionObj.(*SCloudregion)
|
|
data.Add(jsonutils.NewString(region.GetId()), "cloudregion_id")
|
|
if region.isManaged() {
|
|
managerStr := jsonutils.GetAnyString(data, []string{"manager_id", "manager"})
|
|
if len(managerStr) == 0 {
|
|
return nil, httperrors.NewMissingParameterError("manager_id")
|
|
}
|
|
managerObj, err := CloudproviderManager.FetchByIdOrName(userCred, managerStr)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, httperrors.NewResourceNotFoundError2(CloudproviderManager.Keyword(), managerStr)
|
|
} else {
|
|
return nil, httperrors.NewGeneralError(err)
|
|
}
|
|
}
|
|
data.Add(jsonutils.NewString(managerObj.GetId()), "manager_id")
|
|
} else {
|
|
return nil, httperrors.NewNotImplementedError("Cannot create VPC in private cloud")
|
|
}
|
|
|
|
cidrBlock, _ := data.GetString("cidr_block")
|
|
if len(cidrBlock) > 0 {
|
|
blocks := strings.Split(cidrBlock, ",")
|
|
for _, block := range blocks {
|
|
_, err = netutils.NewIPV4Prefix(block)
|
|
if err != nil {
|
|
return nil, httperrors.NewInputParameterError("invalid cidr_block %s", cidrBlock)
|
|
}
|
|
}
|
|
}
|
|
data, err = manager.SEnabledStatusStandaloneResourceBaseManager.ValidateCreateData(ctx, userCred, ownerId, query, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return region.GetDriver().ValidateCreateVpcData(ctx, userCred, data)
|
|
}
|
|
|
|
func (self *SVpc) PostCreate(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) {
|
|
if len(self.ManagerId) == 0 {
|
|
return
|
|
}
|
|
task, err := taskman.TaskManager.NewTask(ctx, "VpcCreateTask", self, userCred, nil, "", "", nil)
|
|
if err != nil {
|
|
log.Errorf("VpcCreateTask newTask error %s", err)
|
|
} else {
|
|
task.ScheduleRun(nil)
|
|
}
|
|
}
|
|
|
|
func (self *SVpc) GetIRegion() (cloudprovider.ICloudRegion, error) {
|
|
region, err := self.GetRegion()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
provider, err := self.GetDriver()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return provider.GetIRegionById(region.GetExternalId())
|
|
}
|
|
|
|
func (self *SVpc) GetIVpc() (cloudprovider.ICloudVpc, error) {
|
|
provider, err := self.GetDriver()
|
|
if err != nil {
|
|
log.Errorf("fail to find cloud provider")
|
|
return nil, err
|
|
}
|
|
var iregion cloudprovider.ICloudRegion
|
|
if provider.GetFactory().IsOnPremise() {
|
|
iregion, err = provider.GetOnPremiseIRegion()
|
|
} else {
|
|
region, err := self.GetRegion()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
iregion, err = provider.GetIRegionById(region.ExternalId)
|
|
}
|
|
if err != nil {
|
|
log.Errorf("fail to find iregion: %s", err)
|
|
return nil, err
|
|
}
|
|
ivpc, err := iregion.GetIVpcById(self.ExternalId)
|
|
if err != nil {
|
|
log.Errorf("fail to find ivpc by id %s %s", self.ExternalId, err)
|
|
return nil, err
|
|
}
|
|
return ivpc, nil
|
|
}
|
|
|
|
func (self *SVpc) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
|
|
log.Infof("SVpc delete do nothing")
|
|
self.SetStatus(userCred, api.VPC_STATUS_START_DELETE, "")
|
|
return nil
|
|
}
|
|
|
|
func (self *SVpc) CustomizeDelete(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) error {
|
|
if len(self.ExternalId) > 0 {
|
|
return self.StartDeleteVpcTask(ctx, userCred)
|
|
} else {
|
|
return self.RealDelete(ctx, userCred)
|
|
}
|
|
}
|
|
|
|
func (self *SVpc) RealDelete(ctx context.Context, userCred mcclient.TokenCredential) error {
|
|
db.OpsLog.LogEvent(self, db.ACT_DELOCATE, self.GetShortDesc(ctx), userCred)
|
|
self.SetStatus(userCred, api.VPC_STATUS_DELETED, "real delete")
|
|
routes := self.GetRouteTables()
|
|
var err error
|
|
for i := 0; i < len(routes); i++ {
|
|
err = routes[i].RealDelete(ctx, userCred)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "delete route table %s failed", routes[i].GetId())
|
|
}
|
|
}
|
|
natgateways, err := self.GetNatgateways()
|
|
if err != nil {
|
|
return errors.Wrap(err, "fetch natgateways failed")
|
|
}
|
|
for i := range natgateways {
|
|
err = natgateways[i].RealDelete(ctx, userCred)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "delete natgateway %s failed", natgateways[i].GetId())
|
|
}
|
|
}
|
|
|
|
return self.SEnabledStatusStandaloneResourceBase.Delete(ctx, userCred)
|
|
}
|
|
|
|
func (self *SVpc) StartDeleteVpcTask(ctx context.Context, userCred mcclient.TokenCredential) error {
|
|
task, err := taskman.TaskManager.NewTask(ctx, "VpcDeleteTask", self, userCred, nil, "", "", nil)
|
|
if err != nil {
|
|
log.Errorf("Start vpcdeleteTask fail %s", err)
|
|
return err
|
|
}
|
|
task.ScheduleRun(nil)
|
|
return nil
|
|
}
|
|
|
|
func (self *SVpc) getPrefix() []netutils.IPV4Prefix {
|
|
if len(self.CidrBlock) > 0 {
|
|
ret := []netutils.IPV4Prefix{}
|
|
blocks := strings.Split(self.CidrBlock, ",")
|
|
for _, block := range blocks {
|
|
prefix, _ := netutils.NewIPV4Prefix(block)
|
|
ret = append(ret, prefix)
|
|
}
|
|
return ret
|
|
}
|
|
return []netutils.IPV4Prefix{{}}
|
|
}
|
|
|
|
func (self *SVpc) getIPRanges() []netutils.IPV4AddrRange {
|
|
ret := []netutils.IPV4AddrRange{}
|
|
prefs := self.getPrefix()
|
|
for _, pref := range prefs {
|
|
ret = append(ret, pref.ToIPRange())
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
func (self *SVpc) AllowPerformPurge(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowPerform(userCred, self, "purge")
|
|
}
|
|
|
|
func (self *SVpc) PerformPurge(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
|
err := self.ValidateDeleteCondition(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
provider := self.GetCloudprovider()
|
|
if provider != nil {
|
|
if provider.Enabled {
|
|
return nil, httperrors.NewInvalidStatusError("Cannot purge vpc on enabled cloud provider")
|
|
}
|
|
}
|
|
err = self.RealDelete(ctx, userCred)
|
|
return nil, err
|
|
}
|
|
|
|
func (manager *SVpcManager) ListItemFilter(ctx context.Context, q *sqlchemy.SQuery, userCred mcclient.TokenCredential, query jsonutils.JSONObject) (*sqlchemy.SQuery, error) {
|
|
var err error
|
|
q, err = managedResourceFilterByAccount(q, query, "", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
q = managedResourceFilterByCloudType(q, query, "", nil)
|
|
|
|
q, err = managedResourceFilterByDomain(q, query, "", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
q, err = manager.SStatusStandaloneResourceBaseManager.ListItemFilter(ctx, q, userCred, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if jsonutils.QueryBoolean(query, "usable", false) {
|
|
regions := CloudregionManager.Query().SubQuery()
|
|
cloudproviders := CloudproviderManager.Query().SubQuery()
|
|
providerSQ := cloudproviders.Query(cloudproviders.Field("id")).Filter(
|
|
sqlchemy.AND(
|
|
sqlchemy.IsTrue(cloudproviders.Field("enabled")),
|
|
sqlchemy.In(cloudproviders.Field("status"), api.CLOUD_PROVIDER_VALID_STATUS),
|
|
sqlchemy.In(cloudproviders.Field("health_status"), api.CLOUD_PROVIDER_VALID_HEALTH_STATUS),
|
|
),
|
|
)
|
|
q = q.Join(regions, sqlchemy.Equals(q.Field("cloudregion_id"), regions.Field("id"))).
|
|
Filter(
|
|
sqlchemy.AND(
|
|
sqlchemy.Equals(regions.Field("status"), api.CLOUD_REGION_STATUS_INSERVER),
|
|
sqlchemy.OR(
|
|
sqlchemy.In(q.Field("manager_id"), providerSQ.SubQuery()),
|
|
sqlchemy.IsNullOrEmpty(q.Field("manager_id")),
|
|
),
|
|
),
|
|
)
|
|
wires := WireManager.Query().SubQuery()
|
|
networks := NetworkManager.Query().SubQuery()
|
|
|
|
sq := wires.Query(wires.Field("vpc_id")).Join(networks, sqlchemy.Equals(wires.Field("id"), networks.Field("wire_id"))).Filter(
|
|
sqlchemy.Equals(networks.Field("status"), api.NETWORK_STATUS_AVAILABLE),
|
|
)
|
|
|
|
q = q.In("id", sq.SubQuery())
|
|
}
|
|
|
|
return q, nil
|
|
}
|
|
|
|
func (manager *SVpcManager) QueryDistinctExtraField(q *sqlchemy.SQuery, field string) (*sqlchemy.SQuery, error) {
|
|
var err error
|
|
q, err = manager.SEnabledStatusStandaloneResourceBaseManager.QueryDistinctExtraField(q, field)
|
|
if err == nil {
|
|
return q, nil
|
|
}
|
|
switch field {
|
|
case "account":
|
|
cloudproviders := CloudproviderManager.Query().SubQuery()
|
|
cloudaccounts := CloudaccountManager.Query("name", "id").Distinct().SubQuery()
|
|
q = q.Join(cloudproviders, sqlchemy.Equals(q.Field("manager_id"), cloudproviders.Field("id")))
|
|
q = q.Join(cloudaccounts, sqlchemy.Equals(cloudproviders.Field("cloudaccount_id"), cloudaccounts.Field("id")))
|
|
q.GroupBy(cloudaccounts.Field("name"))
|
|
q.AppendField(cloudaccounts.Field("name", "account"))
|
|
case "manager":
|
|
cloudproviders := CloudproviderManager.Query("name", "id").Distinct().SubQuery()
|
|
q = q.Join(cloudproviders, sqlchemy.Equals(q.Field("manager_id"), cloudproviders.Field("id")))
|
|
q.GroupBy(cloudproviders.Field("name"))
|
|
q.AppendField(cloudproviders.Field("name", "manager"))
|
|
default:
|
|
return q, httperrors.NewBadRequestError("unsupport field %s", field)
|
|
}
|
|
return q, nil
|
|
}
|
|
|
|
func (self *SVpc) SyncRemoteWires(ctx context.Context, userCred mcclient.TokenCredential) error {
|
|
ivpc, err := self.GetIVpc()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
provider := CloudproviderManager.FetchCloudproviderById(self.ManagerId)
|
|
syncVpcWires(ctx, userCred, nil, provider, self, ivpc, &SSyncRange{})
|
|
|
|
hosts := HostManager.GetHostsByManagerAndRegion(provider.Id, self.CloudregionId)
|
|
for i := 0; i < len(hosts); i += 1 {
|
|
ihost, err := hosts[i].GetIHost()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
syncHostWires(ctx, userCred, nil, provider, &hosts[i], ihost)
|
|
}
|
|
return nil
|
|
}
|