Function main

test/singa/test_ep.cc:38–111

using namespace singa;
int main(int argc, char **argv) {
  char *md = new char[SIZE];
  char *payload = new char[SIZE];

  const char *host = "localhost";
  int port = PORT;

  for (int i = 1; i < argc; ++i) {
    if (strcmp(argv[i], "-p") == 0)
      port = atoi(argv[++i]);
    else if (strcmp(argv[i], "-h") == 0)
      host = argv[++i];
    else
      fprintf(stderr, "Invalid option %s\n", argv[i]);
  }

  memset(md, 'a', SIZE);
  memset(payload, 'b', SIZE);

  NetworkThread *t = new NetworkThread(port);

  EndPointFactory *epf = t->epf_;

  // sleep
  sleep(3);

  EndPoint *ep = epf->getEp(host);

  Message *m[ITER];
  for (int i = 0; i < ITER; ++i) {
    m[i] = new Message();
    m[i]->setMetadata(md, SIZE);
    m[i]->setPayload(payload, SIZE);
  }

  while (1) {
    for (int i = 0; i < ITER; ++i) {
      if (ep->send(m[i]) < 0) return 1;
      delete m[i];
    }

    for (int i = 0; i < ITER; ++i) {
      m[i] = ep->recv();
      if (!m[i]) return 1;
      char *p;
      CHECK_EQ(m[i]->getMetadata((void **)&p), SIZE);
      CHECK_EQ(0, strncmp(p, md, SIZE));
      CHECK_EQ(m[i]->getPayload((void **)&p), SIZE);
      CHECK_EQ(0, strncmp(p, payload, SIZE));
    }
  }

  // while(ep && cnt++ <= 5 && ep->send(m) > 0 ) {

  //    LOG(INFO) << "Send a " << m->getSize() << " bytes message";

  //    Message* m1 = ep->recv();