diff --git a/pkg/apis/compute/nat_skus.go b/pkg/apis/compute/nat_skus.go index 4354e95c56..d2687066ed 100644 --- a/pkg/apis/compute/nat_skus.go +++ b/pkg/apis/compute/nat_skus.go @@ -31,6 +31,11 @@ type NatSkuListInput struct { PostpaidStatus string `json:"postpaid_stauts"` PrepaidStatus string `json:"prepaid_status"` + + Providers []string `json:"providers"` + // swagger:ignore + // Deprecated + Provider []string `json:"provider" yunion-deprecated-by:"providers"` } type NatSkuDetails struct { diff --git a/pkg/compute/models/nat_skus.go b/pkg/compute/models/nat_skus.go index 883c1fb67d..991238f0f7 100644 --- a/pkg/compute/models/nat_skus.go +++ b/pkg/compute/models/nat_skus.go @@ -96,6 +96,9 @@ func (manager *SNatSkuManager) ListItemFilter( if len(query.PrepaidStatus) > 0 { q = q.Equals("prepaid_status", query.PrepaidStatus) } + if len(query.Providers) > 0 { + q = q.In("provider", query.Providers) + } return q, nil }