1 files changed, 52 insertions, 0 deletions
diff --git a/nanomsg/pipeline.c b/nanomsg/pipeline.c
new file mode 100644
index 0000000..37340fa
--- /dev/null
+++ b/nanomsg/pipeline.c
@@ -0,0 +1,52 @@
+#include <assert.h>
+#include <unistd.h>
+#include <string.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <nanomsg/nn.h>
+#include <nanomsg/pipeline.h>
+
+#define NODE0 "node0"
+#define NODE1 "node1"
+
+int node0 (const char *url)
+{
+  int sock = nn_socket (AF_SP, NN_PULL);
+  assert (sock >= 0);
+  assert (nn_bind (sock, url) >= 0);
+  while (1)
+    {
+      char *buf = NULL;
+      int bytes = nn_recv (sock, &buf, NN_MSG, 0);
+      assert (bytes >= 0);
+      printf ("NODE0: RECEIVED \"%s\"\n", buf);
+      nn_freemsg (buf);
+    }
+}
+
+int node1 (const char *url, const char *msg)
+{
+  int sz_msg = strlen (msg) + 1; // '\0' too
+  int sock = nn_socket (AF_SP, NN_PUSH);
+  assert (sock >= 0);
+  assert (nn_connect (sock, url) >= 0);
+  printf ("NODE1: SENDING \"%s\"\n", msg);
+  int bytes = nn_send (sock, msg, sz_msg, 0);
+  assert (bytes == sz_msg);
+  return nn_shutdown (sock, 0);
+}
+
+int main (const int argc, const char **argv)
+{
+  if (strncmp (NODE0, argv[1], strlen (NODE0)) == 0 && argc > 1)
+    return node0 (argv[2]);
+  else if (strncmp (NODE1, argv[1], strlen (NODE1)) == 0 && argc > 2)
+    return node1 (argv[2], argv[3]);
+  else
+    {
+      fprintf (stderr, "Usage: pipeline %s|%s <URL> <ARG> ...'\n",
+               NODE0, NODE1);
+      return 1;
+    }
+}
+