Merge pull request #249 from 0xff-dev/dataset-graphql

feat: dataset graphql api
kubeagi · Nov 21, 2023 · bfa39c9 · bfa39c9
2 parents c5a6da3 + 89baee6
commit bfa39c9
Show file tree

Hide file tree

Showing 13 changed files with 9,295 additions and 3,295 deletions.
diff --git a/api/v1alpha1/versioneddataset_types.go b/api/v1alpha1/versioneddataset_types.go
@@ -48,6 +48,9 @@ type VersionedDatasetSpec struct {
 
 	// Which version of the dataset it inherits from
 	InheritedFrom string `json:"inheritedFrom,omitempty"`
+
+	// DisplayName
+	CommonSpec `json:",inline"`
 }
 
 type FileStatus struct {

diff --git a/api/v1alpha1/zz_generated.deepcopy.go b/api/v1alpha1/zz_generated.deepcopy.go
diff --git a/config/crd/bases/arcadia.kubeagi.k8s.com.cn_versioneddatasets.yaml b/config/crd/bases/arcadia.kubeagi.k8s.com.cn_versioneddatasets.yaml
@@ -42,6 +42,9 @@ spec:
           spec:
             description: VersionedDatasetSpec defines the desired state of VersionedDataset
             properties:
+              creator:
+                description: Creator defines datasource creator (AUTO-FILLED by webhook)
+                type: string
               dataset:
                 description: Dataset which this `VersionedDataset` belongs to
                 properties:
@@ -64,6 +67,12 @@ spec:
                 - kind
                 - name
                 type: object
+              description:
+                description: Description defines datasource description
+                type: string
+              displayName:
+                description: DisplayName defines datasource display name
+                type: string
               fileGroups:
                 description: FileGroups included in this `VersionedDataset` Grouped
                   by Datasource

diff --git a/deploy/charts/arcadia/crds/arcadia.kubeagi.k8s.com.cn_versioneddatasets.yaml b/deploy/charts/arcadia/crds/arcadia.kubeagi.k8s.com.cn_versioneddatasets.yaml
@@ -42,6 +42,9 @@ spec:
           spec:
             description: VersionedDatasetSpec defines the desired state of VersionedDataset
             properties:
+              creator:
+                description: Creator defines datasource creator (AUTO-FILLED by webhook)
+                type: string
               dataset:
                 description: Dataset which this `VersionedDataset` belongs to
                 properties:
@@ -64,6 +67,12 @@ spec:
                 - kind
                 - name
                 type: object
+              description:
+                description: Description defines datasource description
+                type: string
+              displayName:
+                description: DisplayName defines datasource display name
+                type: string
               fileGroups:
                 description: FileGroups included in this `VersionedDataset` Grouped
                   by Datasource

diff --git a/gqlgen.yaml b/gqlgen.yaml
@@ -141,3 +141,39 @@ models:
         resolver: true
       listKnowledgeBases:
         resolver: true
+  DatasetQuery:
+    fields:
+      getDataset:
+        resolver: true
+      listDatasets:
+        resolver: true
+  DatasetMutation:
+    fields:
+      createDataset:
+        resolver: true
+      updateDataset:
+        resolver: true
+      deleteDatasets:
+        resolver: true
+  Dataset:
+    fields:
+      versions:
+        resolver: true
+  VersionedDatasetQuery:
+    fields:
+      getVersionedDataset:
+        resolver: true
+      listVersionedDatasets:
+        resolver: true
+  VersionedDatasetMutation:
+    fields:
+      createVersionedDataset:
+        resolver: true
+      updateVersionedDataset:
+        resolver: true
+      deleteVersionedDatasets:
+        resolver: true
+  VersionedDataset:
+    fields:
+      files:
+        resolver: true