discourse-ai/lib/ai_bot/post_streamer.rb

# frozen_string_literal: true

module DiscourseAi
  module AiBot
    class PostStreamer
      def initialize(delay: 0.5)
        @mutex = Mutex.new
        @callback = nil
        @delay = delay
        @done = false
      end

      def run_later(&callback)
        @mutex.synchronize { @callback = callback }
        ensure_worker!
      end

      def finish(skip_callback: false)
        @mutex.synchronize do
          @callback&.call if skip_callback
          @callback = nil
          @done = true
        end

        begin
          @worker_thread&.wakeup
        rescue StandardError
          ThreadError
        end
        @worker_thread&.join
        @worker_thread = nil
      end

      private

      def run
        while !@done
          @mutex.synchronize do
            callback = @callback
            @callback = nil
            callback&.call
          end
          sleep @delay
        end
      end

      def ensure_worker!
        return if @worker_thread
        @mutex.synchronize do
          return if @worker_thread
          db = RailsMultisite::ConnectionManagement.current_db
          @worker_thread =
            Thread.new { RailsMultisite::ConnectionManagement.with_connection(db) { run } }
        end
      end
    end
  end
end