From 0446859f0ec73920d5b6879ebc5433b89356b4c0 Mon Sep 17 00:00:00 2001
From: Qu Xuan <quxuan@yunionyun.com>
Date: Thu, 18 Mar 2021 16:07:51 +0800
Subject: [PATCH] fix: nat sku env

---
 pkg/apis/compute/nat_skus.go   | 3 +++
 pkg/compute/models/nat_skus.go | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/pkg/apis/compute/nat_skus.go b/pkg/apis/compute/nat_skus.go
index d2687066ed..8c977de005 100644
--- a/pkg/apis/compute/nat_skus.go
+++ b/pkg/apis/compute/nat_skus.go
@@ -42,4 +42,7 @@ type NatSkuDetails struct {
 	apis.EnabledStatusStandaloneResourceDetails
 
 	CloudregionResourceInfo
+
+	// 云环境
+	CloudEnv string `json:"cloud_env"`
 }
diff --git a/pkg/compute/models/nat_skus.go b/pkg/compute/models/nat_skus.go
index 991238f0f7..32faca2a85 100644
--- a/pkg/compute/models/nat_skus.go
+++ b/pkg/compute/models/nat_skus.go
@@ -18,6 +18,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"strings"
 
 	"yunion.io/x/jsonutils"
 	"yunion.io/x/log"
@@ -120,6 +121,8 @@ func (manager *SNatSkuManager) FetchCustomizeColumns(
 			EnabledStatusStandaloneResourceDetails: stdRows[i],
 			CloudregionResourceInfo:                regRows[i],
 		}
+
+		rows[i].CloudEnv = strings.Split(regRows[i].RegionExternalId, "/")[0]
 	}
 
 	return rows