Browse Source

fix bug

tags/v1.21.12.1
lewis 4 years ago
parent
commit
7e5688866b
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      modules/storage/obs.go

+ 5
- 0
modules/storage/obs.go View File

@@ -178,6 +178,7 @@ func GetObsListObject(jobName, parentDir string) ([]FileInfo, error) {
input := &obs.ListObjectsInput{}
input.Bucket = setting.Bucket
input.Prefix = strings.TrimPrefix(path.Join(setting.TrainJobModelPath, jobName, setting.OutPutPath, parentDir), "/")
strPrefix := strings.Split(input.Prefix, "/")
output, err := ObsCli.ListObjects(input)
fileInfos := make([]FileInfo, 0)
if err == nil {
@@ -193,6 +194,10 @@ func GetObsListObject(jobName, parentDir string) ([]FileInfo, error) {
continue
}
} else {
//files in next level dir
if len(str1) - len(strPrefix) > 1 {
continue
}
fileName = str1[len(str1)-1]
isDir = false
}


Loading…
Cancel
Save