sikkienl/dash
1
0
Fork 0
mirror of https://github.com/dashpay/dash.git synced 2024-12-25 03:52:49 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

stats: move _StatsdClientData variables into StatsdClient

Browse Source 
There doesn't seem to be a benefit to keeping these values in a separate
struct.
This commit is contained in:
Kittywhiskers Van Gogh 2024-09-05 07:51:25 +00:00
parent 30c30c1397
commit a9d1b1494d
No known key found for this signature in database
GPG Key ID: 30CD0C065E5C4AAD
2 changed files with 39 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
src/statsd_client.cpp
View File

@ -41,7 +41,6 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
std::unique_ptr<statsd::StatsdClient> g_stats_client;
namespace statsd {
inline bool fequal(float a, float b)
{
const float epsilon = 0.0001;
@ -61,64 +60,39 @@ inline bool should_send(float sample_rate)
return sample_rate > p;
}
struct _StatsdClientData {
SOCKET sock;
struct sockaddr_in server;
std::string ns;
std::string host;
std::string nodename;
short port;
bool init;
char errmsg[1024];
};
StatsdClient::StatsdClient(const std::string& host, const std::string& nodename, int port, const std::string& ns,
StatsdClient::StatsdClient(const std::string& host, const std::string& nodename, short port, const std::string& ns,
bool enabled)
: d{std::make_unique<_StatsdClientData>()}, m_enabled{enabled}
: m_enabled{enabled}, m_port{port}, m_host{host}, m_nodename{nodename}, m_ns{ns}
{
d->sock = INVALID_SOCKET;
config(host, nodename, port, ns);
}
StatsdClient::~StatsdClient()
{
// close socket
CloseSocket(d->sock);
}
void StatsdClient::config(const std::string& host, const std::string& nodename, int port, const std::string& ns)
{
d->ns = ns;
d->host = host;
d->nodename = nodename;
d->port = port;
d->init = false;
CloseSocket(d->sock);
CloseSocket(m_sock);
}
int StatsdClient::init()
{
if ( d->init ) return 0;
if ( m_init ) return 0;
d->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if ( d->sock == INVALID_SOCKET ) {
snprintf(d->errmsg, sizeof(d->errmsg), "could not create socket, err=%m");
m_sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if ( m_sock == INVALID_SOCKET ) {
snprintf(m_errmsg, sizeof(m_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);
memset(&m_server, 0, sizeof(m_server));
m_server.sin_family = AF_INET;
m_server.sin_port = htons(m_port);
CNetAddr netaddr(d->server.sin_addr);
if (!LookupHost(d->host, netaddr, true) || !netaddr.GetInAddr(&d->server.sin_addr)) {
snprintf(d->errmsg, sizeof(d->errmsg), "LookupHost or GetInAddr failed");
CNetAddr netaddr(m_server.sin_addr);
if (!LookupHost(m_host, netaddr, true) || !netaddr.GetInAddr(&m_server.sin_addr)) {
snprintf(m_errmsg, sizeof(m_errmsg), "LookupHost or GetInAddr failed");
return -2;
}
d->init = true;
m_init = true;
return 0;
}
@ -170,8 +144,8 @@ int StatsdClient::send(std::string key, size_t value, const std::string& type, f
}
// partition stats by node name if set
if (!d->nodename.empty())
key = key + "." + d->nodename;
if (!m_nodename.empty())
key = key + "." + m_nodename;
cleanup(key);
@ -179,12 +153,12 @@ int StatsdClient::send(std::string key, size_t value, const std::string& type, f
if ( fequal( sample_rate, 1.0 ) )
{
snprintf(buf, sizeof(buf), "%s%s:%zd|%s",
d->ns.c_str(), key.c_str(), (ssize_t) value, type.c_str());
m_ns.c_str(), key.c_str(), (ssize_t) value, type.c_str());
}
else
{
snprintf(buf, sizeof(buf), "%s%s:%zd|%s|@%.2f",
d->ns.c_str(), key.c_str(), (ssize_t) value, type.c_str(), sample_rate);
m_ns.c_str(), key.c_str(), (ssize_t) value, type.c_str(), sample_rate);
}
return send(buf);
@ -197,8 +171,8 @@ int StatsdClient::sendDouble(std::string key, double value, const std::string& t
}
// partition stats by node name if set
if (!d->nodename.empty())
key = key + "." + d->nodename;
if (!m_nodename.empty())
key = key + "." + m_nodename;
cleanup(key);
@ -206,12 +180,12 @@ int StatsdClient::sendDouble(std::string key, double value, const std::string& t
if ( fequal( sample_rate, 1.0 ) )
{
snprintf(buf, sizeof(buf), "%s%s:%f|%s",
d->ns.c_str(), key.c_str(), value, type.c_str());
m_ns.c_str(), key.c_str(), value, type.c_str());
}
else
{
snprintf(buf, sizeof(buf), "%s%s:%f|%s|@%.2f",
d->ns.c_str(), key.c_str(), value, type.c_str(), sample_rate);
m_ns.c_str(), key.c_str(), value, type.c_str(), sample_rate);
}
return send(buf);
@ -227,10 +201,10 @@ int StatsdClient::send(const std::string& message)
{
return ret;
}
ret = sendto(d->sock, message.data(), message.size(), 0, reinterpret_cast<const sockaddr*>(&d->server), sizeof(d->server));
ret = sendto(m_sock, message.data(), message.size(), 0, reinterpret_cast<const sockaddr*>(&m_server), sizeof(m_server));
if ( ret == -1) {
snprintf(d->errmsg, sizeof(d->errmsg),
"sendto server fail, host=%s:%d, err=%m", d->host.c_str(), d->port);
snprintf(m_errmsg, sizeof(m_errmsg),
"sendto server fail, host=%s:%d, err=%m", m_host.c_str(), m_port);
return -1;
}
return 0;
@ -238,7 +212,6 @@ int StatsdClient::send(const std::string& message)
const char* StatsdClient::errmsg()
{
return d->errmsg;
return m_errmsg;
}
} // namespace statsd

24
src/statsd_client.h
View File

@ -5,6 +5,8 @@
#ifndef BITCOIN_STATSD_CLIENT_H
#define BITCOIN_STATSD_CLIENT_H
#include <util/sock.h>
#include <string>
#include <memory>
@ -20,18 +22,13 @@ static const int MIN_STATSD_PERIOD = 5;
static const int MAX_STATSD_PERIOD = 60 * 60;
namespace statsd {
struct _StatsdClientData;
class StatsdClient {
public:
explicit StatsdClient(const std::string& host, const std::string& nodename, int port, const std::string& ns,
explicit StatsdClient(const std::string& host, const std::string& nodename, short port, const std::string& ns,
bool enabled);
~StatsdClient();
public:
// you can config at anytime; client will use new address (useful for Singleton)
void config(const std::string& host, const std::string& nodename, int port, const std::string& ns);
const char* errmsg();
public:
@ -61,13 +58,18 @@ class StatsdClient {
int init();
static void cleanup(std::string& key);
protected:
const std::unique_ptr<struct _StatsdClientData> d;
private:
const bool m_enabled{false};
};
bool m_init{false};
char m_errmsg[1024];
SOCKET m_sock{INVALID_SOCKET};
struct sockaddr_in m_server;
const bool m_enabled{false};
const short m_port;
const std::string m_host;
const std::string m_nodename;
const std::string m_ns;
};
} // namespace statsd
extern std::unique_ptr<statsd::StatsdClient> g_stats_client;