OpenNebula

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

const char * DefaultQuotas::table    = "system_attributes";

const char * DefaultQuotas::db_names = "name, body";

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

int DefaultQuotas::select(SqlDB *db)

{

    ostringstream oss;

    int rc;

    set_callback(

            static_cast<Callbackable::Callback>(&DefaultQuotas::select_cb));

    oss << "SELECT body FROM " << table << " WHERE name = '" << root_elem << "'";

    rc = db->exec(oss, this);

    unset_callback();

    if (rc != 0)

    {

        return -1;

    }

    return 0;

}

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

int DefaultQuotas::insert_replace(SqlDB *db, bool replace, string& error_str)

{

    ostringstream   oss;

    int    rc;

    string xml_body;

    char * sql_xml;

    sql_xml = db->escape_str(to_xml(xml_body).c_str());

    if ( sql_xml == 0 )

    {

        goto error_body;

    }

/*

    if ( validate_xml(sql_xml) != 0 )

    {

        goto error_xml;

    }

*/

    if ( replace )

    {

        oss << "REPLACE";

    }

    else

    {

        oss << "INSERT";

    }

    // Construct the SQL statement to Insert or Replace

    oss <<" INTO "<<table <<" ("<< db_names <<") VALUES ("

        << "'" <<   root_elem            << "',"

        << "'" <<   sql_xml             << "')";

    rc = db->exec(oss);

    db->free_str(sql_xml);

    return rc;

/*

error_xml:

    db->free_str(sql_xml);

    error_str = "Error transforming the Quotas to XML.";

    goto error_common;

*/

error_body:

    goto error_generic;

error_generic:

    error_str = "Error inserting Quotas in DB.";

error_common:

    return -1;

}

/* -------------------------------------------------------------------------- */

/* -------------------------------------------------------------------------- */

int DefaultQuotas::bootstrap(SqlDB * db)

{

    ostringstream oss;

    oss << "CREATE TABLE IF NOT EXISTS system_attributes ("

            "name VARCHAR(128) PRIMARY KEY, body TEXT)";

    return db->exec(oss);

}