264 строки
9.3 KiB
C
264 строки
9.3 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/** @file:
|
|
*
|
|
* The Open MPI general purpose registry - unit test
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* includes
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "opal/runtime/opal.h"
|
|
#include "opal/util/malloc.h"
|
|
#include "opal/util/output.h"
|
|
|
|
#include "orte/orte_constants.h"
|
|
#include "orte/mca/schema/schema.h"
|
|
#include "orte/mca/schema/base/base.h"
|
|
#include "orte/mca/ns/base/base.h"
|
|
#include "orte/mca/soh/base/base.h"
|
|
#include "orte/mca/rmgr/base/base.h"
|
|
|
|
#include "orte/class/orte_pointer_array.h"
|
|
#include "orte/dss/dss.h"
|
|
#include "orte/runtime/runtime.h"
|
|
#include "orte/util/proc_info.h"
|
|
#include "orte/util/sys_info.h"
|
|
|
|
#include "orte/mca/gpr/base/base.h"
|
|
#include "orte/mca/gpr/replica/api_layer/gpr_replica_api.h"
|
|
#include "orte/mca/gpr/replica/functional_layer/gpr_replica_fn.h"
|
|
#include "orte/mca/gpr/replica/communications/gpr_replica_comm.h"
|
|
#include "orte/mca/gpr/replica/transition_layer/gpr_replica_tl.h"
|
|
|
|
static void test_cbfunc1(orte_gpr_notify_data_t *data, void *user_tag);
|
|
static int test_cbfunc2(orte_gpr_notify_message_t *msg);
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int rc;
|
|
orte_gpr_subscription_id_t id;
|
|
orte_gpr_trigger_id_t id2;
|
|
size_t i;
|
|
char *tokens[5], *keys[5];
|
|
|
|
opal_init();
|
|
|
|
/* register handler for errnum -> string converstion */
|
|
opal_error_register("ORTE", ORTE_ERR_BASE, ORTE_ERR_MAX, orte_err2str);
|
|
|
|
|
|
/* Ensure the process info structure is instantiated and initialized */
|
|
if (ORTE_SUCCESS != (rc = orte_proc_info())) {
|
|
return rc;
|
|
}
|
|
|
|
orte_process_info.seed = true;
|
|
orte_process_info.my_name = (orte_process_name_t*)malloc(sizeof(orte_process_name_t));
|
|
orte_process_info.my_name->cellid = 0;
|
|
orte_process_info.my_name->jobid = 0;
|
|
orte_process_info.my_name->vpid = 0;
|
|
|
|
/* startup the MCA */
|
|
if (OMPI_SUCCESS == mca_base_open()) {
|
|
fprintf(stderr, "MCA started\n");
|
|
} else {
|
|
fprintf(stderr, "MCA could not start\n");
|
|
exit (1);
|
|
}
|
|
|
|
/* open the dss */
|
|
if (ORTE_SUCCESS == orte_dss_open()) {
|
|
fprintf(stderr, "DSS started\n");
|
|
} else {
|
|
fprintf(stderr, "DSS could not start\n");
|
|
exit (1);
|
|
}
|
|
|
|
/* ENSURE THE REPLICA IS ISOLATED */
|
|
setenv("OMPI_MCA_gpr_replica_isolate", "1", 1);
|
|
|
|
/* startup the gpr to register data types */
|
|
if (ORTE_SUCCESS == orte_gpr_base_open()) {
|
|
fprintf(stderr, "GPR opened\n");
|
|
} else {
|
|
fprintf(stderr, "GPR could not open\n");
|
|
exit (1);
|
|
}
|
|
|
|
/* do a select on the registry components */
|
|
if (OMPI_SUCCESS == orte_gpr_base_select()) {
|
|
fprintf(stderr, "GPR selected\n");
|
|
} else {
|
|
fprintf(stderr, "GPR could not select\n");
|
|
exit (1);
|
|
}
|
|
|
|
tokens[0] = strdup("test-token-1");
|
|
tokens[1] = strdup("test-token-2");
|
|
tokens[2] = NULL;
|
|
fprintf(stderr, "quick-subscribe one value with single key\n");
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_1(&id,
|
|
NULL,
|
|
NULL,
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
"test-key", test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key no names failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key no names passed\n");
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_1(&id,
|
|
NULL,
|
|
"orte-std-subscription-single1",
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
"test-key", test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key sub name failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key sub name passed\n");
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_1(&id,
|
|
"orte-std-trigger-single",
|
|
"orte-std-subscription-single2",
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
"test-key", test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key trig and sub names failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of 1 key trig and sub names passed\n");
|
|
}
|
|
free(tokens[0]);
|
|
free(tokens[1]);
|
|
|
|
for (i=0; i < 4; i++) {
|
|
asprintf(&tokens[i], "test-token-%lu", (unsigned long)i);
|
|
asprintf(&keys[i], "test-keys-%lu", (unsigned long)i);
|
|
}
|
|
tokens[4] = NULL;
|
|
keys[4] = NULL;
|
|
fprintf(stderr, "quick-subscribe one value with multiple keyvals\n");
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_N(&id,
|
|
NULL,
|
|
NULL,
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
4, keys, test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key no names failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key no names passed\n");
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_N(&id,
|
|
NULL,
|
|
"orte-std-subscription-multi",
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
4, keys, test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key sub name failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key sub name passed\n");
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.subscribe_N(&id,
|
|
"orte-std-trigger-multi",
|
|
"orte-std-subscription-multi2",
|
|
ORTE_GPR_NOTIFY_VALUE_CHG,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
4, keys, test_cbfunc1, NULL))) {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key trig and sub names failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key trig and sub names passed\n");
|
|
}
|
|
|
|
fprintf(stderr, "quick-define-trigger\n");
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.define_trigger(&id2,
|
|
NULL,
|
|
ORTE_GPR_TRIG_INCLUDE_TRIG_CNTRS,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
4, keys, test_cbfunc2, NULL))) {
|
|
fprintf(stderr, "gpr_test: define_trigger of multi key no names failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: subscribe of multi key no names passed\n");
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.define_trigger(&id,
|
|
"orte-std-trigger",
|
|
ORTE_GPR_TRIG_INCLUDE_TRIG_CNTRS,
|
|
ORTE_GPR_TOKENS_AND,
|
|
"test-put-segment", tokens,
|
|
4, keys, test_cbfunc2, NULL))) {
|
|
fprintf(stderr, "gpr_test: define_trigger of multi key sub name failed with error code %d\n", rc);
|
|
return rc;
|
|
} else {
|
|
fprintf(stderr, "gpr_test: define_trigger of multi key sub name passed\n");
|
|
}
|
|
orte_gpr.dump_subscriptions(0, 0);
|
|
orte_gpr.dump_triggers(0, 0);
|
|
orte_gpr.dump_local_subscriptions(0);
|
|
orte_gpr.dump_local_triggers(0);
|
|
|
|
fprintf(stderr, "now finalize and see if all memory cleared\n");
|
|
orte_gpr_base_close();
|
|
|
|
orte_dss_close();
|
|
mca_base_close();
|
|
opal_malloc_finalize();
|
|
opal_output_finalize();
|
|
opal_class_finalize();
|
|
|
|
return(0);
|
|
}
|
|
|
|
void test_cbfunc1(orte_gpr_notify_data_t *data, void *tag)
|
|
{
|
|
fprintf(stderr, "\n\n\nTRIGGER FIRED AND RECEIVED AT CALLBACK 1\n");
|
|
|
|
orte_gpr.dump_notify_data(data, 0);
|
|
}
|
|
|
|
int test_cbfunc2(orte_gpr_notify_message_t *msg)
|
|
{
|
|
fprintf(stderr, "\n\n\nTRIGGER FIRED AND RECEIVED AT CALLBACK 2\n");
|
|
|
|
orte_gpr.dump_notify_msg(msg, 0);
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|