save document word count updates, too

Author: drusepth

app/jobs/save_document_revision_job.rb

@@ -11,6 +11,14 @@ def perform(*args)
     new_word_count = document.computed_word_count
     document.update(cached_word_count: new_word_count)
 
+    # Save a WordCountUpdate for this document for today
+    update = document.word_count_updates.find_or_initialize_by(
+      for_date: DateTime.current,
+    )
+    update.word_count = new_word_count
+    update.user_id  ||= document.user_id
+    update.save!
+
     # Make sure we're only storing revisions at least every 5 min
     latest_revision = document.document_revisions.order('created_at DESC').limit(1).first
     if latest_revision.present? && latest_revision.created_at > 5.minutes.ago

app/models/documents/document.rb

@@ -26,6 +26,12 @@ class Document < ApplicationRecord
 
   attr_accessor :tagged_text
 
+  # Duplicated from is_content_page since we don't include that here yet
+  has_many :word_count_updates, as: :entity, dependent: :destroy
+  def latest_word_count_cache
+    word_count_updates.order('for_date DESC').limit(1).first.try(:word_count) ||  0
+  end
+
   KEYS_TO_TRIGGER_REVISION_ON_CHANGE = %w(title body synopsis notes_text)
 
   def self.color