thinking-sphinx 2.0.6 → 2.0.7

data/lib/thinking_sphinx/adapters/postgresql_adapter.rb

@@ -0,0 +1,171 @@
+module ThinkingSphinx
+  class PostgreSQLAdapter < AbstractAdapter
+    def setup
+      create_array_accum_function
+      create_crc32_function
+    end
+
+    def sphinx_identifier
+      "pgsql"
+    end
+
+    def concatenate(clause, separator = ' ')
+      if clause[/^COALESCE/]
+        clause.split('), ').join(") || '#{separator}' || ")
+      else
+        clause.split(', ').collect { |field|
+          "CAST(COALESCE(#{field}, '') as varchar)"
+        }.join(" || '#{separator}' || ")
+      end
+    end
+
+    def group_concatenate(clause, separator = ' ')
+      "array_to_string(array_accum(COALESCE(#{clause}, '0')), '#{separator}')"
+    end
+
+    def cast_to_string(clause)
+      clause
+    end
+
+    def cast_to_datetime(clause)
+      if ThinkingSphinx::Configuration.instance.use_64_bit
+        "cast(extract(epoch from #{clause}) as bigint)"
+      else
+        "cast(extract(epoch from #{clause}) as int)"
+      end
+    end
+
+    def cast_to_unsigned(clause)
+      clause
+    end
+
+    def cast_to_int(clause)
+      "#{clause}::INT8"
+    end
+
+    def convert_nulls(clause, default = '')
+      default = case default
+      when String
+        "'#{default}'"
+      when NilClass
+        'NULL'
+      when Fixnum
+        "#{default}::bigint"
+      else
+        default
+      end
+
+      "COALESCE(#{clause}, #{default})"
+    end
+
+    def boolean(value)
+      value ? 'TRUE' : 'FALSE'
+    end
+
+    def crc(clause, blank_to_null = false)
+      clause = "NULLIF(#{clause},'')" if blank_to_null
+      "crc32(#{clause})"
+    end
+
+    def utf8_query_pre
+      nil
+    end
+
+    def time_difference(diff)
+      "current_timestamp - interval '#{diff} seconds'"
+    end
+
+    def utc_query_pre
+      "SET TIME ZONE 'UTC'"
+    end
+
+    private
+
+    def execute(command, output_error = false)
+      if RUBY_PLATFORM == 'java'
+        connection.transaction do
+          execute_command command, output_error
+        end
+      else
+        execute_command command, output_error
+      end
+    end
+
+    def execute_command(command, output_error = false)
+      connection.execute "begin"
+      connection.execute "savepoint ts"
+      begin
+        connection.execute command
+      rescue StandardError => err
+        puts err if output_error
+        connection.execute "rollback to savepoint ts"
+      end
+      connection.execute "release savepoint ts"
+      connection.execute "commit"
+    end
+
+    def create_array_accum_function
+      if connection.raw_connection.respond_to?(:server_version) && connection.raw_connection.server_version > 80200
+        execute <<-SQL
+          CREATE AGGREGATE array_accum (anyelement)
+          (
+              sfunc = array_append,
+              stype = anyarray,
+              initcond = '{}'
+          );
+        SQL
+      else
+        execute <<-SQL
+          CREATE AGGREGATE array_accum
+          (
+              basetype = anyelement,
+              sfunc = array_append,
+              stype = anyarray,
+              initcond = '{}'
+          );
+        SQL
+      end
+    end
+
+    def create_crc32_function
+      execute "CREATE LANGUAGE 'plpgsql';"
+      function = <<-SQL
+        CREATE OR REPLACE FUNCTION crc32(word text)
+        RETURNS bigint AS $$
+          DECLARE tmp bigint;
+          DECLARE i int;
+          DECLARE j int;
+          DECLARE byte_length int;
+          DECLARE word_array bytea;
+          BEGIN
+            IF COALESCE(word, '') = '' THEN
+              return 0;
+            END IF;
+
+            i = 0;
+            tmp = 4294967295;
+            byte_length = bit_length(word) / 8;
+            word_array = decode(replace(word, E'\\\\', E'\\\\\\\\'), 'escape');
+            LOOP
+              tmp = (tmp # get_byte(word_array, i))::bigint;
+              i = i + 1;
+              j = 0;
+              LOOP
+                tmp = ((tmp >> 1) # (3988292384 * (tmp & 1)))::bigint;
+                j = j + 1;
+                IF j >= 8 THEN
+                  EXIT;
+                END IF;
+              END LOOP;
+              IF i >= byte_length THEN
+                EXIT;
+              END IF;
+            END LOOP;
+            return (tmp # 4294967295);
+          END
+        $$ IMMUTABLE LANGUAGE plpgsql;
+      SQL
+      execute function, true
+    end
+  end
+end

data/lib/thinking_sphinx/association.rb

@@ -0,0 +1,229 @@
+module ThinkingSphinx
+  # Association tracks a specific reflection and join to reference data that
+  # isn't in the base model. Very much an internal class for Thinking Sphinx -
+  # perhaps because I feel it's not as strong (or simple) as most of the rest.
+  # 
+  class Association
+    attr_accessor :parent, :reflection, :join
+    
+    # Create a new association by passing in the parent association, and the
+    # corresponding reflection instance. If there is no parent, pass in nil.
+    # 
+    #   top   = Association.new nil, top_reflection
+    #   child = Association.new top, child_reflection
+    # 
+    def initialize(parent, reflection)
+      @parent, @reflection = parent, reflection
+      @children = {}
+    end
+    
+    # Get the children associations for a given association name. The only time
+    # that there'll actually be more than one association is when the
+    # relationship is polymorphic. To keep things simple though, it will always
+    # be an Array that gets returned (an empty one if no matches).
+    #
+    #   # where pages is an association on the class tied to the reflection.
+    #   association.children(:pages)
+    # 
+    def children(assoc)
+      @children[assoc] ||= Association.children(@reflection.klass, assoc, self)
+    end
+    
+    # Get the children associations for a given class, association name and
+    # parent association. Much like the instance method of the same name, it
+    # will return an empty array if no associations have the name, and only
+    # have multiple association instances if the underlying relationship is
+    # polymorphic.
+    # 
+    #   Association.children(User, :pages, user_association)
+    # 
+    def self.children(klass, assoc, parent=nil)
+      ref = klass.reflect_on_association(assoc)
+      
+      return [] if ref.nil?
+      return [Association.new(parent, ref)] unless ref.options[:polymorphic]
+      
+      # association is polymorphic - create associations for each
+      # non-polymorphic reflection.
+      polymorphic_classes(ref).collect { |poly_class|
+        Association.new parent, depolymorphic_reflection(ref, klass, poly_class)
+      }
+    end
+    
+    # Link up the join for this model from a base join - and set parent
+    # associations' joins recursively.
+    #
+    def join_to(base_join)
+      parent.join_to(base_join) if parent && parent.join.nil?
+      
+      @join ||= join_association_class.new(
+        @reflection, base_join, parent ? parent.join : join_parent(base_join)
+      )
+    end
+    
+    def arel_join
+      @join.join_type = Arel::OuterJoin
+      rewrite_conditions
+      
+      @join
+    end
+    
+    # Returns true if the association - or a parent - is a has_many or
+    # has_and_belongs_to_many.
+    # 
+    def is_many?
+      case @reflection.macro
+      when :has_many, :has_and_belongs_to_many
+        true
+      else
+        @parent ? @parent.is_many? : false
+      end
+    end
+    
+    # Returns an array of all the associations that lead to this one - starting
+    # with the top level all the way to the current association object.
+    # 
+    def ancestors
+      (parent ? parent.ancestors : []) << self
+    end
+    
+    def has_column?(column)
+      @reflection.klass.column_names.include?(column.to_s)
+    end
+    
+    def primary_key_from_reflection
+      if @reflection.options[:through]
+        if ThinkingSphinx.rails_3_1?
+          @reflection.source_reflection.foreign_key
+        else
+          @reflection.source_reflection.options[:foreign_key] ||
+          @reflection.source_reflection.primary_key_name
+        end
+      elsif @reflection.macro == :has_and_belongs_to_many
+        @reflection.association_foreign_key
+      else
+        nil
+      end
+    end
+    
+    def table
+      if @reflection.options[:through] ||
+        @reflection.macro == :has_and_belongs_to_many
+        if ThinkingSphinx.rails_3_1?
+          @join.tables.first.name
+        else
+          @join.aliased_join_table_name
+        end
+      else
+        @join.aliased_table_name
+      end
+    end
+    
+    private
+    
+    def self.depolymorphic_reflection(reflection, source_class, poly_class)
+      name = "#{reflection.name}_#{poly_class.name}".to_sym
+      
+      source_class.reflections[name] ||=
+        ::ActiveRecord::Reflection::AssociationReflection.new(
+          reflection.macro, name, casted_options(poly_class, reflection),
+          reflection.active_record
+        )
+    end
+        
+    # Returns all the objects that could be currently instantiated from a
+    # polymorphic association. This is pretty damn fast if there's an index on
+    # the foreign type column - but if there isn't, it can take a while if you
+    # have a lot of data.
+    # 
+    def self.polymorphic_classes(ref)
+      ref.active_record.connection.select_all(
+        "SELECT DISTINCT #{foreign_type(ref)} " +
+        "FROM #{ref.active_record.table_name} " +
+        "WHERE #{foreign_type(ref)} IS NOT NULL"
+      ).collect { |row|
+        row[foreign_type(ref)].constantize
+      }
+    end
+    
+    # Returns a new set of options for an association that mimics an existing
+    # polymorphic relationship for a specific class. It adds a condition to
+    # filter by the appropriate object.
+    # 
+    def self.casted_options(klass, ref)
+      options = ref.options.clone
+      options[:polymorphic]   = nil
+      options[:class_name]    = klass.name
+      options[:foreign_key] ||= "#{ref.name}_id"
+      
+      quoted_foreign_type = klass.connection.quote_column_name foreign_type(ref)
+      case options[:conditions]
+      when nil
+        options[:conditions] = "::ts_join_alias::.#{quoted_foreign_type} = '#{klass.name}'"
+      when Array
+        options[:conditions] << "::ts_join_alias::.#{quoted_foreign_type} = '#{klass.name}'"
+      when Hash
+        options[:conditions].merge!(foreign_type(ref) => klass.name)
+      else
+        options[:conditions] << " AND ::ts_join_alias::.#{quoted_foreign_type} = '#{klass.name}'"
+      end
+      
+      options
+    end
+    
+    def join_association_class
+      if ThinkingSphinx.rails_3_1?
+        ::ActiveRecord::Associations::JoinDependency::JoinAssociation
+      else
+        ::ActiveRecord::Associations::ClassMethods::JoinDependency::JoinAssociation
+      end
+    end
+    
+    def join_parent(join)
+      if ThinkingSphinx.rails_3_1?
+        join.join_parts.first
+      else
+        join.joins.first
+      end
+    end
+    
+    def self.foreign_type(ref)
+      if ThinkingSphinx.rails_3_1?
+        ref.foreign_type
+      else
+        ref.options[:foreign_type]
+      end
+    end
+    
+    def rewrite_conditions
+      @join.options[:conditions] = case @join.options[:conditions]
+      when String
+        rewrite_condition @join.options[:conditions]
+      when Array
+        @join.options[:conditions].collect { |condition|
+          rewrite_condition condition
+        }
+      else
+        @join.options[:conditions]
+      end
+    end
+    
+    def rewrite_condition(condition)
+      return condition unless condition.is_a?(String)
+      
+      if defined?(ActsAsTaggableOn) &&
+        @reflection.klass == ActsAsTaggableOn::Tagging &&
+        @reflection.name.to_s[/_taggings$/]
+        condition = condition.gsub /taggings\./, "#{quoted_alias @join}."
+      end
+      
+      condition.gsub /::ts_join_alias::/, quoted_alias(@join.parent)
+    end
+    
+    def quoted_alias(join)
+      @reflection.klass.connection.quote_table_name(
+        join.aliased_table_name
+      )
+    end
+  end
+end