mirror of
https://github.com/discourse/discourse.git
synced 2024-12-14 03:56:32 +08:00
67 lines
1.6 KiB
Ruby
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
|
|
#
|