See More

#include "histogram.h" // NOLINT(build/include_inline) #include "base_object-inl.h" #include "histogram-inl.h" #include "memory_tracker-inl.h" #include "node_debug.h" #include "node_errors.h" #include "node_external_reference.h" #include "util.h" namespace node { using v8::BigInt; using v8::CFunction; using v8::Context; using v8::FunctionCallbackInfo; using v8::FunctionTemplate; using v8::Integer; using v8::Isolate; using v8::Local; using v8::Map; using v8::Number; using v8::Object; using v8::ObjectTemplate; using v8::String; using v8::Uint32; using v8::Value; Histogram::Histogram(const Options& options) { hdr_histogram* histogram; CHECK_EQ(0, hdr_init(options.lowest, options.highest, options.figures, &histogram)); histogram_.reset(histogram); } void Histogram::MemoryInfo(MemoryTracker* tracker) const { tracker->TrackFieldWithSize("histogram", GetMemorySize()); } HistogramImpl::HistogramImpl(const Histogram::Options& options) : histogram_(new Histogram(options)) {} HistogramImpl::HistogramImpl(std::shared_ptr histogram) : histogram_(std::move(histogram)) {} CFunction HistogramImpl::fast_reset_( CFunction::Make(&HistogramImpl::FastReset)); CFunction HistogramImpl::fast_get_count_( CFunction::Make(&HistogramImpl::FastGetCount)); CFunction HistogramImpl::fast_get_min_( CFunction::Make(&HistogramImpl::FastGetMin)); CFunction HistogramImpl::fast_get_max_( CFunction::Make(&HistogramImpl::FastGetMax)); CFunction HistogramImpl::fast_get_mean_( CFunction::Make(&HistogramImpl::FastGetMean)); CFunction HistogramImpl::fast_get_exceeds_( CFunction::Make(&HistogramImpl::FastGetExceeds)); CFunction HistogramImpl::fast_get_stddev_( CFunction::Make(&HistogramImpl::FastGetStddev)); CFunction HistogramImpl::fast_get_percentile_( CFunction::Make(&HistogramImpl::FastGetPercentile)); CFunction HistogramBase::fast_record_( CFunction::Make(&HistogramBase::FastRecord)); CFunction HistogramBase::fast_record_delta_( CFunction::Make(&HistogramBase::FastRecordDelta)); CFunction IntervalHistogram::fast_start_( CFunction::Make(&IntervalHistogram::FastStart)); CFunction IntervalHistogram::fast_stop_( CFunction::Make(&IntervalHistogram::FastStop)); void HistogramImpl::AddMethods(Isolate* isolate, Local tmpl) { // TODO(@jasnell): The bigint API variations do not yet support fast // variations since v8 will not return a bigint value from a fast method. SetProtoMethodNoSideEffect(isolate, tmpl, "countBigInt", GetCountBigInt); SetProtoMethodNoSideEffect(isolate, tmpl, "exceedsBigInt", GetExceedsBigInt); SetProtoMethodNoSideEffect(isolate, tmpl, "minBigInt", GetMinBigInt); SetProtoMethodNoSideEffect(isolate, tmpl, "maxBigInt", GetMaxBigInt); SetProtoMethodNoSideEffect( isolate, tmpl, "percentileBigInt", GetPercentileBigInt); SetProtoMethodNoSideEffect(isolate, tmpl, "percentiles", GetPercentiles); SetProtoMethodNoSideEffect( isolate, tmpl, "percentilesBigInt", GetPercentilesBigInt); auto instance = tmpl->InstanceTemplate(); SetFastMethodNoSideEffect( isolate, instance, "count", GetCount, &fast_get_count_); SetFastMethodNoSideEffect( isolate, instance, "exceeds", GetExceeds, &fast_get_exceeds_); SetFastMethodNoSideEffect(isolate, instance, "min", GetMin, &fast_get_min_); SetFastMethodNoSideEffect(isolate, instance, "max", GetMax, &fast_get_max_); SetFastMethodNoSideEffect( isolate, instance, "mean", GetMean, &fast_get_mean_); SetFastMethodNoSideEffect( isolate, instance, "stddev", GetStddev, &fast_get_stddev_); SetFastMethodNoSideEffect( isolate, instance, "percentile", GetPercentile, &fast_get_percentile_); SetFastMethod(isolate, instance, "reset", DoReset, &fast_reset_); } void HistogramImpl::RegisterExternalReferences( ExternalReferenceRegistry* registry) { static bool is_registered = false; if (is_registered) return; registry->Register(GetCount); registry->Register(GetCountBigInt); registry->Register(GetExceeds); registry->Register(GetExceedsBigInt); registry->Register(GetMin); registry->Register(GetMinBigInt); registry->Register(GetMax); registry->Register(GetMaxBigInt); registry->Register(GetMean); registry->Register(GetStddev); registry->Register(GetPercentile); registry->Register(GetPercentileBigInt); registry->Register(GetPercentiles); registry->Register(GetPercentilesBigInt); registry->Register(DoReset); registry->Register(fast_reset_); registry->Register(fast_get_count_); registry->Register(fast_get_min_); registry->Register(fast_get_max_); registry->Register(fast_get_mean_); registry->Register(fast_get_exceeds_); registry->Register(fast_get_stddev_); registry->Register(fast_get_percentile_); is_registered = true; } HistogramBase::HistogramBase( Environment* env, Local wrap, const Histogram::Options& options) : BaseObject(env, wrap), HistogramImpl(options) { MakeWeak(); wrap->SetAlignedPointerInInternalField( HistogramImpl::InternalFields::kImplField, static_cast(this), EmbedderDataTag::kDefault); } HistogramBase::HistogramBase( Environment* env, Local wrap, std::shared_ptr histogram) : BaseObject(env, wrap), HistogramImpl(std::move(histogram)) { MakeWeak(); wrap->SetAlignedPointerInInternalField( HistogramImpl::InternalFields::kImplField, static_cast(this), EmbedderDataTag::kDefault); } void HistogramBase::MemoryInfo(MemoryTracker* tracker) const { tracker->TrackField("histogram", histogram()); } void HistogramBase::RecordDelta(const FunctionCallbackInfo& args) { HistogramBase* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, args.This()); (*histogram)->RecordDelta(); } void HistogramBase::FastRecordDelta(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.recordDelta"); HistogramBase* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, receiver); (*histogram)->RecordDelta(); } void HistogramBase::Record(const FunctionCallbackInfo& args) { Environment* env = Environment::GetCurrent(args); CHECK_IMPLIES(!args[0]->IsNumber(), args[0]->IsBigInt()); bool lossless = true; int64_t value = args[0]->IsBigInt() ? args[0].As()->Int64Value(&lossless) : static_cast(args[0].As()->Value()); if (!lossless || value < 1) return THROW_ERR_OUT_OF_RANGE(env, "value is out of range"); HistogramBase* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, args.This()); (*histogram)->Record(value); } void HistogramBase::FastRecord(Local receiver, const int64_t value) { CHECK_GE(value, 1); TRACK_V8_FAST_API_CALL("histogram.record"); HistogramBase* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, receiver); (*histogram)->Record(value); } void HistogramBase::Add(const FunctionCallbackInfo& args) { Environment* env = Environment::GetCurrent(args); HistogramBase* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, args.This()); CHECK(GetConstructorTemplate(env->isolate_data())->HasInstance(args[0])); HistogramBase* other; ASSIGN_OR_RETURN_UNWRAP(&other, args[0]); double count = (*histogram)->Add(*(other->histogram())); args.GetReturnValue().Set(count); } BaseObjectPtr HistogramBase::Create( Environment* env, const Histogram::Options& options) { Local obj; if (!GetConstructorTemplate(env->isolate_data()) ->InstanceTemplate() ->NewInstance(env->context()) .ToLocal(&obj)) { return nullptr; } return MakeBaseObject(env, obj, options); } BaseObjectPtr HistogramBase::Create( Environment* env, std::shared_ptr histogram) { Local obj; if (!GetConstructorTemplate(env->isolate_data()) ->InstanceTemplate() ->NewInstance(env->context()) .ToLocal(&obj)) { return nullptr; } return MakeBaseObject(env, obj, std::move(histogram)); } void HistogramBase::New(const FunctionCallbackInfo& args) { CHECK(args.IsConstructCall()); Environment* env = Environment::GetCurrent(args); CHECK_IMPLIES(!args[0]->IsNumber(), args[0]->IsBigInt()); CHECK_IMPLIES(!args[1]->IsNumber(), args[1]->IsBigInt()); CHECK(args[2]->IsUint32()); int64_t lowest = 1; int64_t highest = std::numeric_limits::max(); bool lossless_ignored; if (args[0]->IsNumber()) { lowest = args[0].As()->Value(); } else if (args[0]->IsBigInt()) { lowest = args[0].As()->Int64Value(&lossless_ignored); } if (args[1]->IsNumber()) { highest = args[1].As()->Value(); } else if (args[1]->IsBigInt()) { highest = args[1].As()->Int64Value(&lossless_ignored); } int32_t figures = args[2].As()->Value(); new HistogramBase(env, args.This(), Histogram::Options { lowest, highest, figures }); } Local HistogramBase::GetConstructorTemplate( IsolateData* isolate_data) { Local tmpl = isolate_data->histogram_ctor_template(); if (tmpl.IsEmpty()) { Isolate* isolate = isolate_data->isolate(); tmpl = NewFunctionTemplate(isolate, New); Local classname = FIXED_ONE_BYTE_STRING(isolate, "Histogram"); tmpl->SetClassName(classname); auto instance = tmpl->InstanceTemplate(); instance->SetInternalFieldCount(HistogramBase::kInternalFieldCount); SetFastMethod(isolate, instance, "record", Record, &fast_record_); SetFastMethod( isolate, instance, "recordDelta", RecordDelta, &fast_record_delta_); SetProtoMethod(isolate, tmpl, "add", Add); HistogramImpl::AddMethods(isolate, tmpl); isolate_data->set_histogram_ctor_template(tmpl); } return tmpl; } void HistogramBase::RegisterExternalReferences( ExternalReferenceRegistry* registry) { registry->Register(New); registry->Register(Add); registry->Register(Record); registry->Register(RecordDelta); registry->Register(fast_record_); registry->Register(fast_record_delta_); HistogramImpl::RegisterExternalReferences(registry); } void HistogramBase::Initialize(IsolateData* isolate_data, Local target) { SetConstructorFunction(isolate_data->isolate(), target, "Histogram", GetConstructorTemplate(isolate_data), SetConstructorFunctionFlag::NONE); } BaseObjectPtr HistogramBase::HistogramTransferData::Deserialize( Environment* env, Local context, std::unique_ptr<:transferdata> self) { return Create(env, std::move(histogram_)); } std::unique_ptr<:transferdata> HistogramBase::CloneForMessaging() const { return std::make_unique(this); } void HistogramBase::HistogramTransferData::MemoryInfo( MemoryTracker* tracker) const { tracker->TrackField("histogram", histogram_); } Local IntervalHistogram::GetConstructorTemplate( Environment* env) { Local tmpl = env->intervalhistogram_constructor_template(); if (tmpl.IsEmpty()) { Isolate* isolate = env->isolate(); tmpl = NewFunctionTemplate(isolate, nullptr); tmpl->Inherit(HandleWrap::GetConstructorTemplate(env)); tmpl->SetClassName(FIXED_ONE_BYTE_STRING(isolate, "Histogram")); auto instance = tmpl->InstanceTemplate(); instance->SetInternalFieldCount(IntervalHistogram::kInternalFieldCount); HistogramImpl::AddMethods(isolate, tmpl); SetFastMethod(isolate, instance, "start", Start, &fast_start_); SetFastMethod(isolate, instance, "stop", Stop, &fast_stop_); env->set_intervalhistogram_constructor_template(tmpl); } return tmpl; } void IntervalHistogram::RegisterExternalReferences( ExternalReferenceRegistry* registry) { registry->Register(Start); registry->Register(Stop); registry->Register(fast_start_); registry->Register(fast_stop_); HistogramImpl::RegisterExternalReferences(registry); } IntervalHistogram::IntervalHistogram( Environment* env, Local wrap, AsyncWrap::ProviderType type, int32_t interval, std::function on_interval, const Histogram::Options& options) : HandleWrap( env, wrap, reinterpret_cast(&timer_), type), HistogramImpl(options), interval_(interval), on_interval_(std::move(on_interval)) { MakeWeak(); wrap->SetAlignedPointerInInternalField( HistogramImpl::InternalFields::kImplField, static_cast(this), EmbedderDataTag::kDefault); uv_timer_init(env->event_loop(), &timer_); } BaseObjectPtr IntervalHistogram::Create( Environment* env, int32_t interval, std::function on_interval, const Histogram::Options& options) { Local obj; if (!GetConstructorTemplate(env) ->InstanceTemplate() ->NewInstance(env->context()).ToLocal(&obj)) { return nullptr; } return MakeBaseObject( env, obj, AsyncWrap::PROVIDER_ELDHISTOGRAM, interval, std::move(on_interval), options); } void IntervalHistogram::TimerCB(uv_timer_t* handle) { IntervalHistogram* histogram = ContainerOf(&IntervalHistogram::timer_, handle); Histogram* h = histogram->histogram().get(); histogram->on_interval_(*h); } void IntervalHistogram::MemoryInfo(MemoryTracker* tracker) const { tracker->TrackField("histogram", histogram()); } void IntervalHistogram::OnStart(StartFlags flags) { if (enabled_ || IsHandleClosing()) return; enabled_ = true; if (flags == StartFlags::RESET) histogram()->Reset(); uv_timer_start(&timer_, TimerCB, interval_, interval_); uv_unref(reinterpret_cast(&timer_)); } void IntervalHistogram::OnStop() { if (!enabled_ || IsHandleClosing()) return; enabled_ = false; uv_timer_stop(&timer_); } void IntervalHistogram::Start(const FunctionCallbackInfo& args) { IntervalHistogram* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, args.This()); histogram->OnStart(args[0]->IsTrue() ? StartFlags::RESET : StartFlags::NONE); } void IntervalHistogram::FastStart(Local receiver, bool reset) { TRACK_V8_FAST_API_CALL("histogram.start"); IntervalHistogram* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, receiver); histogram->OnStart(reset ? StartFlags::RESET : StartFlags::NONE); } void IntervalHistogram::Stop(const FunctionCallbackInfo& args) { IntervalHistogram* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, args.This()); histogram->OnStop(); } void IntervalHistogram::FastStop(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.stop"); IntervalHistogram* histogram; ASSIGN_OR_RETURN_UNWRAP(&histogram, receiver); histogram->OnStop(); } void HistogramImpl::GetCount(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); double value = static_cast((*histogram)->Count()); args.GetReturnValue().Set(value); } void HistogramImpl::GetCountBigInt(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set( BigInt::NewFromUnsigned(args.GetIsolate(), (*histogram)->Count())); } void HistogramImpl::GetMin(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); double value = static_cast((*histogram)->Min()); args.GetReturnValue().Set(value); } void HistogramImpl::GetMinBigInt(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set( BigInt::New(args.GetIsolate(), (*histogram)->Min())); } void HistogramImpl::GetMax(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); double value = static_cast((*histogram)->Max()); args.GetReturnValue().Set(value); } void HistogramImpl::GetMaxBigInt(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set( BigInt::New(args.GetIsolate(), (*histogram)->Max())); } void HistogramImpl::GetMean(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set((*histogram)->Mean()); } void HistogramImpl::GetExceeds(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); double value = static_cast((*histogram)->Exceeds()); args.GetReturnValue().Set(value); } void HistogramImpl::GetExceedsBigInt(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set( BigInt::New(args.GetIsolate(), (*histogram)->Exceeds())); } void HistogramImpl::GetStddev(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); args.GetReturnValue().Set((*histogram)->Stddev()); } void HistogramImpl::GetPercentile(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); CHECK(args[0]->IsNumber()); double percentile = args[0].As()->Value(); double value = static_cast((*histogram)->Percentile(percentile)); args.GetReturnValue().Set(value); } void HistogramImpl::GetPercentileBigInt( const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); CHECK(args[0]->IsNumber()); double percentile = args[0].As()->Value(); int64_t value = (*histogram)->Percentile(percentile); args.GetReturnValue().Set(BigInt::New(args.GetIsolate(), value)); } void HistogramImpl::GetPercentiles(const FunctionCallbackInfo& args) { Environment* env = Environment::GetCurrent(args); HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); CHECK(args[0]->IsMap()); Local map = args[0].As(); (*histogram)->Percentiles([map, env](double key, int64_t value) { USE(map->Set( env->context(), Number::New(env->isolate(), key), Number::New(env->isolate(), static_cast(value)))); }); } void HistogramImpl::GetPercentilesBigInt( const FunctionCallbackInfo& args) { Environment* env = Environment::GetCurrent(args); HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); CHECK(args[0]->IsMap()); Local map = args[0].As(); (*histogram)->Percentiles([map, env](double key, int64_t value) { USE(map->Set( env->context(), Number::New(env->isolate(), key), BigInt::New(env->isolate(), value))); }); } void HistogramImpl::DoReset(const FunctionCallbackInfo& args) { HistogramImpl* histogram = HistogramImpl::FromJSObject(args.This()); (*histogram)->Reset(); } void HistogramImpl::FastReset(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.reset"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); (*histogram)->Reset(); } double HistogramImpl::FastGetCount(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.count"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return static_cast((*histogram)->Count()); } double HistogramImpl::FastGetMin(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.min"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return static_cast((*histogram)->Min()); } double HistogramImpl::FastGetMax(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.max"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return static_cast((*histogram)->Max()); } double HistogramImpl::FastGetMean(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.mean"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return (*histogram)->Mean(); } double HistogramImpl::FastGetExceeds(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.exceeds"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return static_cast((*histogram)->Exceeds()); } double HistogramImpl::FastGetStddev(Local receiver) { TRACK_V8_FAST_API_CALL("histogram.stddev"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return (*histogram)->Stddev(); } double HistogramImpl::FastGetPercentile(Local receiver, const double percentile) { TRACK_V8_FAST_API_CALL("histogram.percentile"); HistogramImpl* histogram = HistogramImpl::FromJSObject(receiver); return static_cast((*histogram)->Percentile(percentile)); } HistogramImpl* HistogramImpl::FromJSObject(Local value) { auto obj = value.As(); DCHECK_GE(obj->InternalFieldCount(), HistogramImpl::kInternalFieldCount); return static_cast(obj->GetAlignedPointerFromInternalField( HistogramImpl::kImplField, EmbedderDataTag::kDefault)); } std::unique_ptr<:transferdata> IntervalHistogram::CloneForMessaging() const { return std::make_unique<:histogramtransferdata>(histogram()); } } // namespace node