fix(chunk): handle incomplete file chunks

Author: j2rong4cn

drivers/chunk/driver.go

@@ -84,7 +84,7 @@ func (d *Chunk) Get(ctx context.Context, path string) (model.Obj, error) {
 	// 0号块默认为-1 以支持空文件
 	chunkSizes := []int64{-1}
 	h := make(map[*utils.HashType]string)
-	var last model.Obj
+	var first model.Obj
 	for _, o := range chunkObjs {
 		if o.IsDir() {
 			continue
@@ -103,9 +103,11 @@ func (d *Chunk) Get(ctx context.Context, path string) (model.Obj, error) {
 		if err != nil {
 			continue
 		}
-		last = o
 		totalSize += o.GetSize()
 		if len(chunkSizes) > idx {
+			if idx == 0 {
+				first = o
+			}
 			chunkSizes[idx] = o.GetSize()
 		} else if len(chunkSizes) == idx {
 			chunkSizes = append(chunkSizes, o.GetSize())
@@ -117,21 +119,32 @@ func (d *Chunk) Get(ctx context.Context, path string) (model.Obj, error) {
 		}
 	}
 	reqDir, _ := stdpath.Split(path)
-	if last == nil {
+	// 文件块不完整时，返回文件夹对象
+	if chunkSizes[0] == -1 {
 		return &model.Object{
 			Path:     stdpath.Join(reqDir, chunkName),
 			Name:     chunkName,
 			IsFolder: true,
 			Modified: d.Modified,
 		}, nil
 	}
+	for i, l := 1, len(chunkSizes)-1; i < l; i++ {
+		if chunkSizes[i] == 0 {
+			return &model.Object{
+				Path:     stdpath.Join(reqDir, chunkName),
+				Name:     chunkName,
+				IsFolder: true,
+				Modified: d.Modified,
+			}, nil
+		}
+	}
 	objRes := chunkObject{
 		Object: model.Object{
 			Path:     stdpath.Join(reqDir, chunkName),
 			Name:     name,
 			Size:     totalSize,
-			Modified: last.ModTime(),
-			Ctime:    last.CreateTime(),
+			Modified: first.ModTime(),
+			Ctime:    first.CreateTime(),
 		},
 		chunkSizes: chunkSizes,
 	}
@@ -167,37 +180,51 @@ func (d *Chunk) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([
 				result = append(result, nil)
 				listG.Go(func(ctx context.Context) error {
 					chunkObjs, err := op.List(ctx, remoteStorage, stdpath.Join(remoteActualDir, rawName), model.ListArgs{
-						ReqPath: stdpath.Join(args.ReqPath, rawName),
 						Refresh: args.Refresh,
 					})
 					if err != nil {
 						return err
 					}
-					totalSize := int64(0)
+					var totalSize int64
+					// 0号块默认为-1 以支持空文件
+					chunkSizes := []int64{-1}
 					h := make(map[*utils.HashType]string)
-					var last model.Obj
+					var first model.Obj
 					for _, o := range chunkObjs {
 						if o.IsDir() {
 							continue
 						}
-						if after, ok := strings.CutPrefix(strings.TrimSuffix(o.GetName(), d.CustomExt), "hash_"); ok {
-							hn, value, ok := strings.Cut(after, "_")
+						if after, ok := strings.CutPrefix(o.GetName(), "hash_"); ok {
+							hn, value, ok := strings.Cut(strings.TrimSuffix(after, d.CustomExt), "_")
 							if ok {
 								ht, ok := utils.GetHashByName(hn)
 								if ok {
 									h[ht] = value
 								}
-								continue
 							}
+							continue
 						}
-						_, err := strconv.Atoi(strings.TrimSuffix(o.GetName(), d.CustomExt))
+						idx, err := strconv.Atoi(strings.TrimSuffix(o.GetName(), d.CustomExt))
 						if err != nil {
 							continue
 						}
-						last = o
 						totalSize += o.GetSize()
+						if len(chunkSizes) > idx {
+							if idx == 0 {
+								first = o
+							}
+							chunkSizes[idx] = o.GetSize()
+						} else if len(chunkSizes) == idx {
+							chunkSizes = append(chunkSizes, o.GetSize())
+						} else {
+							newChunkSizes := make([]int64, idx+1)
+							copy(newChunkSizes, chunkSizes)
+							chunkSizes = newChunkSizes
+							chunkSizes[idx] = o.GetSize()
+						}
 					}
-					if last == nil {
+					// 文件块不完整时，返回文件夹对象
+					if chunkSizes[0] == -1 {
 						result[resultIdx] = &model.Object{
 							Name:     rawName,
 							Size:     obj.GetSize(),
@@ -206,11 +233,22 @@ func (d *Chunk) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([
 						}
 						return nil
 					}
+					for i, l := 1, len(chunkSizes)-1; i < l; i++ {
+						if chunkSizes[i] == 0 {
+							result[resultIdx] = &model.Object{
+								Name:     rawName,
+								Size:     obj.GetSize(),
+								Modified: obj.ModTime(),
+								IsFolder: true,
+							}
+							return nil
+						}
+					}
 					objRes := model.Object{
 						Name:     name,
 						Size:     totalSize,
-						Modified: last.ModTime(),
-						Ctime:    last.CreateTime(),
+						Modified: first.ModTime(),
+						Ctime:    first.CreateTime(),
 					}
 					if len(h) > 0 {
 						objRes.HashInfo = utils.NewHashInfoByMap(h)