axiom-arango-adapter 0.0.1

data/lib/axiom/adapter/arango/visitor/for/binary.rb

@@ -0,0 +1,108 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For
+          # Base class for binary relation visitors
+          class Binary < self
+
+            LOCAL_NAME = AQL.name_node('left')
+
+          private
+
+            # Return iteration source
+            #
+            # @return [AQL::Node]
+            #
+            # @api private
+            #
+            def source
+              visit(input.left)
+            end
+
+            # Return body of for statement
+            #
+            # @return [AQL::Node]
+            #
+            # @api private
+            #
+            def body
+              self.class::Right.new(input.right, self).root
+            end
+
+            # Visitor for right of join
+            class Right < For
+
+              # Return local name
+              #
+              # @return [AQL::Node::Name]
+              #
+              # @api private
+              #
+              def local_name
+                AQL.name_node(:right)
+              end
+              memoize :local_name
+
+            private
+
+              # Return source of iteration
+              #
+              # @return [AQL::Node]
+              #
+              # @api private
+              #
+              def source
+                visit(input)
+              end
+
+              # Return return value
+              #
+              # @return [AQL::Node::Literal::Compised::Document]
+              #
+              # @api private
+              #
+              def return_value
+                AQL::Node::Literal::Composed::Document.new(left_document_attributes + right_document_attributes)
+              end
+
+              # Return right header
+              #
+              # @return [Relation::Header]
+              #
+              # @api private
+              #
+              def right_header
+                input.header
+              end
+
+              # Return left header
+              #
+              # @return [Relation::Header]
+              #
+              # @api private
+              #
+              def left_header
+                context_input.left.header
+              end
+
+              abstract_method :left_document_attributes
+              abstract_method :right_document_attributes
+
+              # Return context input
+              #
+              # @return [Axiom::Relation]
+              #
+              # @api private
+              #
+              def context_input
+                context.input
+              end
+
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/axiom/adapter/arango/visitor/for/binary/join.rb

@@ -0,0 +1,131 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For
+          class Binary
+            # Visitor for emitting joins
+            class Join < self
+
+              handle(Axiom::Algebra::Join)
+
+              # Return local name
+              #
+              # @return [AQL::Node::Name]
+              #
+              # @api private
+              #
+              def local_name
+                AQL::Node::Name.new("local_#{depth}")
+              end
+              memoize :local_name
+
+            private
+
+              # Visitor for right of join
+              class Right < Binary::Right
+
+              private
+
+                # Return filter
+                #
+                # @return [AQL::Node::Operation::Filter]
+                #
+                # @api private
+                #
+                def filter
+                  AQL::Node::Operation::Unary::Filter.new(filter_expression)
+                end
+
+                # Return filter expression
+                #
+                # @return [AQL::Node]
+                #
+                # @api private
+                #
+                def filter_expression
+                  AQL::Node::Operator::Nary::And.new(filter_predicates)
+                end
+
+                # Return filter predicates
+                #
+                # @return [Enumerable<AQL::Node::Operator::Binary::Equality>]
+                #
+                # @api private
+                #
+                def filter_predicates
+                  context_input.join_header.map do |attribute|
+                    filter_predicate(attribute)
+                  end
+                end
+
+                # Return filter predicate for attribute
+                #
+                # @param [Attribute] attribute
+                #
+                # @return [AQL::Node::Operator::Binary::Equality]
+                #
+                # @api private
+                #
+                def filter_predicate(attribute)
+                  AQL::Node::Operator::Binary::Equality.new(visit(attribute, context), visit(attribute))
+                end
+
+                # Return body of for statement
+                #
+                # @return [AQL::Node]
+                #
+                # @api private
+                #
+                def body
+                  AQL::Node::Block.new([filter, return_operation])
+                end
+
+                # Return left document attributes
+                #
+                # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+                #
+                # @api private
+                #
+                def left_document_attributes
+                  visitor(left_header, context).document_attributes
+                end
+
+                # Return left header
+                #
+                # @return [Relation::Header]
+                #
+                # @api private
+                #
+                def left_header
+                  context_input.left.header
+                end
+
+                # Return right document attributes
+                #
+                # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+                #
+                # @api private
+                #
+                def right_document_attributes
+                  visitor(right_document_header).document_attributes
+                end
+
+                # Return right document header
+                #
+                # @return [Relation::Header]
+                #
+                # @api private
+                #
+                def right_document_header
+                  right_header - left_header
+                end
+
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/axiom/adapter/arango/visitor/for/binary/product.rb

@@ -0,0 +1,48 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For
+          class Binary
+            # Visitor for emitting products
+            class Product < self
+
+              handle(Axiom::Algebra::Product)
+
+
+            private
+
+              # Vistior for right of projection
+              class Right < Binary::Right
+
+              private
+
+                # Return left document attributes
+                #
+                # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+                #
+                # @api private
+                #
+                def left_document_attributes
+                  visitor = visitor(left_header, context)
+                  visitor.document_attributes
+                end
+
+                # Return right document attributes
+                #
+                # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+                #
+                # @api private
+                #
+                def right_document_attributes
+                  visitor(right_header).document_attributes
+                end
+
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/axiom/adapter/arango/visitor/for/summarization.rb

