record_store_v1_test_help.h

Go to the documentation of this file.

// record_store_v1_test_help.h

#pragma once

#include <vector>

#include "mongo/db/storage/mmap_v1/data_file.h"
#include "mongo/db/storage/mmap_v1/extent_manager.h"
#include "mongo/db/storage/mmap_v1/record_store_v1_base.h"

namespace mongo {

class DummyRecordStoreV1MetaData : public RecordStoreV1MetaData {
public:
    DummyRecordStoreV1MetaData(bool capped, int userFlags);
    virtual ~DummyRecordStoreV1MetaData() {}

    virtual const DiskLoc& capExtent() const;
    virtual void setCapExtent(OperationContext* opCtx, const DiskLoc& loc);

    virtual const DiskLoc& capFirstNewRecord() const;
    virtual void setCapFirstNewRecord(OperationContext* opCtx, const DiskLoc& loc);

    virtual long long dataSize() const;
    virtual long long numRecords() const;

    virtual void incrementStats(OperationContext* opCtx,
                                long long dataSizeIncrement,
                                long long numRecordsIncrement);

    virtual void setStats(OperationContext* opCtx, long long dataSize, long long numRecords);

    virtual DiskLoc deletedListEntry(int bucket) const;
    virtual void setDeletedListEntry(OperationContext* opCtx, int bucket, const DiskLoc& loc);

    virtual DiskLoc deletedListLegacyGrabBag() const;
    virtual void setDeletedListLegacyGrabBag(OperationContext* opCtx, const DiskLoc& loc);

    virtual void orphanDeletedList(OperationContext* opCtx);

    virtual const DiskLoc& firstExtent(OperationContext* opCtx) const;
    virtual void setFirstExtent(OperationContext* opCtx, const DiskLoc& loc);

    virtual const DiskLoc& lastExtent(OperationContext* opCtx) const;
    virtual void setLastExtent(OperationContext* opCtx, const DiskLoc& loc);

    virtual bool isCapped() const;

    virtual bool isUserFlagSet(int flag) const;
    virtual int userFlags() const {
        return _userFlags;
    }
    virtual bool setUserFlag(OperationContext* opCtx, int flag);
    virtual bool clearUserFlag(OperationContext* opCtx, int flag);
    virtual bool replaceUserFlags(OperationContext* opCtx, int flags);


    virtual int lastExtentSize(OperationContext* opCtx) const;
    virtual void setLastExtentSize(OperationContext* opCtx, int newMax);

    virtual long long maxCappedDocs() const;

protected:
    DiskLoc _capExtent;
    DiskLoc _capFirstNewRecord;

    long long _dataSize;
    long long _numRecords;

    DiskLoc _firstExtent;
    DiskLoc _lastExtent;

    bool _capped;
    int _userFlags;
    long long _maxCappedDocs;

    int _lastExtentSize;
    double _paddingFactor;

    std::vector<DiskLoc> _deletedLists;
    DiskLoc _deletedListLegacyGrabBag;
};

class DummyExtentManager : public ExtentManager {
public:
    virtual ~DummyExtentManager();

    virtual void close(OperationContext* opCtx);

    virtual Status init(OperationContext* opCtx);

    virtual int numFiles() const;
    virtual long long fileSize() const;

    virtual DiskLoc allocateExtent(OperationContext* opCtx,
                                   bool capped,
                                   int size,
                                   bool enforceQuota);

    virtual void freeExtents(OperationContext* opCtx, DiskLoc firstExt, DiskLoc lastExt);

    virtual void freeExtent(OperationContext* opCtx, DiskLoc extent);

    virtual void freeListStats(OperationContext* opCtx,
                               int* numExtents,
                               int64_t* totalFreeSizeBytes) const;

    virtual MmapV1RecordHeader* recordForV1(const DiskLoc& loc) const;

    virtual std::unique_ptr<RecordFetcher> recordNeedsFetch(const DiskLoc& loc) const final;

    virtual Extent* extentForV1(const DiskLoc& loc) const;

    virtual DiskLoc extentLocForV1(const DiskLoc& loc) const;

    virtual Extent* getExtent(const DiskLoc& loc, bool doSanityCheck = true) const;

    virtual int maxSize() const;

    virtual CacheHint* cacheHint(const DiskLoc& extentLoc, const HintType& hint);

    DataFileVersion getFileFormat(OperationContext* opCtx) const final;

    virtual void setFileFormat(OperationContext* opCtx, DataFileVersion newVersion) final;

    const DataFile* getOpenFile(int n) const final;


protected:
    struct ExtentInfo {
        char* data;
        size_t length;
    };

    std::vector<ExtentInfo> _extents;
};

struct LocAndSize {
    DiskLoc loc;
    int size;  // with headers
};

void initializeV1RS(OperationContext* opCtx,
                    const LocAndSize* records,
                    const LocAndSize* drecs,
                    const LocAndSize* legacyGrabBag,
                    DummyExtentManager* em,
                    DummyRecordStoreV1MetaData* md);

void assertStateV1RS(OperationContext* opCtx,
                     const LocAndSize* records,
                     const LocAndSize* drecs,
                     const LocAndSize* legacyGrabBag,
                     const ExtentManager* em,
                     const DummyRecordStoreV1MetaData* md);

}  // namespace mongo