Redis
 sql >> Datenbank >  >> NoSQL >> Redis

Redis-Autovervollständigung

Wenn Sie es mit einem großen Datensatz zu tun haben, würde ich vorschlagen, die Implementierung als Versuch in Betracht zu ziehen. Ich habe ein bisschen Ruby zusammengestellt, das dies tun würde:

    require 'rubygems'
    require 'redis'
    
    class RedisTrie
      TERMINAL = '+'
    
      def initialize(prefix)
        @prefix = prefix
        @r = Redis.new
      end
    
      def add_word(word)
        w = word.gsub(/[^a-zA-Z0-9_-]/, '')
        key = "#{@prefix}:"
    
        w.each_char do |c|
          @r.zset_add key, c.bytes.first, c
          key += c
        end
    
        @r.zset_add key, 0, TERMINAL
      end
    
      def add_words(*words)
        words.flatten.compact.each {|word| add_word word}
      end
    
      def suggest(text)
        @r.zset_range("#{@prefix}:#{text}", 0, -1).map do |c|
          (c == TERMINAL) ? text : suggest(text + c)
        end.flatten
      end
    end
    
    rt = RedisTrie.new('trie')
    
    rt.add_words %w( apple automobile carwash oil-change cranky five ruthie axe auto )
    
    p rt.suggest(ARGV.shift.to_s)

Zum Beispiel:

    $ ruby RedisTrie.rb
    ["apple", "auto", "automobile", "axe", "carwash", "cranky", "five", "oil-change", "ruthie"]
    $ ruby RedisTrie.rb a
    ["apple", "auto", "automobile", "axe"]
    $ ruby RedisTrie.rb au
    ["auto", "automobile"]
    $ ruby RedisTrie.rb aux
    []

Lesen Sie mehr über Tries im Wikipedia-Eintrag zu Tries.

Sie sollten Ihre Suggest-Methode auf jeden Fall so optimieren, dass sie nicht ALLE Werte zurückgibt, sondern nur die ersten X Werte, die sie findet. Es würde den Zweck zunichte machen, die gesamte Datenstruktur zu durchlaufen.