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

lazy generate dirs

This commit is contained in:
Shreyas Goenka 2023-05-22 09:46:26 +02:00
parent 479e2b4e32
commit f6fc63a09e
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
2 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
libs/template/execute.go
View File

@ -27,53 +27,50 @@ func executeTemplate(config map[string]any, templateDefination string) (string,
}
// TODO: allow skipping directories
func generateDirectory(config map[string]any, parentDir, nameTempate string) (string, error) {
dirName, err := executeTemplate(config, nameTempate)
if err != nil {
return "", err
}
err = os.Mkdir(filepath.Join(parentDir, dirName), 0755)
if err != nil {
return "", err
}
return dirName, nil
}
func generateFile(config map[string]any, parentDir, nameTempate, contentTemplate string) error {
func generateFile(config map[string]any, pathTemplate, contentTemplate string) error {
// compute file content
fileContent, err := executeTemplate(config, contentTemplate)
if errors.Is(err, errSkipThisFile) {
// skip this file
return nil
}
if err != nil {
return err
}
// create the file by executing the templatized file name
fileName, err := executeTemplate(config, nameTempate)
// compute the path for this file
path, err := executeTemplate(config, pathTemplate)
if err != nil {
return err
}
// create any intermediate directories required. Directories are lazily generated
// only when they are required for a file.
err = os.MkdirAll(filepath.Dir(path), 0755)
if err != nil {
return err
}
return os.WriteFile(filepath.Join(parentDir, fileName), []byte(fileContent), 0644)
// write content to file
return os.WriteFile(path, []byte(fileContent), 0644)
}
func WalkFileTree(config map[string]any, templatePath, instancePath string) error {
func walkFileTree(config map[string]any, templatePath, instancePath string) error {
entries, err := os.ReadDir(templatePath)
if err != nil {
return err
}
for _, entry := range entries {
if entry.IsDir() {
// case: materialize a template directory
dirName, err := generateDirectory(config, instancePath, entry.Name())
// compute directory name
dirName, err := executeTemplate(config, entry.Name())
if err != nil {
return err
}
// recusive generate files and directories inside inside our newly generated
// recusively generate files and directories inside inside our newly generated
// directory from the template defination
err = WalkFileTree(config, filepath.Join(templatePath, entry.Name()), filepath.Join(instancePath, dirName))
err = walkFileTree(config, filepath.Join(templatePath, entry.Name()), filepath.Join(instancePath, dirName))
if err != nil {
return err
}
@ -85,7 +82,7 @@ func WalkFileTree(config map[string]any, templatePath, instancePath string) erro
}
contentTemplate := string(b)
fileNameTemplate := entry.Name()
err = generateFile(config, instancePath, fileNameTemplate, contentTemplate)
err = generateFile(config, filepath.Join(instancePath, fileNameTemplate), contentTemplate)
if err != nil {
return err
}

2
libs/template/materialize.go
View File

@ -22,5 +22,5 @@ func Materialize(templatePath, instancePath, configPath string) error {
}
// materialize the template
return WalkFileTree(config, filepath.Join(templatePath, templateDirName), instancePath)
return walkFileTree(config, filepath.Join(templatePath, templateDirName), instancePath)
}