#include "UpdateStatementBuilder.h"

namespace db {
  namespace statements {
    namespace builder {
      UpdateStatementBuilder &UpdateStatementBuilder::table(const std::string &table) {
        m_statement.body += "UPDATE " + table;
        return *this;
      }

      UpdateStatementBuilder &UpdateStatementBuilder::set(const std::string &set) {
        m_statement.body += " SET " + set + " ";
        return *this;
      }

      UpdateStatementBuilder &UpdateStatementBuilder::where(const std::string &where) {
        m_statement.body += " WHERE " + where;
        return *this;
      }

      UpdateStatementBuilder &UpdateStatementBuilder::close() {
        m_statement.body += ";\n";
        return *this;
      }
    }
  }
}