feat: versioneddataset return the versions field of the file

kubeagi · Mar 29, 2024 · 2eb96ca · 2eb96ca
1 parent fc3ffd9
commit 2eb96ca
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/apiserver/graph/schema/versioned_dataset.gql b/apiserver/graph/schema/versioned_dataset.gql
@@ -57,6 +57,7 @@ query getVersionedDataset($name: String!, $namespace: String!, $fileInput: FileF
             fileType
             count
             size
+            versions
             creationTimestamp
           }
         }
@@ -96,6 +97,7 @@ query listVersionedDatasets($input: ListVersionedDatasetInput!, $fileInput: File
                   fileType
                   count
                   size
+                  versions
                   creationTimestamp
               }
             }

diff --git a/apiserver/pkg/versioneddataset/versioned_dataset.go b/apiserver/pkg/versioneddataset/versioned_dataset.go
@@ -113,8 +113,9 @@ func VersionFiles(ctx context.Context, _ client.Client, input *generated.Version
 		return nil, err
 	}
 	anyObjectInfoList, err := oss.ListObjects(ctx, input.Namespace, miniogo.ListObjectsOptions{
-		Prefix:    prefix,
-		Recursive: true,
+		Prefix:       prefix,
+		Recursive:    true,
+		WithVersions: true,
 	})
 	if err != nil {
 		return nil, err
@@ -124,13 +125,27 @@ func VersionFiles(ctx context.Context, _ client.Client, input *generated.Version
 		return objectInfoList[i].LastModified.After(objectInfoList[j].LastModified)
 	})
 
+	existMap := make(map[string]struct{})
+	objMap := make(map[string][]string)
+	for _, obj := range objectInfoList {
+		objMap[obj.Key] = append(objMap[obj.Key], obj.VersionID)
+	}
+
 	result := make([]generated.PageNode, 0)
 	for _, obj := range objectInfoList {
+		if _, ok := existMap[obj.Key]; ok {
+			continue
+		}
+		if obj.IsDeleteMarker {
+			continue
+		}
 		if keyword == "" || strings.Contains(obj.Key, keyword) {
+			existMap[obj.Key] = struct{}{}
 			lastModifiedTime := obj.LastModified
 			tf := generated.F{
-				Path: strings.TrimPrefix(obj.Key, prefix),
-				Time: &lastModifiedTime,
+				Path:     strings.TrimPrefix(obj.Key, prefix),
+				Time:     &lastModifiedTime,
+				Versions: objMap[obj.Key],
 			}
 
 			size := utils.BytesToSizedStr(obj.Size)