(ws) add subcommands: cobra subscribe, and cobra subscribe to statsd bridge
This commit is contained in:
Benjamin Sergeant
246
third_party/statsd-client-cpp/src/statsd_client.cpp
vendored
Normal file
246
third_party/statsd-client-cpp/src/statsd_client.cpp
vendored
Normal file
@ -0,0 +1,246 @@
|
||||
#include <math.h>
|
||||
#include <netdb.h>
|
||||
#include <time.h>
|
||||
#include <unistd.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include <netinet/in.h>
|
||||
#include "statsd_client.h"
|
||||
|
||||
using namespace std;
|
||||
namespace statsd {
|
||||
|
||||
inline bool fequal(float a, float b)
|
||||
{
|
||||
const float epsilon = 0.0001;
|
||||
return ( fabs(a - b) < epsilon );
|
||||
}
|
||||
|
||||
inline bool should_send(float sample_rate)
|
||||
{
|
||||
if ( fequal(sample_rate, 1.0) )
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
float p = ((float)random() / RAND_MAX);
|
||||
return sample_rate > p;
|
||||
}
|
||||
|
||||
struct _StatsdClientData {
|
||||
int sock;
|
||||
struct sockaddr_in server;
|
||||
|
||||
string ns;
|
||||
string host;
|
||||
short port;
|
||||
bool init;
|
||||
|
||||
char errmsg[1024];
|
||||
};
|
||||
|
||||
StatsdClient::StatsdClient(const string& host,
|
||||
int port,
|
||||
const string& ns,
|
||||
const bool batching)
|
||||
: batching_(batching), exit_(false)
|
||||
{
|
||||
d = new _StatsdClientData;
|
||||
d->sock = -1;
|
||||
config(host, port, ns);
|
||||
srandom(time(NULL));
|
||||
|
||||
if (batching_) {
|
||||
pthread_mutex_init(&batching_mutex_lock_, nullptr);
|
||||
batching_thread_ = std::thread([this] {
|
||||
while (!exit_) {
|
||||
std::deque<std::string> staged_message_queue;
|
||||
|
||||
pthread_mutex_lock(&batching_mutex_lock_);
|
||||
batching_message_queue_.swap(staged_message_queue);
|
||||
pthread_mutex_unlock(&batching_mutex_lock_);
|
||||
|
||||
while(!staged_message_queue.empty()) {
|
||||
send_to_daemon(staged_message_queue.front());
|
||||
staged_message_queue.pop_front();
|
||||
}
|
||||
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(1000));
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
StatsdClient::~StatsdClient()
|
||||
{
|
||||
if (batching_) {
|
||||
exit_ = true;
|
||||
batching_thread_.join();
|
||||
pthread_mutex_destroy(&batching_mutex_lock_);
|
||||
}
|
||||
|
||||
|
||||
// close socket
|
||||
if (d->sock >= 0) {
|
||||
close(d->sock);
|
||||
d->sock = -1;
|
||||
delete d;
|
||||
d = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
void StatsdClient::config(const string& host, int port, const string& ns)
|
||||
{
|
||||
d->ns = ns;
|
||||
d->host = host;
|
||||
d->port = port;
|
||||
d->init = false;
|
||||
if ( d->sock >= 0 ) {
|
||||
close(d->sock);
|
||||
}
|
||||
d->sock = -1;
|
||||
}
|
||||
|
||||
int StatsdClient::init()
|
||||
{
|
||||
if ( d->init ) return 0;
|
||||
|
||||
d->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||
if ( d->sock == -1 ) {
|
||||
snprintf(d->errmsg, sizeof(d->errmsg), "could not create socket, err=%m");
|
||||
return -1;
|
||||
}
|
||||
|
||||
memset(&d->server, 0, sizeof(d->server));
|
||||
d->server.sin_family = AF_INET;
|
||||
d->server.sin_port = htons(d->port);
|
||||
|
||||
int ret = inet_aton(d->host.c_str(), &d->server.sin_addr);
|
||||
if ( ret == 0 )
|
||||
{
|
||||
// host must be a domain, get it from internet
|
||||
struct addrinfo hints, *result = NULL;
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
hints.ai_family = AF_INET;
|
||||
hints.ai_socktype = SOCK_DGRAM;
|
||||
|
||||
ret = getaddrinfo(d->host.c_str(), NULL, &hints, &result);
|
||||
if ( ret ) {
|
||||
close(d->sock);
|
||||
d->sock = -1;
|
||||
snprintf(d->errmsg, sizeof(d->errmsg),
|
||||
"getaddrinfo fail, error=%d, msg=%s", ret, gai_strerror(ret) );
|
||||
return -2;
|
||||
}
|
||||
struct sockaddr_in* host_addr = (struct sockaddr_in*)result->ai_addr;
|
||||
memcpy(&d->server.sin_addr, &host_addr->sin_addr, sizeof(struct in_addr));
|
||||
freeaddrinfo(result);
|
||||
}
|
||||
|
||||
d->init = true;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* will change the original string */
|
||||
void StatsdClient::cleanup(string& key)
|
||||
{
|
||||
size_t pos = key.find_first_of(":|@");
|
||||
while ( pos != string::npos )
|
||||
{
|
||||
key[pos] = '_';
|
||||
pos = key.find_first_of(":|@");
|
||||
}
|
||||
}
|
||||
|
||||
int StatsdClient::dec(const string& key, float sample_rate)
|
||||
{
|
||||
return count(key, -1, sample_rate);
|
||||
}
|
||||
|
||||
int StatsdClient::inc(const string& key, float sample_rate)
|
||||
{
|
||||
return count(key, 1, sample_rate);
|
||||
}
|
||||
|
||||
int StatsdClient::count(const string& key, size_t value, float sample_rate)
|
||||
{
|
||||
return send(key, value, "c", sample_rate);
|
||||
}
|
||||
|
||||
int StatsdClient::gauge(const string& key, size_t value, float sample_rate)
|
||||
{
|
||||
return send(key, value, "g", sample_rate);
|
||||
}
|
||||
|
||||
int StatsdClient::timing(const string& key, size_t ms, float sample_rate)
|
||||
{
|
||||
return send(key, ms, "ms", sample_rate);
|
||||
}
|
||||
|
||||
int StatsdClient::send(string key, size_t value, const string &type, float sample_rate)
|
||||
{
|
||||
if (!should_send(sample_rate)) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
cleanup(key);
|
||||
|
||||
char buf[256];
|
||||
if ( fequal( sample_rate, 1.0 ) )
|
||||
{
|
||||
snprintf(buf, sizeof(buf), "%s%s:%zd|%s",
|
||||
d->ns.c_str(), key.c_str(), value, type.c_str());
|
||||
}
|
||||
else
|
||||
{
|
||||
snprintf(buf, sizeof(buf), "%s%s:%zd|%s|@%.2f",
|
||||
d->ns.c_str(), key.c_str(), value, type.c_str(), sample_rate);
|
||||
}
|
||||
|
||||
return send(buf);
|
||||
}
|
||||
|
||||
int StatsdClient::send(const string &message)
|
||||
{
|
||||
if (batching_) {
|
||||
pthread_mutex_lock(&batching_mutex_lock_);
|
||||
if (batching_message_queue_.empty() ||
|
||||
batching_message_queue_.back().length() > max_batching_size) {
|
||||
batching_message_queue_.push_back(message);
|
||||
} else {
|
||||
(*batching_message_queue_.rbegin()).append("\n").append(message);
|
||||
}
|
||||
pthread_mutex_unlock(&batching_mutex_lock_);
|
||||
|
||||
return 0;
|
||||
} else {
|
||||
return send_to_daemon(message);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int StatsdClient::send_to_daemon(const string &message) {
|
||||
std::cout << "send_to_daemon: " << message.length() << " B" << std::endl;
|
||||
int ret = init();
|
||||
if ( ret )
|
||||
{
|
||||
return ret;
|
||||
}
|
||||
ret = sendto(d->sock, message.data(), message.size(), 0, (struct sockaddr *) &d->server, sizeof(d->server));
|
||||
if ( ret == -1) {
|
||||
snprintf(d->errmsg, sizeof(d->errmsg),
|
||||
"sendto server fail, host=%s:%d, err=%m", d->host.c_str(), d->port);
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char* StatsdClient::errmsg()
|
||||
{
|
||||
return d->errmsg;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
66
third_party/statsd-client-cpp/src/statsd_client.h
vendored
Normal file
66
third_party/statsd-client-cpp/src/statsd_client.h
vendored
Normal file
@ -0,0 +1,66 @@
|
||||
|
||||
#ifndef STATSD_CLIENT_H
|
||||
#define STATSD_CLIENT_H
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/socket.h>
|
||||
#include <pthread.h>
|
||||
#include <string>
|
||||
#include <thread>
|
||||
#include <deque>
|
||||
#include <iostream>
|
||||
|
||||
namespace statsd {
|
||||
|
||||
struct _StatsdClientData;
|
||||
|
||||
class StatsdClient {
|
||||
public:
|
||||
StatsdClient(const std::string& host="127.0.0.1", int port=8125, const std::string& ns = "", const bool batching = false);
|
||||
~StatsdClient();
|
||||
|
||||
public:
|
||||
// you can config at anytime; client will use new address (useful for Singleton)
|
||||
void config(const std::string& host, int port, const std::string& ns = "");
|
||||
const char* errmsg();
|
||||
int send_to_daemon(const std::string &);
|
||||
|
||||
public:
|
||||
int inc(const std::string& key, float sample_rate = 1.0);
|
||||
int dec(const std::string& key, float sample_rate = 1.0);
|
||||
int count(const std::string& key, size_t value, float sample_rate = 1.0);
|
||||
int gauge(const std::string& key, size_t value, float sample_rate = 1.0);
|
||||
int timing(const std::string& key, size_t ms, float sample_rate = 1.0);
|
||||
|
||||
public:
|
||||
/**
|
||||
* (Low Level Api) manually send a message
|
||||
* which might be composed of several lines.
|
||||
*/
|
||||
int send(const std::string& message);
|
||||
|
||||
/* (Low Level Api) manually send a message
|
||||
* type = "c", "g" or "ms"
|
||||
*/
|
||||
int send(std::string key, size_t value,
|
||||
const std::string& type, float sample_rate);
|
||||
|
||||
protected:
|
||||
int init();
|
||||
void cleanup(std::string& key);
|
||||
|
||||
protected:
|
||||
struct _StatsdClientData* d;
|
||||
|
||||
bool batching_;
|
||||
bool exit_;
|
||||
pthread_mutex_t batching_mutex_lock_;
|
||||
std::thread batching_thread_;
|
||||
std::deque<std::string> batching_message_queue_;
|
||||
const uint64_t max_batching_size = 32768;
|
||||
};
|
||||
|
||||
}; // end namespace
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user