feat: add region api

Signed-off-by: jwcesign <jwcesign@cloudpilot.ai>
cloudpilot-ai · Oct 9, 2024 · 2b52935 · 2b52935
1 parent 93dab41
commit 2b52935
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/hack/tools/demo/client-demo.go b/hack/tools/demo/client-demo.go
@@ -7,11 +7,13 @@ import (
 )
 
 func main() {
-	c, err := tools.NewQueryClient("https://pre-price.cloudpilot.ai", "alibabacloud", "cn-beijing")
+	c, err := tools.NewQueryClient("https://pre-price.cloudpilot.ai", "alibabacloud", "")
 	if err != nil {
 		panic(err)
 	}
 
+	regions := c.ListRegions()
+	fmt.Println(regions)
 	d := c.ListInstancesDetails("cn-beijing")
 	fmt.Println(d)
 }
diff --git a/pkg/tools/queryclient.go b/pkg/tools/queryclient.go
@@ -17,6 +17,7 @@ import (
 
 type QueryClientInterface interface {
 	Run(ctx context.Context)
+	ListRegions() []string
 	ListInstancesDetails(region string) *apis.RegionalInstancePrice
 	GetInstanceDetails(region, instanceType string) *apis.InstanceTypePrice
 	TriggerRefreshData()
@@ -163,6 +164,17 @@ func (q *QueryClientImpl) refreshData() error {
 	return nil
 }
 
+func (q *QueryClientImpl) ListRegions() []string {
+	q.awsMutex.Lock()
+	defer q.awsMutex.Unlock()
+
+	var ret []string
+	for region := range q.priceData {
+		ret = append(ret, region)
+	}
+	return ret
+}
+
 func (q *QueryClientImpl) ListInstancesDetails(region string) *apis.RegionalInstancePrice {
 	q.awsMutex.Lock()
 	defer q.awsMutex.Unlock()