1 | package Etsy::StatsD;
|
2 | use strict;
|
3 | use warnings;
|
4 | use IO::Socket;
|
5 | use Carp;
|
6 |
|
7 | our $VERSION = 1.000002;
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 | sub new {
|
26 | my ( $class, $host, $port, $sample_rate ) = @_;
|
27 | $host = 'localhost' unless defined $host;
|
28 | $port = 8125 unless defined $port;
|
29 |
|
30 | my $sock = new IO::Socket::INET(
|
31 | PeerAddr => $host,
|
32 | PeerPort => $port,
|
33 | Proto => 'udp',
|
34 | ) or croak "Failed to initialize socket: $!";
|
35 |
|
36 | bless { socket => $sock, sample_rate => $sample_rate }, $class;
|
37 | }
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 | sub timing {
|
46 | my ( $self, $stat, $time, $sample_rate ) = @_;
|
47 | $self->send( { $stat => "$time|ms" }, $sample_rate );
|
48 | }
|
49 |
|
50 |
|
51 |
|
52 |
|
53 |
|
54 |
|
55 |
|
56 | sub increment {
|
57 | my ( $self, $stats, $sample_rate ) = @_;
|
58 | $self->update( $stats, 1, $sample_rate );
|
59 | }
|
60 |
|
61 |
|
62 |
|
63 |
|
64 |
|
65 |
|
66 |
|
67 | sub decrement {
|
68 | my ( $self, $stats, $sample_rate ) = @_;
|
69 | $self->update( $stats, -1, $sample_rate );
|
70 | }
|
71 |
|
72 |
|
73 |
|
74 |
|
75 |
|
76 |
|
77 |
|
78 | sub update {
|
79 | my ( $self, $stats, $delta, $sample_rate ) = @_;
|
80 | $delta = 1 unless defined $delta;
|
81 | my %data;
|
82 | if ( ref($stats) eq 'ARRAY' ) {
|
83 | %data = map { $_ => "$delta|c" } @$stats;
|
84 | }
|
85 | else {
|
86 | %data = ( $stats => "$delta|c" );
|
87 | }
|
88 | $self->send( \%data, $sample_rate );
|
89 | }
|
90 |
|
91 |
|
92 |
|
93 |
|
94 |
|
95 |
|
96 |
|
97 |
|
98 |
|
99 | sub send {
|
100 | my ( $self, $data, $sample_rate ) = @_;
|
101 | $sample_rate = $self->{sample_rate} unless defined $sample_rate;
|
102 |
|
103 | my $sampled_data;
|
104 | if ( defined($sample_rate) and $sample_rate < 1 ) {
|
105 | while ( my ( $stat, $value ) = each %$data ) {
|
106 | $sampled_data->{$stat} = "$value|\@$sample_rate" if rand() <= $sample_rate;
|
107 | }
|
108 | }
|
109 | else {
|
110 | $sampled_data = $data;
|
111 | }
|
112 |
|
113 | return '0 but true' unless keys %$sampled_data;
|
114 |
|
115 |
|
116 | my $count = 0;
|
117 | my $socket = $self->{socket};
|
118 | while ( my ( $stat, $value ) = each %$sampled_data ) {
|
119 | _send_to_sock($socket, "$stat:$value\n");
|
120 | ++$count;
|
121 | }
|
122 | return $count;
|
123 | }
|
124 |
|
125 | sub _send_to_sock( $$ ) {
|
126 | my ($sock,$msg) = @_;
|
127 | CORE::send( $sock, $msg, 0 );
|
128 | }
|
129 |
|
130 |
|
131 |
|
132 |
|
133 |
|
134 |
|
135 |
|
136 |
|
137 |
|
138 |
|
139 |
|
140 | 1;
|