src/graph/example/FileReader.h

#ifndef HT_EXAMPLE_FILE_READER_
#define HT_EXAMPLE_FILE_READER_

#include <graph/NodeBase.h>
#include <fstream>

namespace ht {
    namespace example {

        struct ByteBuffer
        {
            char buffer[1024];
            size_t count;
        };

        class BlockReader : public ht::graph::NodeBase
        {
        public:
            virtual void process(const ByteBuffer & buf) = 0;
        };

        class FileReader : public ht::graph::NodeBase
        {
            std::fstream stream_;
            BlockReader * next_;
        public:
            FileReader();

            /* Configure our input file, or use stdin. */
            void configure(const ht::options::Config *);

            /* Read blocks of data from the input and send it to the next node. */
            void start();

            /* Verify that we have somewhere to send blocks of data. */
            void configured();

            /* Connect this node to the next node in the graph. */
            void connect(ht::graph::Node *n);
        };
    }
}


#endif

---