mysql5/mysql-5.7.27/storage/ndb/test/include/NDBT_Workingdir.hpp

62 lines
1.6 KiB
C++

/*
Copyright 2009 Sun Microsystems, Inc.
All rights reserved. Use is subject to license terms.
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; version 2 of the License.
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef NDBT_WORKINGDIR_HPP
#define NDBT_WORKINGDIR_HPP
#include <NdbDir.hpp>
#include <BaseString.hpp>
class NDBT_Workingdir
{
NdbDir::Temp m_temp;
BaseString m_wd;
public:
NDBT_Workingdir(const char* dirname)
{
const char* tmp_path = m_temp.path();
char* ndbt_tmp = getenv("NDBT_TMP_DIR");
if (ndbt_tmp)
tmp_path = ndbt_tmp;
require(tmp_path);
m_wd.assfmt("%s%s%s%d", tmp_path, DIR_SEPARATOR, dirname,
(int)NdbProcess::getpid());
if (access(m_wd.c_str(), F_OK) == 0)
NdbDir::remove_recursive(m_wd.c_str());
if (!NdbDir::create(m_wd.c_str()))
abort();
}
~NDBT_Workingdir()
{
if (access(m_wd.c_str(), F_OK) == 0)
NdbDir::remove_recursive(m_wd.c_str());
}
const char* path(void) const
{
return m_wd.c_str();
}
};
#endif