forked from Cavemanon/cavepaintings
PSR-2. I'm not a huge fan, but ugly consistency beats no consistency...
This commit is contained in:
@@ -11,82 +11,90 @@
|
||||
|
||||
_d("STATSD_HOST", null);
|
||||
|
||||
function dstat($name, $val) {
|
||||
StatsDInterface::$stats["shimmie.$name"] = $val;
|
||||
function dstat($name, $val)
|
||||
{
|
||||
StatsDInterface::$stats["shimmie.$name"] = $val;
|
||||
}
|
||||
|
||||
class StatsDInterface extends Extension {
|
||||
public static $stats = array();
|
||||
class StatsDInterface extends Extension
|
||||
{
|
||||
public static $stats = [];
|
||||
|
||||
private function _stats(string $type) {
|
||||
global $_shm_event_count, $database, $_shm_load_start;
|
||||
$time = microtime(true) - $_shm_load_start;
|
||||
StatsDInterface::$stats["shimmie.$type.hits"] = "1|c";
|
||||
StatsDInterface::$stats["shimmie.$type.time"] = "$time|ms";
|
||||
StatsDInterface::$stats["shimmie.$type.time-db"] = "{$database->dbtime}|ms";
|
||||
StatsDInterface::$stats["shimmie.$type.memory"] = memory_get_peak_usage(true)."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.files"] = count(get_included_files())."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.queries"] = $database->query_count."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.events"] = $_shm_event_count."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.cache-hits"] = $database->cache->get_hits()."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.cache-misses"] = $database->cache->get_misses()."|c";
|
||||
}
|
||||
private function _stats(string $type)
|
||||
{
|
||||
global $_shm_event_count, $database, $_shm_load_start;
|
||||
$time = microtime(true) - $_shm_load_start;
|
||||
StatsDInterface::$stats["shimmie.$type.hits"] = "1|c";
|
||||
StatsDInterface::$stats["shimmie.$type.time"] = "$time|ms";
|
||||
StatsDInterface::$stats["shimmie.$type.time-db"] = "{$database->dbtime}|ms";
|
||||
StatsDInterface::$stats["shimmie.$type.memory"] = memory_get_peak_usage(true)."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.files"] = count(get_included_files())."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.queries"] = $database->query_count."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.events"] = $_shm_event_count."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.cache-hits"] = $database->cache->get_hits()."|c";
|
||||
StatsDInterface::$stats["shimmie.$type.cache-misses"] = $database->cache->get_misses()."|c";
|
||||
}
|
||||
|
||||
public function onPageRequest(PageRequestEvent $event) {
|
||||
$this->_stats("overall");
|
||||
public function onPageRequest(PageRequestEvent $event)
|
||||
{
|
||||
$this->_stats("overall");
|
||||
|
||||
if($event->page_matches("post/view")) { # 40%
|
||||
$this->_stats("post-view");
|
||||
}
|
||||
else if($event->page_matches("post/list")) { # 30%
|
||||
$this->_stats("post-list");
|
||||
}
|
||||
else if($event->page_matches("user")) {
|
||||
$this->_stats("user");
|
||||
}
|
||||
else if($event->page_matches("upload")) {
|
||||
$this->_stats("upload");
|
||||
}
|
||||
else if($event->page_matches("rss")) {
|
||||
$this->_stats("rss");
|
||||
}
|
||||
else if($event->page_matches("api")) {
|
||||
$this->_stats("api");
|
||||
}
|
||||
else {
|
||||
#global $_shm_load_start;
|
||||
#$time = microtime(true) - $_shm_load_start;
|
||||
#file_put_contents("data/other.log", "{$_SERVER['REQUEST_URI']} $time\n", FILE_APPEND);
|
||||
$this->_stats("other");
|
||||
}
|
||||
if ($event->page_matches("post/view")) { # 40%
|
||||
$this->_stats("post-view");
|
||||
} elseif ($event->page_matches("post/list")) { # 30%
|
||||
$this->_stats("post-list");
|
||||
} elseif ($event->page_matches("user")) {
|
||||
$this->_stats("user");
|
||||
} elseif ($event->page_matches("upload")) {
|
||||
$this->_stats("upload");
|
||||
} elseif ($event->page_matches("rss")) {
|
||||
$this->_stats("rss");
|
||||
} elseif ($event->page_matches("api")) {
|
||||
$this->_stats("api");
|
||||
} else {
|
||||
#global $_shm_load_start;
|
||||
#$time = microtime(true) - $_shm_load_start;
|
||||
#file_put_contents("data/other.log", "{$_SERVER['REQUEST_URI']} $time\n", FILE_APPEND);
|
||||
$this->_stats("other");
|
||||
}
|
||||
|
||||
$this->send(StatsDInterface::$stats, 1.0);
|
||||
StatsDInterface::$stats = array();
|
||||
}
|
||||
$this->send(StatsDInterface::$stats, 1.0);
|
||||
StatsDInterface::$stats = [];
|
||||
}
|
||||
|
||||
public function onUserCreation(UserCreationEvent $event) {
|
||||
StatsDInterface::$stats["shimmie_events.user_creations"] = "1|c";
|
||||
}
|
||||
public function onUserCreation(UserCreationEvent $event)
|
||||
{
|
||||
StatsDInterface::$stats["shimmie_events.user_creations"] = "1|c";
|
||||
}
|
||||
|
||||
public function onDataUpload(DataUploadEvent $event) {
|
||||
StatsDInterface::$stats["shimmie_events.uploads"] = "1|c";
|
||||
}
|
||||
public function onDataUpload(DataUploadEvent $event)
|
||||
{
|
||||
StatsDInterface::$stats["shimmie_events.uploads"] = "1|c";
|
||||
}
|
||||
|
||||
public function onCommentPosting(CommentPostingEvent $event) {
|
||||
StatsDInterface::$stats["shimmie_events.comments"] = "1|c";
|
||||
}
|
||||
public function onCommentPosting(CommentPostingEvent $event)
|
||||
{
|
||||
StatsDInterface::$stats["shimmie_events.comments"] = "1|c";
|
||||
}
|
||||
|
||||
public function onImageInfoSet(ImageInfoSetEvent $event) {
|
||||
StatsDInterface::$stats["shimmie_events.info-sets"] = "1|c";
|
||||
}
|
||||
public function onImageInfoSet(ImageInfoSetEvent $event)
|
||||
{
|
||||
StatsDInterface::$stats["shimmie_events.info-sets"] = "1|c";
|
||||
}
|
||||
|
||||
public function get_priority(): int {return 99;}
|
||||
public function get_priority(): int
|
||||
{
|
||||
return 99;
|
||||
}
|
||||
|
||||
private function send(array $data, int $sampleRate=1) {
|
||||
if (!STATSD_HOST) { return; }
|
||||
private function send(array $data, int $sampleRate=1)
|
||||
{
|
||||
if (!STATSD_HOST) {
|
||||
return;
|
||||
}
|
||||
|
||||
// sampling
|
||||
$sampledData = array();
|
||||
$sampledData = [];
|
||||
|
||||
if ($sampleRate < 1) {
|
||||
foreach ($data as $stat => $value) {
|
||||
@@ -98,7 +106,9 @@ class StatsDInterface extends Extension {
|
||||
$sampledData = $data;
|
||||
}
|
||||
|
||||
if (empty($sampledData)) { return; }
|
||||
if (empty($sampledData)) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Wrap this in a try/catch - failures in any of this should be silently ignored
|
||||
try {
|
||||
@@ -106,7 +116,9 @@ class StatsDInterface extends Extension {
|
||||
$host = $parts[0];
|
||||
$port = $parts[1];
|
||||
$fp = fsockopen("udp://$host", $port, $errno, $errstr);
|
||||
if (! $fp) { return; }
|
||||
if (! $fp) {
|
||||
return;
|
||||
}
|
||||
foreach ($sampledData as $stat => $value) {
|
||||
fwrite($fp, "$stat:$value");
|
||||
}
|
||||
|
Reference in New Issue
Block a user