diff --git a/pkg/database/mongodb.go b/pkg/database/mongodb.go index ed75e75..c884b1b 100644 --- a/pkg/database/mongodb.go +++ b/pkg/database/mongodb.go @@ -175,6 +175,7 @@ func newMongoClientFromComponents(ctx context.Context, options *MongoOptions) (D }, err } +// Ping pings the MongoDB server to check connectivity func (m *MongoClient) Ping(ctx context.Context) error { err := m.Client.Ping(ctx, nil) return err @@ -185,6 +186,11 @@ func (m *MongoClient) Disconnect(ctx context.Context) error { return m.Client.Disconnect(ctx) } +// GetTimeout returns the timeout duration for the MongoDB client +func (m *MongoClient) GetTimeout() time.Duration { + return time.Duration(m.Options.Timeout) * time.Millisecond +} + // Find executes a find query on the specified database and collection func (m *MongoClient) Find(ctx context.Context, db string, collection string, filter any, opts ...any) (any, error) { coll := m.Client.Database(db).Collection(collection)