influxdb-arel 0.0.1

data/lib/influxdb-arel.rb

@@ -0,0 +1 @@
+require 'influxdb'

data/lib/influxdb.rb

@@ -0,0 +1,4 @@
+require 'influxdb/arel'
+
+module Influxdb
+end

data/lib/influxdb/arel.rb

@@ -0,0 +1,38 @@
+require "influxdb/arel/version"
+
+require 'influxdb/arel/expressions'
+require 'influxdb/arel/predications'
+require 'influxdb/arel/math'
+require 'influxdb/arel/alias_predication'
+require 'influxdb/arel/table'
+require 'influxdb/arel/attributes'
+
+require 'influxdb/arel/visitor'
+
+require 'influxdb/arel/tree_manager'
+require 'influxdb/arel/select_manager'
+# require 'influxdb/arel/delete_manager'
+require 'influxdb/arel/nodes'
+
+module Influxdb
+  module Arel
+    extend self
+
+    def sql(raw_sql)
+      Nodes::SqlLiteral.new(raw_sql)
+    end
+
+    def star
+      sql('*')
+    end
+
+    def now
+      Influxdb::Arel::Nodes::Now.new
+    end
+
+    def time(duration)
+      duration = sql(duration) if String === duration
+      Influxdb::Arel::Nodes::Time.new(duration)
+    end
+  end
+end

data/lib/influxdb/arel/alias_predication.rb

@@ -0,0 +1,9 @@
+module Influxdb
+  module Arel
+    module AliasPredication
+      def as(other)
+        Nodes::As.new(self, Arel.sql(other.to_s))
+      end
+    end
+  end
+end

data/lib/influxdb/arel/attributes.rb

@@ -0,0 +1 @@
+require 'influxdb/arel/attributes/attribute'

data/lib/influxdb/arel/attributes/attribute.rb

@@ -0,0 +1,74 @@
+module Influxdb
+  module Arel
+    module Attributes
+      class Attribute < Struct.new(:relation, :name)
+        include Expressions
+        include Predications
+        include AliasPredication
+        include Math
+
+        def self.encode(value)
+          "'#{value.to_s}'"
+        end
+      end
+
+      class Time < Attribute
+        def self.encode(value)
+          value.to_i * 1_000_000
+        end
+      end
+
+      class Date < Time; end
+      class DateTime < Time; end
+
+      class BigDecimal < Attribute
+        def self.encode(value)
+          value.to_s('F')
+        end
+      end
+
+      class Boolean < Attribute
+        def self.encode(value)
+          value.inspect
+        end
+      end
+
+      class FalseClass < Boolean; end
+      class TrueClass < Boolean; end
+
+      class Float < Attribute
+        def self.encode(value)
+          value
+        end
+      end
+
+      class Integer < Attribute
+        def self.encode(value)
+          value
+        end
+      end
+
+      class Regexp < Attribute
+        def self.encode(value)
+          value.inspect
+        end
+      end
+
+      class Hash < Attribute
+        def self.encode(value)
+          value = value.to_json if value.respond_to?(:to_json)
+          super(value)
+        end
+      end
+
+      class NilClass < Attribute
+        def self.encode(value)
+          'null'
+        end
+      end
+    end
+
+    Attribute = Attributes::Attribute
+  end
+end
+

data/lib/influxdb/arel/core_extensions.rb

@@ -0,0 +1,49 @@
+class Integer
+  def u
+    Influxdb::Arel::Nodes::Duration.new(self, 'u')
+  end
+
+  def s
+    Influxdb::Arel::Nodes::Duration.new(self, 's')
+  end
+
+  def m
+    Influxdb::Arel::Nodes::Duration.new(self, 'm')
+  end
+
+  def h
+    Influxdb::Arel::Nodes::Duration.new(self, 'h')
+  end
+
+  def d
+    Influxdb::Arel::Nodes::Duration.new(self, 'd')
+  end
+
+  def w
+    Influxdb::Arel::Nodes::Duration.new(self, 'w')
+  end
+end
+
+class String
+  def to_influxdb_arel
+    Influxdb::Arel::Nodes::SqlLiteral.new(self)
+  end
+
+  alias :to_arel :to_influxdb_arel unless method_defined?(:to_arel)
+
+  def as(other)
+    Influxdb::Arel::Nodes::As.new(to_influxdb_arel, other.to_influxdb_arel)
+  end
+end
+
+class Symbol
+  def to_influxdb_arel
+    Influxdb::Arel::Table.new(self.to_s)
+  end
+
+  alias :to_arel :to_influxdb_arel unless method_defined?(:to_arel)
+
+  def as(other)
+    to_influxdb_arel.as(other)
+  end
+end

data/lib/influxdb/arel/expressions.rb

@@ -0,0 +1,73 @@
+module Influxdb
+  module Arel
+    module Expressions
+      def count
+        Nodes::Count.new([self])
+      end
+
+      def sum
+        Nodes::Sum.new([self])
+      end
+
+      def max
+        Nodes::Max.new([self])
+      end
+
+      def min
+        Nodes::Min.new([self])
+      end
+
+      def mean
+        Nodes::Mean.new([self])
+      end
+
+      def mode
+        Nodes::Mode.new([self])
+      end
+
+      def median
+        Nodes::Median.new([self])
+      end
+
+      def distinct
+        Nodes::Distinct.new([self])
+      end
+
+      def percentile(nth)
+        Nodes::Percentile.new([self, nth])
+      end
+
+      def histogram(bucket_size = nil)
+        Nodes::Histogram.new([self, bucket_size || 1.0])
+      end
+
+      def derivative
+        Nodes::Derivative.new([self])
+      end
+
+      def stddev
+        Nodes::Stddev.new([self])
+      end
+
+      def first
+        Nodes::First.new([self])
+      end
+
+      def last
+        Nodes::Last.new([self])
+      end
+
+      def difference
+        Nodes::Difference.new([self])
+      end
+
+      def top(size)
+        Nodes::Top.new([self, size])
+      end
+
+      def bottom(size)
+        Nodes::Bottom.new([self, size])
+      end
+    end
+  end
+end

