soa/logger/file_output.h

/* file_output.h                                                   -*- C++ -*-
   Jeremy Barnes, 29 May 2011
   Copyright (c) 2011 Datacratic.  All rights reserved.

   Get the output from the given file.
*/

#ifndef __logger__file_output_h__
#define __logger__file_output_h__

#include "logger.h"
#include "rotating_output.h"
#include "compressing_output.h"
#include "soa/service/s3.h"


namespace Datacratic {


/*****************************************************************************/
/* FILE OUTPUT                                                               */
/*****************************************************************************/

struct NamedOutput : public CompressingOutput {

    NamedOutput(size_t ringBufferSize = 65536);

    virtual ~NamedOutput();

    virtual void open(const std::string & filename,
                      const std::string & compression = "",
                      int level = -1);

    void rotate(const std::string & newFilename,
                const std::string & newCompression = "",
                int newLevel = -1);

    virtual void close();

    boost::function<void (std::string)> onPreFileOpen;
    boost::function<void (std::string)> onPostFileOpen;
    boost::function<void (std::string)> onPreFileClose;
    boost::function<void (std::string)> onPostFileClose;

protected:

    virtual std::shared_ptr<Sink>
    createSink(const std::string & filename, bool append) = 0;

    void switchFile(const std::string & filename,
                    const std::string & compression,
                    int level);

    void closeFile();
};


/*****************************************************************************/
/* FILE SINK                                                                 */
/*****************************************************************************/

struct FileSink : public CompressingOutput::Sink {
    FileSink(const std::string & filename = "",
             bool append = true,
             bool disambiguate = true);

    virtual ~FileSink();

    void open(const std::string & filename,
              bool append,
              bool disambiguate);

    virtual void close();
        
    virtual size_t write(const char * data, size_t size);

    virtual size_t flush(FileFlushLevel flushLevel);

    std::string currentFilename;

    int fd;
};


/*****************************************************************************/
/* FILE OUTPUT                                                               */
/*****************************************************************************/

struct FileOutput : public NamedOutput {

    FileOutput(const std::string & filename = "",
               size_t ringBufferSize = 65536);

    virtual ~FileOutput();

    virtual std::shared_ptr<Sink>
    createSink(const std::string & filename, bool append);
};


/*****************************************************************************/
/* ROTATING FILE OUTPUT                                                      */
/*****************************************************************************/

struct RotatingFileOutput : public RotatingOutputAdaptor {

    RotatingFileOutput();

    virtual ~RotatingFileOutput();
    
    void open(const std::string & filenamePattern,
              const std::string & periodPattern,
              const std::string & compression = "",
              int level = -1);
    
private:
    FileOutput * createFile(const std::string & filename);

    std::string compression;
    int level;
};

} // namespace Datacratic


#endif /* __logger__file_output_h__ */