Golang Mongodb dynamic auto indexing

helpers.go

package helpers

import (
	"context"
	"log"
	"time"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

func AutoIndexingCollection[T any](db *mongo.Database, key []string, collectionName string) error {
	collection := db.Collection(collectionName)

	var indexModels []mongo.IndexModel
	for _, k := range key {
		indexModel := mongo.IndexModel{
			Keys:    bson.D{{Key: k, Value: 1}},
			Options: options.Index().SetUnique(true),
		}
		indexModels = append(indexModels, indexModel)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	_, err := collection.Indexes().CreateMany(ctx, indexModels)
	if err != nil {
		log.Printf("Error creating index on %v: %v\n", key, err)
		return err
	}

	log.Printf("Successfully created unique index on %s for collection %s\n", key, collectionName)
	return nil
}