sq_detailed_metrics 0.1.0

data/sq_detailed_metrics.cpp

@@ -0,0 +1,248 @@
+extern "C" {
+    #include <ruby.h>
+    #include <ruby/version.h>
+    #include <ruby/encoding.h>
+    #include "extconf.h"
+}
+
+#include <sstream>
+#include "metrics.hpp"
+#include "json_conv.hpp"
+
+using namespace sq_detailed_metrics;
+
+namespace sq_detailed_metrics {
+    MetricsIndex *metrics_index;
+}
+
+static void request_free(void *ptr);
+static size_t request_dsize(const void *ptr);
+
+#if RUBY_API_VERSION_CODE < 20200
+static inline VALUE rb_utf8_str_new(const char *ptr, long len)
+{
+    return rb_enc_str_new(ptr, len, rb_utf8_encoding());
+}
+#endif
+
+static const rb_data_type_t request_type = {
+    "SqDetailedMetrics::Request",
+    {
+        nullptr, // dmark
+        request_free,
+        request_dsize
+    },
+    nullptr,
+    nullptr
+#ifdef RUBY_TYPED_FREE_IMMEDIATELY
+    , RUBY_TYPED_FREE_IMMEDIATELY
+#endif
+};
+
+// RequestCollection
+
+static void request_collection_free(void *ptr);
+static size_t request_collection_dsize(const void *ptr);
+
+static const rb_data_type_t request_collection_type = {
+    "SqDetailedMetrics::RequestCollection",
+    {
+        nullptr, // dmark
+        request_collection_free,
+        request_collection_dsize
+    },
+    nullptr, // parent
+    nullptr // arbitrary data
+#ifdef RUBY_TYPED_FREE_IMMEDIATELY
+    , RUBY_TYPED_FREE_IMMEDIATELY
+#endif
+};
+
+static void request_collection_free(void *ptr) {
+    RequestCollection *coll = static_cast<RequestCollection*>(ptr);
+    delete coll;
+}
+
+static size_t request_collection_dsize(const void *ptr) {
+    RequestCollection const* coll = static_cast<RequestCollection const*>(ptr);
+    return coll->mem_size();
+}
+
+static VALUE request_collection_alloc(VALUE klass) {
+    RequestCollection *coll = new RequestCollection();
+    VALUE obj = TypedData_Wrap_Struct(klass, &request_collection_type, coll);
+
+    return obj;
+}
+
+static VALUE request_collection_serialize(VALUE self) {
+    RequestCollection *coll;
+    TypedData_Get_Struct(self, RequestCollection,
+                         &request_collection_type, coll);
+    std::string str;
+    bool res = coll->flush(str);
+    if (!res) {
+        return Qnil;
+    }
+    return rb_enc_str_new(&str[0], static_cast<long>(str.length()),
+            rb_ascii8bit_encoding());
+}
+
+static VALUE request_collection_add(VALUE self, VALUE req_arg) {
+    RequestCollection *coll;
+    TypedData_Get_Struct(self, RequestCollection,
+                         &request_collection_type, coll);
+
+    Request *req;
+    TypedData_Get_Struct(req_arg, Request,
+                         &request_type, req);
+
+    coll->operator<<(std::move(*req));
+
+    return Qnil;
+}
+
+// Request
+
+static void request_free(void *ptr) {
+    Request *req = static_cast<Request*>(ptr);
+    delete req;
+}
+
+static size_t request_dsize(const void *ptr) {
+    Request const *req = static_cast<Request const*>(ptr);
+    return req->mem_size();
+}
+
+static VALUE request_alloc(VALUE klass) {
+    Request *req = new Request();
+    VALUE obj = TypedData_Wrap_Struct(klass, &request_type, req);
+
+    return obj;
+}
+
+static VALUE request_set_route(VALUE self, VALUE route_arg) {
+    Request *req;
+    TypedData_Get_Struct(self, Request,
+                         &request_type, req);
+
+    if (route_arg == Qnil) {
+        req->set_route({});
+        return Qnil;
+    }
+
+    Check_Type(route_arg, T_STRING);
+
+    long route_arg_len = RSTRING_LEN(route_arg);
+    std::string route{
+        RSTRING_PTR(route_arg),
+        static_cast<size_t>(route_arg_len)};
+
+    req->set_route(std::move(route));
+
+    return Qnil;
+}
+
+static VALUE request_set_overtime_cb(VALUE self, VALUE cb_arg) {
+    Request *req;
+    TypedData_Get_Struct(self, Request,
+                         &request_type, req);
+
+    if (cb_arg == Qnil) {
+        req->set_overtime_cb({});
+        return Qnil;
+    }
+
+    Check_Type(cb_arg, T_STRING);
+
+    long cb_arg_len = RSTRING_LEN(cb_arg);
+    std::string cb{
+        RSTRING_PTR(cb_arg),
+        static_cast<size_t>(cb_arg_len)};
+
+    req->set_overtime_cb(cb);
+
+    return Qnil;
+}
+
+static VALUE request_add_measurement(VALUE self,
+                                     VALUE category_arg,
+                                     VALUE flowt_arg) {
+    Check_Type(category_arg, T_STRING);
+    Check_Type(flowt_arg, T_FLOAT);
+
+    Request *req;
+    TypedData_Get_Struct(self, Request,
+                         &request_type, req);
+
+    long categ_arg_len = RSTRING_LEN(category_arg);
+    std::string category{
+        RSTRING_PTR(category_arg),
+        static_cast<size_t>(categ_arg_len)};
+
+    req->add(category, NUM2DBL(flowt_arg));
+
+    return Qnil;
+}
+
+static VALUE to_json(VALUE self, VALUE data) {
+    (void)self;
+
+    Check_Type(data, T_STRING);
+
+    VALUE err_msg = Qnil;
+
+    {
+        std::ostringstream stream;
+        JsonConv conv{stream};
+
+        long data_len = RSTRING_LEN(data);
+        try {
+            conv.parse(RSTRING_PTR(data), static_cast<size_t>(data_len));
+            std::string s = stream.str();
+            return rb_utf8_str_new(&s[0], static_cast<long>(s.size()));
+        } catch (const std::exception& e) {
+            err_msg = rb_str_buf_new_cstr(e.what());
+        }
+    } // all C++ objects must be destroyed before rb_raise is called
+
+    if (err_msg != Qnil) {
+        rb_raise(rb_eRuntimeError, "%s", RSTRING_PTR(err_msg));
+    }
+    return Qnil; // never reached
+}
+
+extern "C" void Init_sq_detailed_metrics(void);
+extern "C" void Init_sq_detailed_metrics()
+{
+    metrics_index = MetricsIndex_create();
+
+    VALUE mSqlDetailedMetrics = rb_define_module("SqDetailedMetrics");
+    VALUE cRequestCollection = rb_define_class_under(
+                mSqlDetailedMetrics, "RequestCollection", rb_cData);
+    rb_define_alloc_func(cRequestCollection, request_collection_alloc);
+    rb_define_method(cRequestCollection, "serialize",
+                     reinterpret_cast<VALUE (*)(...)>(request_collection_serialize),
+                     0);
+    rb_define_method(cRequestCollection, "<<",
+                     reinterpret_cast<VALUE (*)(...)>(request_collection_add),
+                     1);
+
+    VALUE cRequest = rb_define_class_under(
+                mSqlDetailedMetrics, "Request", rb_cData);
+    rb_define_alloc_func(cRequest, request_alloc);
+
+    rb_define_method(cRequest, "route=",
+                     reinterpret_cast<VALUE (*)(...)>(request_set_route),
+                     1);
+    rb_define_method(cRequest, "overtime_cb=",
+                     reinterpret_cast<VALUE (*)(...)>(request_set_overtime_cb),
+                     1);
+    rb_define_method(cRequest, "add_measurement",
+                     reinterpret_cast<VALUE (*)(...)>(request_add_measurement),
+                     2);
+
+    rb_define_singleton_method(mSqlDetailedMetrics, "to_json",
+                               reinterpret_cast<VALUE (*)(...)>(to_json),
+                               1);
+}

