mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
127 lines
3.2 KiB
C++
127 lines
3.2 KiB
C++
#include "../../Interface/Server.h"
|
|
#include "../../Interface/Database.h"
|
|
#include "../../Interface/Types.h"
|
|
#include "../server_settings.h"
|
|
#include "../../urbackupcommon/os_functions.h"
|
|
#include "../../stringtools.h"
|
|
#include "../server_cleanup.h"
|
|
|
|
#include "../database.h"
|
|
#include <stdlib.h>
|
|
|
|
void open_server_database(bool &use_berkeleydb);
|
|
void open_settings_database(bool use_berkeleydb);
|
|
|
|
int64 cleanup_amount(std::string cleanup_pc, IDatabase *db)
|
|
{
|
|
ServerSettings settings(db);
|
|
|
|
int64 total_space=os_total_space(settings.getSettings()->backupfolder);
|
|
|
|
if(total_space==-1)
|
|
{
|
|
Server->Log("Error getting free space", LL_ERROR);
|
|
return -1;
|
|
}
|
|
|
|
strupper(&cleanup_pc);
|
|
|
|
std::wstring wcleanup_pc=widen(cleanup_pc);
|
|
|
|
int64 cleanup_bytes=0;
|
|
if(cleanup_pc.find("%")!=std::string::npos)
|
|
{
|
|
double pc=atof(getuntil("%", cleanup_pc).c_str());
|
|
Server->Log("Cleaning up "+nconvert(pc)+" percent", LL_INFO);
|
|
|
|
cleanup_bytes=(int64)(pc*total_space+0.5);
|
|
}
|
|
else if(cleanup_pc.find("K")!=std::string::npos)
|
|
{
|
|
cleanup_bytes=watoi64(getuntil(L"K", wcleanup_pc))*1024;
|
|
}
|
|
else if(cleanup_pc.find("M")!=std::string::npos)
|
|
{
|
|
cleanup_bytes=watoi64(getuntil(L"M", wcleanup_pc))*1024*1024;
|
|
}
|
|
else if(cleanup_pc.find("G")!=std::string::npos)
|
|
{
|
|
cleanup_bytes=watoi64(getuntil(L"G", wcleanup_pc))*1024*1024*1024;
|
|
}
|
|
else if(cleanup_pc.find("T")!=std::string::npos)
|
|
{
|
|
cleanup_bytes=watoi64(getuntil(L"T", wcleanup_pc))*1024*1024*1024*1024;
|
|
}
|
|
else
|
|
{
|
|
cleanup_bytes=watoi64(getuntil(L"T", wcleanup_pc));
|
|
}
|
|
|
|
if(cleanup_bytes>total_space)
|
|
{
|
|
cleanup_bytes=total_space;
|
|
}
|
|
|
|
return cleanup_bytes;
|
|
}
|
|
|
|
int cleanup_cmd(void)
|
|
{
|
|
Server->Log("Shutting down all database instances...", LL_INFO);
|
|
Server->destroyAllDatabases();
|
|
|
|
Server->Log("Opening urbackup server database...", LL_INFO);
|
|
bool use_bdb;
|
|
open_server_database(use_bdb);
|
|
open_settings_database(use_bdb);
|
|
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER);
|
|
if(db==NULL)
|
|
{
|
|
Server->Log("Could not open database", LL_ERROR);
|
|
return 1;
|
|
}
|
|
|
|
Server->Log("Transitioning urbackup server database to different journaling mode...", LL_INFO);
|
|
db->Write("PRAGMA journal_mode = DELETE");
|
|
|
|
std::string cleanup_pc=Server->getServerParameter("cleanup_amount");
|
|
|
|
if(cleanup_pc=="true" || cleanup_pc.empty())
|
|
{
|
|
cleanup_pc="10%";
|
|
}
|
|
|
|
int64 cleanup_bytes=cleanup_amount(cleanup_pc, db);
|
|
|
|
if(cleanup_bytes<0)
|
|
{
|
|
return 3;
|
|
}
|
|
|
|
Server->Log("Cleaning up "+PrettyPrintBytes(cleanup_bytes)+" on backup storage", LL_INFO);
|
|
|
|
{
|
|
ServerSettings settings(db);
|
|
size_t cache_size=settings.getSettings()->update_stats_cachesize;
|
|
Server->Log("Database cache size is "+PrettyPrintBytes(cache_size*1024), LL_INFO);
|
|
}
|
|
|
|
Server->Log("Starting cleanup...", LL_INFO);
|
|
|
|
bool b = ServerCleanupThread::cleanupSpace(cleanup_bytes);
|
|
|
|
if(!b)
|
|
{
|
|
Server->Log("Cleanup failed. Could not remove backups. Please lower the minimal number of backups", LL_ERROR);
|
|
return 2;
|
|
}
|
|
|
|
Server->Log("Cleanup successfull.", LL_INFO);
|
|
|
|
Server->Log("Freeing database connections...", LL_INFO);
|
|
|
|
Server->destroyAllDatabases();
|
|
|
|
return 0;
|
|
} |