Update example to show tests of various APIs

Signed-off-by: Ralph Castain <[email protected]>

Author: Ralph Castain

orte/test/mpi/interlib.c

@@ -65,13 +65,33 @@ static void opcbfunc(int status, void *cbdata)
     OPAL_PMIX_WAKEUP_THREAD(lock);
 }
 
+static void infocb(int status,
+                   opal_list_t *info,
+                   void *cbdata,
+                   opal_pmix_release_cbfunc_t release_fn,
+                   void *release_cbdata)
+{
+    opal_pmix_lock_t *lock = (opal_pmix_lock_t*)cbdata;
+    opal_value_t *kv;
+
+    OPAL_LIST_FOREACH(kv, info, opal_value_t) {
+        opal_output(0, "QUERY DATA KEY: %s VALUE %s", kv->key, kv->data.string);
+    }
+    if (NULL != release_fn) {
+        release_fn(release_cbdata);
+    }
+    OPAL_PMIX_WAKEUP_THREAD(lock);
+}
+
 static void *mylib(void *ptr)
 {
     opal_list_t info, directives;
     opal_value_t *kv;
     int ret;
     opal_pmix_lock_t lock;
     bool init = false;
+    opal_pmix_query_t *query;
+    opal_pmix_pdata_t *pdata;
 
     OPAL_PMIX_CONSTRUCT_LOCK(&thread_complete);
 
@@ -150,10 +170,57 @@ static void *mylib(void *ptr)
                                  (void*)&lock);
     OPAL_PMIX_WAIT_THREAD(&lock);
     OPAL_PMIX_DESTRUCT_LOCK(&lock);
+    OPAL_LIST_DESTRUCT(&info);
+    OPAL_LIST_DESTRUCT(&directives);
 
     /* wait for the model callback */
     OPAL_PMIX_WAIT_THREAD(&thread_complete);
 
+    /* let's do a couple of operations just to verify we can,
+     * starting with a query */
+    OBJ_CONSTRUCT(&info, opal_list_t);
+    query = OBJ_NEW(opal_pmix_query_t);
+    opal_argv_append_nosize(&query->keys, OPAL_PMIX_QUERY_NAMESPACES);
+    opal_list_append(&info, &query->super);
+    OPAL_PMIX_CONSTRUCT_LOCK(&lock);
+    opal_pmix.query(&info, infocb, &lock);
+    OPAL_PMIX_WAIT_THREAD(&lock);
+    OPAL_PMIX_DESTRUCT_LOCK(&lock);
+    OPAL_LIST_DESTRUCT(&info);
+
+    /* Get something */
+    opal_pmix.get(&orte_process_info.my_name,
+                  "WASSUP", NULL, &kv);
+    if (NULL == kv) {
+        fprintf(stderr, "ERROR GETTING WASSUP\n");
+    } else {
+        fprintf(stderr, "THREAD WASSUP: %s\n", kv->data.string);
+        OBJ_RELEASE(kv);
+    }
+
+    /* lookup something published by the main thread */
+    OBJ_CONSTRUCT(&info, opal_list_t);
+    pdata = OBJ_NEW(opal_pmix_pdata_t);
+    pdata->proc = orte_process_info.my_name;
+    pdata->value.key = strdup("SOMETHING");
+    opal_list_append(&info, &pdata->super);
+    /* tell the call to wait for the data to be published */
+    OBJ_CONSTRUCT(&directives, opal_list_t);
+    kv = OBJ_NEW(opal_value_t);
+    kv->key = strdup(OPAL_PMIX_WAIT);
+    kv->type = OPAL_INT;
+    kv->data.integer = 0;  // wait for all
+    opal_list_append(&directives, &kv->super);
+
+    if (OPAL_SUCCESS != opal_pmix.lookup(&info, &directives)) {
+        fprintf(stderr, "LOOKUP FAILED\n");
+    } else {
+        pdata = (opal_pmix_pdata_t*)opal_list_get_first(&info);
+        fprintf(stderr, "LOOKUP RETURNED %s\n", pdata->value.data.string);
+    }
+    OPAL_LIST_DESTRUCT(&info);
+    OPAL_LIST_DESTRUCT(&directives);
+
     if (init) {
         /* need to finalize to maintain refcount */
         opal_pmix.finalize();
@@ -170,12 +237,23 @@ int main(int argc, char* argv[])
     char *bindings = NULL;
     pid_t pid;
     pthread_t mythread;
+    opal_value_t kv, *kptr;
+    opal_list_t list;
 
     MPI_Init(&argc, &argv);
     MPI_Comm_rank(MPI_COMM_WORLD, &rank);
     MPI_Comm_size(MPI_COMM_WORLD, &size);
     pid = getpid();
 
+    /* push something the thread can recognize */
+    OBJ_CONSTRUCT(&kv, opal_value_t);
+    kv.key = strdup("WASSUP");
+    kv.type = OPAL_STRING;
+    kv.data.string = strdup("nothing");
+    opal_pmix.put(OPAL_PMIX_LOCAL, &kv);
+    OBJ_DESTRUCT(&kv);
+    /* no need to commit it as this is strictly within ourselves */
+
     /* spin up a thread */
     if (pthread_create(&mythread, NULL, mylib, NULL)) {
         fprintf(stderr, "Error creating thread\n");
@@ -194,6 +272,16 @@ int main(int argc, char* argv[])
            rank, size, orte_process_info.num_local_peers, rc,
            (NULL == bindings) ? "NULL" : bindings);
 
+    /* publish something */
+    OBJ_CONSTRUCT(&list, opal_list_t);
+    kptr = OBJ_NEW(opal_value_t);
+    kptr->key = strdup("SOMETHING");
+    kptr->type = OPAL_STRING;
+    kptr->data.string = strdup("SILLY-THING");
+    opal_list_append(&list, &kptr->super);
+    opal_pmix.publish(&list);
+    OPAL_LIST_DESTRUCT(&list);
+
     /* wait for the thread to finish */
     if (pthread_join(mythread, NULL)) {
         fprintf(stderr, "Error joining thread\n");