Skip to content

Commit

Permalink
first comment
Browse files Browse the repository at this point in the history
  • Loading branch information
shreyas-goenka committed Sep 9, 2024
1 parent 02e8387 commit f772ce4
Show file tree
Hide file tree
Showing 3 changed files with 84 additions and 0 deletions.
1 change: 1 addition & 0 deletions bundle/config/resources.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ type Resources struct {
RegisteredModels map[string]*resources.RegisteredModel `json:"registered_models,omitempty"`
QualityMonitors map[string]*resources.QualityMonitor `json:"quality_monitors,omitempty"`
Schemas map[string]*resources.Schema `json:"schemas,omitempty"`
Volumes map[string]*resources.Volume `json:"volumes,omitempty"`
}

type ConfigResource interface {
Expand Down
27 changes: 27 additions & 0 deletions bundle/config/resources/volume.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
package resources

import (
"github.com/databricks/databricks-sdk-go/marshal"
"github.com/databricks/databricks-sdk-go/service/catalog"
)

type Volume struct {
// List of grants to apply on this schema.
Grants []Grant `json:"grants,omitempty"`

// Full name of the schema (catalog_name.schema_name.volume_name). This value is read from
// the terraform state after deployment succeeds.
ID string `json:"id,omitempty" bundle:"readonly"`

*catalog.CreateVolumeRequestContent

ModifiedStatus ModifiedStatus `json:"modified_status,omitempty" bundle:"internal"`
}

func (v *Volume) UnmarshalJSON(b []byte) error {
return marshal.Unmarshal(b, v)
}

func (v Volume) MarshalJSON() ([]byte, error) {
return marshal.Marshal(v)
}
56 changes: 56 additions & 0 deletions bundle/deploy/terraform/tfdyn/convert_volume.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
package tfdyn

import (
"context"
"fmt"

"github.com/databricks/cli/bundle/internal/tf/schema"
"github.com/databricks/cli/libs/dyn"
"github.com/databricks/cli/libs/dyn/convert"
"github.com/databricks/cli/libs/log"
)

func convertVolumeResource(ctx context.Context, vin dyn.Value) (dyn.Value, error) {
// Normalize the output value to the target schema.
v, diags := convert.Normalize(schema.ResourceVolume{}, vin)
for _, diag := range diags {
log.Debugf(ctx, "volume normalization diagnostic: %s", diag.Summary)
}

// TODO: What happens if I try to delete a UC volume that has data in it?
// Do I need force destroy functionality here.

// We always set force destroy as it allows DABs to manage the lifecycle
// of the schema. It's the responsibility of the CLI to ensure the user
// is adequately warned when they try to delete a UC Volume.
vout, err := dyn.SetByPath(v, dyn.MustPathFromString("force_destroy"), dyn.V(true))
if err != nil {
return dyn.InvalidValue, err
}

return vout, nil
}

type volumeConverter struct{}

func (volumeConverter) Convert(ctx context.Context, key string, vin dyn.Value, out *schema.Resources) error {
vout, err := convertVolumeResource(ctx, vin)
if err != nil {
return err
}

// Add the converted resource to the output.
out.Schema[key] = vout.AsAny()

// Configure grants for this resource.
if grants := convertGrantsResource(ctx, vin); grants != nil {
grants.Schema = fmt.Sprintf("${databricks_schema.%s.id}", key)
out.Grants["schema_"+key] = grants
}

return nil
}

func init() {
registerConverter("volumes", volumeConverter{})
}

0 comments on commit f772ce4

Please sign in to comment.