dbldots_oedipus 0.0.16

data/ext/oedipus/oedipus.h

@@ -0,0 +1,58 @@
+/*-- encoding: utf-8 --*/
+
+/*
+ * Oedipus Sphinx 2 Search.
+ * Copyright © 2012 Chris Corbyn.
+ *
+ * See LICENSE file for details.
+ */
+
+#include <ruby.h>
+#include <mysql.h>
+
+// Macros for lazy fingers
+#define ODP_TO_S(v)                  rb_funcall(v, rb_intern("to_s"), 0)
+#define ODP_TO_F(n)                  rb_funcall(n, rb_intern("to_f"), 0)
+#define ODP_KIND_OF_P(v, type)       (rb_funcall(v, rb_intern("kind_of?"), 1, type) == Qtrue)
+
+/*! Internal struct used to reference a mysql connection */
+typedef struct {
+  /*! Boolean representing the connected state */
+  int     connected;
+  /*! The actual pointer allocated by mysql_init() */
+  MYSQL * ptr;
+} OdpMysql;
+
+/* -- Public methods -- */
+
+/*! Allocate and initialize a new mysql client */
+static VALUE odp_new(VALUE klass, VALUE host, VALUE port);
+
+/*! Initialize a new mysql client */
+static VALUE odp_initialize(VALUE self, VALUE host, VALUE port);
+
+/*! Connect, or reconnect to mysql */
+static VALUE odp_open(VALUE self);
+
+/*! Disconnect from mysql */
+static VALUE odp_close(VALUE self);
+
+/*! Execute an SQL non-read query and return the number of rows affected */
+static VALUE odp_execute(int argc, VALUE * args, VALUE self);
+
+/*! Execute several SQL read queries and return the result sets  */
+static VALUE odp_query(int argc, VALUE * args, VALUE self);
+
+/* -- Internal methods -- */
+
+/*! Generic method to raise a connection error */
+static void odp_raise(VALUE self, const char *msg);
+
+/*! Free memory allocated to mysql */
+static void odp_free(OdpMysql * conn);
+
+/*! Substitute all ? markers with the values in bind_values */
+static VALUE odp_replace_bind_values(OdpMysql * conn, VALUE sql, VALUE * bind_values, int num_values);
+
+/*! Cast the given field to a ruby data type */
+static VALUE odp_cast_value(MYSQL_FIELD f, char * v, unsigned long len);

data/lib/oedipus.rb

@@ -0,0 +1,40 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+require "oedipus/version"
+
+require "oedipus/oedipus"
+
+require "oedipus/comparison"
+require "oedipus/comparison/equal"
+require "oedipus/comparison/not_equal"
+require "oedipus/comparison/between"
+require "oedipus/comparison/outside"
+require "oedipus/comparison/in"
+require "oedipus/comparison/not_in"
+require "oedipus/comparison/gte"
+require "oedipus/comparison/gt"
+require "oedipus/comparison/lte"
+require "oedipus/comparison/lt"
+require "oedipus/comparison/not"
+require "oedipus/comparison/shortcuts"
+
+require "oedipus/query_builder"
+
+require "oedipus/connection_error"
+require "oedipus/connection"
+require "oedipus/connection/pool"
+require "oedipus/connection/registry"
+
+require "oedipus/index"
+
+module Oedipus
+  extend Comparison::Shortcuts
+  extend Connection::Registry
+end

data/lib/oedipus/comparison.rb

