instream_document_trec.h

Go to the documentation of this file.

/*
    INSTREAM_DOCUMENT_TREC.H
    ------------------------
    Copyright (c) 2016 Andrew Trotman
    Released under the 2-clause BSD license (See:https://en.wikipedia.org/wiki/BSD_licenses)
*/
#pragma once

#include <stdint.h>

#include <string>

#include "slice.h"
#include "instream.h"

namespace JASS
    {
    /*
        CLASS INSTREAM_DOCUMENT_TREC
        ----------------------------
    */
    class instream_document_trec : public instream
        {
        protected:
            size_t buffer_size;                             

        protected:
            uint8_t *buffer;                                    
            uint8_t *buffer_end;                                
            size_t buffer_used;                             
        
            std::string document_start_tag;             
            std::string document_end_tag;                   
            std::string primary_key_start_tag;          
            std::string primary_key_end_tag;                

        protected:
            /*
                INSTREAM_DOCUMENT_TREC::INSTREAM_DOCUMENT_TREC()
                ------------------------------------------------
            */
            instream_document_trec(std::shared_ptr<instream> &source, size_t buffer_size, const std::string &document_tag, const std::string &document_primary_key_tag);
            
            /*
                INSTREAM_DOCUMENT_TREC::SET_TAGS()
                ----------------------------------
            */
            void set_tags(const std::string &document_tag, const std::string &primary_key_tag);

            /*
                INSTREAM_DOCUMENT_TREC::FETCH()
                -------------------------------
            */
            void fetch(void *buffer, size_t bytes)
                {
                buffer_end = (uint8_t *)buffer + source->fetch(buffer, bytes);
                }

        public:
            /*
                INSTREAM_DOCUMENT_TREC::INSTREAM_DOCUMENT_TREC()
                ------------------------------------------------
            */
            instream_document_trec(const instream_document_trec &previous) = delete;

            /*
                INSTREAM_DOCUMENT_TREC::INSTREAM_DOCUMENT_TREC()
                ------------------------------------------------
            */
            instream_document_trec(std::shared_ptr<instream> &source, const std::string &document_tag = "DOC", const std::string &document_primary_key_tag = "DOCNO");
            
            /*
                INSTREAM_DOCUMENT_TREC::INSTREAM_DOCUMENT_TREC()
                ------------------------------------------------
            */
            virtual ~instream_document_trec();

            /*
                INSTREAM_DOCUMENT_TREC::READ()
                ------------------------------
            */
            virtual void read(document &buffer);
            
            /*
                INSTREAM_DOCUMENT_TREC::UNITTEST()
                ----------------------------------
            */
            static void unittest(void);
        } ;
}