opensource/cloudpods
3
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-09-24 16:03:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feature(region): Add bind/unbind-snapshotpolicy api for disk

Browse Source
This commit is contained in:
Rain
2020-01-04 20:31:02 +08:00
parent 7b794390cb
commit 8df61823e4
3 changed files with 96 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/compute/models/disks.go
+93
View File
@@ -2476,3 +2476,96 @@ func (disk *SDisk) GetUsages() []db.IUsage {
&usage,
}
}
func (disk *SDisk) AllowPerformBindSnapshotpolicy(ctx context.Context, userCred mcclient.TokenCredential,
query jsonutils.JSONObject) bool {
return disk.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, disk, "bind-snapshotpolicy")
}
func (disk *SDisk) PerformBindSnapshotpolicy(ctx context.Context, userCred mcclient.TokenCredential,
query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
spIden, err := data.GetString("snapshotpolicy")
if err != nil {
return nil, httperrors.NewMissingParameterError("miss snapshotpolicy")
}
// check snapshotpolicy
imodel, err := db.FetchByIdOrName(SnapshotPolicyManager, userCred, spIden)
if errors.Cause(err) == sql.ErrNoRows {
return nil, httperrors.NewInputParameterError("no such snapshotpolicy %s", spIden)
}
if err != nil {
return nil, errors.Wrap(err, "db.FetchByIdOrName")
}
snapshotpolicy := imodel.(*SSnapshotPolicy)
// try to bind
spd, err := SnapshotPolicyDiskManager.newSnapshotpolicyDisk(ctx, userCred, snapshotpolicy, disk)
if errors.Cause(err) == ErrExistSD {
if spd.Status != api.SNAPSHOT_POLICY_DISK_INIT {
return nil, nil
}
} else if err != nil {
return nil, errors.Wrap(err, "SnapshotPolicyDiskManager.newSnapshotpolicyDisk")
}
// start up SnapshotPolicyApplyTask
taskData := jsonutils.NewDict()
taskData.Add(jsonutils.Marshal(spd), "snapshotPolicyDisk")
taskData.Add(jsonutils.Marshal(snapshotpolicy), "snapshotPolicy")
if task, err := taskman.TaskManager.NewTask(ctx, "SnapshotPolicyApplyTask", disk, userCred, nil, "", "",
nil); err != nil {
return nil, errors.Wrap(err, "fail to start up SnapshotPolicyApplyTask")
} else {
task.ScheduleRun(taskData)
}
return nil, nil
}
func (disk *SDisk) AllowPerformUnbindSnapshotpolicy(ctx context.Context, userCred mcclient.TokenCredential,
query jsonutils.JSONObject) bool {
return disk.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, disk, "unbind-snapshotpolicy")
}
func (disk *SDisk) PerformUnbindSnapshotpolicy(ctx context.Context, userCred mcclient.TokenCredential,
query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
spIden, err := data.GetString("snapshotpolicy")
if err != nil {
return nil, httperrors.NewMissingParameterError("miss snapshotpolicy")
}
// check snapshotpolicy
imodel, err := db.FetchByIdOrName(SnapshotPolicyManager, userCred, spIden)
if errors.Cause(err) == sql.ErrNoRows {
return nil, httperrors.NewInputParameterError("no such snapshotpolicy %s", spIden)
}
if err != nil {
return nil, errors.Wrap(err, "db.FetchByIdOrName")
}
snapshotpolicy := imodel.(*SSnapshotPolicy)
spd, err := SnapshotPolicyDiskManager.FetchBySnapshotPolicyDisk(snapshotpolicy.GetId(), disk.GetId())
if err != nil {
return nil, errors.Wrap(err, "SnapshotPolicyDiskManager.FetchBySnapshotPolicyDisk")
}
if spd == nil {
// has been detach
return nil, nil
}
// start up SnapshotPolicyCancelTask
taskdata := jsonutils.NewDict()
taskdata.Add(jsonutils.NewString(snapshotpolicy.Id), "snapshot_policy_id")
taskdata.Add(jsonutils.Marshal(spd), "snapshotPolicyDisk")
if task, err := taskman.TaskManager.NewTask(ctx, "SnapshotPolicyCancelTask", disk, userCred, nil, "", "",
nil); err != nil {
return nil, errors.Wrap(err, "fail to start up SnapshotPolicyCancelTask")
} else {
spd.SetStatus(userCred, api.SNAPSHOT_POLICY_DISK_DELETING, "")
task.ScheduleRun(taskdata)
}
return nil, nil
}
pkg/compute/models/snapshotpolicy.go
+2 -2
View File
@@ -287,7 +287,7 @@ func (sp *SSnapshotPolicy) DetachAfterDelete(ctx context.Context, userCred mccli
}
func (sp *SSnapshotPolicy) CustomizeDelete(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.
JSONObject, data jsonutils.JSONObject) error {
JSONObject, data jsonutils.JSONObject) error {
// check if sp bind to some disks
sds, err := SnapshotPolicyDiskManager.FetchAllBySnapshotpolicyID(ctx, userCred, sp.GetId())
@@ -674,7 +674,7 @@ func (manager *SSnapshotPolicyManager) sSnapshotPolicyCreateInputToInternal(inpu
}
func (manager *SSnapshotPolicyManager) sSnapshotPolicyCreateInputFromInternal(input *api.
SSnapshotPolicyCreateInternalInput) *api.SSnapshotPolicyCreateInput {
SSnapshotPolicyCreateInternalInput) *api.SSnapshotPolicyCreateInput {
return nil
}
pkg/compute/models/snapshotpolicydisks.go
+1 -1
View File
@@ -367,7 +367,7 @@ func (sd *SSnapshotPolicyDisk) DetachBySnapshotpolicy(ctx context.Context, userC
// ==================================================== create =========================================================
var ErrExistSD = fmt.Errorf("snapshotpolicy disk has been exist")
const ErrExistSD = errors.Error("snapshotpolicy disk has been exist")
func (self *SSnapshotPolicyDiskManager) newSnapshotpolicyDisk(ctx context.Context, userCred mcclient.TokenCredential,
sp *SSnapshotPolicy, disk *SDisk) (*SSnapshotPolicyDisk, error) {