@@ -0,0 +1,88 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Represents a comparison operator and value.
+  class Comparison
+    class << self
+      # Return a suitable comparison object for +v+.
+      #
+      # The conversions are:
+      #
+      #   - leave Comparison objects unchanged
+      #   - convert real ranges to Between comparisons
+      #   - convert Infinity-bounded exclusive ranges to GT/LT comparisons
+      #   - convert Infinity-bounded inclusive ranges to GTE/LTE comparisons.
+      #   - convert everything else to an Equal comparison
+      #
+      # @param [Object] v
+      #   a ruby object to be compared
+      #
+      # @param [Comparison]
+      #   a comparison suitable for comparing the input
+      def of(v)
+        case v
+        when Comparison
+          v
+        when Range
+          if v.end == Float::INFINITY
+            v.exclude_end? ? Shortcuts.gt(v.first) : Shortcuts.gte(v.first)
+          elsif v.first == -Float::INFINITY
+            v.exclude_end? ? Shortcuts.lt(v.end) : Shortcuts.lte(v.end)
+          else
+            Shortcuts.between(v)
+          end
+        when Enumerable
+          Shortcuts.in(v)
+        else
+          Shortcuts.eq(v)
+        end
+      end
+    end
+
+    attr_reader :v
+
+    # Initialize a new Comparison for +v+.
+    #
+    # @param [Object] v
+    #   any ruby object to compare
+    def initialize(v)
+      @v = v
+    end
+
+    # Compare two comparisons for equality.
+    #
+    # @param [Comparison] other
+    #   another comparison to check
+    #
+    # @return [Boolean]
+    #   true if the comparisons are the same
+    def ==(other)
+      other.class == self.class && other.v == v
+    end
+
+    alias_method :eql?, :==
+
+    # Return the exact inverse of this comparison.
+    #
+    # @return [Comparison]
+    #   the inverse of the current comparison
+    def inverse
+      raise NotImplementedError, "Comparison#inverse must be defined by subclasses"
+    end
+
+    # Represent the comparison as SQL arguments.
+    #
+    # @return [Array]
+    #   an SQL expression to compare a LHS against v
+    def to_sql
+      raise NotImplementedError, "Comparison#to_sql must be defined by subclasses"
+    end
+  end
+end

data/lib/oedipus/comparison/between.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Between comparison of range.
+  class Comparison::Between < Comparison
+    def to_sql
+      ["BETWEEN ? AND ?", v.first, v.exclude_end? ? v.end - 1 : v.end]
+    end
+
+    def inverse
+      Comparison::Outside.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/equal.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Equality comparison of value.
+  class Comparison::Equal < Comparison
+    def to_sql
+      ["= ?", v]
+    end
+
+    def inverse
+      Comparison::NotEqual.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/gt.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Greater than comparison of +v+.
+  class Comparison::GT < Comparison
+    def to_sql
+      ["> ?", v]
+    end
+
+    def inverse
+      Comparison::LTE.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/gte.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Greater than or equal comparison of +v+.
+  class Comparison::GTE < Comparison
+    def to_sql
+      [">= ?", v]
+    end
+
+    def inverse
+      Comparison::LT.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/in.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # IN comparison of +v+.
+  class Comparison::In < Comparison
+    def to_sql
+      ["IN (#{v.map{'?'}.join(', ')})", *v]
+    end
+
+    def inverse
+      Comparison::NotIn.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/lt.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Less than comparison of +v+.
+  class Comparison::LT < Comparison
+    def to_sql
+      ["< ?", v]
+    end
+
+    def inverse
+      Comparison::GTE.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/lte.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Less than or equal comparison of +v+.
+  class Comparison::LTE < Comparison
+    def to_sql
+      ["<= ?", v]
+    end
+
+    def inverse
+      Comparison::GT.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/not.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Negation comparison of value.
+  class Comparison::Not < Comparison
+    def initialize(v)
+      super(Comparison.of(v))
+    end
+
+    def to_sql
+      v.inverse.to_sql
+    end
+
+    def inverse
+      v
+    end
+  end
+end

data/lib/oedipus/comparison/not_equal.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Negation comparison of value.
+  class Comparison::NotEqual < Comparison
+    def to_sql
+      ["!= ?", v]
+    end
+
+    def inverse
+      Comparison::Equal.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/not_in.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # NOT IN comparison of +v+.
+  class Comparison::NotIn < Comparison
+    def to_sql
+      ["NOT IN (#{v.map{'?'}.join(', ')})", *v]
+    end
+
+    def inverse
+      Comparison::In.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/outside.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  # Outside comparison of range.
+  class Comparison::Outside < Comparison
+    def to_sql
+      ["NOT BETWEEN ? AND ?", v.first, v.exclude_end? ? v.end - 1 : v.end]
+    end
+
+    def inverse
+      Comparison::Between.new(v)
+    end
+  end
+end