data/lib/influxdb/arel/math.rb

@@ -0,0 +1,21 @@
+module Influxdb
+  module Arel
+    module Math
+      def *(other)
+        Nodes::Multiplication.new(self, other)
+      end
+
+      def +(other)
+        Nodes::Grouping.new(Nodes::Addition.new(self, other))
+      end
+
+      def -(other)
+        Nodes::Grouping.new(Nodes::Subtraction.new(self, other))
+      end
+
+      def /(other)
+        Nodes::Division.new(self, other)
+      end
+    end
+  end
+end

data/lib/influxdb/arel/nodes.rb

@@ -0,0 +1,27 @@
+require 'influxdb/arel/nodes/node'
+require 'influxdb/arel/nodes/now'
+require 'influxdb/arel/nodes/select_statement'
+# require 'influxdb/arel/nodes/update_statement'
+
+# unary
+require 'influxdb/arel/nodes/unary'
+require 'influxdb/arel/nodes/grouping'
+require 'influxdb/arel/nodes/time'
+
+
+# binary
+require 'influxdb/arel/nodes/binary'
+require 'influxdb/arel/nodes/duration'
+require 'influxdb/arel/nodes/equality'
+require 'influxdb/arel/nodes/in'
+# require 'influxdb/arel/nodes/delete_statement'
+require 'influxdb/arel/nodes/table_alias'
+require 'influxdb/arel/nodes/infix_operation'
+
+# nary
+require 'influxdb/arel/nodes/and'
+
+
+require 'influxdb/arel/nodes/function'
+# require 'influxdb/arel/nodes/named_function'
+require 'influxdb/arel/nodes/sql_literal'

data/lib/influxdb/arel/nodes/and.rb

@@ -0,0 +1,32 @@
+module Influxdb
+  module Arel
+    module Nodes
+      class And < Node
+        attr_reader :children
+
+        def initialize(children)
+          super()
+          @children = children
+        end
+
+        def left
+          children.first
+        end
+
+        def right
+          children[1]
+        end
+
+        def hash
+          children.hash
+        end
+
+        def eql?(other)
+          self.class == other.class && children == other.children
+        end
+
+        alias :== :eql?
+      end
+    end
+  end
+end

data/lib/influxdb/arel/nodes/binary.rb

@@ -0,0 +1,47 @@
+module Influxdb
+  module Arel
+    module Nodes
+      class Binary < Node
+        attr_accessor :left, :right
+
+        def initialize(left, right)
+          super()
+          self.left = left
+          self.right = right
+        end
+
+        def initialize_copy(other)
+          super
+          self.left = left.clone if left
+          self.right = right.clone if right
+        end
+
+        def hash
+          [left, right].hash
+        end
+
+        def eql?(other)
+          self.class == other.class && left == other.left && right == other.right
+        end
+
+        alias :== :eql?
+      end
+
+      %w{
+        As
+        DoesNotMatch
+        GreaterThan
+        GreaterThanOrEqual
+        Join
+        LessThan
+        LessThanOrEqual
+        Matches
+        Merge
+        NotEqual
+        Or
+      }.each do |name|
+        const_set(name, Class.new(Binary))
+      end
+    end
+  end
+end

data/lib/influxdb/arel/nodes/duration.rb

@@ -0,0 +1,30 @@
+module Influxdb
+  module Arel
+    module Nodes
+      class Duration < Binary
+        POSSIBLE_SUFFIXES = %w{u s m h d w}.freeze
+
+        alias :value :left
+        alias :suffix :right
+
+        def initialize(value, suffix)
+          suffix = suffix.to_s
+          suffix = 'u' unless POSSIBLE_SUFFIXES.include?(suffix)
+          super(value.to_i, suffix)
+        end
+
+        def time
+          Arel.time(self)
+        end
+
+        def ago
+          Arel.now - self
+        end
+
+        def since
+          Arel.now + self
+        end
+      end
+    end
+  end
+end

data/lib/influxdb/arel/nodes/equality.rb

@@ -0,0 +1,11 @@
+module Influxdb
+  module Arel
+    module Nodes
+      class Equality < Binary
+        def operator; :== end
+        alias :operand1 :left
+        alias :operand2 :right
+      end
+    end
+  end
+end

data/lib/influxdb/arel/nodes/function.rb

@@ -0,0 +1,47 @@
+module Influxdb
+  module Arel
+    module Nodes
+      class Function < Node
+        include AliasPredication
+        include Math
+
+        attr_accessor :expressions
+
+        def initialize(expr)
+          super()
+          self.expressions = expr
+        end
+
+        def hash
+          expressions.hash
+        end
+
+        def eql?(other)
+          self.class == other.class && expressions == other.expressions
+        end
+      end
+
+      %w{
+        Count
+        Sum
+        Max
+        Min
+        Mean
+        Mode
+        Median
+        Distinct
+        Percentile
+        Histogram
+        Derivative
+        Stddev
+        First
+        Last
+        Difference
+        Top
+        Bottom
+      }.each do |name|
+        const_set(name, Class.new(Function))
+      end
+    end
+  end
+end