feat: add totalCount field to DatasetSchema type of GraphQL schema (#154

) * feat: impl totalCount resolver to count dataset * chore: run `go generate` * Update internal/infrastructure/memory/dataset.go Co-authored-by: rot1024 <aayhrot@gmail.com> * Update internal/infrastructure/mongo/dataset.go Co-authored-by: rot1024 <aayhrot@gmail.com> Co-authored-by: rot1024 <aayhrot@gmail.com>
reearth · Jun 3, 2022 · ab6334a · ab6334a
1 parent e444e4e
commit ab6334a
Show file tree

Hide file tree

Showing 11 changed files with 164 additions and 0 deletions.
diff --git a/gqlgen.yml b/gqlgen.yml
@@ -27,3 +27,7 @@ models:
     model: github.com/reearth/reearth-backend/internal/adapter/gql/gqlmodel.Lang
   ID:
     model: github.com/reearth/reearth-backend/internal/adapter/gql/gqlmodel.ID
+  DatasetSchema:
+    fields:
+      totalCount:
+        resolver: true
diff --git a/internal/adapter/gql/generated.go b/internal/adapter/gql/generated.go
diff --git a/internal/adapter/gql/gqlmodel/models_gen.go b/internal/adapter/gql/gqlmodel/models_gen.go
diff --git a/internal/adapter/gql/loader_dataset.go b/internal/adapter/gql/loader_dataset.go
@@ -177,6 +177,20 @@ func (c *DatasetLoader) FindBySchema(ctx context.Context, dsid gqlmodel.ID, firs
 	return conn, nil
 }
 
+func (c *DatasetLoader) CountBySchema(ctx context.Context, dsid gqlmodel.ID) (int, error) {
+	id, err := gqlmodel.ToID[id.DatasetSchema](dsid)
+	if err != nil {
+		return 0, err
+	}
+
+	cnt, err := c.usecase.CountBySchema(ctx, id)
+	if err != nil {
+		return 0, err
+	}
+
+	return cnt, nil
+}
+
 // data loader
 
 type DatasetDataLoader interface {

diff --git a/internal/adapter/gql/resolver_dataset_schema.go b/internal/adapter/gql/resolver_dataset_schema.go
@@ -38,6 +38,10 @@ func (r *datasetSchemaResolver) Datasets(ctx context.Context, obj *gqlmodel.Data
 	return loaders(ctx).Dataset.FindBySchema(ctx, obj.ID, first, last, before, after)
 }
 
+func (r *datasetSchemaResolver) TotalCount(ctx context.Context, obj *gqlmodel.DatasetSchema) (int, error) {
+	return loaders(ctx).Dataset.CountBySchema(ctx, obj.ID)
+}
+
 type datasetSchemaFieldResolver struct{ *Resolver }
 
 func (r *datasetSchemaFieldResolver) Schema(ctx context.Context, obj *gqlmodel.DatasetSchemaField) (*gqlmodel.DatasetSchema, error) {

diff --git a/internal/infrastructure/memory/dataset.go b/internal/infrastructure/memory/dataset.go
@@ -87,6 +87,21 @@ func (r *Dataset) FindBySchema(ctx context.Context, id id.DatasetSchemaID, p *us
 	), nil
 }
 
+func (r *Dataset) CountBySchema(ctx context.Context, id id.DatasetSchemaID) (int, error) {
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	n := 0
+	for _, dataset := range r.data {
+		if dataset.Schema() == id {
+			if r.f.CanRead(dataset.Scene()) {
+				n++
+			}
+		}
+	}
+	return n, nil
+}
+
 func (r *Dataset) FindBySchemaAll(ctx context.Context, id id.DatasetSchemaID) (dataset.List, error) {
 	r.lock.Lock()
 	defer r.lock.Unlock()

diff --git a/internal/infrastructure/mongo/dataset.go b/internal/infrastructure/mongo/dataset.go
@@ -69,6 +69,16 @@ func (r *datasetRepo) FindBySchema(ctx context.Context, schemaID id.DatasetSchem
 	}, pagination)
 }
 
+func (r *datasetRepo) CountBySchema(ctx context.Context, id id.DatasetSchemaID) (int, error) {
+	res, err := r.client.Count(ctx, r.readFilter(bson.M{
+		"schema": id.String(),
+	}))
+	if err != nil {
+		return 0, err
+	}
+	return int(res), nil
+}
+
 func (r *datasetRepo) FindBySchemaAll(ctx context.Context, schemaID id.DatasetSchemaID) (dataset.List, error) {
 	return r.find(ctx, nil, bson.M{
 		"schema": schemaID.String(),

diff --git a/internal/usecase/interactor/dataset.go b/internal/usecase/interactor/dataset.go
@@ -405,6 +405,10 @@ func (i *Dataset) FindBySchema(ctx context.Context, ds id.DatasetSchemaID, p *us
 	return i.datasetRepo.FindBySchema(ctx, ds, p)
 }
 
+func (i *Dataset) CountBySchema(ctx context.Context, id id.DatasetSchemaID) (int, error) {
+	return i.datasetRepo.CountBySchema(ctx, id)
+}
+
 func (i *Dataset) FindSchemaByScene(ctx context.Context, sid id.SceneID, p *usecase.Pagination, operator *usecase.Operator) (dataset.SchemaList, *usecase.PageInfo, error) {
 	if err := i.CanReadScene(sid, operator); err != nil {
 		return nil, nil, err

diff --git a/internal/usecase/interfaces/dataset.go b/internal/usecase/interfaces/dataset.go
@@ -69,6 +69,7 @@ type Dataset interface {
 	AddDynamicDatasetSchema(context.Context, AddDynamicDatasetSchemaParam) (*dataset.Schema, error)
 	AddDynamicDataset(context.Context, AddDynamicDatasetParam) (*dataset.Schema, *dataset.Dataset, error)
 	FindBySchema(context.Context, id.DatasetSchemaID, *usecase.Pagination, *usecase.Operator) (dataset.List, *usecase.PageInfo, error)
+	CountBySchema(context.Context, id.DatasetSchemaID) (int, error)
 	FindSchemaByScene(context.Context, id.SceneID, *usecase.Pagination, *usecase.Operator) (dataset.SchemaList, *usecase.PageInfo, error)
 	FindDynamicSchemaByScene(context.Context, id.SceneID) (dataset.SchemaList, error)
 	RemoveDatasetSchema(context.Context, RemoveDatasetSchemaParam, *usecase.Operator) (id.DatasetSchemaID, error)

diff --git a/internal/usecase/repo/dataset.go b/internal/usecase/repo/dataset.go
@@ -13,6 +13,7 @@ type Dataset interface {
 	FindByID(context.Context, id.DatasetID) (*dataset.Dataset, error)
 	FindByIDs(context.Context, id.DatasetIDList) (dataset.List, error)
 	FindBySchema(context.Context, id.DatasetSchemaID, *usecase.Pagination) (dataset.List, *usecase.PageInfo, error)
+	CountBySchema(context.Context, id.DatasetSchemaID) (int, error)
 	FindBySchemaAll(context.Context, id.DatasetSchemaID) (dataset.List, error)
 	FindGraph(context.Context, id.DatasetID, id.DatasetFieldIDList) (dataset.List, error)
 	Save(context.Context, *dataset.Dataset) error

diff --git a/schema.graphql b/schema.graphql
@@ -598,6 +598,7 @@ type DatasetSchema implements Node {
   name: String!
   sceneId: ID!
   fields: [DatasetSchemaField!]!
+  totalCount: Int!
   representativeFieldId: ID
   dynamic: Boolean
   datasets(