steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

two functions

This commit is contained in:
Shreyas Goenka 2025-01-07 15:54:33 +05:30
parent fcc47de4fe
commit 4d09201588
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
1 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
bundle/config/mutator/capture_schema_dependency.go
View File

@ -51,33 +51,43 @@ func resolveVolume(v *resources.Volume, b *bundle.Bundle) {
v.SchemaName = fmt.Sprintf("${resources.schemas.%s.name}", schemaK) v.SchemaName = fmt.Sprintf("${resources.schemas.%s.name}", schemaK)
} }
func resolvePipeline(p *resources.Pipeline, b *bundle.Bundle) { func resolvePipelineTarget(p *resources.Pipeline, b *bundle.Bundle) {
if p.PipelineSpec == nil { if p.PipelineSpec == nil {
return return
} }
// schema and target have the same semantics in the DLT API but are mutually if p.Schema == "" {
// exclusive. If schema is set, the pipeline is in direct publishing mode return
// and can write tables to multiple schemas (vs target which is limited to a single schema).
schemaName := p.Schema
if schemaName == "" {
schemaName = p.Target
} }
schemaK, schema := findSchema(b, p.Catalog, p.Schema)
schemaK, schema := findSchema(b, p.Catalog, schemaName)
if schema == nil { if schema == nil {
return return
} }
if p.Schema != "" { p.Schema = fmt.Sprintf("${resources.schemas.%s.name}", schemaK)
p.Schema = fmt.Sprintf("${resources.schemas.%s.name}", schemaK) }
} else if p.Target != "" {
p.Target = fmt.Sprintf("${resources.schemas.%s.name}", schemaK) func resolvePipelineSchema(p *resources.Pipeline, b *bundle.Bundle) {
if p.PipelineSpec == nil {
return
} }
if p.Target == "" {
return
}
schemaK, schema := findSchema(b, p.Catalog, p.Target)
if schema == nil {
return
}
p.Target = fmt.Sprintf("${resources.schemas.%s.name}", schemaK)
} }
func (m *captureSchemaDependency) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics { func (m *captureSchemaDependency) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
for _, p := range b.Config.Resources.Pipelines { for _, p := range b.Config.Resources.Pipelines {
resolvePipeline(p, b) // "schema" and "target" have the same semantics in the DLT API but are mutually
// exclusive i.e. only one can be set at a time. If schema is set, the pipeline
// is in direct publishing mode and can write tables to multiple schemas
// (vs target which is limited to a single schema).
resolvePipelineTarget(p, b)
resolvePipelineSchema(p, b)
} }
for _, v := range b.Config.Resources.Volumes { for _, v := range b.Config.Resources.Volumes {
resolveVolume(v, b) resolveVolume(v, b)