Create mongod indexes for each query schema field (#45)

internal/api/swagger.yaml

@@ -9,7 +9,7 @@ info:
   license:
     name: Apache 2.0
     url: http://www.apache.org/licenses/LICENSE-2.0.html
-  version: 1.14.3
+  version: 1.15.0
 externalDocs:
   description: Find out more about spec
   url: 'https://moonlogs.pages.dev'

internal/persistence/mongodb.go

@@ -3,9 +3,9 @@ package persistence
 import (
 	"context"
 	"fmt"
+	"moonlogs/internal/shared"
 	"time"
 
-	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
@@ -37,20 +37,7 @@ func initMongoDB(dataSourceName string) (*mongo.Client, error) {
 
 func createIndexes(client *mongo.Client) error {
 	collection := client.Database(MONGODB_DATABASE_NAME).Collection("records")
-
 	indexNames := []string{"schema_name", "id", "group_hash"}
 
-	for _, name := range indexNames {
-		indexModel := mongo.IndexModel{
-			Keys:    bson.D{{Key: name, Value: 1}},
-			Options: options.Index().SetUnique(false),
-		}
-
-		_, err := collection.Indexes().CreateOne(context.Background(), indexModel)
-		if err != nil {
-			return fmt.Errorf("creating index `%s` on `records` collection: %w", name, err)
-		}
-	}
-
-	return nil
+	return shared.CreateIndexes(collection, indexNames)
 }

internal/shared/mongo.go

@@ -0,0 +1,26 @@
+package shared
+
+import (
+	"context"
+	"fmt"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func CreateIndexes(collection *mongo.Collection, indexNames []string) error {
+	for _, name := range indexNames {
+		indexModel := mongo.IndexModel{
+			Keys:    bson.D{{Key: name, Value: 1}},
+			Options: options.Index().SetUnique(false),
+		}
+
+		_, err := collection.Indexes().CreateOne(context.Background(), indexModel)
+		if err != nil {
+			return fmt.Errorf("creating index `%s` on `records` collection: %w", name, err)
+		}
+	}
+
+	return nil
+}

internal/storage/mongodb_adapter/schema.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"moonlogs/internal/entities"
+	"moonlogs/internal/shared"
 	"moonlogs/internal/storage"
 
 	"go.mongodb.org/mongo-driver/bson"
@@ -49,6 +50,12 @@ func (s *SchemaStorage) CreateSchema(ctx context.Context, schema entities.Schema
 		return nil, fmt.Errorf("failed querying inserted schema: %w", err)
 	}
 
+	indexNames := make([]string, 0)
+	for _, field := range schema.Fields {
+		indexNames = append(indexNames, fmt.Sprintf("query.%s", field.Name))
+	}
+	_ = shared.CreateIndexes(s.collection, indexNames)
+
 	return &sm, nil
 }
 
@@ -67,6 +74,12 @@ func (s *SchemaStorage) UpdateSchemaByID(ctx context.Context, id int, schema ent
 		return nil, fmt.Errorf("failed updating schema: %w", err)
 	}
 
+	indexNames := make([]string, 0)
+	for _, field := range schema.Fields {
+		indexNames = append(indexNames, fmt.Sprintf("query.%s", field.Name))
+	}
+	_ = shared.CreateIndexes(s.collection, indexNames)
+
 	return s.GetById(ctx, id)
 }