ibrcommon/ibrcommon/data/BLOB.cpp

Go to the documentation of this file.

/*
 * BLOB.cpp
 *
 *  Created on: 15.12.2009
 *      Author: morgenro
 */

#include "ibrcommon/data/BLOB.h"
#include "ibrcommon/thread/MutexLock.h"
#include "ibrcommon/Exceptions.h"
#include "ibrcommon/Logger.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <cstring>
#include <cerrno>

#ifdef __DEVELOPMENT_ASSERTIONS__
#include <cassert>
#endif

namespace ibrcommon
{
        ibrcommon::Semaphore BLOB::_filelimit(10);

        File BLOB::tmppath;

        BLOB::BLOB()
         : ibrcommon::Mutex(ibrcommon::Mutex::MUTEX_NORMAL), _locked(false), _refcount(0), _reflock(ibrcommon::Mutex::MUTEX_NORMAL)
        { }

        BLOB::~BLOB()
        {
        }

        std::ostream& BLOB::copy(std::ostream &output, std::istream &input, const size_t size, const size_t buffer_size)
        {
                // read payload
                char buffer[buffer_size];
                size_t remain = size;

                while (remain > 0)
                {
                        // something bad happened, abort!
                        if (input.bad())
                        {
                                std::stringstream errmsg; errmsg << "input stream went bad [" << std::strerror(errno) << "]; " << (size-remain) << " of " << size << " bytes copied";
                                throw ibrcommon::IOException(errmsg.str());
                        }

                        // reached EOF too early
                        if (input.eof())
                        {
                                std::stringstream errmsg; errmsg << "input stream reached EOF [" << std::strerror(errno) << "]; " << (size-remain) << " of " << size << " bytes copied";
                                throw ibrcommon::IOException(errmsg.str());
                        }

                        // check if the destination stream is ok
                        if (output.bad())
                        {
                                std::stringstream errmsg; errmsg << "output stream went bad [" << std::strerror(errno) << "]; " << (size-remain) << " of " << size << " bytes copied";
                                throw ibrcommon::IOException(errmsg.str());
                        }

                        // retry if the read failed
                        if (input.fail())
                        {
                                IBRCOMMON_LOGGER(warning) << "input stream failed [" << std::strerror(errno) << "]; " << (size-remain) << " of " << size << " bytes copied" << IBRCOMMON_LOGGER_ENDL;
                                input.clear();
                        }

                        // if the last write failed, then retry
                        if (output.fail())
                        {
                                IBRCOMMON_LOGGER(warning) << "output stream failed [" << std::strerror(errno) << "]; " << (size-remain) << " of " << size << " bytes copied" << IBRCOMMON_LOGGER_ENDL;
                                output.clear();
                        }
                        else
                        {
                                // read the full buffer size of less?
                                if (remain > buffer_size)
                                {
                                        input.read(buffer, buffer_size);
                                }
                                else
                                {
                                        input.read(buffer, remain);
                                }

                                // retry if the read failed
                                if (input.fail()) continue;
                        }

                        // write the bytes to the BLOB
                        output.write(buffer, input.gcount());

                        // shrink the remaining bytes by the red bytes
                        remain -= input.gcount();
                }

                return output;
        }

        BLOB::Reference::Reference(const Reference &ref)
         : ibrcommon::Mutex(MUTEX_NORMAL), _blob(ref._blob)
        {
                MutexLock l(_blob->_reflock);
                _blob->_refcount++;
        }

        BLOB::Reference::~Reference()
        {
                // check if this is the last reference
                {
                        MutexLock l(_blob->_reflock);
                        if ((_blob->_refcount--) > 1) return;
                }

                delete _blob;
        }

        BLOB::iostream BLOB::Reference::iostream()
        {
                return BLOB::iostream(_blob);
        }

        std::iostream& BLOB::Reference::operator*()
        {
#ifdef __DEVELOPMENT_ASSERTIONS__
                assert(_blob->_locked);
#endif
                return _blob->__get_stream();
        }

        BLOB::Reference::Reference(BLOB *blob)
         : _blob(blob)
        {
                MutexLock l(_blob->_reflock);
                _blob->_refcount++;
        }

        void BLOB::Reference::enter() throw (ibrcommon::MutexException)
        {
                // lock the stream
                _blob->enter();
                _blob->_locked = true;

                // open the stream
                _blob->open();
        }

        void BLOB::Reference::leave() throw (ibrcommon::MutexException)
        {
                _blob->_locked = false;

                // close the stream
                _blob->close();

                // unlock the stream
                _blob->leave();
        }

