fix(aliyun): support change vm billing type

yunionio · Feb 11, 2025 · 7a81c58 · 7a81c58
1 parent fc323d7
commit 7a81c58
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 0 deletions.
diff --git a/pkg/cloudprovider/resources.go b/pkg/cloudprovider/resources.go
@@ -63,6 +63,7 @@ type IBillingResource interface {
 	GetExpiredAt() time.Time
 	SetAutoRenew(bc billing.SBillingCycle) error
 	Renew(bc billing.SBillingCycle) error
+	ChangeBillingType(billType string) error
 	IsAutoRenew() bool
 }
 

diff --git a/pkg/multicloud/aliyun/instance.go b/pkg/multicloud/aliyun/instance.go
@@ -995,6 +995,30 @@ func (self *SInstance) GetBillingType() string {
 	return convertChargeType(self.InstanceChargeType)
 }
 
+func (self *SInstance) ChangeBillingType(billingType string) error {
+	return self.host.zone.region.ModifyInstanceChargeType(self.InstanceId, billingType)
+}
+
+func (region *SRegion) ModifyInstanceChargeType(vmId string, billingType string) error {
+	params := map[string]string{
+		"RegionId":         region.RegionId,
+		"IncludeDataDisks": "true",
+		"AutoPay":          "true",
+		"ClientToken":      utils.GenRequestId(20),
+		"InstanceIds":      jsonutils.Marshal([]string{vmId}).String(),
+	}
+	switch billingType {
+	case billing_api.BILLING_TYPE_POSTPAID:
+		params["InstanceChargeType"] = "PostPaid"
+	case billing_api.BILLING_TYPE_PREPAID:
+		params["InstanceChargeType"] = "PrePaid"
+	default:
+		return fmt.Errorf("invalid billing_type %s", billingType)
+	}
+	_, err := region.ecsRequest("ModifyInstanceChargeType", params)
+	return err
+}
+
 func (self *SInstance) GetCreatedAt() time.Time {
 	return self.CreationTime
 }

diff --git a/pkg/multicloud/aliyun/shell/instance.go b/pkg/multicloud/aliyun/shell/instance.go
@@ -224,4 +224,13 @@ func init() {
 		return nil
 	})
 
+	type InstanceChangeBillingType struct {
+		ID          string
+		BillingType string `choices:"postpaid|prepaid" default:"prepaid"`
+	}
+
+	shellutils.R(&InstanceChangeBillingType{}, "instance-change-billing-type", "change instance billing type", func(cli *aliyun.SRegion, args *InstanceChangeBillingType) error {
+		return cli.ModifyInstanceChargeType(args.ID, args.BillingType)
+	})
+
 }
diff --git a/pkg/multicloud/billing_base.go b/pkg/multicloud/billing_base.go
@@ -44,3 +44,7 @@ func (self *SBillingBase) IsAutoRenew() bool {
 func (self *SBillingBase) Renew(bc billing.SBillingCycle) error {
 	return errors.Wrap(cloudprovider.ErrNotImplemented, "Renew")
 }
+
+func (self *SBillingBase) ChangeBillingType(billType string) error {
+	return errors.Wrap(cloudprovider.ErrNotImplemented, "ChangeBillingType")
+}
diff --git a/pkg/multicloud/huawei/modelarts_pool.go b/pkg/multicloud/huawei/modelarts_pool.go
@@ -34,6 +34,7 @@ import (
 type SModelartsPool struct {
 	region *SRegion
 	multicloud.SResourceBase
+	multicloud.SBillingBase
 
 	Metadata     SModelartsPoolMetadata `json:"metadata"`
 	Spec         SModelartsPoolSpec     `json:"spec"`