metadata

@@ -0,0 +1,199 @@
+--- !ruby/object:Gem::Specification
+name: sq_detailed_metrics
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Gustavo Lopes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-09-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: minitest-junit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hashdiff
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: Detailed per-request metrics for Sqreen
+email:
+- gustavo@sqreen.io
+executables: []
+extensions:
+- extconf.rb
+extra_rdoc_files: []
+files:
+- extconf.rb
+- include/half.hpp
+- include/msgpack.h
+- include/msgpack/fbuffer.h
+- include/msgpack/gcc_atomic.h
+- include/msgpack/object.h
+- include/msgpack/pack.h
+- include/msgpack/pack_define.h
+- include/msgpack/pack_template.h
+- include/msgpack/sbuffer.h
+- include/msgpack/sysdep.h
+- include/msgpack/timestamp.h
+- include/msgpack/unpack.h
+- include/msgpack/unpack_define.h
+- include/msgpack/unpack_template.h
+- include/msgpack/util.h
+- include/msgpack/version.h
+- include/msgpack/version_master.h
+- include/msgpack/vrefbuffer.h
+- include/msgpack/zbuffer.h
+- include/msgpack/zone.h
+- include/rapidjson/allocators.h
+- include/rapidjson/document.h
+- include/rapidjson/encodedstream.h
+- include/rapidjson/encodings.h
+- include/rapidjson/error/en.h
+- include/rapidjson/error/error.h
+- include/rapidjson/filereadstream.h
+- include/rapidjson/filewritestream.h
+- include/rapidjson/fwd.h
+- include/rapidjson/internal/biginteger.h
+- include/rapidjson/internal/diyfp.h
+- include/rapidjson/internal/dtoa.h
+- include/rapidjson/internal/ieee754.h
+- include/rapidjson/internal/itoa.h
+- include/rapidjson/internal/meta.h
+- include/rapidjson/internal/pow10.h
+- include/rapidjson/internal/regex.h
+- include/rapidjson/internal/stack.h
+- include/rapidjson/internal/strfunc.h
+- include/rapidjson/internal/strtod.h
+- include/rapidjson/internal/swap.h
+- include/rapidjson/istreamwrapper.h
+- include/rapidjson/memorybuffer.h
+- include/rapidjson/memorystream.h
+- include/rapidjson/msinttypes/inttypes.h
+- include/rapidjson/msinttypes/stdint.h
+- include/rapidjson/ostreamwrapper.h
+- include/rapidjson/pointer.h
+- include/rapidjson/prettywriter.h
+- include/rapidjson/rapidjson.h
+- include/rapidjson/reader.h
+- include/rapidjson/schema.h
+- include/rapidjson/stream.h
+- include/rapidjson/stringbuffer.h
+- include/rapidjson/writer.h
+- include/xxhash.h
+- json_conv.cpp
+- json_conv.hpp
+- metrics.cpp
+- metrics.hpp
+- msgpack/objectc.c
+- msgpack/unpack.c
+- msgpack/version.c
+- msgpack/vrefbuffer.c
+- msgpack/zone.c
+- sq_detailed_metrics.cpp
+homepage: 
+licenses:
+- Proprietary
+metadata:
+  homepage_uri: https://sqreen.com
+  documentation_uri: https://docs.sqreen.com/
+  changelog_uri: https://docs.sqreen.com/ruby/release-notes/
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.3
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.2
+signing_key: 
+specification_version: 4
+summary: Detailed per-request metrics for Sqreen
+test_files: []