data/lib/oedipus/comparison/shortcuts.rb

@@ -0,0 +1,144 @@
+# encoding: utf-8
+
+##
+# Oedipus Sphinx 2 Search.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module Oedipus
+  class Comparison
+    # Provides short methods for casting values to Comparisons.
+    module Shortcuts
+      extend self
+
+      # Return the Comparison for equality of +v+.
+      #
+      # @param [Object] v
+      #   any ruby object to compare in a query
+      #
+      # @return [Comparison::Equal]
+      #   an equality comparison of v
+      def eq(v)
+        Equal.new(v)
+      end
+
+      # Return the Comparison for inequality of +v+.
+      #
+      # @param [Object] v
+      #   any ruby object to compare in a query
+      #
+      # @return [Comparison::Equal]
+      #   an inequality comparison of v
+      def neq(v)
+        NotEqual.new(v)
+      end
+
+      # Return the Comparison for negation of +v+.
+      #
+      # @param [Object] v
+      #   any ruby object to compare in a query
+      #
+      # @return [Comparison::Not]
+      #   an negated comparison of v
+      def not(v)
+        Not.new(v)
+      end
+
+      # Return the Comparison for the range a..b.
+      #
+      # @param [Object] v
+      #   either a Range, or a number
+      #
+      # @param [Fixnum] b
+      #   if the first argument was a number, the other bound
+      #
+      # @return [Comparison::Between]
+      #   an between comparison of a..b
+      def between(a, b = nil)
+        Between.new(a.kind_of?(Range) ? a : a..b)
+      end
+
+      # Return the Comparison to exclude the range a..b.
+      #
+      # @param [Object] v
+      #   either a Range, or a number
+      #
+      # @param [Fixnum] b
+      #   if the first argument was a number, the other bound
+      #
+      # @return [Comparison::Outside]
+      #   an outside comparison of a..b
+      def outside(a, b = nil)
+        Outside.new(a.kind_of?(Range) ? a : a..b)
+      end
+
+      # Return the Comparison for any value in the set +v+.
+      #
+      # @param [Object] v
+      #   any ruby object to compare
+      #
+      # @return [Comparison::In]
+      #   the IN comparison for the values in v
+      def in(*v)
+        In.new(v.map { |el| el.respond_to?(:to_a) ? el.to_a : el }.flatten)
+      end
+
+      # Return the Comparison for any value NOT in the set +v+.
+      #
+      # @param [Object] v
+      #   any ruby object to compare
+      #
+      # @return [Comparison::NotIn]
+      #   the NOT IN comparison for the values in v
+      def not_in(*v)
+        NotIn.new(v.map { |el| el.respond_to?(:to_a) ? el.to_a : el }.flatten)
+      end
+
+      # Return the Comparison for >= +v+.
+      #
+      # @param [Object] v
+      #   a number to compare
+      #
+      # @return [Comparison::GTE]
+      #   a greater than or equal comparison for v
+      def gte(v)
+        GTE.new(v)
+      end
+
+      # Return the Comparison for > +v+.
+      #
+      # @param [Object] v
+      #   a number to compare
+      #
+      # @return [Comparison::GT]
+      #   a greater than comparison for v
+      def gt(v)
+        GT.new(v)
+      end
+
+      # Return the Comparison for <= +v+.
+      #
+      # @param [Object] v
+      #   a number to compare
+      #
+      # @return [Comparison::LTE]
+      #   a less than or equal comparison for v
+      def lte(v)
+        LTE.new(v)
+      end
+
+      # Return the Comparison for < +v+.
+      #
+      # @param [Object] v
+      #   a number to compare
+      #
+      # @return [Comparison::LT]
+      #   a less than comparison for v
+      def lt(v)
+        LT.new(v)
+      end
+    end
+  end
+end