mirror of https://github.com/databricks/cli.git
45 lines
1.7 KiB
Go
45 lines
1.7 KiB
Go
package libraries
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/dyn"
|
|
"github.com/databricks/cli/libs/dyn/dynvar"
|
|
"github.com/databricks/cli/libs/filer"
|
|
)
|
|
|
|
func filerForVolume(ctx context.Context, b *bundle.Bundle) (filer.Filer, string, diag.Diagnostics) {
|
|
w := b.WorkspaceClient()
|
|
uploadPath := path.Join(b.Config.Workspace.ArtifactPath, InternalDirName)
|
|
f, err := filer.NewFilesClient(w, uploadPath)
|
|
return f, uploadPath, diag.FromErr(err)
|
|
}
|
|
|
|
func FindVolumeInBundle(r config.Root, catalogName, schemaName, volumeName string) (dyn.Path, []dyn.Location, bool) {
|
|
volumes := r.Resources.Volumes
|
|
for k, v := range volumes {
|
|
if v.CatalogName != catalogName || v.Name != volumeName {
|
|
continue
|
|
}
|
|
// UC schemas can be defined in the bundle itself, and thus might be interpolated
|
|
// at runtime via the ${resources.schemas.<name>} syntax. Thus we match the volume
|
|
// definition if the schema name is the same as the one in the bundle, or if the
|
|
// schema name is interpolated.
|
|
// We only have to check for ${resources.schemas...} references because any
|
|
// other valid reference (like ${var.foo}) would have been interpolated by this point.
|
|
p, ok := dynvar.PureReferenceToPath(v.SchemaName)
|
|
isSchemaDefinedInBundle := ok && p.HasPrefix(dyn.Path{dyn.Key("resources"), dyn.Key("schemas")})
|
|
if v.SchemaName != schemaName && !isSchemaDefinedInBundle {
|
|
continue
|
|
}
|
|
pathString := fmt.Sprintf("resources.volumes.%s", k)
|
|
return dyn.MustPathFromString(pathString), r.GetLocations(pathString), true
|
|
}
|
|
return nil, nil, false
|
|
}
|