void _initialize(char* argv[])
{
slurm_conf_reinit(argv[1]); //load the slurm configuration file
self_id = strdup(slurmctld_conf.control_machine); //get the controller id
num_ctrl = str_to_int(argv[2]); // get the number of controllers
mem_list = c_calloc_2(num_ctrl, 30);
_read_memlist(argv[3]); // read the controller membership list
part_size = str_to_int(argv[4]); //get the partition size
source = c_calloc_2(part_size + 1, 30);
/* fill the node list of the controller (part_size, node_1, node_2, ...) */
resource = c_calloc((part_size + 1) * 30);
strcat(resource, argv[4]); strcat(resource, ",");
/* read workload and fill it in the job queue*/
job_queue = init_queue();
_read_workload(argv[5]);
num_job = job_queue->queue_length;
c_zht_init(argv[6], argv[7]); // initialize the controller as a ZHT client
max_proc_thread = str_to_int(argv[8]);
char *job_output_path = c_calloc(100);
strcpy(job_output_path, "./job_output_");
strcat(job_output_path, argv[9]);
job_output_file = fopen(job_output_path, "w");
c_free(job_output_path);
}
int main(int argc, char **argv) {
if (argc < 4) {
printf("Usage: %s", "###.exe neighbor zht.cfg TCP\n");
return -1;
}
bool useTCP = false;
char *tcpFlag = argv[3];
if (!strcmp("TCP", tcpFlag)) {
useTCP = true;
} else {
useTCP = false;
}
c_zht_init(argv[1], argv[2], useTCP); //neighbor zht.cfg TCP
// test_large_keyvalue();
// test_common_usecase();
test_pass_package();
c_zht_teardown();
return 0;
}
int zht_init()
{
/* use TCP by default */
c_zht_init("/dev/shm/neighbor", "/dev/shm/zht.cfg", true);
// /* DFZ: debug info */
// printf("\n =====DFZ debug: %s \n", "zht_init() succeeded. ");
return 0;
}
int zht_init()
{
/* use TCP by default */
c_zht_init("/intrepid-fs0/users/dzhao/persistent/neighbor", "/intrepid-fs0/users/dzhao/persistent/zht.cfg", true);
// /* DFZ: debug info */
// printf("\n =====DFZ debug: %s \n", "zht_init() succeeded. ");
return 0;
}
extern int _zht_client_init(char* memlist, char* config, char* tcp_flag)
{
int use_tcp = 0;
if (!strcmp("TCP", tcp_flag))
{
use_tcp = 1;
}
else
{
use_tcp = 0;
}
return c_zht_init(memlist, config, use_tcp);
}
int main(int argc, char **argv) {
if (argc < 4) {
printf("Usage: %s", "###.exe neighbor zht.cfg TCP\n");
return -1;
}
bool useTCP = false;
char *tcpFlag = argv[3];
if (!strcmp("TCP", tcpFlag)) {
useTCP = true;
} else {
useTCP = false;
}
c_zht_init(argv[1], argv[2], useTCP); //neighbor zht.cfg TCP
const char *key = "hello";
const char *value = "zht";
int iret = c_zht_insert2(key, value);
fprintf(stderr, "c_zht_insert, return code: %d\n", iret);
size_t n;
char *result = (char*) calloc(LOOKUP_SIZE, sizeof(char));
if (result != NULL) {
int lret = c_zht_lookup2(key, result, &n);
fprintf(stderr, "c_zht_lookup, return code: %d\n", lret);
fprintf(stderr, "c_zht_lookup, return value: %lu, %s\n", n, result);
}
int rret = c_zht_remove2(key);
fprintf(stderr, "c_zht_remove, return code: %d\n", rret);
c_zht_teardown();
free(result);
return 0;
}
int main(int argc, char **argv) {
CUR_PID = getpid();
extern char *optarg;
double us = 0;
int printHelp = 0;
char *neighbor = NULL;
char *zht_cfg = NULL;
char *protocol = NULL;
int is_init = 0;
int is_lookup = 0;
int c;
while ((c = getopt(argc, argv, "n:z:p:ilh")) != -1) {
switch (c) {
case 'n':
neighbor = optarg;
break;
case 'z':
zht_cfg = optarg;
break;
case 'p':
protocol = optarg;
break;
case 'i':
is_init = 1;
break;
case 'l':
is_lookup = 1;
break;
case 'h':
printHelp = 1;
break;
default:
fprintf(stdout, "Illegal argument \"%c\"\n", c);
printUsage(argv[0]);
exit(1);
}
}
if (printHelp) {
printUsage(argv[0]);
exit(1);
}
if (neighbor != NULL && zht_cfg != NULL && protocol != NULL) {
bool useTCP = false;
if (!strcmp("TCP", protocol)) {
useTCP = true;
} else {
useTCP = false;
}
/*init...*/
c_zht_init(neighbor, zht_cfg, useTCP); //neighbor zht.cfg TCP
if (is_init) {
insert_resource();
} else if (is_lookup) {
lookup_nodes();
} else {
test_compare_swap();
}
/*clear...*/
c_zht_teardown();
} else {
printUsage(argv[0]);
exit(1);
}
return 0;
}
