Override default ephemeral database uri by using env variable

This commit is contained in:
2022-01-16 23:06:34 +01:00
parent eab00f99b5
commit 6b34ad5a09
5 changed files with 19 additions and 4 deletions
+15
View File
@@ -2,6 +2,7 @@ package database
import (
"context"
"os"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
@@ -11,14 +12,28 @@ import (
"github.com/mainnika/mongox-go-driver/v2/mongox/database"
)
// defaultURI is a mongodb uri that is being used by tests
var defaultURI = "mongodb://localhost"
// EphemeralDatabase is a temporary database connection that will be destroyed after close
type EphemeralDatabase struct {
mongox.Database
}
func init() {
envURI := os.Getenv("MONGODB_URI")
if envURI != "" {
defaultURI = envURI
}
}
// NewEphemeral creates new mongo connection
func NewEphemeral(URI string) (db *EphemeralDatabase, err error) {
if URI == "" {
URI = defaultURI
}
name := primitive.NewObjectID().Hex()
opts := options.Client().ApplyURI(URI)
client, err := mongo.Connect(context.Background(), opts)