deferred_writer.h

Go to the documentation of this file.

#pragma once

#include "mongo/db/catalog/collection.h"
#include "mongo/db/namespace_string.h"
#include "mongo/db/operation_context.h"
#include "mongo/stdx/mutex.h"

namespace mongo {

class AutoGetCollection;
class ThreadPool;

class DeferredWriter {
    MONGO_DISALLOW_COPYING(DeferredWriter);

public:
    DeferredWriter(NamespaceString nss, CollectionOptions opts, int64_t maxSize);

    void startup(std::string workerName);

    void shutdown(void);

    ~DeferredWriter();

    bool insertDocument(BSONObj obj);

    int64_t getDroppedEntries();

private:
    void _logFailure(const Status& status);

    void _logDroppedEntry();

    Status _makeCollection(OperationContext* opCtx);

    StatusWith<std::unique_ptr<AutoGetCollection>> _getCollection(OperationContext* opCtx);

    void _worker(InsertStatement stmt);

    const CollectionOptions _collectionOptions;

    const int64_t _maxNumBytes;

    const NamespaceString _nss;

    std::unique_ptr<ThreadPool> _pool;

    stdx::mutex _mutex;

    int64_t _numBytes;

    int64_t _droppedEntries;

    using TimePoint = stdx::chrono::time_point<stdx::chrono::system_clock>;
    TimePoint _lastLogged;
    TimePoint _lastLoggedDrop;
};

}  // namespace mongo