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

clean up

This commit is contained in:
Denis Bilenko 2025-03-03 16:12:32 +01:00
parent 965b250d89
commit 8bcf9ac2fe
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/patchwheel/patchwheel.go
View File

@ -42,19 +42,22 @@ func readMetadataAndRecord(r *zip.ReadCloser) (metadataFile, recordFile *zip.Fil
return metadataFile, recordFile, oldDistInfoPrefix, nil return metadataFile, recordFile, oldDistInfoPrefix, nil
} }
const versionKey = "Version:"
// parseMetadata scans the METADATA content for the "Version:" and "Name:" fields. // parseMetadata scans the METADATA content for the "Version:" and "Name:" fields.
func parseMetadata(content []byte) (version, distribution string, err error) { func parseMetadata(content []byte) (version, distribution string, err error) {
scanner := bufio.NewScanner(bytes.NewReader(content)) scanner := bufio.NewScanner(bytes.NewReader(content))
for scanner.Scan() { for scanner.Scan() {
line := scanner.Text() line := scanner.Text()
if strings.HasPrefix(line, "Version:") { if strings.HasPrefix(line, versionKey) {
v := strings.TrimSpace(strings.TrimPrefix(line, "Version:")) v := strings.TrimSpace(strings.TrimPrefix(line, versionKey))
// If there's a '+' in the version, strip it off. // If there's a '+' in the version, strip it off.
if strings.Contains(v, "+") { if strings.Contains(v, "+") {
v = strings.SplitN(v, "+", 2)[0] v = strings.SplitN(v, "+", 2)[0]
} }
version = v version = v
} else if strings.HasPrefix(line, "Name:") { } else if strings.HasPrefix(line, "Name:") {
// AI TODO: apply similar approach as for Version:
parts := strings.SplitN(line, ":", 2) parts := strings.SplitN(line, ":", 2)
if len(parts) == 2 { if len(parts) == 2 {
distribution = strings.TrimSpace(parts[1]) distribution = strings.TrimSpace(parts[1])