Savepoint.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac 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 Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "Savepoint.hpp"
#include "tools/GenGuid.h"
#include <boost/algorithm/string.hpp>
#include <spdlog/details/format.h>

namespace Leosac
{
namespace db
{

static std::string fixup_uuid(const std::string &uuid)
{
    return boost::replace_all_copy(uuid, "-", "_");
}

Savepoint::Savepoint(odb::database &db)
    : Savepoint(db, fixup_uuid(gen_uuid()))
{
}

Savepoint::Savepoint(odb::database &db, const std::string &name)
    : db_(db)
    , name_(name)
{
    db_.execute(fmt::format("SAVEPOINT {}", name_));
}

Savepoint::~Savepoint()
{
    release();
}

void Savepoint::release()
{
    if (released_)
        return;

    db_.execute(fmt::format("RELEASE SAVEPOINT {}", name_));
    released_ = true;
}

bool Savepoint::rollback_to()
{
    if (released_)
        return false;

    db_.execute(fmt::format("ROLLBACK TO SAVEPOINT {}", name_));
    return true;
}
}
}