fix/as: Set ScalingPolicy's project as its ScalingGroup's project

yunionio · Apr 28, 2020 · 352e912 · 352e912
1 parent be5c539
commit 352e912
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/compute/models/scaling_policy.go b/pkg/compute/models/scaling_policy.go
@@ -239,6 +239,17 @@ func (spm *SScalingPolicyManager) ValidateCreateData(ctx context.Context, userCr
 	return input, err
 }
 
+func (sp *SScalingPolicy) CustomizeCreate(ctx context.Context, userCred mcclient.TokenCredential,
+	ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) error {
+	// sp.Project must be same with sp.ScalingGroup
+	sg, err := sp.ScalingGroup()
+	if err != nil {
+		return err
+	}
+	ownerId = sg.GetOwnerId()
+	return sp.SVirtualResourceBase.CustomizeCreate(ctx, userCred, ownerId, query, data)
+}
+
 func (sp *SScalingPolicy) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
 	// do nothing
 	sp.SetStatus(userCred, api.SP_STATUS_DELETING, "")