fix(server): add wrappers to prevent index-dropping initialization code [VIZ-2228] (#118)

Author: wilfredmulenga

server/internal/infrastructure/mongo/container.go

@@ -14,7 +14,6 @@ import (
 	"github.com/reearth/reearth/server/pkg/property"
 	"github.com/reearth/reearth/server/pkg/scene"
 	"github.com/reearth/reearthx/account/accountdomain/user"
-	"github.com/reearth/reearthx/account/accountinfrastructure/accountmongo"
 	"github.com/reearth/reearthx/account/accountusecase/accountrepo"
 	"github.com/reearth/reearthx/authserver"
 	"github.com/reearth/reearthx/log"
@@ -53,14 +52,14 @@ func New(ctx context.Context, db *mongo.Database, account *accountrepo.Container
 		Scene:          NewScene(client),
 		Tag:            NewTag(client),
 		SceneLock:      NewSceneLock(client),
-		Permittable:    accountmongo.NewPermittable(client), // TODO: Delete this once the permission check migration is complete.
+		Permittable:    NewPermittableWrapper(client), // TODO: Delete this once the permission check migration is complete.
 		Policy:         NewPolicy(client),
-		Role:           accountmongo.NewRole(client), // TODO: Delete this once the permission check migration is complete.
+		Role:           NewRoleWrapper(client), // TODO: Delete this once the permission check migration is complete.
 		Storytelling:   NewStorytelling(client),
 		Lock:           lock,
 		Transaction:    client.Transaction(),
 		Workspace:      NewWorkspaceWrapper(client),
-		User:           account.User,
+		User:           NewUserWrapper(client),
 	}
 
 	// init
@@ -126,16 +125,16 @@ func Init(r *repo.Container) error {
 		func() error { return r.Dataset.(*Dataset).Init(ctx) },
 		func() error { return r.DatasetSchema.(*DatasetSchema).Init(ctx) },
 		func() error { return r.Layer.(*Layer).Init(ctx) },
-		func() error { return r.Permittable.(*accountmongo.Permittable).Init(ctx) }, // TODO: Delete this once the permission check migration is complete.
+		func() error { return r.Permittable.(*PermittableWrapper).Init(ctx) }, // TODO: Delete this once the permission check migration is complete.
 		func() error { return r.Plugin.(*Plugin).Init(ctx) },
 		func() error { return r.Policy.(*Policy).Init(ctx) },
 		func() error { return r.Project.(*Project).Init(ctx) },
 		func() error { return r.Property.(*Property).Init(ctx) },
 		func() error { return r.PropertySchema.(*PropertySchema).Init(ctx) },
-		func() error { return r.Role.(*accountmongo.Role).Init(ctx) }, // TODO: Delete this once the permission check migration is complete.
+		func() error { return r.Role.(*RoleWrapper).Init(ctx) }, // TODO: Delete this once the permission check migration is complete.
 		func() error { return r.Scene.(*Scene).Init(ctx) },
 		func() error { return r.Tag.(*Tag).Init(ctx) },
-		func() error { return r.User.(*accountmongo.User).Init() },
+		func() error { return r.User.(*UserWrapper).Init() },
 		func() error { return r.Workspace.(*WorkspaceWrapper).Init() },
 	)
 }

server/internal/infrastructure/mongo/permittable_wrapper.go

@@ -0,0 +1,28 @@
+package mongo
+
+import (
+	"context"
+
+	"github.com/reearth/reearthx/account/accountinfrastructure/accountmongo"
+	"github.com/reearth/reearthx/account/accountusecase/accountrepo"
+	"github.com/reearth/reearthx/mongox"
+)
+
+// PermittableWrapper wraps accountmongo.Permittable to prevent index dropping
+type PermittableWrapper struct {
+	*accountmongo.Permittable
+	client *mongox.Collection
+}
+
+func NewPermittableWrapper(client *mongox.Client) accountrepo.Permittable {
+	original := accountmongo.NewPermittable(client)
+
+	return &PermittableWrapper{
+		Permittable: original.(*accountmongo.Permittable),
+		client:      client.WithCollection("permittable"),
+	}
+}
+
+func (p *PermittableWrapper) Init(ctx context.Context) error {
+	return createIndexesOnly(ctx, p.client, nil, []string{"id"})
+}

server/internal/infrastructure/mongo/role_wrapper.go

@@ -0,0 +1,28 @@
+package mongo
+
+import (
+	"context"
+
+	"github.com/reearth/reearthx/account/accountinfrastructure/accountmongo"
+	"github.com/reearth/reearthx/account/accountusecase/accountrepo"
+	"github.com/reearth/reearthx/mongox"
+)
+
+// RoleWrapper wraps accountmongo.Role to prevent index dropping
+type RoleWrapper struct {
+	*accountmongo.Role
+	client *mongox.Collection
+}
+
+func NewRoleWrapper(client *mongox.Client) accountrepo.Role {
+	original := accountmongo.NewRole(client)
+
+	return &RoleWrapper{
+		Role:   original.(*accountmongo.Role),
+		client: client.WithCollection("role"),
+	}
+}
+
+func (r *RoleWrapper) Init(ctx context.Context) error {
+	return createIndexesOnly(ctx, r.client, nil, []string{"id"})
+}

server/internal/infrastructure/mongo/user_wrapper.go

@@ -0,0 +1,30 @@
+package mongo
+
+import (
+	"context"
+
+	"github.com/reearth/reearthx/account/accountinfrastructure/accountmongo"
+	"github.com/reearth/reearthx/account/accountusecase/accountrepo"
+	"github.com/reearth/reearthx/mongox"
+)
+
+// UserWrapper wraps accountmongo.User to prevent index dropping
+type UserWrapper struct {
+	*accountmongo.User
+	client *mongox.Collection
+}
+
+func NewUserWrapper(client *mongox.Client) accountrepo.User {
+	original := accountmongo.NewUser(client)
+
+	return &UserWrapper{
+		User:   original.(*accountmongo.User),
+		client: client.WithCollection("user"),
+	}
+}
+
+func (u *UserWrapper) Init() error {
+	ctx := context.Background()
+
+	return createIndexesOnly(ctx, u.client, nil, []string{"id"})
+}