Minor changes

src/client.c

@@ -8,32 +8,40 @@
 
 #include "utils.h"
 
-
 int main (int argc, char *argv [])
 {
-    //  Socket to talk to server
+  if(argc != 3){
-    printf ("Collecting updates from weather server...\n");
+    printf("Usage: %s <address> <port>",argv[0]);
-    void *context = zmq_ctx_new ();
+    exit(1);
-    void *subscriber = zmq_socket (context, ZMQ_SUB);
+  }
-    int rc = zmq_connect (subscriber, "tcp://"STRINGIFY(PUBLISHER_ADDR)":"STRINGIFY(PUBLISHER_PORT));
-    assert (rc == 0);
 
-    //  Subscribe to zipcode, default is NYC, 10001
+  char *ip=argv[1];
- //  Subscribe to zipcode, default is NYC, 10001
+  int port=atoi(argv[2]);
-    const char *filter = "Hello";
-    rc = zmq_setsockopt (subscriber, ZMQ_SUBSCRIBE,
-                         filter, strlen(filter));
-    assert (rc == 0);
 
-    //  Process 100 updates
+  //  Socket to talk to server
-    int update_nbr;
+  void *context = zmq_ctx_new ();
-    char buffer[10];
+  void *subscriber = zmq_socket (context, ZMQ_SUB);
-    for (update_nbr = 0; update_nbr < 100; update_nbr++) {
+  char bindto[30];
-      zmq_recv (subscriber, buffer, 10, 0);
+  sprintf(bindto,"tcp://%s:%d",ip,port);
-      printf("Received!");
+  int rc = zmq_connect (subscriber, bindto);
-    }
+  if(rc!=0){
+    printf("Failed to bind zmq on %s\n",bindto);
+    exit(1);
+  }
+  rc = zmq_setsockopt (subscriber, ZMQ_SUBSCRIBE,
+                        ZMQ_TOKEN, strlen(ZMQ_TOKEN));
 
-    zmq_close (subscriber);
+
-    zmq_ctx_destroy (context);
+
-    return 0;
+  char buffer[10];
+  zmq_recv (subscriber, buffer, 10, 0);
+  printf("Received!");
+
+
+
+
+  zmq_close (subscriber);
+  zmq_ctx_destroy (context);
+
+  return 0;
 }

src/publisher.c

@@ -14,7 +14,7 @@ char *__logdir;
 int __loginterval;
 int __port;
 
-void publish(char *filepath, char* client, int interval);
+void publish(void *publisher, char *filepath, char* client, int interval);
 
 int main (int argc, char *argv [])
 {
@@ -60,7 +60,7 @@ int main (int argc, char *argv [])
                 }
                 // Send current one
                 if(FILE_EXISTS(logfile)){
-                    publish(logfile,client,interval);
+                    publish(publisher,logfile,client,interval);
                     remove(logfile);
                 }
             }
@@ -88,6 +88,7 @@ int main (int argc, char *argv [])
     return 0;
 }
 
-void publish(char *filepath, char* client, int interval){
+void publish(void *publisher, char *filepath, char* client, int interval){
     printf("Publish!\n");
+    zmq_send (publisher, ZMQ_TOKEN, strlen(ZMQ_TOKEN), 0);
 }

src/utils.h

@@ -6,9 +6,9 @@
 #include <string.h>
 
 #define MIN_INTERVAL 30
-
 #define INA260_SYSFS "/sys/kernel/ina260"
 #define INA260_POWER_REGISTER "registers/power"
+#define ZMQ_TOKEN "ina260-zmq-publisher"
 
 #define STR(symbol) #symbol
 #define STRINGIFY(symbol) STR(symbol)