Socket: Ignore ECONNREFUSED on write to datagram socket
[senf.git] / Examples / RateStuffer / ratestuffer.cc
index ea29427..7969d59 100644 (file)
 #include "PPI/Module.hh"
 #include "PPI/IntervalTimer.hh"
 #include "PPI/Joins.hh"
+#include "PPI/ThrottleBarrier.hh"
 #include "PPI/PassiveQueue.hh"
+#include "PPI/Queueing.hh"
+#include "PPI/CloneSource.hh"
 #include "PPI/Setup.hh"
 
 //#include "ppitest.mpp"
@@ -46,61 +49,90 @@ namespace module = senf::ppi::module;
 namespace connector = senf::ppi::connector;
 namespace ppi = senf::ppi;
 
-namespace {
+// ////////////////////////////////////////////////////////////////////////
+// RateFilter
 
-    class RateFilter
-        : public module::Module
-    {
-        SENF_PPI_MODULE(RateFilter);
-    public:
-
-        connector::ActiveInput input;
-        connector::ActiveOutput output;
+class RateFilter
+    : public module::Module
+{
+    SENF_PPI_MODULE(RateFilter);
+public:
 
-        RateFilter(senf::ClockService::clock_type interval) : timer(interval) {
-            route(input,output);
-            route(input,timer);
-            registerEvent(&RateFilter::timeout, timer);
-        }
+    connector::ActiveInput input;
+    connector::ActiveOutput output;
 
-    private:
-        void timeout() {
-            output(input());
-        }
+    RateFilter(senf::ClockService::clock_type interval);
 
-        ppi::IntervalTimer timer;
-    };
+private:
+    void timeout();
 
-    class CopyPacketGenerator
-        : public module::Module
-    {
-        SENF_PPI_MODULE(CopyPacketGenerator);
-    public:
+    ppi::IntervalTimer timer;
+};
 
-        connector::PassiveOutput output;
+RateFilter::RateFilter(senf::ClockService::clock_type interval)
+    : timer(interval) 
+{
+    route(input,timer);
+    route(timer,output);
+    registerEvent(&RateFilter::timeout, timer);
+}
 
-        CopyPacketGenerator(senf::Packet p) : packet(p) {
-            noroute(output);
-            output.onRequest(&CopyPacketGenerator::request);
-        }
+void RateFilter::timeout()
+{
+    output(input());
+}
 
-    private:
-        void request() {
-            output(packet);
-        }
+// ////////////////////////////////////////////////////////////////////////
 
-        senf::Packet packet;
-    };
-}
+class RateStuffer
+{
+    module::ThrottleBarrier barrier;
+    module::PassiveQueue    queue;
+    module::CloneSource     generator;
+    module::PriorityJoin    join;
+    RateFilter              rateFilter;
+
+public:
+    connector::PassiveInput & input;
+    connector::ActiveOutput & output;
+
+    RateStuffer(senf::ClockService::clock_type interval, 
+                senf::Packet packet,
+                unsigned high = 1,
+                unsigned low  = 0)
+    :   barrier    (),
+        queue      (),
+        generator  ( packet ),
+        join       (),
+        rateFilter ( interval ),
+        input      ( barrier.input ),
+        output     ( rateFilter.output )
+    {
+        ppi::connect( barrier,    queue      );
+        ppi::connect( queue,      join       );
+        ppi::connect( generator,  join       );
+        ppi::connect( join,       rateFilter );
+
+        queue.qdisc(ppi::ThresholdQueueing(high,low));
+    }
+};
+        
+// ////////////////////////////////////////////////////////////////////////
+// ////////////////////////////////////////////////////////////////////////
 
 // Module setup:
 //
 // 'O'        = active connector
 // '>' or '<' = input connector
 //
-// [ udpReader ] O--> [ queue ] -->O [      ]
-//                                   [ join ] -->O [ rateFilter] O--> [ udpWriter ]
-//                [ generator ] -->O [      ]
+//                   +----------------------------------------------------+
+//                   | stuffer                                            |
+//                   |                                                    |
+// [ udpReader ] O-->:---> [ queue ] -->O [      ]                        |
+//                   |                    [ join ] -->O [ rateFilter] O-->:O--> [ udpWriter ]
+//                   | [ generator ] -->O [      ]                        |
+//                   |                                                    |
+//                   +----------------------------------------------------+
 
 int main(int argc, char * argv[])
 {
@@ -110,18 +142,14 @@ int main(int argc, char * argv[])
     senf::ConnectedUDPv4ClientSocketHandle outputSocket(
         senf::INet4SocketAddress("localhost:44345"));
 
-    module::ActiveSocketReader<>  udpReader  (inputSocket);
-    module::PassiveQueue          queue;
-    CopyPacketGenerator           generator  (senf::DataPacket::create(std::string("<idle>\n")));
-    module::PriorityJoin          join;
-    RateFilter                    rateFilter (1000000000ul);
-    module::PassiveSocketWriter<> udpWriter  (outputSocket);
-
-    ppi::connect( udpReader,  queue      );
-    ppi::connect( queue,      join       );
-    ppi::connect( generator,  join       );
-    ppi::connect( join,       rateFilter );
-    ppi::connect( rateFilter, udpWriter  );
+    module::ActiveSocketReader<>  udpReader  ( inputSocket );
+    RateStuffer                   stuffer    ( 1000000000ul, 
+                                               senf::DataPacket::create(std::string("<idle>\n")),
+                                               2u, 1u );
+    module::PassiveSocketWriter<> udpWriter  ( outputSocket );
+
+    ppi::connect( udpReader, stuffer   );
+    ppi::connect( stuffer,   udpWriter );
 
     ppi::run();