trifle-stats 1.0.0 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 43ed8b8a05c933e5397b59bce9dc18eaee146ac62d2df67eccfbd9a53caad619
-  data.tar.gz: 59f6d4ceba6de8977fcedf6f153fd621bd78a7055102080475c0d1f0e98f952f
+  metadata.gz: d2ab7e87008ccd472bcbb2f821db69a304c0d2be078827150d828459c62b9ffa
+  data.tar.gz: 7e6b5b77c70845d2c2d7b5b29f7a9811bba32f166f3a6204fb13e931c29ab90e
 SHA512:
-  metadata.gz: 3a6fc973aaedd3167ea5c1b4d2f2cb7ebd8376de153f212005aae43320e59acb992d1034f8a83c5d6063e9b314f916603cd98714ecb9d0a66b02e4dd8dc8fb33
-  data.tar.gz: adfbb9e6625f1f05b20c7f420c6a029b94086882ebd7e02fb3223449fbbc22c6b42c2ad49218b76ad8406570b1c9e2263b677ae93150fe1a6200e00dadddc8e1
+  metadata.gz: 49870c3899c775ce1a91a857cc39f9cc92c05b6bb7ffe211ecf52adf033436942f5db44942cd4e2f84143b8111e586b25254f88f4d9c4cf8c9123adac20503e8
+  data.tar.gz: 3dee093eed89e39a2dbc0c867dba0529359a9cba84f11c021fad8a78246e2e0d1115119ee896658e8ed917546757423432078faece8c582f559923b973170938

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    trifle-stats (1.0.0)
+    trifle-stats (1.1.1)
       tzinfo (~> 2.0)
 
 GEM

data/lib/trifle/stats/configuration.rb

@@ -6,12 +6,13 @@ module Trifle
   module Stats
     class Configuration
       attr_writer :driver
-      attr_accessor :track_ranges, :time_zone, :beginning_of_week
+      attr_accessor :track_ranges, :time_zone, :beginning_of_week, :designator
 
       def initialize
         @ranges = %i[minute hour day week month quarter year]
         @beginning_of_week = :monday
         @time_zone = 'GMT'
+        @designator = nil
       end
 
       def tz

data/lib/trifle/stats/designator/custom.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Trifle
+  module Stats
+    class Designator
+      class Custom
+        attr_reader :buckets
+
+        def initialize(buckets:)
+          @buckets = buckets.sort
+        end
+
+        def designate(value:)
+          return buckets.first.to_s if value <= buckets.first
+          return "#{buckets.last}+" if value > buckets.last
+
+          (buckets.find { |b| value.ceil < b }).to_s
+        end
+      end
+    end
+  end
+end

data/lib/trifle/stats/designator/geometric.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Trifle
+  module Stats
+    class Designator
+      class Geometric
+        attr_reader :min, :max
+
+        def initialize(min:, max:)
+          @min = min.negative? ? 0 : min
+          @max = max
+        end
+
+        def designate(value:) # rubocop:disable Metrics/AbcSize
+          return min.to_f.to_s if value <= min
+          return "#{max.to_f}+" if value > max
+          return (10**value.floor.to_s.length).to_f.to_s if value > 1
+          return 1.0.to_s if value > 0.1 # ugh?
+
+          (1.0 / 10**value.to_s.gsub('0.', '').split(/[1-9]/).first.length).to_s
+        end
+      end
+    end
+  end
+end

data/lib/trifle/stats/designator/linear.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Trifle
+  module Stats
+    class Designator
+      class Linear
+        attr_reader :min, :max, :step
+
+        def initialize(min:, max:, step:)
+          @min = min
+          @max = max
+          @step = step.to_i
+        end
+
+        def designate(value:) # rubocop:disable Metrics/AbcSize
+          return min.to_s if value <= min
+          return "#{max}+" if value > max
+
+          (value.ceil / step * step + ((value.ceil % step).zero? ? 0 : step)).to_s
+        end
+      end
+    end
+  end
+end

data/lib/trifle/stats/driver/README.md

@@ -2,8 +2,8 @@
 
 Driver is a wrapper class that persists and retrieves values from backend. It needs to implement:
 
-- `inc(key:, **values)` method increment values
-- `set(key:, **values)` method set values
+- `inc(keys:, **values)` method increment values
+- `set(keys:, **values)` method set values
 - `get(keys:)` method to retrieve values
 
 ## Documentation

data/lib/trifle/stats/driver/mongo.rb

