- (&key n-senders n-receivers n-messages mailbox interruptor)
- (let* ((cnt (make-counter))
- (mbox (or mailbox (make-mailbox)))
- (senders
- (make-threads n-senders "SENDER"
- #'(lambda ()
- (dotimes (i n-messages)
- (send-message mbox i)
- (sleep (random 0.001))))))
- (receivers
- ;; We have three groups of receivers, one using
- ;; RECEIVE-MESSAGE, one RECEIVE-MESSAGE-NO-HANG, and another
- ;; one RECEIVE-PENDING-MESSAGES.
- (let* ((aux (floor n-receivers 2))
- (n-recv-msg (- n-receivers aux))
- (n-recv-pend-msgs (floor aux 3))
- (n-recv-msg-n-h (- aux n-recv-pend-msgs)))
- (append
- (make-threads n-recv-msg "RECV-MSG"
- #'(lambda ()
- (sleep (random 0.001))
- (handler-case
- (loop
- (sb-sys:with-deadline (:seconds 1.0)
- (let ((msg (receive-message mbox)))
- (sb-ext:atomic-incf (counter-ref cnt))
- (unless (< -1 msg n-messages)
- (hang)))))
- (sb-ext:timeout ()))))
- (make-threads n-recv-pend-msgs "RECV-PEND-MSGS"
- #'(lambda ()
- (sleep (random 0.001))
- (dotimes (i 10)
- (thread-yield)
- (let ((msgs (receive-pending-messages mbox (random 5))))
- (mapc #'(lambda (msg)
- (sb-ext:atomic-incf (counter-ref cnt))
- (unless (< -1 msg n-messages)
- (hang)))
- msgs)))))
- (make-threads n-recv-msg-n-h "RECV-MSG-NO-HANG"
- #'(lambda ()
- (sleep (random 0.001))
- (dotimes (i 30)
- (thread-yield)
- (multiple-value-bind (msg ok)
- (receive-message-no-hang mbox)
- (when ok
- (sb-ext:atomic-incf (counter-ref cnt))
- (unless (< -1 msg n-messages)
- (hang))))))))))
- (threads (append receivers senders)))
- (when interruptor (funcall interruptor threads))
- (mapc #'timed-join-thread threads)
- (values mbox (counter-ref cnt) (* n-senders n-messages))))
+ (&key n-senders n-receivers n-messages interruptor)
+ (let ((mbox (make-mailbox))
+ (counter (make-counter))
+ (+sleep+ 0.0001)
+ (+fin-token+ :finish) ; end token for receivers to stop
+ (+blksize+ 5)) ; "block size" for RECEIVE-PENDING-MESSAGES
+ (multiple-value-bind (n-recv-msg
+ n-recv-msg-n-h
+ n-recv-pend-msgs)
+ ;; We have three groups of receivers, one using
+ ;; RECEIVE-MESSAGE, one RECEIVE-MESSAGE-NO-HANG, and
+ ;; another one RECEIVE-PENDING-MESSAGES.
+ (receiver-distribution n-receivers)
+ (let ((senders
+ (make-threads n-senders "SENDER"
+ #'(lambda ()
+ (dotimes (i n-messages t)
+ (send-message mbox i)
+ (sleep (random +sleep+))))))
+ (receivers
+ (flet ((process-msg (msg out)
+ (cond
+ ((eq msg +fin-token+)
+ (funcall out t))
+ ((not (< -1 msg n-messages))
+ (funcall out nil))
+ (t
+ (sb-ext:atomic-incf (counter-ref counter))))))
+ (append
+ (make-threads n-recv-msg "RECV-MSG"
+ #'(lambda ()
+ (sleep (random +sleep+))
+ (loop (process-msg (receive-message mbox)
+ #'(lambda (x) (return x))))))
+ (make-threads n-recv-pend-msgs "RECV-PEND-MSGS"
+ #'(lambda ()
+ (loop
+ (sleep (random +sleep+))
+ (mapc #'(lambda (msg)
+ (process-msg msg #'(lambda (x) (return x))))
+ (receive-pending-messages mbox +blksize+)))))
+ (make-threads n-recv-msg-n-h "RECV-MSG-NO-HANG"
+ #'(lambda ()
+ (loop
+ (sleep (random +sleep+))
+ (multiple-value-bind (msg ok)
+ (receive-message-no-hang mbox)
+ (when ok
+ (process-msg msg #'(lambda (x)
+ (return x))))))))))))
+
+ (when interruptor
+ (funcall interruptor (append receivers senders)))
+ (let ((garbage 0)
+ (errors 0)
+ (timeouts 0))
+ (flet ((wait-for (threads)
+ (mapc #'(lambda (thread)
+ (ecase (timed-join-thread thread)
+ ((t))
+ ((nil) (incf garbage))
+ ((:aborted) (incf errors))
+ ((:timeout) (incf timeouts)
+ (kill-thread thread))))
+ threads)))
+ ;; First wait until all messages are propagating.
+ (wait-for senders)
+ ;; Senders are finished, inform and wait for the
+ ;; receivers.
+ (loop repeat (+ n-recv-msg
+ n-recv-msg-n-h
+ (* n-recv-pend-msgs +blksize+))
+ ;; The number computed above is an upper bound; if
+ ;; we send as many FINs as that, we can be sure that
+ ;; every receiver must have got at least one FIN.
+ do (send-message mbox +fin-token+))
+ (wait-for receivers)
+ ;; We may in fact have sent too many FINs, so make sure
+ ;; it's only FINs in the mailbox now.
+ (mapc #'(lambda (msg) (unless (eq msg +fin-token+)
+ (incf garbage)))
+ (list-mailbox-messages mbox))
+ (values `(:received . ,(counter-ref counter))
+ `(:garbage . ,garbage)
+ `(:errors . ,errors)
+ `(:timeouts . ,timeouts))))))))
+