// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

#include "removedocumentsoperation.h"
#include <vespa/searchlib/query/base.h>

namespace proton {

class PruneRemovedDocumentsOperation : public RemoveDocumentsOperation
{
private:
    uint32_t _subDbId;
public:
    using UP = std::unique_ptr<PruneRemovedDocumentsOperation>;

    PruneRemovedDocumentsOperation();
    PruneRemovedDocumentsOperation(search::DocumentIdT docIdLimit, uint32_t subDbId);

    uint32_t getSubDbId() const { return _subDbId; }

    void setLidsToRemove(const LidVectorContext::SP &lidsToRemove) {
        RemoveDocumentsOperation::setLidsToRemove(_subDbId, lidsToRemove);
    }

    const LidVectorContext::SP
    getLidsToRemove() const {
        return RemoveDocumentsOperation::getLidsToRemove(_subDbId);
    }

    void serialize(vespalib::nbostream &os) const override;
    void deserialize(vespalib::nbostream &is, const document::DocumentTypeRepo &repo) override;
    vespalib::string toString() const override;
};

} // namespace proton