26 lines
719 B
Go
26 lines
719 B
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// GetDB connects to DB and returns a client that will disconnect when the passed in context is cancelled
|
|
func GetDB(appCtx context.Context, log *zap.Logger, uri, databaseName string) (*mongo.Database, error) {
|
|
cli, err := mongo.Connect(appCtx, options.Client().ApplyURI(uri))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go func() {
|
|
<-appCtx.Done()
|
|
ctx, _ := context.WithDeadline(context.Background(), time.Now().Add(30*time.Second))
|
|
err := cli.Disconnect(ctx)
|
|
log.Error("error disconnecting from db", zap.Error(err))
|
|
}()
|
|
return cli.Database(databaseName), err
|
|
}
|