elasticsearch for topics (#1318)

app/models/topic.rb

@@ -5,6 +5,7 @@ class Topic < ApplicationRecord
   include Commentable
   include Moderatable
   include Viewable
+  include ElasticsearchConcern
 
   NEWS_WALL = /[\r\n]*\[wall[\s\S]+\[\/wall\]\Z/
 

app/services/elasticsearch/data/topic.rb

@@ -0,0 +1,19 @@
+class Elasticsearch::Data::Topic < Elasticsearch::Data::DataBase
+  name_fields %i[name]
+  data_fields %i[locale forum_id]
+  track_changes_fields %i[name forum_id]
+
+private
+
+  def name
+    fix Topics::TopicViewFactory.new(true, true).build(@entry).topic_title
+  end
+
+  def locale
+    @entry.locale
+  end
+
+  def forum_id
+    @entry.forum_id
+  end
+end

app/services/elasticsearch/query/topic.rb

@@ -0,0 +1,25 @@
+class Elasticsearch::Query::Topic < Elasticsearch::Query::QueryBase
+  method_object %i[phrase locale forum_id limit]
+
+private
+
+  def query
+    {
+      bool: {
+        must: [name_fields_query, locale_query, forum_id_query]
+      }
+    }
+  end
+
+  def locale_query
+    { term: { locale: @locale } }
+  end
+
+  def forum_id_query
+    { term: { forum_id: @forum_id } }
+  end
+
+  def cache_key
+    super + [@locale, @forum_id]
+  end
+end

app/services/elasticsearch/reindex.rb

@@ -3,10 +3,10 @@ class Elasticsearch::Reindex
 
   CACHE_KEY = 'elastic_reindex'
 
-  TYPES = %i[anime manga ranobe character person user club collection]
+  TYPES = %i[anime manga ranobe character person user club collection topic]
 
   def call
-    @types.each do |type|
+    Array(@types).each do |type|
       respond_to?("fill_#{type}") ? send("fill_#{type}") : fill_type(type)
     end
   end

app/workers/elasticsearch/create.rb

@@ -7,12 +7,13 @@ def perform entry_id, entry_type
   end
 
   def sync entry
-    type = entry.class.name.pluralize.downcase
+    klass = entry.is_a?(Topic) ? Topic : entry.class
+    type = klass.name.pluralize.downcase
 
     Elasticsearch::Client.instance.send(
       self.class::METHOD,
       "#{INDEX}_#{type}/#{type}/#{entry.id}",
-      "Elasticsearch::Data::#{entry.class.name}".constantize.call(entry)
+      "Elasticsearch::Data::#{klass.name}".constantize.call(entry)
     )
   end
 end

config/app/elasticsearch.yml

@@ -169,3 +169,8 @@
       properties:
         name:
           <<: *name_properties
+    :topics:
+      <<: *default_type_mappings
+      properties:
+        name:
+          <<: *name_properties

spec/models/topic_spec.rb

@@ -290,4 +290,6 @@
       end
     end
   end
+
+  it_behaves_like :elasticsearch_concern, :anime
 end

spec/services/elasticsearch/query/topic_spec.rb

@@ -0,0 +1,21 @@
+describe Elasticsearch::Query::Topic do
+  let(:service) do
+    Elasticsearch::Query::Topic.new(
+      phrase: phrase,
+      locale: locale,
+      forum_id: forum_id,
+      limit: limit
+    )
+  end
+
+  describe '#call', :vcr do
+    let(:phrase) { 'kai' }
+    let(:locale) { 'ru' }
+    let(:forum_id) { 1 }
+    let(:limit) { 10 }
+
+    subject { service.call }
+
+    it { is_expected.to have(limit).items }
+  end
+end