octocore-mongo 0.0.6

data/lib/octocore-mongo/models/enterprise/funnels.rb

@@ -0,0 +1,129 @@
+require 'mongo_mapper'
+require 'octocore-mongo/record'
+require 'set'
+
+module Octo
+
+  # Stores the funnel for the enterprise
+  class Funnel
+
+    Octo.featureflag self, true
+
+    include MongoMapper::Document
+    include Octo::Record
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :name_slug, String
+    key :funnel, Array
+
+    key :name, String
+    key :active, Boolean
+
+    timestamps!
+
+    before_create :create_name_slug, :activate_funnel
+
+    after_create :populate_with_fake_data
+
+    # Creates name slug
+    def create_name_slug
+      self.name_slug = self.name.to_slug
+    end
+
+    # Activates a funnel
+    def activate_funnel
+      self.active = true
+    end
+
+    # Generates a new funnel from the pages provided
+    # @param [Array] pages The pages array. This array could contain instances
+    #   of either String, Octo::Product or Octo::Page.
+    #   If string, it will be assumed that these are routeurls for pages or
+    #   products.
+    #   If the class is explicitly specified, it will be used.
+    # @param [Hash] opts The options for creating funnel
+    # @option opts [String] :name The name of the funnel
+    # @option opts [String] :enterprise_id The enterprise ID for whom funnel is
+    #   being created
+    # @return [Octo::Funnel] The funnel created
+    def self.from_pages(pages, opts = {})
+      funnel_length = pages.count
+      return nil if funnel_length.zero?
+
+      funnel = Array.new
+      enterprise_id = opts.fetch(:enterprise_id, nil)
+
+      # Check if they are Octo::Product or Octo::Page instantces and handle
+      if ::Set.new([Octo::Product, Octo::Page]).include?(pages[0].class)
+        funnel = pages.collect { |p| p.routeurl }
+        enterprise_id = pages[0].enterprise_id
+      elsif pages[0].class == String
+        funnel = pages
+      end
+
+      # Create a new funnel
+      self.new(
+        enterprise_id: enterprise_id,
+        name: opts.fetch(:name),
+        funnel: funnel
+      ).save!
+    end
+
+    # Populates a newly created funnel with some fake data
+    # @param [Fixnum] days The number of days for which data to be faked
+    def populate_with_fake_data(interval_days = 7)
+      if self.enterprise.fakedata?
+        today = Time.now.beginning_of_day
+        (today - interval_days.days).to(today, 24.hour).each do |ts|
+          Octo::FunnelData.new(
+            enterprise_id: self.enterprise_id,
+            funnel_slug: self.name_slug,
+            ts: ts,
+            value: fake_data(self.funnel.count)
+          ).save!
+        end
+      end
+    end
+
+    # Returns data for a funnel
+    # @return [Octo::FunnelData] The Octo funnel data
+    def data(ts = Time.now.floor)
+      args = {
+        enterprise_id: self.enterprise._id,
+        funnel_slug: self.name_slug,
+        ts: ts
+      }
+      res = Octo::FunnelData.where(args)
+      if res.count > 0
+        res.first
+      elsif self.enterprise.fakedata?
+        args.merge!({ value: fake_data(self.funnel.count) })
+        Octo::FunnelData.new(args).save!
+      end
+    end
+
+    private
+
+    # Generates fake data for funnel
+    # @param [Fixnum] n The length of funnel
+    # @return [Array] An array containing the funnel value
+    def fake_data(n)
+      fun = Array.new(n)
+      max_dropoff = 100/n
+      n.times do |i|
+        if i == 0
+          fun[i] = 100.0
+        else
+          fun[i] = fun[i-1] - rand(1..max_dropoff)
+          if fun[i] < 0
+            fun[i] = rand(0...fun[i].abs)
+          end
+        end
+      end
+      fun
+    end
+
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/gcm.rb

@@ -0,0 +1,21 @@
+require 'mongo_mapper'
+
+module Octo
+
+  # Storage for Notifications
+  class GcmNotification
+    include MongoMapper::Document
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :gcmid, String
+    key :userid, Integer
+
+    key :score, Float
+    key :ack, Boolean
+    key :sent_at, Time
+    key :recieved_at, Time
+
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/newsfeed_hit.rb

@@ -0,0 +1,52 @@
+require 'mongo_mapper'
+
+require 'octocore-mongo/counter'
+require 'octocore-mongo/schedeuleable'
+
+module Octo
+
+  class NewsfeedHit
+    include MongoMapper::Document
+    extend Octo::Counter
+
+    extend Octo::Scheduleable
+
+    COUNTERS = 20
+
+    # Specify that we do not want any granularity more than
+    # TYPE_DAY
+    max_type Octo::Counter::TYPE_DAY
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    countables
+
+    def self.fakedata(args)
+      res = self.where(args)
+
+      enterprise = Octo::Enterprise.find_by_id(args[:enterprise_id])
+      if res.count == 0 and enterprise.fakedata?
+        unless args.has_key?(:uid)
+          args[:uid] = 'newsfeed'
+        end
+        res = []
+        ts = args.fetch(:ts, 7.days.ago..Time.now)
+        if ts.class == Range
+          start_ts = ts.begin.beginning_of_day
+          end_ts = ts.end.end_of_day
+          start_ts.to(end_ts, 1.day).each do |_ts|
+            _args = args.merge({ ts: _ts, count: rand(400..700) })
+            res << self.new(_args).save!
+          end
+        elsif ts.class == Time
+          _args = args.merge({ count: rand(400..800) })
+          res << self.new(_args).save!
+        end
+      end
+      res
+    end
+
+
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/notification_hit.rb

