mirror of https://github.com/BOINC/boinc.git
108 lines
2.6 KiB
C++
108 lines
2.6 KiB
C++
|
// make_work.C
|
||
|
|
//
|
||
|
|
// Create result records as needed to maintain a pool to send
|
||
|
|
//
|
||
|
|
// This reads a result record from the DB, then makes clones of it.
|
||
|
|
// Assumes the result has a single output file,
|
||
|
|
// so overwrites the first <name> element with a new name
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
#include "db.h"
|
||
|
|
#include "config.h"
|
||
|
|
|
||
|
|
#define TRIGGER_FILENAME "stop_server"
|
||
|
|
|
||
|
|
void check_trigger() {
|
||
|
|
FILE* f = fopen(TRIGGER_FILENAME, "r");
|
||
|
|
if (!f) return;
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
void replace_element(char* buf, char* start, char* end, char* replacement) {
|
||
|
|
char temp[MAX_BLOB_SIZE], *p, *q;
|
||
|
|
|
||
|
|
p = strstr(buf, start);
|
||
|
|
p += strlen(start);
|
||
|
|
q = strstr(p, end);
|
||
|
|
strcpy(temp, q);
|
||
|
|
strcpy(p, replacement);
|
||
|
|
strcat(p, temp);
|
||
|
|
}
|
||
|
|
|
||
|
|
void make_work() {
|
||
|
|
CONFIG config;
|
||
|
|
RESULT result;
|
||
|
|
int retval, i=time(0), n;
|
||
|
|
char buf[256];
|
||
|
|
|
||
|
|
retval = config.parse_file();
|
||
|
|
if (retval) {
|
||
|
|
fprintf(stderr, "make_work: can't read config file\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
retval = db_open(config.db_name, config.db_passwd);
|
||
|
|
if (retval) {
|
||
|
|
fprintf(stderr, "make_work: can't open db\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
retval = db_result(1, result);
|
||
|
|
if (retval) {
|
||
|
|
fprintf(stderr, "make_work: can't read result\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
while (true) {
|
||
|
|
fflush(stdout);
|
||
|
|
retval = db_result_count_state(RESULT_STATE_UNSENT, n);
|
||
|
|
if (retval) {
|
||
|
|
fprintf(stderr, "make_work: can't counts results\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
printf("make_work: %d results\n", n);
|
||
|
|
if (n > 10) {
|
||
|
|
sleep(1);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
result.id = 0;
|
||
|
|
result.create_time = time(0);
|
||
|
|
sprintf(result.name, "result_%d", i++);
|
||
|
|
result.state = RESULT_STATE_UNSENT;
|
||
|
|
result.validate_state = VALIDATE_STATE_INITIAL;
|
||
|
|
replace_element(result.xml_doc_in, "<name>", "</name>", result.name);
|
||
|
|
replace_element(result.xml_doc_in, "<file_name>", "</file_name>", result.name);
|
||
|
|
retval = db_result_new(result);
|
||
|
|
if (retval) {
|
||
|
|
fprintf(stderr, "make_work: can't create result\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
printf("make_work: added a result\n");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char** argv) {
|
||
|
|
bool asynch = false;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
unlink(TRIGGER_FILENAME);
|
||
|
|
for (i=1; i<argc; i++) {
|
||
|
|
if (!strcmp(argv[i], "-asynch")) {
|
||
|
|
asynch = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (asynch) {
|
||
|
|
if (!fork()) {
|
||
|
|
make_work();
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
make_work();
|
||
|
|
}
|
||
|
|
}
|