blob: 868765655e204f75a413d9b11663c631e00d5563 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
;;; Simple ping client implementation
;; Copyright (C) 2012 Free Software Foundation, Inc.
;; This library is free software; you can redistribute it and/or
;; modify it under the terms of the GNU Lesser General Public
;; License as published by the Free Software Foundation; either
;; version 3 of the License, or (at your option) any later version.
;;
;; This library is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;; Lesser General Public License for more details.
;;
;; You should have received a copy of the GNU Lesser General Public
;; License along with this library; if not, write to the Free Software
;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
;; 02110-1301 USA
(use-modules (rnrs bytevectors)
(fibers)
(fibers channels)
(ice-9 binary-ports)
(ice-9 textual-ports)
(ice-9 rdelim)
(ice-9 match))
(define (set-nonblocking! port)
(fcntl port F_SETFL (logior O_NONBLOCK (fcntl port F_GETFL)))
(setvbuf port 'block 1024))
(define (connect-to-server addrinfo)
(let ((port (socket (addrinfo:fam addrinfo)
(addrinfo:socktype addrinfo)
(addrinfo:protocol addrinfo))))
;; Disable Nagle's algorithm. We buffer ourselves.
(setsockopt port IPPROTO_TCP TCP_NODELAY 0)
(set-nonblocking! port)
(connect port (addrinfo:addr addrinfo))
port))
(define (client-loop addrinfo n num-connections)
(let ((port (connect-to-server addrinfo))
(test (string-append "test-" (number->string n))))
(let lp ((m 0))
(when (< m num-connections)
(put-string port test)
(put-char port #\newline)
(force-output port)
(let ((response (read-line port)))
(unless (equal? test response)
(close-port port)
(error "Bad response: ~A (expected ~A)" response test))
(lp (1+ m)))))
(close-port port)))
(define (run-ping-test num-clients num-connections)
;; The getaddrinfo call blocks, unfortunately. Call it once before
;; spawning clients.
(let ((addrinfo (car (getaddrinfo "localhost" (number->string 11211)))))
(map get-message
(map (lambda (n)
(let ((ch (make-channel)))
(spawn-fiber
(lambda ()
(client-loop addrinfo n num-connections)
(put-message ch 'done))
#:parallel? #t)
ch))
(iota num-clients)))))
(run-fibers
(lambda ()
(apply run-ping-test (map string->number (cdr (program-arguments))))))
|
