Utils: Implement more flexible SystemException

[senf.git] / Scheduler / WriteHelper.ct

diff --git a/Scheduler/WriteHelper.ct b/Scheduler/WriteHelper.ct
index 8ef626d..afc09c6 100644 (file)
--- a/Scheduler/WriteHelper.ct
+++ b/Scheduler/WriteHelper.ct
@@ -1,53 +1,70 @@
 // $Id$
 //
-// Copyright (C) 2006 
+// Copyright (C) 2006 Stefan Bund <g0dil@berlios.de>
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; either version 2 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the
+// Free Software Foundation, Inc.,
+// 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+// Copyright (C) 2006
 
-// Definition of non-inline template functions
+/** \file
+    \brief WriteHelper non-inline template implementation */
 
 //#include "WriteHelper.ih"
 
 // Custom includes
 #include <errno.h>
-#include "Utils/Exception.hh"
+#include "../Utils/Exception.hh"
 #include "Scheduler.hh"
 
 #define prefix_
 ///////////////////////////////ct.p////////////////////////////////////////
 
 template <class Handle>
-prefix_ satcom::lib::WriteHelper<Handle>::WriteHelper(Handle handle, std::string data,
+prefix_ senf::WriteHelper<Handle>::WriteHelper(Handle handle, std::string const & data,
                                                       Callback callback)
     : handle_(handle), data_(data), callback_(callback),
-      offset_(0), errno_(0)
+      offset_(data_.begin()), errno_(0)
 {
-    satcom::lib::Scheduler::instance()
-       .add(handle_, boost::bind(&WriteHelper::dispatchProcess, ptr(this), _1, _2),
-            satcom::lib::Scheduler::EV_WRITE);
+    senf::Scheduler::instance()
+        .add(handle_, boost::bind(&WriteHelper::dispatchProcess, ptr(this), _1, _2),
+             senf::Scheduler::EV_WRITE);
 }
 
 template <class Handle>
-prefix_ std::string const & satcom::lib::WriteHelper<Handle>::data()
+prefix_ std::string const & senf::WriteHelper<Handle>::data()
     const
 {
-    if (offset_ > 0) {
-       data_.erase(0,offset_);
-       offset_ = 0;
+    if (offset_ > data_.begin()) {
+        data_.erase(data_.begin(),offset_);
+        offset_ = data_.begin();
     }
     return data_;
 }
 
 template <class Handle>
-prefix_ void satcom::lib::WriteHelper<Handle>::revoke()
+prefix_ void senf::WriteHelper<Handle>::revoke()
 {
     ptr guard (this); // To ensure, 'this' is deleted only after this method terminates ...
-    satcom::lib::Scheduler::instance()
-       .remove(handle_, satcom::lib::Scheduler::EV_WRITE);
+    senf::Scheduler::instance()
+        .remove(handle_, senf::Scheduler::EV_WRITE);
 }
 
 template <class Handle>
 prefix_ void
-satcom::lib::WriteHelper<Handle>::dispatchProcess(ptr helper, Handle handle,
-                                                  satcom::lib::Scheduler::EventId event)
+senf::WriteHelper<Handle>::dispatchProcess(ptr helper, Handle handle,
+                                                  senf::Scheduler::EventId event)
 {
     // since we have a 'ptr' argument, the instance cannot be deleted
     // before this method returns
@@ -55,26 +72,31 @@ satcom::lib::WriteHelper<Handle>::dispatchProcess(ptr helper, Handle handle,
 }
 
 template <class Handle>
-prefix_ void satcom::lib::WriteHelper<Handle>::process(Handle handle,
-                                                       satcom::lib::Scheduler::EventId event)
+prefix_ void senf::WriteHelper<Handle>::process(Handle handle,
+                                                       senf::Scheduler::EventId event)
 {
+    bool complete_ (false);
     try {
-       if (event != satcom::lib::Scheduler::EV_WRITE)
-           throw satcom::lib::SystemException(EPIPE);
-       offset_ += handle.write(data_.data()+offset_,data_.size()-offset_);
-       if (offset_ >= data_.size()) {
-           data_.erase();
-           done();
-       }
+        if (event != senf::Scheduler::EV_WRITE)
+            throwErrno(EPIPE);
+        offset_ = handle.write(std::make_pair(offset_,data_.end()));
+        if (offset_ == data_.end()) {
+            data_.erase();
+            offset_ = data_.begin();
+            complete_ = true;
+        }
     }
-    catch (satcom::lib::SystemException const & ex) {
-       errno_ = ex.err;
-       done();
+    catch (senf::SystemException const & ex) {
+        errno_ = ex.code();
+        done();
+        return;
     }
+    if (complete_)
+        done();
 }
 
 template <class Handle>
-prefix_ void satcom::lib::WriteHelper<Handle>::done()
+prefix_ void senf::WriteHelper<Handle>::done()
 {
     revoke();
     callback_(ptr(this));
@@ -86,4 +108,10 @@ prefix_ void satcom::lib::WriteHelper<Handle>::done()
 \f
 // Local Variables:
 // mode: c++
+// fill-column: 100
+// c-file-style: "senf"
+// indent-tabs-mode: nil
+// ispell-local-dictionary: "american"
+// compile-command: "scons -u test"
+// comment-column: 40
 // End: