When closing unused inputs, remember to take down their sockets.

[cubemap] / udpinput.h

diff --git a/udpinput.h b/udpinput.h
index 0f89d6828de79b9be3e27db77ed2d7efcf93fb70..dc91bc2bfd6d3c5befbee6df37635c435b113786 100644 (file)
--- a/udpinput.h
+++ b/udpinput.h
@@ -1,8 +1,8 @@
 #ifndef _UDPINPUT_H
 #define _UDPINPUT_H 1
 
-#include <vector>
 #include <string>
+#include <vector>
 
 #include "input.h"
 
@@ -10,7 +10,7 @@ class InputProto;
 
 class UDPInput : public Input {
 public:
-       UDPInput(const std::string &stream_id, const std::string &url);
+       UDPInput(const std::string &url);
 
        // Serialization/deserialization.
        UDPInput(const InputProto &serialized);
@@ -19,6 +19,11 @@ public:
        virtual std::string get_url() const { return url; }
        virtual void close_socket();
 
+       virtual void add_destination(const std::string &stream_id)
+       {
+               stream_ids.push_back(stream_id);
+       }
+
 private:
        // Actually gets the packets.
        virtual void do_work();
@@ -26,7 +31,7 @@ private:
        // Create the HTTP header.
        void construct_header();
 
-       std::string stream_id;
+       std::vector<std::string> stream_ids;
 
        // The URL and its parsed components.
        std::string url;