        void BLOB::Reference::trylock() throw (ibrcommon::MutexException)
        {
                // lock the stream
                _blob->trylock();
                _blob->_locked = true;

                // open the stream
                _blob->open();
        }

        size_t BLOB::Reference::getSize() const
        {
#ifdef __DEVELOPMENT_ASSERTIONS__
                assert(_blob->_locked);
#endif
                return _blob->__get_size();
        }

        void BLOB::Reference::clear()
        {
#ifdef __DEVELOPMENT_ASSERTIONS__
                assert(_blob->_locked);
#endif
                _blob->clear();
        }

        BLOB::Reference StringBLOB::create()
        {
                BLOB::Reference ref(new StringBLOB());
                return ref;
        }

        void StringBLOB::clear()
        {
                _stringstream.str("");
        }

        StringBLOB::StringBLOB()
         : BLOB(), _stringstream()
        {

        }

        StringBLOB::~StringBLOB()
        {
        }

        void StringBLOB::open()
        {
                // set pointer to the beginning of the stream and remove any error flags
                _stringstream.clear();
                _stringstream.seekp(0);
                _stringstream.seekg(0);
        }

        void StringBLOB::close()
        {
        }

        size_t StringBLOB::__get_size()
        {
                // store current position
                size_t pos = _stringstream.tellg();

                _stringstream.seekg(0, std::ios_base::end);
                size_t size = _stringstream.tellg();
                _stringstream.seekg(pos);

                return size;
        }

        BLOB::Reference FileBLOB::create(const File &f)
        {
                return BLOB::Reference(new FileBLOB(f));
        }

        void FileBLOB::clear()
        {
                throw ibrcommon::IOException("clear is not possible on a read only file");
        }

        FileBLOB::FileBLOB(const File &f)
         : BLOB(), _filestream(), _file(f)
        {
                if (!f.exists())
                {
                        throw ibrcommon::FileNotExistsException(f);
                }
        }

        FileBLOB::~FileBLOB()
        {
        }

        void FileBLOB::open()
        {
                BLOB::_filelimit.wait();

                // open the file
                _filestream.open(_file.getPath().c_str(), ios::in|ios::binary);

                if (!_filestream.is_open())
                {
                        throw ibrcommon::CanNotOpenFileException(_file);
                }
        }

        void FileBLOB::close()
        {
                // close the file
                _filestream.close();

                BLOB::_filelimit.post();
        }

        size_t FileBLOB::__get_size()
        {
                return _file.size();
        }

        BLOB::Reference TmpFileBLOB::create()
        {
                try {
                        return BLOB::Reference(new TmpFileBLOB());
                } catch (const ibrcommon::IOException&) {
                        return StringBLOB::create();
                }
        }

        void TmpFileBLOB::clear()
        {
                // close the file
                _filestream.close();

                // open temporary file
                _filestream.open(_tmpfile.getPath().c_str(), ios::in | ios::out | ios::trunc | ios::binary );

                if (!_filestream.is_open())
                {
                        IBRCOMMON_LOGGER(error) << "can not open temporary file " << _tmpfile.getPath() << IBRCOMMON_LOGGER_ENDL;
                        throw ibrcommon::CanNotOpenFileException(_tmpfile);
                }
        }

        TmpFileBLOB::TmpFileBLOB()
         : BLOB(), _filestream()
        {
                // check if path is a directory
                if (!BLOB::tmppath.isDirectory())
                {
                        throw ibrcommon::IOException("BLOB::tmppath not initialized or path is not a directory.");
                }

                createtmpfile();
        }

        void TmpFileBLOB::createtmpfile()
        {
                _tmpfile = ibrcommon::TemporaryFile(BLOB::tmppath, "blob");
                _tmpfile.update();
        }

        TmpFileBLOB::~TmpFileBLOB()
        {
                // delete the file if the last reference is destroyed
                _tmpfile.remove();
        }

        void TmpFileBLOB::open()
        {
                BLOB::_filelimit.wait();

                // open temporary file
                _filestream.open(_tmpfile.getPath().c_str(), ios::in | ios::out | ios::binary );

                if (!_filestream.is_open())
                {
                        IBRCOMMON_LOGGER(error) << "can not open temporary file " << _tmpfile.getPath() << IBRCOMMON_LOGGER_ENDL;
                        throw ibrcommon::CanNotOpenFileException(_tmpfile);
                }
        }

        void TmpFileBLOB::close()
        {
                // flush the filestream
                _filestream.flush();

                // close the file
                _filestream.close();

                BLOB::_filelimit.post();
        }

        size_t TmpFileBLOB::__get_size()
        {
                return _tmpfile.size();
        }
}