Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
branch: master
100 lines (84 sloc) 3.171 kb
#include <v8.h>
#include <node.h>
#include <uv.h>
#include <stdlib.h>
#if defined(_WIN32)
#include <time.h>
#else
#include <sys/time.h>
#endif
using namespace v8;
static const unsigned int POLL_PERIOD_MS = 500;
static unsigned int HIGH_WATER_MARK_MS = 70;
// A dampening factor. When determining average calls per second or
// current lag, we weigh the current value against the previous value 2:1
// to smooth spikes.
static const unsigned int AVG_DECAY_FACTOR = 3;
//static uv_idle_t s_idler;
static uv_timer_t s_timer;
static uint32_t s_currentLag;
static uint64_t s_lastMark;
Handle<Value> TooBusy(const Arguments& args) {
// No HandleScope required, because this function allocates no
// v8 classes that reside on the heap.
bool block = false;
if (s_currentLag > HIGH_WATER_MARK_MS) {
// probabilistically block requests proportional to how
// far behind we are.
double pctToBlock = ((s_currentLag - HIGH_WATER_MARK_MS) /
(double) HIGH_WATER_MARK_MS) * 100.0;
double r = (rand() / (double) RAND_MAX) * 100.0;
if (r < pctToBlock) block = true;
}
return block ? True() : False();
}
Handle<Value> ShutDown(const Arguments& args) {
// No HandleScope required, because this function allocates no
// v8 classes that reside on the heap.
uv_timer_stop(&s_timer);
return Undefined();
}
Handle<Value> Lag(const Arguments& args) {
HandleScope scope;
return scope.Close(Integer::New(s_currentLag));
}
Handle<Value> HighWaterMark(const Arguments& args) {
HandleScope scope;
if (args.Length() >= 1) {
if (!args[0]->IsNumber()) {
return v8::ThrowException(
v8::Exception::Error(
v8::String::New("expected numeric first argument")));
}
int hwm = args[0]->Int32Value();
if (hwm < 10) {
return v8::ThrowException(
v8::Exception::Error(
v8::String::New("maximum lag should be greater than 10ms")));
}
HIGH_WATER_MARK_MS = hwm;
}
return scope.Close(Number::New(HIGH_WATER_MARK_MS));
}
static void every_second(uv_timer_t* handle, int status)
{
uint64_t now = uv_hrtime();
if (s_lastMark > 0) {
// keep track of (dampened) average lag.
uint32_t lag = (uint32_t) ((now - s_lastMark) / 1000000);
lag = (lag < POLL_PERIOD_MS) ? 0 : lag - POLL_PERIOD_MS;
s_currentLag = (lag + (s_currentLag * (AVG_DECAY_FACTOR-1))) /
AVG_DECAY_FACTOR;
}
s_lastMark = now;
};
extern "C" void init(Handle<Object> target) {
HandleScope scope;
target->Set(String::New("toobusy"), FunctionTemplate::New(TooBusy)->GetFunction());
target->Set(String::New("shutdown"), FunctionTemplate::New(ShutDown)->GetFunction());
target->Set(String::New("lag"), FunctionTemplate::New(Lag)->GetFunction());
target->Set(String::New("maxLag"), FunctionTemplate::New(HighWaterMark)->GetFunction());
uv_timer_init(uv_default_loop(), &s_timer);
uv_timer_start(&s_timer, every_second, POLL_PERIOD_MS, POLL_PERIOD_MS);
};
NODE_MODULE(toobusy, init);
Jump to Line
Something went wrong with that request. Please try again.