x3c/include/libIndexer/CIndexWriter.h

#ifndef __CINDEXWRITER_H__
#define __CINDEXWRITER_H__

#include <vector>
#include <cassert>
#include "IdxFileFormat.h"
#include "CIndexer.h"

namespace x3c {
    namespace indexer {

        class CIndexWriter
        {
        public:

            ~CIndexWriter();

            void close();

            void  addIndexRecord(UINT64 timestamp, UINT64 offset);

            void  addIndexRecord(IndexRecord const& record);

            UINT32 numRecords() const;

            bool  flushToDisk();

            std::string const& filename() const;

       private:

            CIndexWriter& operator=(CIndexWriter const&);

            CIndexWriter(CIndexWriter const&);

            CIndexWriter(CIndexer<CIndexWriter> const& parent,
                                 UINT32 reservedMem);

            bool create(std::string const& filename);

            bool writeHeader();

            bool writeIndexRecords();

            bool writeTrailer();

            std::string         mFilename;

            IndexHeader         mHeader;

            std::vector<IndexRecord>  mIndexRecords;
            
            IndexTrailer        mTrailer;

            CIndexer<CIndexWriter> const& mParent;

            mutable INT32       mFd;

            friend class CIndexer<CIndexWriter>;
        };

        //======================================================================
        // Inline members
        //


        inline void CIndexWriter::addIndexRecord(IndexRecord const& record)
        {
            mIndexRecords.push_back(record);
        }

        inline void CIndexWriter::addIndexRecord(UINT64 timestamp,
                                                 UINT64 offset)
        {
            // TODO: support the IndexRecord::packet_number field
            IndexRecord idx = { timestamp, offset, 0 };
            addIndexRecord(idx);
        }

        inline std::string const& CIndexWriter::filename() const
        {
            return mFilename;
        }

        inline UINT32 CIndexWriter::numRecords() const
        {
            return mIndexRecords.size();
        }

    }   /* namespace indexer */
}  /* namespace x3c */

#endif /*  __CINDEXWRITER_H__ */