@@ -0,0 +1,113 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For
+          # Visitor for axiom summarizations
+          class Summarization < self
+            handle(Axiom::Algebra::Summarization)
+
+            LOCAL_NAME = AQL.name_node('summarization')
+            COLLECT_NAME = AQL.name_node('collect')
+
+          private
+
+            # Return body of for operation
+            #
+            # @return [AQL::Node::Block]
+            #
+            # @api private
+            #
+            def body
+              AQL::Node::Block.new([
+                collect_operation,
+                return_operation
+              ])
+            end
+
+            # Return collection assignments
+            #
+            # @return [Enumerable<AQL::Node::Operator::Assignment>]
+            #
+            # @api private
+            #
+            def assignments
+              input.summarize_per.header.map do |attribute|
+                name = AQL::Node::Name.new(attribute.name.to_s)
+                value = AQL::Node::Attribute.new(LOCAL_NAME, name)
+                AQL::Node::Operator::Assignment.new(
+                  name,
+                  value
+                )
+              end
+            end
+            memoize :assignments
+
+            # Return AQL return value
+            #
+            # @return [AQL::Node::Literal::Composed::Document]
+            #
+            # @api private
+            #
+            def return_value
+              AQL::Node::Literal::Composed::Document.new(projected_document_attributes + extension_document_attributes)
+            end
+
+            # Return extension document attributes
+            #
+            # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+            #
+            # @api private
+            #
+            def extension_document_attributes
+              input.summarizers.map do |attribute, summarizer|
+                extension_document_attribute(attribute, summarizer)
+              end
+            end
+
+            # Return extension document attribute
+            #
+            # @param [Attribute] attribute
+            # @param [Aggregate] summarizer
+            #
+            # @return [AQL::Node::Literal::Composed::Document::Attribute]
+            #
+            # @api private
+            #
+            def extension_document_attribute(attribute, summarizer)
+              key = AQL::Node::Literal::Primitive::String.new(attribute.name.to_s)
+              value = visit(summarizer)
+              AQL::Node::Literal::Composed::Document::Attribute.new(key, value)
+            end
+
+            # Return projected document attributes
+            #
+            # @return [Enumerable<AQL::Node::Literal::Composed::Document::Attribute>]
+            #
+            # @api private
+            #
+            def projected_document_attributes
+              assignments.map do |assignment|
+                value = assignment.name
+                key   = AQL::Node::Literal::Primitive::String.new(value.name)
+                AQL::Node::Literal::Composed::Document::Attribute.new(key, value)
+              end
+            end
+            memoize :projected_document_attributes
+
+            # Return collect operation
+            #
+            # @return [AQL::Node::Operation::Nary::Collect::Into]
+            #
+            # @api private
+            #
+            def collect_operation
+              AQL::Node::Operation::Nary::Collect::Into.new(assignments, COLLECT_NAME)
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/axiom/adapter/arango/visitor/for/unary.rb

@@ -0,0 +1,29 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For 
+          # Base class for visitors that emit from unary axiom relations
+          class Unary < self
+
+          private
+
+            # Return body
+            #
+            # @return [AQL::Node]
+            #
+            # @api private
+            #
+            def body
+              AQL::Node::Block.new([
+                operation,
+                return_operation
+              ])
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/axiom/adapter/arango/visitor/for/unary/extension.rb

@@ -0,0 +1,72 @@
+module Axiom
+  module Adapter
+    module Arango
+      class Visitor
+        class For
+          class Unary
+            # Visitor for emitting extensions
+            class Extension < self
+
+              handle(Algebra::Extension)
+
+              LOCAL_NAME = AQL.name_node('extension')
+
+            private
+
+              # Return for body
+              #
+              # @return [AQL::Node]
+              #
+              # @api private
+              #
+              def body
+                return_operation
+              end
+
+              # Return return value
+              #
+              # @return [AQL::Node::Call]
+              #
+              # @api private
+              #
+              def return_value
+                Node::Call.new('MERGE', [local_name, Node::Literal::Composed::Document.new(extended_attributes)])
+              end
+
+              # Return document attribute node
+              #
+              # @return [Axiom::Attribute] attribute
+              #
+              # @return [AQL::Node]
+              #
+              # @api private
+              #
+              def extended_attributes
+                input.extensions.map do |attribute, function|
+                  document_attribute(attribute, function)
+                end
+              end
+
+              # Return document attribute
+              #
+              # @param [Axiom::Attribute] attribute
+              # @param [Axiom::Function] function
+              #
+              # @return [AQL::Node::Literal::Composed::Document::Attribute>]
+              #
+              # @api private
+              #
+              def document_attribute(attribute, function)
+                Node::Literal::Composed::Document::Attribute.new(
+                  Node::Literal::Primitive::String.new(attribute.name.to_s),
+                  visit(function)
+                )
+              end
+
+            end
+          end
+        end
+      end
+    end
+  end
+end