Merge pull request #373 from 0xff-dev/main

fix: delete model files when model got deleted
kubeagi · Dec 14, 2023 · b9bed4d · b9bed4d
2 parents 5f5cb2d + 7c54031
commit b9bed4d
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pkg/datasource/oss.go b/pkg/datasource/oss.go
@@ -138,12 +138,22 @@ func (oss *OSS) Stat(ctx context.Context, info any) error {
 	return oss.statObject(ctx, ossInfo)
 }
 
-// TODO: implement `Remove` against info
 func (oss *OSS) Remove(ctx context.Context, info any) error {
 	ossInfo, err := oss.preCheck(info)
 	if err != nil {
 		return err
 	}
+	if strings.HasSuffix(ossInfo.Object, "/") {
+		var resultErr error
+		for e := range oss.Client.RemoveObjects(
+			ctx,
+			ossInfo.Bucket,
+			oss.Client.ListObjects(ctx, ossInfo.Bucket, minio.ListObjectsOptions{Prefix: ossInfo.Object, Recursive: true}),
+			minio.RemoveObjectsOptions{}) {
+			resultErr = e.Err
+		}
+		return resultErr
+	}
 
 	return oss.Client.RemoveObject(ctx, ossInfo.Bucket, ossInfo.Object, minio.RemoveObjectOptions{ForceDelete: true})
 }