ephemeral_for_test_recovery_unit.h

Go to the documentation of this file.

// ephemeral_for_test_recovery_unit.h

#pragma once

#include <vector>

#include "mongo/db/record_id.h"
#include "mongo/db/storage/recovery_unit.h"
#include "mongo/stdx/functional.h"

namespace mongo {

class SortedDataInterface;

class EphemeralForTestRecoveryUnit : public RecoveryUnit {
public:
    EphemeralForTestRecoveryUnit(stdx::function<void()> cb = nullptr)
        : _waitUntilDurableCallback(cb) {}

    void beginUnitOfWork(OperationContext* opCtx) final{};
    void commitUnitOfWork() final;
    void abortUnitOfWork() final;

    virtual bool waitUntilDurable() {
        if (_waitUntilDurableCallback) {
            _waitUntilDurableCallback();
        }
        return true;
    }

    virtual void abandonSnapshot() {}

    Status obtainMajorityCommittedSnapshot() final;

    virtual void registerChange(Change* change) {
        _changes.push_back(ChangePtr(change));
    }

    virtual void* writingPtr(void* data, size_t len) {
        MONGO_UNREACHABLE;
    }

    virtual void setRollbackWritesDisabled() {}

    virtual SnapshotId getSnapshotId() const {
        return SnapshotId();
    }

    virtual void setOrderedCommit(bool orderedCommit) {}

    virtual void setPrepareTimestamp(Timestamp) {}

private:
    typedef std::shared_ptr<Change> ChangePtr;
    typedef std::vector<ChangePtr> Changes;

    Changes _changes;
    stdx::function<void()> _waitUntilDurableCallback;
};

}  // namespace mongo