boinc/db/boinc_db.C

974 lines
28 KiB
C

// The contents of this file are subject to the BOINC Public License
// Version 1.0 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// http://boinc.berkeley.edu/license_1.0.txt
//
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
// License for the specific language governing rights and limitations
// under the License.
//
// The Original Code is the Berkeley Open Infrastructure for Network Computing.
//
// The Initial Developer of the Original Code is the SETI@home project.
// Portions created by the SETI@home project are Copyright (C) 2002
// University of California at Berkeley. All Rights Reserved.
//
// Contributor(s):
//
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <unistd.h>
#include <cmath>
#include "util.h"
#include "boinc_db.h"
#ifdef _USING_FCGI_
#include "fcgi_stdio.h"
#endif
DB_CONN boinc_db;
static struct random_init {
random_init() {
srand48(getpid() + time(0));
}
} random_init;
#define ESCAPE(x) escape_string(x, sizeof(x))
#define UNESCAPE(x) unescape_string(x, sizeof(x))
void PLATFORM::clear() {memset(this, 0, sizeof(*this));}
void CORE_VERSION::clear() {memset(this, 0, sizeof(*this));}
void APP::clear() {memset(this, 0, sizeof(*this));}
void APP_VERSION::clear() {memset(this, 0, sizeof(*this));}
void USER::clear() {memset(this, 0, sizeof(*this));}
void TEAM::clear() {memset(this, 0, sizeof(*this));}
void HOST::clear() {memset(this, 0, sizeof(*this));}
void RESULT::clear() {memset(this, 0, sizeof(*this));}
void WORKUNIT::clear() {memset(this, 0, sizeof(*this));}
void MSG_FROM_HOST::clear() {memset(this, 0, sizeof(*this));}
void MSG_TO_HOST::clear() {memset(this, 0, sizeof(*this));}
void TRANSITIONER_ITEM::clear() {memset(this, 0, sizeof(*this));}
void SCHED_RESULT_ITEM::clear() {memset(this, 0, sizeof(*this));}
DB_PLATFORM::DB_PLATFORM() : DB_BASE(boinc_db, "platform"){}
DB_CORE_VERSION::DB_CORE_VERSION() : DB_BASE(boinc_db, "core_version"){}
DB_APP::DB_APP() : DB_BASE(boinc_db, "app"){}
DB_APP_VERSION::DB_APP_VERSION() : DB_BASE(boinc_db, "app_version"){}
DB_USER::DB_USER() : DB_BASE(boinc_db, "user"){}
DB_TEAM::DB_TEAM() : DB_BASE(boinc_db, "team"){}
DB_HOST::DB_HOST() : DB_BASE(boinc_db, "host"){}
DB_WORKUNIT::DB_WORKUNIT() : DB_BASE(boinc_db, "workunit"){}
DB_RESULT::DB_RESULT() : DB_BASE(boinc_db, "result"){}
DB_MSG_FROM_HOST::DB_MSG_FROM_HOST() : DB_BASE(boinc_db, "msg_from_host"){}
DB_MSG_TO_HOST::DB_MSG_TO_HOST() : DB_BASE(boinc_db, "msg_to_host"){}
DB_TRANSITIONER_ITEM_SET::DB_TRANSITIONER_ITEM_SET() : DB_BASE_SPECIAL(boinc_db){}
DB_WORK_ITEM::DB_WORK_ITEM() : DB_BASE_SPECIAL(boinc_db){}
DB_SCHED_RESULT_ITEM_SET::DB_SCHED_RESULT_ITEM_SET() : DB_BASE_SPECIAL(boinc_db){}
int DB_PLATFORM::get_id() {return id;}
int DB_CORE_VERSION::get_id() {return id;}
int DB_APP::get_id() {return id;}
int DB_APP_VERSION::get_id() {return id;}
int DB_USER::get_id() {return id;}
int DB_TEAM::get_id() {return id;}
int DB_HOST::get_id() {return id;}
int DB_WORKUNIT::get_id() {return id;}
int DB_RESULT::get_id() {return id;}
int DB_MSG_FROM_HOST::get_id() {return id;}
int DB_MSG_TO_HOST::get_id() {return id;}
void DB_PLATFORM::db_print(char* buf){
sprintf(buf,
"create_time=%d, name='%s', user_friendly_name='%s', "
"deprecated=%d",
create_time, name, user_friendly_name,
deprecated
);
}
void DB_PLATFORM::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time=atol(r[i++]);
strcpy2(name, r[i++]);
strcpy2(user_friendly_name, r[i++]);
deprecated=atol(r[i++]);
}
void DB_CORE_VERSION::db_print(char* buf) {
sprintf(buf,
"create_time=%d, version_num=%d, platformid=%d, "
"xml_doc='%s', message='%s', deprecated=%d",
create_time, version_num, platformid,
xml_doc, message, deprecated?1:0
);
}
void DB_CORE_VERSION::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
version_num = atoi(r[i++]);
platformid = atoi(r[i++]);
strcpy2(xml_doc, r[i++]);
strcpy2(message, r[i++]);
deprecated = atoi(r[i++]);
}
void DB_APP::db_print(char* buf){
sprintf(buf,
"create_time=%d, name='%s', min_version=%d, "
"deprecated=%d, user_friendly_name='%s'",
create_time, name, min_version,
deprecated, user_friendly_name
);
}
void DB_APP::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
strcpy2(name, r[i++]);
min_version = atoi(r[i++]);
deprecated = atoi(r[i++]);
strcpy2(user_friendly_name, r[i++]);
}
void DB_APP_VERSION::db_print(char* buf){
sprintf(buf,
"create_time=%d, appid=%d, version_num=%d, platformid=%d, "
"xml_doc='%s', "
"min_core_version=%d, max_core_version=%d, deprecated=%d",
create_time, appid, version_num, platformid,
xml_doc,
min_core_version, max_core_version, deprecated
);
}
void DB_APP_VERSION::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
appid = atoi(r[i++]);
version_num = atoi(r[i++]);
platformid = atoi(r[i++]);
strcpy2(xml_doc, r[i++]);
min_core_version = atoi(r[i++]);
max_core_version = atoi(r[i++]);
deprecated = atoi(r[i++]);
}
void DB_USER::db_print(char* buf){
ESCAPE(email_addr);
ESCAPE(name);
ESCAPE(country);
ESCAPE(postal_code);
ESCAPE(global_prefs);
ESCAPE(project_prefs);
ESCAPE(url);
ESCAPE(signature);
sprintf(buf,
"create_time=%d, email_addr='%s', name='%s', "
"authenticator='%s', "
"country='%s', postal_code='%s', "
"total_credit=%.15e, expavg_credit=%.15e, expavg_time=%.15e, "
"global_prefs='%s', project_prefs='%s', "
"teamid=%d, venue='%s', url='%s', send_email=%d, show_hosts=%d, "
"posts=%d, "
"seti_id=%d, seti_nresults=%d, seti_last_result_time=%d, "
"seti_total_cpu=%.15e, signature='%s', has_profile=%d, "
"cross_project_id='%s'",
create_time, email_addr, name,
authenticator,
country, postal_code,
total_credit, expavg_credit, expavg_time,
global_prefs, project_prefs,
teamid, venue, url, send_email, show_hosts,
posts,
seti_id, seti_nresults, seti_last_result_time,
seti_total_cpu, signature, has_profile,
cross_project_id
);
UNESCAPE(email_addr);
UNESCAPE(name);
UNESCAPE(country);
UNESCAPE(postal_code);
UNESCAPE(global_prefs);
UNESCAPE(project_prefs);
UNESCAPE(url);
UNESCAPE(signature);
}
void DB_USER::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id = atoi(r[i++]);
create_time = atoi(r[i++]);
strcpy2(email_addr, r[i++]);
strcpy2(name, r[i++]);
strcpy2(authenticator, r[i++]);
strcpy2(country, r[i++]);
strcpy2(postal_code, r[i++]);
total_credit = atof(r[i++]);
expavg_credit = atof(r[i++]);
expavg_time = atof(r[i++]);
strcpy2(global_prefs, r[i++]);
strcpy2(project_prefs, r[i++]);
teamid = atoi(r[i++]);
strcpy2(venue, r[i++]);
strcpy2(url, r[i++]);
send_email = atoi(r[i++]);
show_hosts = atoi(r[i++]);
posts = safe_atoi(r[i++]);
seti_id = safe_atoi(r[i++]);
seti_nresults = safe_atoi(r[i++]);
seti_last_result_time = safe_atoi(r[i++]);
seti_total_cpu = safe_atof(r[i++]);
strcpy2(signature, r[i++]);
has_profile = atoi(r[i++]);
strcpy2(cross_project_id, r[i++]);
}
void DB_TEAM::db_print(char* buf){
ESCAPE(name);
ESCAPE(name_lc);
ESCAPE(url);
ESCAPE(name_html);
ESCAPE(description);
sprintf(buf,
"create_time=%d, userid=%d, name='%s', "
"name_lc='%s', url='%s', "
"type=%d, name_html='%s', description='%s', nusers=%d, "
"country='%s', "
"total_credit=%.15e, expavg_credit=%.15e, expavg_time=%.15e, "
"seti_id=%d",
create_time,
userid,
name,
name_lc,
url,
type,
name_html,
description,
nusers,
country,
total_credit,
expavg_credit,
expavg_time,
seti_id
);
UNESCAPE(name);
UNESCAPE(name_lc);
UNESCAPE(url);
UNESCAPE(name_html);
UNESCAPE(description);
}
void DB_TEAM::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id = atoi(r[i++]);
create_time = atoi(r[i++]);
userid = atoi(r[i++]);
strcpy2(name, r[i++]);
strcpy2(name_lc, r[i++]);
strcpy2(url, r[i++]);
type = atoi(r[i++]);
strcpy2(name_html, r[i++]);
strcpy2(description, r[i++]);
nusers = atoi(r[i++]);
strcpy2(country, r[i++]);
total_credit = atof(r[i++]);
expavg_credit = atof(r[i++]);
expavg_time = atof(r[i++]);
seti_id = safe_atoi(r[i++]);
}
// set NaNs to a reasonable value
void HOST::fix_nans() {
if (isnan(p_fpops)) p_fpops = 0;
if (isnan(p_iops)) p_iops = 0;
if (isnan(p_membw)) p_membw = 0;
if (isnan(m_nbytes)) m_nbytes = 0;
if (isnan(m_cache)) m_cache = 0;
if (isnan(m_swap)) m_swap = 0;
if (isnan(d_total)) d_total = 0;
if (isnan(d_free)) d_free = 0;
if (isnan(d_boinc_used_total)) d_boinc_used_total = 0;
if (isnan(d_boinc_used_project)) d_boinc_used_project = 0;
if (isnan(d_boinc_max)) d_boinc_max = 0;
if (isnan(n_bwup)) n_bwup = 0;
if (isnan(n_bwdown)) n_bwdown = 0;
}
void DB_HOST::db_print(char* buf){
ESCAPE(domain_name);
ESCAPE(serialnum);
ESCAPE(last_ip_addr);
ESCAPE(p_vendor);
ESCAPE(p_model);
ESCAPE(os_name);
ESCAPE(os_version);
sprintf(buf,
"create_time=%d, userid=%d, "
"rpc_seqno=%d, rpc_time=%d, "
"total_credit=%.12e, expavg_credit=%.12e, expavg_time=%.15e, "
"timezone=%d, domain_name='%s', serialnum='%s', "
"last_ip_addr='%s', nsame_ip_addr=%d, "
"on_frac=%.15e, connected_frac=%.15e, active_frac=%.15e, "
"p_ncpus=%d, p_vendor='%s', p_model='%s', "
"p_fpops=%.15e, p_iops=%.15e, p_membw=%.15e, "
"os_name='%s', os_version='%s', "
"m_nbytes=%.15e, m_cache=%.15e, m_swap=%.15e, "
"d_total=%.15e, d_free=%.15e, "
"d_boinc_used_total=%.15e, d_boinc_used_project=%.15e, d_boinc_max=%.15e, "
"n_bwup=%.15e, n_bwdown=%.15e, "
"credit_per_cpu_sec=%.15e, "
"venue='%s', projects='%s', nresults_today=%d",
create_time, userid,
rpc_seqno, rpc_time,
total_credit, expavg_credit, expavg_time,
timezone, domain_name, serialnum,
last_ip_addr, nsame_ip_addr,
on_frac, connected_frac, active_frac,
p_ncpus, p_vendor, p_model,
p_fpops, p_iops, p_membw,
os_name, os_version,
m_nbytes, m_cache, m_swap,
d_total, d_free,
d_boinc_used_total, d_boinc_used_project, d_boinc_max,
n_bwup, n_bwdown,
credit_per_cpu_sec,
venue, projects, nresults_today
);
UNESCAPE(domain_name);
UNESCAPE(serialnum);
UNESCAPE(last_ip_addr);
UNESCAPE(p_vendor);
UNESCAPE(p_model);
UNESCAPE(os_name);
UNESCAPE(os_version);
}
void DB_HOST::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
userid = atoi(r[i++]);
rpc_seqno = atoi(r[i++]);
rpc_time = atoi(r[i++]);
total_credit = atof(r[i++]);
expavg_credit = atof(r[i++]);
expavg_time = atof(r[i++]);
timezone = atoi(r[i++]);
strcpy2(domain_name, r[i++]);
strcpy2(serialnum, r[i++]);
strcpy2(last_ip_addr, r[i++]);
nsame_ip_addr = atoi(r[i++]);
on_frac = atof(r[i++]);
connected_frac = atof(r[i++]);
active_frac = atof(r[i++]);
p_ncpus = atoi(r[i++]);
strcpy2(p_vendor, r[i++]);
strcpy2(p_model, r[i++]);
p_fpops = atof(r[i++]);
p_iops = atof(r[i++]);
p_membw = atof(r[i++]);
strcpy2(os_name, r[i++]);
strcpy2(os_version, r[i++]);
m_nbytes = atof(r[i++]);
m_cache = atof(r[i++]);
m_swap = atof(r[i++]);
d_total = atof(r[i++]);
d_free = atof(r[i++]);
d_boinc_used_total = atof(r[i++]);
d_boinc_used_project = atof(r[i++]);
d_boinc_max = atof(r[i++]);
n_bwup = atof(r[i++]);
n_bwdown = atof(r[i++]);
credit_per_cpu_sec = atof(r[i++]);
strcpy2(venue, r[i++]);
strcpy2(projects, r[i++]);
nresults_today = atoi(r[i++]);
}
void DB_WORKUNIT::db_print(char* buf){
sprintf(buf,
"create_time=%d, appid=%d, "
"name='%s', xml_doc='%s', batch=%d, "
"rsc_fpops_est=%.15e, rsc_fpops_bound=%.15e, "
"rsc_memory_bound=%.15e, rsc_disk_bound=%.15e, "
"need_validate=%d, "
"canonical_resultid=%d, canonical_credit=%.15e, "
"transition_time=%d, delay_bound=%d, "
"error_mask=%d, file_delete_state=%d, assimilate_state=%d, "
"workseq_next=%d, opaque=%f, "
"min_quorum=%d, target_nresults=%d, max_error_results=%d, "
"max_total_results=%d, max_success_results=%d, "
"result_template_file='%s'",
create_time, appid,
name, xml_doc, batch,
rsc_fpops_est, rsc_fpops_bound, rsc_memory_bound, rsc_disk_bound,
need_validate,
canonical_resultid, canonical_credit,
transition_time, delay_bound,
error_mask, file_delete_state, assimilate_state,
workseq_next, opaque,
min_quorum,
target_nresults,
max_error_results,
max_total_results,
max_success_results,
result_template_file
);
}
void DB_WORKUNIT::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
appid = atoi(r[i++]);
strcpy2(name, r[i++]);
strcpy2(xml_doc, r[i++]);
batch = atoi(r[i++]);
rsc_fpops_est = atof(r[i++]);
rsc_fpops_bound = atof(r[i++]);
rsc_memory_bound = atof(r[i++]);
rsc_disk_bound = atof(r[i++]);
need_validate = atoi(r[i++]);
canonical_resultid = atoi(r[i++]);
canonical_credit = atof(r[i++]);
transition_time = atoi(r[i++]);
delay_bound = atoi(r[i++]);
error_mask = atoi(r[i++]);
file_delete_state = atoi(r[i++]);
assimilate_state = atoi(r[i++]);
workseq_next = atoi(r[i++]);
opaque = atof(r[i++]);
min_quorum = atoi(r[i++]);
target_nresults = atoi(r[i++]);
max_error_results = atoi(r[i++]);
max_total_results = atoi(r[i++]);
max_success_results = atoi(r[i++]);
strcpy2(result_template_file, r[i++]);
}
void DB_RESULT::db_print(char* buf){
ESCAPE(xml_doc_out);
ESCAPE(stderr_out);
sprintf(
buf,
"create_time=%d, workunitid=%d, "
"server_state=%d, outcome=%d, client_state=%d, "
"hostid=%d, userid=%d, "
"report_deadline=%d, sent_time=%d, received_time=%d, "
"name='%s', cpu_time=%.15e, "
"xml_doc_in='%s', xml_doc_out='%s', stderr_out='%s', "
"batch=%d, file_delete_state=%d, validate_state=%d, "
"claimed_credit=%.15e, granted_credit=%.15e, opaque=%f, random=%d, "
"app_version_num=%d, appid=%d, exit_status=%d, teamid=%d",
create_time, workunitid,
server_state, outcome, client_state,
hostid, userid,
report_deadline, sent_time, received_time,
name, cpu_time,
xml_doc_in, xml_doc_out, stderr_out,
batch, file_delete_state, validate_state,
claimed_credit, granted_credit, opaque, random,
app_version_num, appid, exit_status, teamid
);
UNESCAPE(xml_doc_out);
UNESCAPE(stderr_out);
}
void DB_RESULT::db_print_values(char* buf){
ESCAPE(xml_doc_out);
ESCAPE(stderr_out);
sprintf(
buf,
"(0, %d, %d, "
"%d, %d, %d, "
"%d, %d, "
"%d, %d, %d, "
"'%s', %.15e, "
"'%s', '%s', '%s', "
"%d, %d, %d, "
"%.15e, %.15e, %f, %d, "
"%d, %d, %d, %d)",
create_time, workunitid,
server_state, outcome, client_state,
hostid, userid,
report_deadline, sent_time, received_time,
name, cpu_time,
xml_doc_in, xml_doc_out, stderr_out,
batch, file_delete_state, validate_state,
claimed_credit, granted_credit, opaque, random,
app_version_num, appid, exit_status, teamid
);
UNESCAPE(xml_doc_out);
UNESCAPE(stderr_out);
}
int DB_RESULT::update_subset() {
char query[MAX_QUERY_LEN];
sprintf(query,
"update result set server_state=%d, hostid=%d, userid=%d, sent_time=%d, report_deadline=%d where id=%d",
server_state, hostid, userid, sent_time, report_deadline, id
);
return db->do_query(query);
}
void DB_RESULT::db_parse(MYSQL_ROW &r) {
int i=0;
clear();
id=atol(r[i++]);
create_time = atoi(r[i++]);
workunitid = atoi(r[i++]);
server_state = atoi(r[i++]);
outcome = atoi(r[i++]);
client_state = atoi(r[i++]);
hostid = atoi(r[i++]);
userid = atoi(r[i++]);
report_deadline = atoi(r[i++]);
sent_time = atoi(r[i++]);
received_time = atoi(r[i++]);
strcpy2(name, r[i++]);
cpu_time = atof(r[i++]);
strcpy2(xml_doc_in, r[i++]);
strcpy2(xml_doc_out, r[i++]);
strcpy2(stderr_out, r[i++]);
batch = atoi(r[i++]);
file_delete_state = atoi(r[i++]);
validate_state = atoi(r[i++]);
claimed_credit = atof(r[i++]);
granted_credit = atof(r[i++]);
opaque = atof(r[i++]);
random = atoi(r[i++]);
app_version_num = atoi(r[i++]);
appid = atoi(r[i++]);
exit_status = atoi(r[i++]);
teamid = atoi(r[i++]);
}
void DB_MSG_FROM_HOST::db_print(char* buf) {
ESCAPE(xml);
sprintf(buf,
"create_time=%d, "
"hostid=%d, variety='%s', "
"handled=%d, xml='%s'",
create_time,
hostid, variety,
handled, xml
);
UNESCAPE(xml);
}
void DB_MSG_FROM_HOST::db_parse(MYSQL_ROW& r) {
int i=0;
clear();
id = atol(r[i++]);
create_time = atol(r[i++]);
hostid = atol(r[i++]);
strcpy2(variety, r[i++]);
handled = atoi(r[i++]);
strcpy2(xml, r[i++]);
}
void DB_MSG_TO_HOST::db_print(char* buf) {
ESCAPE(xml);
sprintf(buf,
"create_time=%d, "
"hostid=%d, variety='%s', "
"handled=%d, xml='%s'",
create_time,
hostid, variety,
handled, xml
);
UNESCAPE(xml);
}
void DB_MSG_TO_HOST::db_parse(MYSQL_ROW& r) {
int i=0;
clear();
id = atol(r[i++]);
create_time = atol(r[i++]);
hostid = atol(r[i++]);
strcpy2(variety, r[i++]);
handled = atol(r[i++]);
strcpy2(xml, r[i++]);
}
void TRANSITIONER_ITEM::parse(MYSQL_ROW& r) {
int i=0;
clear();
id = atoi(r[i++]);
strcpy2(name, r[i++]);
appid = atoi(r[i++]);
min_quorum = atoi(r[i++]);
canonical_resultid = atoi(r[i++]);
transition_time = atoi(r[i++]);
delay_bound = atoi(r[i++]);
error_mask = atoi(r[i++]);
max_error_results = atoi(r[i++]);
max_total_results = atoi(r[i++]);
file_delete_state = atoi(r[i++]);
assimilate_state = atoi(r[i++]);
target_nresults = atoi(r[i++]);
strcpy2(result_template_file, r[i++]);
res_id = safe_atoi(r[i++]);
strcpy2(res_name, r[i++]);
res_report_deadline = safe_atoi(r[i++]);
res_server_state = safe_atoi(r[i++]);
res_outcome = safe_atoi(r[i++]);
res_validate_state = safe_atoi(r[i++]);
res_file_delete_state = safe_atoi(r[i++]);
res_sent_time = safe_atoi(r[i++]);
}
int DB_TRANSITIONER_ITEM_SET::enumerate(
int transition_time, int nresult_limit,
std::vector<TRANSITIONER_ITEM>& items
) {
int x;
char query[MAX_QUERY_LEN];
char priority[256];
MYSQL_ROW row;
TRANSITIONER_ITEM new_item;
if (!cursor.active) {
strcpy(priority, "");
if (db->mysql) strcpy(priority, "HIGH_PRIORITY");
sprintf(query,
"SELECT %s "
" wu.id, "
" wu.name, "
" wu.appid, "
" wu.min_quorum, "
" wu.canonical_resultid, "
" wu.transition_time, "
" wu.delay_bound, "
" wu.error_mask, "
" wu.max_error_results, "
" wu.max_total_results, "
" wu.file_delete_state, "
" wu.assimilate_state, "
" wu.target_nresults, "
" wu.result_template_file, "
" res.id, "
" res.name, "
" res.report_deadline, "
" res.server_state, "
" res.outcome, "
" res.validate_state, "
" res.file_delete_state, "
" res.sent_time "
"FROM "
" workunit AS wu "
" LEFT JOIN result AS res ON wu.id = res.workunitid "
"WHERE "
" wu.transition_time < %d "
"LIMIT "
" %d ",
priority, transition_time, nresult_limit);
x = db->do_query(query);
if (x) return mysql_errno(db->mysql);
// the following stores the entire result set in memory
cursor.rp = mysql_store_result(db->mysql);
if (!cursor.rp) return mysql_errno(db->mysql);
cursor.active = true;
row = mysql_fetch_row(cursor.rp);
if (!row) {
mysql_free_result(cursor.rp);
cursor.active = false;
return -1;
}
last_item.parse(row);
nitems_this_query = 1;
}
items.clear();
while (true) {
items.push_back(last_item);
row = mysql_fetch_row(cursor.rp);
if (!row) {
mysql_free_result(cursor.rp);
cursor.active = false;
// if got fewer rows than requested, last group is complete
//
if (nitems_this_query < nresult_limit) {
return 0;
} else {
return -1;
}
}
new_item.parse(row);
nitems_this_query++;
if (new_item.id != last_item.id) {
last_item = new_item;
return 0;
}
last_item = new_item;
}
return 0;
}
int DB_TRANSITIONER_ITEM_SET::update_result(TRANSITIONER_ITEM& ti) {
char query[MAX_QUERY_LEN];
sprintf(query,
"update result set server_state=%d, outcome=%d, validate_state=%d, file_delete_state=%d where id=%d",
ti.res_server_state,
ti.res_outcome,
ti.res_validate_state,
ti.res_file_delete_state,
ti.res_id
);
return db->do_query(query);
}
int DB_TRANSITIONER_ITEM_SET::update_workunit(TRANSITIONER_ITEM& ti) {
char query[MAX_QUERY_LEN];
sprintf(query,
"update workunit set need_validate=%d, error_mask=%d, assimilate_state=%d, file_delete_state=%d, transition_time=%d where id=%d",
ti.need_validate,
ti.error_mask,
ti.assimilate_state,
ti.file_delete_state,
ti.transition_time,
ti.id
);
return db->do_query(query);
}
void WORK_ITEM::parse(MYSQL_ROW& r) {
int i=0;
memset(this, 0, sizeof(WORK_ITEM));
res_id = atoi(r[i++]);
wu.id=atol(r[i++]);
wu.create_time = atoi(r[i++]);
wu.appid = atoi(r[i++]);
strcpy2(wu.name, r[i++]);
strcpy2(wu.xml_doc, r[i++]);
wu.batch = atoi(r[i++]);
wu.rsc_fpops_est = atof(r[i++]);
wu.rsc_fpops_bound = atof(r[i++]);
wu.rsc_memory_bound = atof(r[i++]);
wu.rsc_disk_bound = atof(r[i++]);
wu.need_validate = atoi(r[i++]);
wu.canonical_resultid = atoi(r[i++]);
wu.canonical_credit = atof(r[i++]);
wu.transition_time = atoi(r[i++]);
wu.delay_bound = atoi(r[i++]);
wu.error_mask = atoi(r[i++]);
wu.file_delete_state = atoi(r[i++]);
wu.assimilate_state = atoi(r[i++]);
wu.workseq_next = atoi(r[i++]);
wu.opaque = atof(r[i++]);
wu.min_quorum = atoi(r[i++]);
wu.target_nresults = atoi(r[i++]);
wu.max_error_results = atoi(r[i++]);
wu.max_total_results = atoi(r[i++]);
wu.max_success_results = atoi(r[i++]);
strcpy2(wu.result_template_file, r[i++]);
}
int DB_WORK_ITEM::enumerate(int limit) {
char query[MAX_QUERY_LEN];
int retval;
MYSQL_ROW row;
if (!cursor.active) {
sprintf(query,
"select high_priority result.id, workunit.* from result left join workunit "
"on workunit.id = result.workunitid "
"where result.server_state=%d "
"limit %d",
RESULT_SERVER_STATE_UNSENT, limit
);
retval = db->do_query(query);
if (retval) return mysql_errno(db->mysql);
cursor.rp = mysql_store_result(db->mysql);
if (!cursor.rp) return mysql_errno(db->mysql);
cursor.active = true;
}
row = mysql_fetch_row(cursor.rp);
if (!row) {
mysql_free_result(cursor.rp);
cursor.active = false;
return 1;
} else {
parse(row);
}
return 0;
}
void SCHED_RESULT_ITEM::parse(MYSQL_ROW& r) {
int i=0;
clear();
id = atoi(r[i++]);
strcpy2(name, r[i++]);
workunitid = atoi(r[i++]);
server_state = atoi(r[i++]);
hostid = atoi(r[i++]);
userid = atoi(r[i++]);
received_time = atoi(r[i++]);
}
int DB_SCHED_RESULT_ITEM_SET::add_result(char* result_name) {
SCHED_RESULT_ITEM result;
strcpy2(result.queried_name, result_name);
results.push_back(result);
return 0;
}
int DB_SCHED_RESULT_ITEM_SET::enumerate() {
char query[MAX_QUERY_LEN];
int retval;
unsigned int i;
MYSQL_RES* rp;
MYSQL_ROW row;
SCHED_RESULT_ITEM ri;
strcpy2(query,
"SELECT "
" id, "
" name, "
" workunitid, "
" server_state, "
" hostid, "
" userid, "
" received_time "
"FROM "
" result "
"WHERE "
" name IN ( "
);
for (i=0; i<results.size(); i++) {
if (i>0) strcat(query, ",");
strcat(query, "'");
strcat(query, results[i].queried_name);
strcat(query, "'");
}
strcat(query, ")");
retval = db->do_query(query);
if (retval) return retval;
// the following stores the entire result set in memory
//
rp = mysql_store_result(db->mysql);
if (!rp) return mysql_errno(db->mysql);
do {
row = mysql_fetch_row(rp);
if (!row) {
mysql_free_result(rp);
} else {
ri.parse(row);
for (i=0; i<results.size(); i++) {
if (!strcmp(results[i].queried_name, ri.name)) {
results[i] = ri;
}
}
}
} while (row);
return 0;
}
int DB_SCHED_RESULT_ITEM_SET::lookup_result(char* result_name, SCHED_RESULT_ITEM** rip) {
unsigned int i;
for (i=0; i<results.size(); i++) {
if (!strcmp(results[i].name, result_name)) {
*rip = &results[i];
return 0;
}
}
return -1;
}
int DB_SCHED_RESULT_ITEM_SET::update_result(SCHED_RESULT_ITEM& ri) {
char query[MAX_QUERY_LEN];
int retval;
ESCAPE(ri.xml_doc_out);
ESCAPE(ri.stderr_out);
sprintf(query,
"UPDATE result SET "
" hostid=%d, "
" received_time=%d, "
" client_state=%d, "
" cpu_time=%.15e, "
" exit_status=%d, "
" app_version_num=%d, "
" claimed_credit=%.15e, "
" server_state=%d, "
" outcome=%d, "
" stderr_out='%s', "
" xml_doc_out='%s', "
" validate_state=%d, "
" teamid=%d "
"WHERE "
" id=%d",
ri.hostid,
ri.received_time,
ri.client_state,
ri.cpu_time,
ri.exit_status,
ri.app_version_num,
ri.claimed_credit,
ri.server_state,
ri.outcome,
ri.stderr_out,
ri.xml_doc_out,
ri.validate_state,
ri.teamid,
ri.id
);
retval = db->do_query(query);
UNESCAPE(ri.xml_doc_out);
UNESCAPE(ri.stderr_out);
return retval;
}
int DB_SCHED_RESULT_ITEM_SET::update_workunits() {
char query[MAX_QUERY_LEN], buf[256];
unsigned int i;
sprintf(query,
"UPDATE workunit SET transition_time=%d WHERE id in (",
(int)time(0)
);
for (i=0; i<results.size(); i++) {
if (i>0) strcat(query, ",");
sprintf(buf, "%d", results[i].workunitid);
strcat(query, buf);
}
strcat(query, ")");
return db->do_query(query);
}