Merge pull request #125 from djellemah/master

add Socket.set_stream_notify

zmq/src/caml_zmq_stubs.c

@@ -249,6 +249,7 @@ static int const native_int_option_for[] = {
     ZMQ_REQ_CORRELATE,
     ZMQ_REQ_RELAXED,
     ZMQ_CONFLATE,
+    ZMQ_STREAM_NOTIFY
 };
 
 CAMLprim value caml_zmq_set_int_option(value socket, value option_name, value socket_option) {

zmq/src/zmq.ml

@@ -191,6 +191,7 @@ module Socket = struct
   | ZMQ_REQ_CORRELATE
   | ZMQ_REQ_RELAXED
   | ZMQ_CONFLATE
+  | ZMQ_STREAM_NOTIFY
   [@@@warning "+37"]
 
   external set_int_option :
@@ -483,6 +484,9 @@ module Socket = struct
   let set_conflate socket flag =
     set_int_option socket ZMQ_CONFLATE (if flag then 1 else 0)
 
+  let set_stream_notify socket stream_notify_flag =
+    set_int_option socket ZMQ_STREAM_NOTIFY (if stream_notify_flag then 1 else 0)
+
   external get_fd : 'a t -> Unix.file_descr = "caml_zmq_get_fd"
 
   type event = No_event | Poll_in | Poll_out | Poll_in_out | Poll_error

zmq/src/zmq.mli

@@ -232,6 +232,7 @@ module Socket : sig
   val set_zap_domain : 'a t -> string -> unit
   val get_zap_domain : 'a t -> string
   val set_conflate : [< `Pull | `Push | `Sub | `Pub | `Dealer] t -> bool -> unit
+  val set_stream_notify : [< `Stream ] t -> bool -> unit
 
   val get_fd : 'a t -> Unix.file_descr
 

zmq/test/zmq_test.ml

@@ -79,6 +79,13 @@ let test_socket_options () =
   test_set_get_value "Tcp keepalive interval" set_tcp_keepalive_interval get_tcp_keepalive_interval socket (`Value 1000);
 
   Socket.close socket;
+
+  (* Must be a `Stream. No getter so just test that it doesn't explode. *)
+  let stream_socket = create ctx stream in
+  set_stream_notify stream_socket true;
+  set_stream_notify stream_socket false;
+  Socket.close stream_socket;
+
   Context.terminate ctx;
   ()