discourse/app/models/draft_sequence.rb

67 lines
1.6 KiB
Ruby

# 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
#