868
|
1 #!/usr/bin/perl
|
|
2
|
|
3 # (C) Sergey Kandaurov
|
|
4 # (C) Nginx, Inc.
|
|
5
|
|
6 # Stream tests for upstream least_conn balancer module with datagrams.
|
|
7
|
|
8 ###############################################################################
|
|
9
|
|
10 use warnings;
|
|
11 use strict;
|
|
12
|
|
13 use Test::More;
|
|
14
|
|
15 BEGIN { use FindBin; chdir($FindBin::Bin); }
|
|
16
|
|
17 use lib 'lib';
|
|
18 use Test::Nginx;
|
|
19 use Test::Nginx::Stream qw/ dgram /;
|
|
20
|
|
21 ###############################################################################
|
|
22
|
|
23 select STDERR; $| = 1;
|
|
24 select STDOUT; $| = 1;
|
|
25
|
|
26 my $t = Test::Nginx->new()->has(qw/stream stream_upstream_least_conn/)
|
|
27 ->write_file_expand('nginx.conf', <<'EOF');
|
|
28
|
|
29 %%TEST_GLOBALS%%
|
|
30
|
|
31 daemon off;
|
|
32
|
|
33 events {
|
|
34 }
|
|
35
|
|
36 stream {
|
|
37 proxy_responses 1;
|
|
38 proxy_timeout 1s;
|
|
39
|
|
40 upstream u {
|
|
41 least_conn;
|
|
42 server 127.0.0.1:8081;
|
|
43 server 127.0.0.1:8082;
|
|
44 }
|
|
45
|
|
46 server {
|
|
47 listen 127.0.0.1:8080 udp;
|
|
48 proxy_pass u;
|
|
49 }
|
|
50 }
|
|
51
|
|
52 EOF
|
|
53
|
|
54 $t->run_daemon(\&udp_daemon, 8081, $t);
|
|
55 $t->run_daemon(\&udp_daemon, 8082, $t);
|
|
56 $t->try_run('no stream udp')->plan(2);
|
|
57
|
|
58 $t->waitforfile($t->testdir . '/8081');
|
|
59 $t->waitforfile($t->testdir . '/8082');
|
|
60
|
|
61 ###############################################################################
|
|
62
|
|
63 is(many('.', 10), '8081: 5, 8082: 5', 'balanced');
|
|
64
|
|
65 my @sockets;
|
|
66 for (1 .. 2) {
|
|
67 my $s = dgram();
|
|
68 $s->write('w');
|
|
69 push @sockets, $s;
|
|
70 }
|
|
71
|
|
72 select undef, undef, undef, 0.2;
|
|
73
|
|
74 is(many('.', 10), '8082: 10', 'least_conn');
|
|
75
|
|
76 ###############################################################################
|
|
77
|
|
78 sub many {
|
|
79 my ($data, $count, %opts) = @_;
|
|
80 my (%ports);
|
|
81
|
|
82 for (1 .. $count) {
|
|
83 if (dgram()->io($data) =~ /(\d+)/) {
|
|
84 $ports{$1} = 0 unless defined $ports{$1};
|
|
85 $ports{$1}++;
|
|
86 }
|
|
87 }
|
|
88
|
|
89 return join ', ', map { $_ . ": " . $ports{$_} } sort keys %ports;
|
|
90 }
|
|
91
|
|
92 ###############################################################################
|
|
93
|
|
94 sub udp_daemon {
|
|
95 my ($port, $t) = @_;
|
|
96
|
|
97 my $server = IO::Socket::INET->new(
|
|
98 Proto => 'udp',
|
|
99 LocalAddr => '127.0.0.1:' . $port,
|
|
100 Reuse => 1,
|
|
101 )
|
|
102 or die "Can't create listening socket: $!\n";
|
|
103
|
|
104 # signal we are ready
|
|
105
|
|
106 open my $fh, '>', $t->testdir() . '/' . $port;
|
|
107 close $fh;
|
|
108
|
|
109 while (1) {
|
|
110 $server->recv(my $buffer, 65536);
|
|
111
|
|
112 my $port = $server->sockport();
|
|
113
|
|
114 if ($buffer =~ /w/ && $port == 8081) {
|
|
115 select undef, undef, undef, 2.5;
|
|
116 }
|
|
117
|
|
118 $buffer = $port;
|
|
119
|
|
120 $server->send($buffer);
|
|
121 }
|
|
122 }
|
|
123
|
|
124 ###############################################################################
|