34 lines
796 B
Go
34 lines
796 B
Go
package mongo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.acooldomain.co/server-manager/backend/models"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func getMongoCollection(config models.MongoDBConfig) (*mongo.Collection, error) {
|
|
clientOptions := options.Client().ApplyURI(config.Url)
|
|
|
|
if config.Username != "" || config.Password != "" {
|
|
clientOptions = clientOptions.SetAuth(options.Credential{
|
|
Username: config.Username,
|
|
Password: config.Password,
|
|
})
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeoutCause(context.Background(), 30*time.Second, fmt.Errorf("Timeout"))
|
|
defer cancel()
|
|
|
|
client, err := mongo.Connect(ctx, clientOptions)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return client.Database(config.Database).Collection(config.Collection), nil
|
|
}
|