Migrate databricks_shares data source to sdk

databricks · Mar 6, 2023 · 506a8bb · 506a8bb
1 parent 29cea04
commit 506a8bb
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/catalog/data_shares.go b/catalog/data_shares.go
@@ -3,22 +3,21 @@ package catalog
 import (
 	"context"
 
+	"github.com/databricks/databricks-sdk-go"
 	"github.com/databricks/terraform-provider-databricks/common"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 )
 
 func DataSourceShares() *schema.Resource {
-	type sharesData struct {
+	return common.WorkspaceData(func(ctx context.Context, data *struct {
 		Shares []string `json:"shares,omitempty" tf:"computed,slice_set"`
-	}
-	return common.DataResource(sharesData{}, func(ctx context.Context, e any, c *common.DatabricksClient) error {
-		data := e.(*sharesData)
-		sharesAPI := NewSharesAPI(ctx, c)
-		shares, err := sharesAPI.list()
+	}, w *databricks.WorkspaceClient) error {
+
+		shares, err := w.Shares.ListAll(ctx)
 		if err != nil {
 			return err
 		}
-		for _, share := range shares.Shares {
+		for _, share := range shares {
 			data.Shares = append(data.Shares, share.Name)
 		}
 		return nil