diff options
author | Scott Hernandez <scotthernandez@tart.local> | 2016-06-10 16:12:01 -0400 |
---|---|---|
committer | Scott Hernandez <scotthernandez@tart.local> | 2016-06-17 11:53:01 -0400 |
commit | 42eb9464500381ef8034070472badded4b427cf5 (patch) | |
tree | 9f7aaf66732d0352f37ced532f6c0fc99cd5793d /src/mongo/db/repl/collection_bulk_loader_impl.h | |
parent | c59f5ade57e41b6a50f40999ea14883da691e951 (diff) | |
download | mongo-42eb9464500381ef8034070472badded4b427cf5.tar.gz |
SERVER-23059: storage interface improvements
Diffstat (limited to 'src/mongo/db/repl/collection_bulk_loader_impl.h')
-rw-r--r-- | src/mongo/db/repl/collection_bulk_loader_impl.h | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/src/mongo/db/repl/collection_bulk_loader_impl.h b/src/mongo/db/repl/collection_bulk_loader_impl.h new file mode 100644 index 00000000000..1650d98f5fb --- /dev/null +++ b/src/mongo/db/repl/collection_bulk_loader_impl.h @@ -0,0 +1,90 @@ +/** + * Copyright (C) 2015 MongoDB Inc. + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License, version 3, + * as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + * + * As a special exception, the copyright holders give permission to link the + * code of portions of this program with the OpenSSL library under certain + * conditions as described in each individual source file and distribute + * linked combinations including the program with the OpenSSL library. You + * must comply with the GNU Affero General Public License in all respects for + * all of the code used other than as permitted herein. If you modify file(s) + * with this exception, you may extend this exception to your version of the + * file(s), but you are not obligated to do so. If you do not wish to do so, + * delete this exception statement from your version. If you delete this + * exception statement from all source files in the program, then also delete + * it in the license file. + */ + + +#pragma once + +#include "mongo/base/disallow_copying.h" +#include "mongo/base/status.h" +#include "mongo/base/status_with.h" +#include "mongo/bson/bsonobj.h" +#include "mongo/db/catalog/index_create.h" +#include "mongo/db/db_raii.h" +#include "mongo/db/namespace_string.h" +#include "mongo/db/repl/collection_bulk_loader.h" +#include "mongo/db/repl/storage_interface.h" +#include "mongo/db/repl/task_runner.h" +#include "mongo/util/concurrency/old_thread_pool.h" + +namespace mongo { +namespace repl { + +/** + * Class in charge of building a collection during data loading (like initial sync). + * + * Note: Call commit when done inserting documents. + */ +class CollectionBulkLoaderImpl : public CollectionBulkLoader { + MONGO_DISALLOW_COPYING(CollectionBulkLoaderImpl); + +public: + CollectionBulkLoaderImpl(OperationContext* txn, + TaskRunner* runner, + Collection* coll, + const BSONObj idIndexSpec, + std::unique_ptr<AutoGetOrCreateDb> autoDB, + std::unique_ptr<AutoGetCollection> autoColl); + virtual ~CollectionBulkLoaderImpl(); + + virtual Status init(OperationContext* txn, + Collection* coll, + const std::vector<BSONObj>& secondaryIndexSpecs) override; + + virtual Status insertDocuments(const std::vector<BSONObj>::const_iterator begin, + const std::vector<BSONObj>::const_iterator end) override; + virtual Status commit() override; + + virtual std::string toString() const override; + virtual BSONObj toBSON() const override; + +private: + TaskRunner* _runner; + std::unique_ptr<AutoGetCollection> _autoColl; + std::unique_ptr<AutoGetOrCreateDb> _autoDB; + OperationContext* _txn = nullptr; + Collection* _coll = nullptr; + NamespaceString _nss; + MultiIndexBlock _idIndexBlock; + MultiIndexBlock _secondaryIndexesBlock; + bool _hasSecondaryIndexes = false; + BSONObj _idIndexSpec; + bool _callAbortOnDestructor = false; +}; + +} // namespace repl +} // namespace mongo |