mobile_record_store.h

Go to the documentation of this file.

#pragma once

#include <memory>
#include <string>

#include "mongo/db/catalog/collection_options.h"
#include "mongo/db/operation_context.h"
#include "mongo/db/storage/capped_callback.h"
#include "mongo/db/storage/mobile/mobile_sqlite_statement.h"
#include "mongo/db/storage/record_store.h"
#include "mongo/platform/atomic_word.h"

namespace mongo {

class MobileRecordStore : public RecordStore {
public:
    explicit MobileRecordStore(OperationContext* opCtx,
                               StringData ns,
                               const std::string& path,
                               const std::string& ident,
                               const CollectionOptions& options);

    const char* name() const override;

    RecordData dataFor(OperationContext* opCtx, const RecordId& recId) const override;

    bool findRecord(OperationContext* opCtx, const RecordId& recId, RecordData* rd) const override;

    void deleteRecord(OperationContext* opCtx, const RecordId& dl) override;

    StatusWith<RecordId> insertRecord(OperationContext* opCtx,
                                      const char* data,
                                      int len,
                                      Timestamp timestamp,
                                      bool enforceQuota) override;

    Status insertRecordsWithDocWriter(OperationContext* opCtx,
                                      const DocWriter* const* docs,
                                      const Timestamp* timestamps,
                                      size_t nDocs,
                                      RecordId* idsOut) override;

    Status updateRecord(OperationContext* opCtx,
                        const RecordId& oldLocation,
                        const char* data,
                        int len,
                        bool enforceQuota,
                        UpdateNotifier* notifier) override;

    bool updateWithDamagesSupported() const override;

    StatusWith<RecordData> updateWithDamages(OperationContext* opCtx,
                                             const RecordId& recId,
                                             const RecordData& oldRec,
                                             const char* damageSource,
                                             const mutablebson::DamageVector& damages) override;

    std::unique_ptr<SeekableRecordCursor> getCursor(OperationContext* opCtx,
                                                    bool forward) const override;

    Status truncate(OperationContext* opCtx) override;

    void cappedTruncateAfter(OperationContext* opCtx, RecordId end, bool inclusive) override;

    bool compactSupported() const override {
        return true;
    }

    bool compactsInPlace() const override {
        return true;
    }

    Status compact(OperationContext* opCtx,
                   RecordStoreCompactAdaptor* adaptor,
                   const CompactOptions* options,
                   CompactStats* stats) override;

    Status validate(OperationContext* opCtx,
                    ValidateCmdLevel level,
                    ValidateAdaptor* adaptor,
                    ValidateResults* results,
                    BSONObjBuilder* output) override;

    void appendCustomStats(OperationContext* opCtx,
                           BSONObjBuilder* result,
                           double scale) const override;

    Status touch(OperationContext* opCtx, BSONObjBuilder* output) const override;

    int64_t storageSize(OperationContext* opCtx,
                        BSONObjBuilder* extraInfo = NULL,
                        int infoLevel = 0) const override;

    long long dataSize(OperationContext* opCtx) const override;

    long long numRecords(OperationContext* opCtx) const override;

    boost::optional<RecordId> oplogStartHack(OperationContext* opCtx,
                                             const RecordId& startingPosition) const override;

    void waitForAllEarlierOplogWritesToBeVisible(OperationContext* opCtx) const override {}

    void updateStatsAfterRepair(OperationContext* opCtx,
                                long long numRecords,
                                long long dataSize) override {}

    bool isCapped() const override {
        return _isCapped;
    }

    void setCappedCallback(CappedCallback* cb) override {
        _cappedCallback = cb;
    }

    Status updateCappedSize(OperationContext* opCtx, long long cappedSize) override;

    // Not in record store API.

    static void create(OperationContext* opCtx, const std::string& ident);

private:
    class InsertChange;
    class RemoveChange;
    class TruncateChange;

    class NumRecsChange;
    class DataSizeChange;

    class Cursor;

    RecordId _nextId();

    const std::string _path;
    const std::string _ident;

    // True if the namespace of this record store starts with "local.oplog.", and false otherwise.
    const bool _isOplog;

    bool _isCappedAndNeedsDelete(int64_t numRecs, int64_t numBytes);

    void _notifyCappedCallbackIfNeeded_inlock(OperationContext* opCtx,
                                              RecordId recId,
                                              const RecordData& recData);

    void _doCappedDelete(OperationContext* opCtx,
                         SqliteStatement& stmt,
                         const std::string& direction,
                         int64_t startRecId = 0);

    void _cappedDeleteIfNeeded(OperationContext* opCtx);

    const bool _isCapped;
    int64_t _cappedMaxSize;
    const int64_t _cappedMaxDocs;
    // Mutex that protects _cappedCallback
    stdx::mutex _cappedCallbackMutex;
    CappedCallback* _cappedCallback = nullptr;

    AtomicInt64 _nextIdNum;

    void _initNumRecsIfNeeded_inlock(OperationContext* opCtx) const;

    void _changeNumRecs(OperationContext* opCtx, int64_t diff);

    bool _resetNumRecsIfNeeded(OperationContext* opCtx, int64_t newNumRecs);

    mutable int64_t _numRecs;
    mutable stdx::mutex _numRecsMutex;
    mutable bool _isNumRecsInitialized = false;

    void _initDataSizeIfNeeded_inlock(OperationContext* opCtx) const;

    void _changeDataSize(OperationContext* opCtx, int64_t diff);

    bool _resetDataSizeIfNeeded(OperationContext* opCtx, int64_t newDataSize);

    mutable int64_t _dataSize;
    mutable stdx::mutex _dataSizeMutex;
    mutable bool _isDataSizeInitialized = false;
};

}  // namespace mongo