patty 0.0.1

data/.travis.yml

@@ -0,0 +1,6 @@
+language: ruby
+rvm:
+  - 1.9.2
+  - 1.9.3
+notifications:
+  email: false

data/Gemfile

@@ -0,0 +1,5 @@
+source "http://rubygems.org"
+
+gem 'rake'
+gem 'mocha'
+gem 'patty', :git => 'git://github.com/peanut/patty.git'

data/README.md

@@ -0,0 +1,5 @@
+# Patty
+
+Framework for creating Waffle events handlers.
+
+[![Build Status](https://secure.travis-ci.org/peanut/patty.png?branch=master)](http://travis-ci.org/peanut/patty)

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task :default => :test

data/lib/patty/aggregated_storage.rb

@@ -0,0 +1,47 @@
+require 'yajl'
+
+module Patty
+  class AggregatedStorage
+
+    def initialize(storage = nil)
+      @storage = storage
+    end
+
+    def put(signature = nil, value = nil, marker = 'main')
+      validate_signature signature
+      signature.parents.each{ |p| del p }
+      @storage.put key(signature, marker), encode(value)
+    end
+
+    def get(signature = nil, marker = 'main')
+      validate_signature signature
+      decode @storage.get(key(signature, marker))
+    end
+
+    def del(signature = nil, marker = 'main')
+      validate_signature signature
+      @storage.del key(signature, marker)
+    end
+
+    private
+
+      def encode(data = nil)
+        Yajl::Encoder.encode data
+      end
+
+      def decode(data = '')
+        Yajl::Parser.parse data
+      end
+
+      def key(signature = nil, marker = 'main')
+        "#{marker}|#{signature}"
+      end
+
+      def validate_signature(signature = nil)
+        if signature.nil? || !signature.is_a?(Patty::TimeSignature)
+          raise "key validation fail for #{signature}"
+        end
+      end
+
+  end
+end

data/lib/patty/base.rb

@@ -0,0 +1,72 @@
+require 'date'
+require 'waffle'
+require 'patty/time_signature'
+
+module Patty
+  class Base
+
+    def initialize(storage = nil)
+      @storage = storage
+    end
+
+    def flow
+      'events'
+    end
+
+    def build_signature(datetime_string = '')
+      Patty::TimeSignature.new(datetime_string[0, 16]).align
+    end
+
+    def emit(signature = nil, value = nil, marker = nil)
+      if signature.present?
+        signature = Patty::TimeSignature.from_datetime.align
+      end
+
+      current_value = fetch signature, marker
+
+      unless current_value.present?
+        current_value = value
+      else
+        current_value = reduce [current_value, value]
+      end
+
+      @storage.put signature, current_value, marker
+    end
+
+    def map(flow_title = '', signature = nil, record = nil)
+      emit(signature, value, record['marker'])
+    end
+
+    def reduce(data = [])
+      data.inject(0){ |acc, item| acc += item }
+    end
+
+    def fetch(signature = nil, marker = nil)
+      current_value = @storage.get signature, marker
+
+      unless current_value.present?
+        data = signature.children.inject([]) do |acc, child|
+          child_value = fetch(child, marker)
+          acc << child_value unless child_value.nil?
+          acc
+        end
+
+        if data.size > 0
+          current_value = reduce data
+          @storage.put signature, current_value, marker
+        end
+      end
+
+      current_value
+    end
+
+    def run
+      transport = Waffle::Base.new eval("Waffle::Transports::#{Waffle::Config.transport.capitalize}").new
+
+      transport.subscribe flow do |flow_title, event|
+        map flow_title, build_signature(event['occured_at']), event
+      end
+    end
+
+  end
+end

data/lib/patty/storages/riak.rb

@@ -0,0 +1,32 @@
+require 'riak'
+
+module Patty
+  module Storages
+    class Riak
+
+      def initialize(bucket_name = 'events')
+        @client = ::Riak::Client.new
+        @bucket = @client.bucket bucket_name
+      end
+
+      def put(key = nil, value = nil)
+        record = @bucket.get_or_new key
+        record.data = value
+        record.store
+      end
+
+      def get(key = nil)
+        if @bucket.exists? key
+          @bucket.get(key).data
+        else
+          ''
+        end
+      end
+
+      def del(key = nil)
+        @bucket.delete key
+      end
+
+    end
+  end
+end

data/lib/patty/time_signature.rb

@@ -0,0 +1,117 @@
+require 'date'
+
+module Patty
+  class TimeSignature
+
+    MINUTES = ['00', '10', '20', '30', '40', '50']
+    HOURS   = (0..23).map { |h| "%02d" % h }
+    MONTHS  = (1..12).map { |m| "%02d" % m }
+
+    class << self
+      def from_datetime(datetime = lambda{ DateTime.now })
+        if datetime.is_a?(Proc)
+          datetime = datetime.call
+        end
+        self.new(datetime.strftime("%Y-%m-%d %H:%M")).align
+      end
+    end
+
+    attr_reader :signature
+
+    def initialize(signature = '')
+      if signature.empty?
+        raise ArgumentError, 'signature can not be empty'
+      end
+      @signature = signature
+    end
+
+    def ==(other)
+      signature == other.signature
+    end
+
+    def to_s
+      signature
+    end
+
+    def kind
+      case signature.size
+        when 16
+          :minute
+        when 13
+          :hour
+        when 10
+          :day
+        when 7
+          :month
+        when 4
+          :year
+        else
+          raise 'broken signature'
+      end
+    end
+
+    def align
+      self.class.new "%04d-%02d-%02d %02d:%02d" % [year, month, day, hour, aligned_minute]
+    end
+
+    def cut(from = 0, length = 2)
+      signature.size > from ? signature[from, length] : ''
+    end
+
+    def crop(to = 13)
+      signature.size > to ? signature[0, to] : signature
+    end
+
+    def year
+      cut(0, 4).to_i
+    end
+
+    def month
+      cut(5).to_i
+    end
+
+    def day
+      cut(8).to_i
+    end
+
+    def hour
+      cut(11).to_i
+    end
+
+    def minute
+      cut(14).to_i
+    end
+
+    def aligned_minute
+      if kind == :minute
+        "%-1d0" % (minute / 10)
+      else
+        minute
+      end
+    end
+
+    def days_count
+      Date.new(year, month, -1).day
+    end
+
+    def parents
+      [crop(13), crop(10), crop(7), crop(4)].delete_if{ |p| p.empty? }.map{ |p| self.class.new p }
+    end
+
+    def children
+      case kind
+        when :hour
+          MINUTES.map{ |m| self.class.new "#{signature}:#{m}" }
+        when :day
+          HOURS.map{ |h| self.class.new "#{signature} #{h}" }
+        when :month
+          (1..days_count).map{ |d| self.class.new "#{signature}-" + '%02d' % d }
+        when :year
+          MONTHS.map{ |m| self.class.new "#{signature}-#{m}" }
+        else
+          []
+      end
+    end
+
+  end
+end

data/lib/patty/version.rb

@@ -0,0 +1,5 @@
+module Patty
+
+  VERSION = '0.0.1'
+
+end

data/lib/patty.rb

@@ -0,0 +1,9 @@
+require 'patty/version'
+require 'patty/base'
+require 'patty/aggregated_storage'
+
+module Patty
+  module Storages
+    autoload :Riak, 'patty/storages/riak'
+  end
+end

data/patty.gemspec

@@ -0,0 +1,20 @@
+$:.push File.expand_path("../lib", __FILE__)
+
+require "patty/version"
+
+Gem::Specification.new do |s|
+  s.name    = 'patty'
+  s.version = Patty::VERSION
+
+  s.homepage = 'http://github.com/peanut/patty'
+  s.authors  = ['Alexander Lomakin']
+  s.email    = 'alexander.lomakin@gmail.com'
+
+  s.summary     = 'Framework for creating Waffle events handlers'
+  s.description = 'Server part of Patty statistics server'
+
+  s.files = `git ls-files`.split("\n")
+
+  s.add_runtime_dependency 'yajl-ruby'
+  s.add_runtime_dependency 'riak-client'
+end

data/test/test_time_signature.rb

@@ -0,0 +1,67 @@
+require 'test/unit'
+require 'patty/time_signature'
+
+class TimeSignatureTest < Test::Unit::TestCase
+
+  def setup
+    @time_signature = Patty::TimeSignature.new('2005-10-10 18:55')
+  end
+
+  def test_to_s
+    assert_equal '2005-10-10 18:55', @time_signature.to_s
+  end
+
+  def test_kind
+    assert_equal :minute, @time_signature.kind
+  end
+
+  def test_cut
+    assert_equal '10', @time_signature.cut(5, 2)
+    assert_equal '', @time_signature.cut(1000, 2)
+  end
+
+  def test_crop
+    assert_equal '2005-10-10', @time_signature.crop(10)
+    assert_equal '2005-10-10 18:55', @time_signature.crop(100)
+  end
+
+  def test_parents
+    expected_parents = [
+      Patty::TimeSignature.new('2005-10-10 18'),
+      Patty::TimeSignature.new('2005-10-10'),
+      Patty::TimeSignature.new('2005-10'),
+      Patty::TimeSignature.new('2005')
+    ]
+    assert_equal expected_parents, @time_signature.parents
+  end
+
+  def test_children_for_minute
+    assert_equal 0, @time_signature.children.size
+  end
+
+  def test_children_for_hour
+    @time_signature = Patty::TimeSignature.new('2005-10-10 18')
+    assert_equal 6, @time_signature.children.size
+  end
+
+  def test_children_for_day
+    @time_signature = Patty::TimeSignature.new('2005-10-10')
+    assert_equal 24, @time_signature.children.size
+  end
+
+  def test_children_for_month
+    @time_signature = Patty::TimeSignature.new('2012-02')
+    assert_equal 29, @time_signature.children.size
+  end
+
+  def test_children_for_year
+    @time_signature = Patty::TimeSignature.new('2005')
+    assert_equal 12, @time_signature.children.size
+  end
+
+  def test_from_datetime
+    @time_signature = Patty::TimeSignature.from_datetime DateTime.new(2012, 02, 29, 12, 35)
+    assert_equal '2012-02-29 12:30', @time_signature.signature
+  end
+
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: patty
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Alexander Lomakin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-02 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: &70183040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70183040
+- !ruby/object:Gem::Dependency
+  name: riak-client
+  requirement: &70182340 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70182340
+description: Server part of Patty statistics server
+email: alexander.lomakin@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .travis.yml
+- Gemfile
+- README.md
+- Rakefile
+- lib/patty.rb
+- lib/patty/aggregated_storage.rb
+- lib/patty/base.rb
+- lib/patty/storages/riak.rb
+- lib/patty/time_signature.rb
+- lib/patty/version.rb
+- patty.gemspec
+- test/test_time_signature.rb
+homepage: http://github.com/peanut/patty
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Framework for creating Waffle events handlers
+test_files: []