[ES] Improved code search

Gemfile

@@ -111,7 +111,7 @@ gem 'seed-fu', '~> 2.3.5'
 # Search
 gem 'elasticsearch-model'
 gem 'elasticsearch-rails'
-gem 'gitlab-elasticsearch-git', '~> 0.0.16', require: "elasticsearch/git"
+gem 'gitlab-elasticsearch-git', '~> 0.0.17', require: "elasticsearch/git"
 
 # Markdown and HTML processing
 gem 'html-pipeline', '~> 1.11.0'

Gemfile.lock

@@ -282,7 +282,7 @@ GEM
       mime-types (>= 1.19)
       rugged (>= 0.23.0b)
     github-markup (1.4.0)
-    gitlab-elasticsearch-git (0.0.16)
+    gitlab-elasticsearch-git (0.0.17)
       activemodel (~> 4.2)
       activesupport (~> 4.2)
       charlock_holmes (~> 0.7)
@@ -898,7 +898,7 @@ DEPENDENCIES
   gemojione (~> 3.0)
   github-linguist (~> 4.7.0)
   github-markup (~> 1.4)
-  gitlab-elasticsearch-git (~> 0.0.16)
+  gitlab-elasticsearch-git (~> 0.0.17)
   gitlab-flowdock-git-hook (~> 1.0.1)
   gitlab-license (~> 1.0)
   gitlab_git (~> 10.4.5)

lib/tasks/gitlab/elastic.rake

@@ -79,10 +79,10 @@ namespace :gitlab do
       [Project, Issue, MergeRequest, Snippet, Note, Milestone].each do |klass|
         print "Indexing #{klass} records... "
 
-        case klass
-        when Note
+        case klass.to_s
+        when 'Note'
           Note.searchable.import_with_parent
-        when Project, Snippet
+        when 'Project', 'Snippet'
           klass.import
         else
           klass.import_with_parent