@@ -16,24 +16,27 @@ module Trifle
           @separator = '::'
         end
 
-        def inc(key:, **values)
-          pkey = key.join(separator)
+        def inc(keys:, **values)
+          data = self.class.pack(hash: { data: values })
 
           collection.bulk_write(
-            [upsert_operation('$inc', pkey: pkey, values: values)]
+            keys.map do |key|
+              upsert_operation('$inc', pkey: key.join(separator), data: data)
+            end
           )
         end
 
-        def set(key:, **values)
-          pkey = key.join(separator)
+        def set(keys:, **values)
+          data = self.class.pack(hash: { data: values })
 
           collection.bulk_write(
-            [upsert_operation('$set', pkey: pkey, values: values)]
+            keys.map do |key|
+              upsert_operation('$set', pkey: key.join(separator), data: data)
+            end
           )
         end
 
-        def upsert_operation(operation, pkey:, values:)
-          data = self.class.pack(hash: { data: values })
+        def upsert_operation(operation, pkey:, data:)
           {
             update_many: {
               filter: { key: pkey },

data/lib/trifle/stats/driver/postgres.rb

@@ -16,11 +16,13 @@ module Trifle
           @separator = '::'
         end
 
-        def inc(key:, **values)
-          pkey = key.join(separator)
+        def inc(keys:, **values)
+          keys.map do |key|
+            pkey = key.join(separator)
 
-          self.class.pack(hash: values).each do |k, c|
-            _inc_one(key: pkey, name: k, value: c)
+            self.class.pack(hash: values).each do |k, c|
+              _inc_one(key: pkey, name: k, value: c)
+            end
           end
         end
 
@@ -31,10 +33,12 @@ module Trifle
           client.exec(query)
         end
 
-        def set(key:, **values)
-          pkey = key.join(separator)
+        def set(keys:, **values)
+          keys.map do |key|
+            pkey = key.join(separator)
 
-          _set_all(key: pkey, **values)
+            _set_all(key: pkey, **values)
+          end
         end
 
         def _set_all(key:, **values)

data/lib/trifle/stats/driver/process.rb

@@ -12,19 +12,23 @@ module Trifle
           @separator = '::'
         end
 
-        def inc(key:, **values)
-          self.class.pack(hash: values).each do |k, c|
-            d = @data.fetch(key.join(@separator), {})
-            d[k] = d[k].to_i + c
-            @data[key.join(@separator)] = d
+        def inc(keys:, **values)
+          keys.map do |key|
+            self.class.pack(hash: values).each do |k, c|
+              d = @data.fetch(key.join(@separator), {})
+              d[k] = d[k].to_i + c
+              @data[key.join(@separator)] = d
+            end
           end
         end
 
-        def set(key:, **values)
-          self.class.pack(hash: values).each do |k, c|
-            d = @data.fetch(key.join(@separator), {})
-            d[k] = c
-            @data[key.join(@separator)] = d
+        def set(keys:, **values)
+          keys.map do |key|
+            self.class.pack(hash: values).each do |k, c|
+              d = @data.fetch(key.join(@separator), {})
+              d[k] = c
+              @data[key.join(@separator)] = d
+            end
           end
         end
 

data/lib/trifle/stats/driver/redis.rb

@@ -16,18 +16,22 @@ module Trifle
           @separator = '::'
         end
 
-        def inc(key:, **values)
-          pkey = ([prefix] + key).join(separator)
+        def inc(keys:, **values)
+          keys.map do |key|
+            pkey = ([prefix] + key).join(separator)
 
-          self.class.pack(hash: values).each do |k, c|
-            client.hincrby(pkey, k, c)
+            self.class.pack(hash: values).each do |k, c|
+              client.hincrby(pkey, k, c)
+            end
           end
         end
 
-        def set(key:, **values)
-          pkey = ([prefix] + key).join(separator)
+        def set(keys:, **values)
+          keys.map do |key|
+            pkey = ([prefix] + key).join(separator)
 
-          client.hmset(pkey, *self.class.pack(hash: values))
+            client.hmset(pkey, *self.class.pack(hash: values))
+          end
         end
 
         def get(keys:)

data/lib/trifle/stats/operations/timeseries/classify.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Trifle
+  module Stats
+    module Operations
+      module Timeseries
+        class Classify
+          attr_reader :key, :values
+
+          def initialize(**keywords)
+            @key = keywords.fetch(:key)
+            @at = keywords.fetch(:at)
+            @values = keywords.fetch(:values)
+            @config = keywords[:config]
+          end
+
+          def config
+            @config || Trifle::Stats.default
+          end
+
+          def deep_classify(hash)
+            hash.transform_values do |value|
+              next deep_classify(value) if value.is_a?(Hash)
+
+              { classify(value) => 1 }
+            end
+          end
+
+          def classify(value)
+            config.designator.designate(value: value).to_s.gsub('.', '_')
+          end
+
+          def key_for(range:)
+            at = Nocturnal.new(@at, config: config).send(range)
+            [key, range, at.to_i]
+          end
+
+          def perform
+            config.driver.inc(
+              keys: config.ranges.map { |range| key_for(range: range) },
+              **deep_classify(values)
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/trifle/stats/operations/timeseries/increment.rb

@@ -18,14 +18,16 @@ module Trifle
             @config || Trifle::Stats.default
           end
 
+          def key_for(range:)
+            at = Nocturnal.new(@at, config: config).send(range)
+            [key, range, at.to_i]
+          end
+
           def perform
-            config.ranges.map do |range|
-              at = Nocturnal.new(@at, config: config).send(range)
-              config.driver.inc(
-                key: [key, range, at.to_i],
-                **values
-              )
-            end
+            config.driver.inc(
+              keys: config.ranges.map { |range| key_for(range: range) },
+              **values
+            )
           end
         end
       end

data/lib/trifle/stats/operations/timeseries/set.rb

@@ -18,14 +18,16 @@ module Trifle
             @config || Trifle::Stats.default
           end
 
+          def key_for(range:)
+            at = Nocturnal.new(@at, config: config).send(range)
+            [key, range, at.to_i]
+          end
+
           def perform
-            config.ranges.map do |range|
-              at = Nocturnal.new(@at, config: config).send(range)
-              config.driver.set(
-                key: [key, range, at.to_i],
-                **values
-              )
-            end
+            config.driver.set(
+              keys: config.ranges.map { |range| key_for(range: range) },
+              **values
+            )
           end
         end
       end

data/lib/trifle/stats/version.rb

@@ -2,6 +2,6 @@
 
 module Trifle
   module Stats
-    VERSION = '1.0.0'
+    VERSION = '1.1.1'
   end
 end

data/lib/trifle/stats.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require 'trifle/stats/designator/custom'
+require 'trifle/stats/designator/geometric'
+require 'trifle/stats/designator/linear'
 require 'trifle/stats/driver/mongo'
 require 'trifle/stats/driver/postgres'
 require 'trifle/stats/driver/process'
@@ -7,6 +10,7 @@ require 'trifle/stats/driver/redis'
 require 'trifle/stats/mixins/packer'
 require 'trifle/stats/nocturnal'
 require 'trifle/stats/configuration'
+require 'trifle/stats/operations/timeseries/classify'
 require 'trifle/stats/operations/timeseries/increment'
 require 'trifle/stats/operations/timeseries/set'
 require 'trifle/stats/operations/timeseries/values'
@@ -45,6 +49,15 @@ module Trifle
       ).perform
     end
 
+    def self.assort(key:, at:, values:, config: nil)
+      Trifle::Stats::Operations::Timeseries::Classify.new(
+        key: key,
+        at: at,
+        values: values,
+        config: config
+      ).perform
+    end
+
     def self.values(key:, from:, to:, range:, config: nil)
       Trifle::Stats::Operations::Timeseries::Values.new(
         key: key,

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trifle-stats
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.1
 platform: ruby
 authors:
 - Jozef Vaclavik
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-06-18 00:00:00.000000000 Z
+date: 2022-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -183,6 +183,9 @@ files:
 - bin/setup
 - lib/trifle/stats.rb
 - lib/trifle/stats/configuration.rb
+- lib/trifle/stats/designator/custom.rb
+- lib/trifle/stats/designator/geometric.rb
+- lib/trifle/stats/designator/linear.rb
 - lib/trifle/stats/driver/README.md
 - lib/trifle/stats/driver/mongo.rb
 - lib/trifle/stats/driver/postgres.rb
@@ -190,6 +193,7 @@ files:
 - lib/trifle/stats/driver/redis.rb
 - lib/trifle/stats/mixins/packer.rb
 - lib/trifle/stats/nocturnal.rb
+- lib/trifle/stats/operations/timeseries/classify.rb
 - lib/trifle/stats/operations/timeseries/increment.rb
 - lib/trifle/stats/operations/timeseries/set.rb
 - lib/trifle/stats/operations/timeseries/values.rb