TempFile.h

// -*- mode: c++; c-basic-offset:4 -*-
 
// This file is part of the BES, A C++ implementation of the OPeNDAP Data
// Access Protocol.
 
// Copyright (c) 2018 OPeNDAP, Inc.
// Author: Nathan Potter <ndp@opendap.org>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//
// You can contact OPeNDAP, Inc. at PO Box 112, Saunderstown, RI. 02874-0112.
 
#ifndef DAP_TEMPFILE_H_
#define DAP_TEMPFILE_H_
 
#include <unistd.h>
 
#include <vector>
#include <string>
#include <map>
#include <mutex>
#include <memory>
 
namespace bes {
 
class TempFile {
private:
    // Lifecycle controls
    static struct sigaction cached_sigpipe_handler;
    mutable std::recursive_mutex d_tf_lock_mutex;
    static void init();
    static std::once_flag d_init_once;
 
    // Holds the static list of all open files
    static std::unique_ptr< std::map<std::string, int> > open_files;
 
    // Instance variables
    int d_fd = -1;
    std::string d_fname;
    bool d_keep_temps;
 
    static void mk_temp_dir(const std::string &dir_name = "/tmp/hyrax_tmp") ;
 
    friend class TemporaryFileTest;
 
public:
    // Odd, but even with TemporaryFileTest declared as a friend, the tests won't
    // compile unless this is declared public.
    static void sigpipe_handler(int signal);
 
    explicit TempFile(bool keep_temps = false);
    ~TempFile();
 
    std::string create(const std::string &dir_name = "/tmp/hyrax_tmp", const std::string &path_template = "opendap");
 
    int get_fd() const { return d_fd; }
 
    std::string get_name() const { return d_fname; }
};
 
} // namespace bes
 
#endif /* DAP_TEMPFILE_H_ */