WriteHelper.ct

Go to the documentation of this file.

//
// Copyright (c) 2020 Fraunhofer Institute for Applied Information Technology (FIT)
//                    Network Research Group (NET)
//                    Schloss Birlinghoven, 53754 Sankt Augustin, GERMANY
//                    Contact: support@wiback.org
//
// This file is part of the SENF code tree.
// It is licensed under the 3-clause BSD License (aka New BSD License).
// See LICENSE.txt in the top level directory for details or visit
// https://opensource.org/licenses/BSD-3-Clause
//


/** \file
    \brief WriteHelper non-inline template implementation */

//#include "WriteHelper.ih"

// Custom includes
#include <boost/bind.hpp>
#include <senf/Utils/Exception.hh>

#define prefix_
//-/////////////////////////////////////////////////////////////////////////////////////////////////

template <class Handle>
prefix_ senf::WriteHelper<Handle>::WriteHelper(Handle handle, std::string const & data,
                                                      Callback callback)
    : handle_(handle),
      fde_("WriteHelper", boost::bind(&WriteHelper::dispatchProcess, ptr(this), _1, _2),
           handle, scheduler::FdEvent::EV_WRITE),
      data_(data), callback_(callback), offset_(data_.begin()), errno_(0)
{}

template <class Handle>
prefix_ std::string const & senf::WriteHelper<Handle>::data()
    const
{
    if (offset_ > data_.begin()) {
        data_.erase(data_.begin(),offset_);
        offset_ = data_.begin();
    }
    return data_;
}

template <class Handle>
prefix_ void senf::WriteHelper<Handle>::revoke()
{
    ptr guard (this); // To ensure, 'this' is deleted only after this method terminates ...
    fde_.disable();
    fde_.action(0); // To remove the smart pointer reference to this
}

template <class Handle>
prefix_ void senf::WriteHelper<Handle>::dispatchProcess(ptr helper, Handle handle,
                                                        scheduler::FdEvent::Events event)
{
    // since we have a 'ptr' argument, the instance cannot be deleted
    // before this method returns
    helper->process(handle, event);
}

template <class Handle>
prefix_ void senf::WriteHelper<Handle>::process(Handle handle,
                                                scheduler::FdEvent::Events event)
{
    bool complete_ (false);
    try {
        if (event != scheduler::FdEvent::EV_WRITE)
            throw SystemException(EPIPE SENF_EXC_DEBUGINFO);
        offset_ = handle.write(std::make_pair(offset_,data_.end()));
        if (offset_ == data_.end()) {
            data_.erase();
            offset_ = data_.begin();
            complete_ = true;
        }
    }
    catch (senf::SystemException const & ex) {
        errno_ = ex.errorNumber();
        done();
        return;
    }
    if (complete_)
        done();
}

template <class Handle>
prefix_ void senf::WriteHelper<Handle>::done()
{
    revoke();
    callback_(ptr(this));
}

//-/////////////////////////////////////////////////////////////////////////////////////////////////
#undef prefix_


// 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: