discourse/app/models/draft_sequence.rb

# frozen_string_literal: true

class DraftSequence < ActiveRecord::Base
  def self.next!(user, key)
    return nil if !user

    user_id = user
    user_id = user.id unless user.is_a?(Integer)

    return 0 if !User.human_user_id?(user_id)

    sequence = DB.query_single(<<~SQL, user_id: user_id, draft_key: key).first
        INSERT INTO draft_sequences (user_id, draft_key, sequence)
        VALUES (:user_id, :draft_key, 1)
        ON CONFLICT (user_id, draft_key) DO
        UPDATE
        SET sequence = draft_sequences.sequence + 1
        WHERE draft_sequences.user_id = :user_id
        AND draft_sequences.draft_key = :draft_key
        RETURNING sequence
      SQL

    DB.exec(
      "DELETE FROM drafts WHERE user_id = :user_id AND draft_key = :draft_key AND sequence < :sequence",
      draft_key: key,
      user_id: user_id,
      sequence: sequence,
    )

    UserStat.update_draft_count(user_id)

    sequence
  end

  def self.current(user, key)
    return nil if !user

    user_id = user
    user_id = user.id unless user.is_a?(Integer)

    return 0 if !User.human_user_id?(user_id)

    # perf critical path
    r, _ =
      DB.query_single(
        "select sequence from draft_sequences where user_id = ? and draft_key = ?",
        user_id,
        key,
      )
    r.to_i
  end
end

# == Schema Information
#
# Table name: draft_sequences
#
#  id        :integer          not null, primary key
#  user_id   :integer          not null
#  draft_key :string           not null
#  sequence  :bigint           not null
#
# Indexes
#
#  index_draft_sequences_on_user_id_and_draft_key  (user_id,draft_key) UNIQUE
#