leon 1 mês atrás
pai
commit
82c4c41b9a

+ 4 - 4
src/main.cpp

@@ -41,16 +41,16 @@ int main()
     std::shared_ptr<Node::DrawNode> draw_node     = std::make_shared<Node::DrawNode>("draw");
     std::shared_ptr<Node::HttpPushNode> push_node = std::make_shared<Node::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
     Node::LinkNode(src_node0, infer_node);
-    // Node::LinkNode(src_node1, infer_node);
-    // Node::LinkNode(src_node2, infer_node);
+    Node::LinkNode(src_node1, infer_node);
+    Node::LinkNode(src_node2, infer_node);
     Node::LinkNode(infer_node, draw_node);
     Node::LinkNode(draw_node, push_node);
     push_node->start();
     draw_node->start();
     infer_node->start();
     src_node0->start();
-    src_node1->start();
-    src_node2->start();
+    // src_node1->start();
+    // src_node2->start();
     getchar();
     return 0;
 }

+ 1 - 1
src/nodes/draw/drawNode.hpp

@@ -11,7 +11,7 @@ class DrawNode : public BaseNode
 public:
     DrawNode() = delete;
     DrawNode(const std::string& name) : BaseNode(name, NODE_TYPE::DES_NODE) {}
-    virtual ~DrawNode() { };
+    virtual ~DrawNode() { running_ = false; };
 
     void work() override;
 };

+ 1 - 1
src/nodes/infer/inferNode.hpp

@@ -13,7 +13,7 @@ class InferNode : public BaseNode
 public:
     InferNode() = delete;
     InferNode(const std::string& name) : BaseNode(name, NODE_TYPE::MID_NODE) {}
-    virtual ~InferNode() { };
+    virtual ~InferNode()  { running_ = false; };
 
     void set_model_instance(std::shared_ptr<Infer> model)
     {

+ 1 - 1
src/nodes/stream/streamNode.hpp

@@ -61,7 +61,7 @@ public:
             status_ = StreamStatus::OPENED;
         }
     }
-    virtual ~StreamNode() { };
+    virtual ~StreamNode() { running_ = false; };
 
     void set_stream_url(const std::string& stream_url)
     {