rpc_example9.cpp

/*
 * Copyright (c) 2009 Carnegie Mellon University.
 *     All rights reserved.
 *
 *  Licensed under the Apache License, Version 2.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://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an "AS
 *  IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 *  express or implied.  See the License for the specific language
 *  governing permissions and limitations under the License.
 *
 * For more about this software visit:
 *
 *      http://www.graphlab.ml.cmu.edu
 *
 */

#include <vector>
#include <iostream>
#include <graphlab/rpc/dc.hpp>
#include <graphlab/serialization/iarchive.hpp>
using namespace graphlab;


struct test_struct {
  dc_dist_object<test_struct> rmi;
  test_struct(distributed_control &dc):rmi(dc, this) {
    dc.barrier();
  }

  void test_blob(size_t len, wild_pointer w) {
    assert(len == sizeof(procid_t));
    std::cout << "split call from : " << *reinterpret_cast<const procid_t*>(w.ptr) << "\n";
  }

  void print(int val) {
    std::cout << rmi.procid() << ": Receiving print with value : " << val << std::endl;
  }

  void test() {
    if (rmi.procid() == 0) {
      std::cout << "First set of calls... Proc 1 and 3 should receive" << std::endl;
      std::vector<procid_t> s;
      s.push_back(1); s.push_back(3);
      rmi.pod_call(s.begin(), s.end(), &test_struct::print, 1);
    }
    rmi.full_barrier();

    if (rmi.procid() == 0) {
      std::cout << "Second set of calls... Proc 0 and 2 should receive" << std::endl;
      std::vector<procid_t> s;
      s.push_back(2); s.push_back(0);
      rmi.pod_call(s.begin(), s.end(), &test_struct::print, 1);
    }
    rmi.full_barrier();

    oarchive* oarc = rmi.split_call_begin(&test_struct::test_blob);
    (*oarc) << rmi.procid();
    rmi.split_call_end(1, oarc);
    rmi.full_barrier();
  }
};

int main(int argc, char ** argv) {
  mpi_tools::init(argc, argv);
  distributed_control dc;

  if (dc.numprocs() != 4) {
    std::cout<< "RPC Example 8: Basic Broadcast Test\n";
    std::cout << "Run with exactly 4 MPI nodes.\n";
    return 0;
  }
  test_struct ts(dc);
  ts.test();

  // terminate MPI
  mpi_tools::finalize();
}