feat(workspace): use custom url types

microsoft · Nov 12, 2024 · 7f2e117 · 7f2e117
1 parent 81f4780
commit 7f2e117
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/internal/services/workspace/data_workspace.go b/internal/services/workspace/data_workspace.go
@@ -92,10 +92,12 @@ func (d *dataSourceWorkspace) Schema(ctx context.Context, _ datasource.SchemaReq
 					"blob_endpoint": schema.StringAttribute{
 						MarkdownDescription: "The OneLake API endpoint available for Blob API operations.",
 						Computed:            true,
+						CustomType:          customtypes.URLType{},
 					},
 					"dfs_endpoint": schema.StringAttribute{
 						MarkdownDescription: "The OneLake API endpoint available for Distributed File System (DFS) or ADLSgen2 filesystem API operations.",
 						Computed:            true,
+						CustomType:          customtypes.URLType{},
 					},
 				},
 			},

diff --git a/internal/services/workspace/models.go b/internal/services/workspace/models.go
@@ -83,13 +83,13 @@ func (to *workspaceIdentityModel) set(from *fabcore.WorkspaceIdentity) {
 }
 
 type oneLakeEndpointsModel struct {
-	BlobEndpoint types.String `tfsdk:"blob_endpoint"`
-	DfsEndpoint  types.String `tfsdk:"dfs_endpoint"`
+	BlobEndpoint customtypes.URL `tfsdk:"blob_endpoint"`
+	DfsEndpoint  customtypes.URL `tfsdk:"dfs_endpoint"`
 }
 
 func (to *oneLakeEndpointsModel) set(from *fabcore.OneLakeEndpoints) {
-	to.BlobEndpoint = types.StringPointerValue(from.BlobEndpoint)
-	to.DfsEndpoint = types.StringPointerValue(from.DfsEndpoint)
+	to.BlobEndpoint = customtypes.NewURLPointerValue(from.BlobEndpoint)
+	to.DfsEndpoint = customtypes.NewURLPointerValue(from.DfsEndpoint)
 }
 
 func checkWorkspaceType(entity fabcore.WorkspaceInfo) diag.Diagnostics {

diff --git a/internal/services/workspace/resource_workspace.go b/internal/services/workspace/resource_workspace.go
@@ -111,10 +111,12 @@ func (r *resourceWorkspace) Schema(ctx context.Context, _ resource.SchemaRequest
 					"blob_endpoint": schema.StringAttribute{
 						MarkdownDescription: "The OneLake API endpoint available for Blob API operations.",
 						Computed:            true,
+						CustomType:          customtypes.URLType{},
 					},
 					"dfs_endpoint": schema.StringAttribute{
 						MarkdownDescription: "The OneLake API endpoint available for Distributed File System (DFS) or ADLSgen2 filesystem API operations.",
 						Computed:            true,
+						CustomType:          customtypes.URLType{},
 					},
 				},
 			},