51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package initialize
|
|
|
|
import (
|
|
"codeword/internal/repository"
|
|
mdb "codeword/pkg/mongo"
|
|
"context"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.uber.org/zap"
|
|
"time"
|
|
)
|
|
|
|
func MongoDB(ctx context.Context, cfg Config) (*mongo.Database, error) {
|
|
dbConfig := &mdb.Configuration{
|
|
MongoHost: cfg.MongoHost,
|
|
MongoPort: cfg.MongoPort,
|
|
MongoUser: cfg.MongoUser,
|
|
MongoPassword: cfg.MongoPassword,
|
|
MongoDatabase: cfg.MongoDatabase,
|
|
MongoAuth: cfg.MongoAuth,
|
|
}
|
|
|
|
newCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
|
|
defer cancel()
|
|
|
|
mongoDeps := &mdb.ConnectDeps{
|
|
Configuration: dbConfig,
|
|
Timeout: 10 * time.Second,
|
|
}
|
|
|
|
db, err := mdb.Connect(newCtx, mongoDeps)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = db.Client().Ping(newCtx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func InitDatabaseIndexes(ctx context.Context, mdb *mongo.Database, logger *zap.Logger) error {
|
|
if err := repository.InitPromoCodeIndexes(ctx, mdb.Collection("promoCodes")); err != nil {
|
|
logger.Error("Failed to initialize promoCodes indexes", zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|