@@ -0,0 +1,42 @@
+require 'mongo_mapper'
+
+require 'octocore-mongo/counter'
+require 'octocore-mongo/schedeuleable'
+
+module Octo
+
+  # Counters for notifications sent
+  class NotificationHit
+
+    include MongoMapper::Document
+    extend Octo::Counter
+    extend Octo::Scheduleable
+
+    max_type Octo::Counter::TYPE_DAY
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    countables
+
+    def self.time_slots
+      Array.new(6) { |i| "s_#{ i }" }
+    end
+
+    def self.fakedata(args)
+      opts = {
+        bod: false,
+        step: 1.day
+      }
+      self.time_slots.concat([:ios, :android]).inject([]) do |res, uid|
+        values = {
+            count: rand(500..900),
+          }
+        _args = args.merge(uid: uid)
+        res << self.fake_data_with(_args, values, opts)
+        res.flatten
+      end
+    end
+
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/page.rb

@@ -0,0 +1,15 @@
+require 'mongo_mapper'
+
+module Octo
+  class Page
+    include MongoMapper::Document
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :routeurl, String
+
+    key :categories, Array
+    key :tags, Array
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/page_view.rb

@@ -0,0 +1,14 @@
+require 'mongo_mapper'
+
+module Octo
+  class PageView
+    include MongoMapper::Document
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :userid,     Integer
+    key :created_at, Time, order: :desc
+
+    key :routeurl, String
+  end
+end

data/lib/octocore-mongo/models/enterprise/pageload_time.rb

@@ -0,0 +1,43 @@
+require 'mongo_mapper'
+
+module Octo
+
+  class PageloadTime
+    include MongoMapper::Document
+
+    # Types of pageloads
+    NEWSFEED = 0
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+    key :pageload_type, Integer
+    key :counter_type, Integer
+    key :ts, Time
+
+    key :time, Integer
+
+    timestamps!
+
+    def self.fakedata(args)
+      res = self.where(args)
+      enterprise = Octo::Enterprise.find_by_id(args[:enterprise_id])
+      if res.count == 0 and enterprise.fakedata?
+        res = []
+        ts = args.fetch(ts, 7.days.ago..Time.now.floor)
+        if ts.class == Range
+          start_time = ts.begin.beginning_of_day
+          end_time = ts.end.end_of_day
+          start_time.to(end_time, 1.day).each do |_ts|
+            _args = args.merge({ ts: _ts, time: (rand(5.0..7.0)*60).to_i })
+            res << self.new(_args).save!
+          end
+        elsif ts.class == Time
+          _args = args.merge({ time: (rand(5.0..7.0)).to_i })
+          res << self.new(args).save!
+        end
+      end
+      res
+    end
+
+  end
+end
+

data/lib/octocore-mongo/models/enterprise/product.rb

@@ -0,0 +1,22 @@
+require 'mongo_mapper'
+require 'octocore-mongo/record'
+
+module Octo
+  class Product
+    include MongoMapper::Document
+    include Octo::Record
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :_id, Integer
+
+    key :price, Float
+    key :name, String
+    key :routeurl, String
+
+    key :categories, Array
+    key :tags, Array
+
+    timestamps!
+  end
+end

data/lib/octocore-mongo/models/enterprise/product_baseline.rb

@@ -0,0 +1,20 @@
+require 'mongo_mapper'
+
+require 'octocore-mongo/baseline'
+require 'octocore-mongo/schedeuleable'
+
+module Octo
+  class ProductBaseline
+    include MongoMapper::Document
+    extend Octo::Baseline
+    extend Octo::Scheduleable
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    baselineable
+
+    # Specify the type of counter
+    baseline_for 'Octo::ProductHit'
+
+  end
+end

data/lib/octocore-mongo/models/enterprise/product_hit.rb

@@ -0,0 +1,26 @@
+require 'mongo_mapper'
+
+require 'octocore-mongo/counter'
+require 'octocore-mongo/trendable'
+require 'octocore-mongo/schedeuleable'
+
+module Octo
+
+  class ProductHit
+    include MongoMapper::Document
+    extend Octo::Counter
+    extend Octo::Trendable
+    extend Octo::Scheduleable
+
+    COUNTERS = 30
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    countables
+    trendables
+
+    baseline 'Octo::ProductBaseline'
+    trends_class 'Octo::ProductTrend'
+
+  end
+end

data/lib/octocore-mongo/models/enterprise/product_page_view.rb

@@ -0,0 +1,13 @@
+require 'mongo_mapper'
+
+module Octo
+  class ProductPageView
+    include MongoMapper::Document
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+    
+    key :userid, Integer
+    key :created_at, Time, order: :desc
+    
+    key :product_id, Integer
+  end
+end

data/lib/octocore-mongo/models/enterprise/product_trend.rb

@@ -0,0 +1,18 @@
+require 'mongo_mapper'
+require 'octocore-mongo/trends'
+
+module Octo
+
+  # Class for storing trending product
+  class ProductTrend
+    include MongoMapper::Document
+    extend Octo::Trends
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    trendable
+
+    trend_for 'Octo::ProductHit'
+    trend_class 'Octo::Product'
+  end
+end

data/lib/octocore-mongo/models/enterprise/push_key.rb

@@ -0,0 +1,15 @@
+require 'mongo_mapper'
+
+module Octo
+  class PushKey
+    include MongoMapper::Document
+
+    belongs_to :enterprise, class_name: 'Octo::Enterprise'
+
+    key :push_type, Integer
+    key :key, String
+
+    timestamps!
+  end
+end
+