#include "nodes/base/base.hpp"
#include "nodes/infer/inferNode.hpp"
#include <unordered_map>

namespace Node
{

void InferNode::work()
{
    printf("InferNode %s\n", name_.c_str());
    while (running_)
    {
        
        for (auto& input_buffer : input_buffers_)
        {
            std::shared_ptr<meta::MetaData> metaData;
            if (!input_buffer.second->try_pop(metaData))
            {
                continue;
            }
            printf("Node %s get data from %s\n", name_.c_str(), input_buffer.first.c_str());
            // do something
            for (auto& output_buffer : output_buffers_)
            {
                printf("Node %s push data to %s\n", name_.c_str(), output_buffer.first.c_str());
                output_buffer.second->push(metaData);
            }
        }
    }
};

}   // namespace Node