Redis
 sql >> Base de données >  >> NoSQL >> Redis

Autocomplétion Redis

Si vous avez affaire à un grand ensemble de données, je suggérerais d'envisager de l'implémenter comme un trie. J'ai rassemblé un petit morceau de Ruby qui ferait ceci :

    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)

Par exemple :

    $ 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
    []

En savoir plus sur les essais à l'entrée de Wikipedia sur les essais.

Vous voudrez certainement optimiser votre méthode de suggestion pour ne pas renvoyer TOUTES les valeurs, mais uniquement renvoyer les premières valeurs X qu'elle trouve. Cela irait à l'encontre de l'objectif d'itérer toute la structure de données.