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

#pragma once

#include <vespa/searchcorespi/flush/iflushtarget.h>
#include <vespa/searchlib/common/tunefileinfo.h>
#include <vespa/searchcore/proton/common/hw_info.h>

namespace search::common { class FileHeaderContext; }

namespace proton {

class AttributeDirectory;
class AttributeDiskLayout;
class DocumentMetaStore;
class ITlsSyncer;
class TransientResourceUsage;

/**
 * Implementation of IFlushTarget interface for document meta store.
 **/
class DocumentMetaStoreFlushTarget : public searchcorespi::LeafFlushTarget
{
private:
    /**
     * Task performing the actual flushing to disk.
     **/
    class Flusher;
    using DocumentMetaStoreSP = std::shared_ptr<DocumentMetaStore>;
    using FlushStats = searchcorespi::FlushStats;

    DocumentMetaStoreSP                      _dms;
    ITlsSyncer                              &_tlsSyncer;
    vespalib::string                         _baseDir;
    bool                                     _cleanUpAfterFlush;
    FlushStats                               _lastStats;
    const search::TuneFileAttributes         _tuneFileAttributes;
    const search::common::FileHeaderContext &_fileHeaderContext;
    HwInfo                                   _hwInfo;
    std::shared_ptr<AttributeDiskLayout>     _diskLayout;
    std::shared_ptr<AttributeDirectory>      _dmsDir;

public:
    using SP = std::shared_ptr<DocumentMetaStoreFlushTarget>;

    /**
     * Creates a new instance using the given attribute vector and the
     * given base dir where all attribute vectors are located.
     **/
    DocumentMetaStoreFlushTarget(const DocumentMetaStoreSP dms, ITlsSyncer &tlsSyncer,
                                 const vespalib::string &baseDir, const search::TuneFileAttributes &tuneFileAttributes,
                                 const search::common::FileHeaderContext &fileHeaderContext, const HwInfo &hwInfo);

    ~DocumentMetaStoreFlushTarget() override;

    void setCleanUpAfterFlush(bool cleanUp) { _cleanUpAfterFlush = cleanUp; }

    TransientResourceUsage get_transient_resource_usage() const;

    MemoryGain getApproxMemoryGain() const override;
    DiskGain getApproxDiskGain() const override;
    Time getLastFlushTime() const override;
    SerialNum getFlushedSerialNum() const override;
    Task::UP initFlush(SerialNum currentSerial, std::shared_ptr<search::IFlushToken> flush_token) override;
    FlushStats getLastFlushStats() const override { return _lastStats; }

    static void initCleanup(const vespalib::string &baseDir);
    uint64_t getApproxBytesToWriteToDisk() const override;
};

} // namespace proton