Files
multica/server/pkg/db/queries/channel.sql
T
2026-08-31 17:06:17 +08:00

1207 lines
56 KiB
SQL

-- Platform-agnostic inbound channel queries (MUL-3515). These operate on
-- the channel_* tables created in migration 124. Each installation carries
-- a `channel_type` discriminator and a JSONB `config` blob for
-- platform-specific identifiers/credentials; the cross-platform columns
-- stay flat. The Go layer owns building/parsing config — these queries
-- treat it as opaque JSON except for the routing index on config->>'app_id'.
--
-- No foreign keys exist on these tables (MUL-3515 §4): the integrity the
-- old composite FKs enforced (binding workspace matches installation;
-- binding dies with membership / chat_session) is maintained in the
-- application layer via the membership check in the inbound identity step
-- and the *DeleteChannel*BindingsBy* cleanup queries below.
-- =====================
-- channel_installation
-- =====================
-- name: UpsertChannelInstallation :one
-- Install / re-install path. `config` is the opaque per-channel JSONB the
-- Go layer assembles (for feishu: app_id, app_secret_encrypted, tenant_key,
-- bot_open_id, bot_union_id, region). Re-installing the same agent on the
-- same channel_type replaces the whole config and forces status back to
-- 'active'. The conflict key is (workspace_id, agent_id, channel_type) so an
-- agent may hold one installation per channel_type (feishu + slack + ...)
-- without one install clobbering another. The WS lease is intentionally NOT
-- reset here — the inbound hub owns lease lifecycle.
INSERT INTO channel_installation (
workspace_id, agent_id, channel_type, config, installer_user_id
) VALUES (
$1, $2, $3, $4, $5
)
ON CONFLICT (workspace_id, agent_id, channel_type) DO UPDATE SET
channel_type = EXCLUDED.channel_type,
config = EXCLUDED.config,
installer_user_id = EXCLUDED.installer_user_id,
status = 'active',
installed_at = now(),
updated_at = now()
RETURNING *;
-- name: UpsertChannelInstallationByAppID :one
-- Team-keyed install / re-install for channels whose natural identity is the
-- platform workspace, not the (agent) pairing. Slack: one Slack workspace
-- (team_id, stored as config->>'app_id') maps to exactly one installation, so
-- re-connecting it — even to represent a DIFFERENT agent in the SAME Multica
-- workspace — UPDATES the existing row (moving agent_id) instead of colliding
-- with the (channel_type, app_id) unique index. Contrast UpsertChannelInstallation,
-- whose conflict key is (workspace_id, agent_id, channel_type): right for Feishu
-- (one app per agent), wrong for Slack.
--
-- The `WHERE channel_installation.workspace_id = EXCLUDED.workspace_id` fences
-- the conflict update to the SAME Multica workspace: a team already owned by a
-- DIFFERENT workspace updates no row and RETURNING is empty (pgx.ErrNoRows),
-- which the caller maps to ErrTeamOwnedByAnotherWorkspace. This is the ATOMIC
-- cross-workspace guard — a plain SELECT before the upsert cannot stop two
-- workspaces racing to OAuth the same team (both read no rows, then one inserts
-- and the other's conflict-update would silently steal it). A re-connect that
-- would move the team to an agent already holding a different Slack install in
-- the same workspace still trips the (workspace_id, agent_id, channel_type)
-- unique constraint — a genuine conflict the OAuth callback turns into a redirect.
INSERT INTO channel_installation (
workspace_id, agent_id, channel_type, config, installer_user_id
) VALUES (
$1, $2, $3, $4, $5
)
ON CONFLICT (channel_type, (config ->> 'app_id')) DO UPDATE SET
agent_id = EXCLUDED.agent_id,
config = EXCLUDED.config,
installer_user_id = EXCLUDED.installer_user_id,
status = 'active',
installed_at = now(),
updated_at = now()
WHERE channel_installation.workspace_id = EXCLUDED.workspace_id
RETURNING *;
-- name: GetChannelInstallation :one
-- Scoped by channel_type: a per-channel caller (e.g. the Feishu store)
-- must never resolve another channel's installation by guessing its UUID.
SELECT * FROM channel_installation
WHERE id = sqlc.arg('id') AND channel_type = sqlc.arg('channel_type');
-- name: GetChannelInstallationInWorkspace :one
SELECT * FROM channel_installation
WHERE id = sqlc.arg('id')
AND workspace_id = sqlc.arg('workspace_id')
AND channel_type = sqlc.arg('channel_type');
-- name: GetChannelInstallationByAppID :one
-- Inbound routing. The platform event carries only the channel's app
-- identifier (Feishu app_id); the dispatcher's installation resolver routes
-- on (channel_type, config->>'app_id'). Backed by the functional unique
-- index idx_channel_installation_type_appid.
--
-- Both params are named + explicitly typed: `config ->> 'app_id'` makes sqlc
-- attribute a bare `$2` to the JSONB `config` column (it would emit
-- `Config []byte`), so we pin the app_id arg to ::text to get AppID string.
SELECT * FROM channel_installation
WHERE channel_type = sqlc.arg('channel_type')
AND config ->> 'app_id' = sqlc.arg('app_id')::text;
-- name: GetChannelInstallationOwnerByAppID :one
-- Identifies the LIVE owner of a (channel_type, config->>'app_id') routing slot
-- so the install path can refuse a rebind with an ACCURATE message instead of the
-- old catch-all "connected to a different Multica workspace". Meant to be read
-- only after ReclaimDeadChannelInstallationByAppID has removed every DEAD owner,
-- so a returned row is a live active owner. `agent_archived` distinguishes an
-- archived (reversible) owner — its bot stays owned, recovered by unarchiving the
-- agent or disconnecting the bot — from a plain active one. The JOIN drops a row
-- whose agent no longer exists (an orphan the reclaim gate should already have
-- cleared), so a missing row (pgx.ErrNoRows) means "no live owner". The caller
-- reads agent_archived_at.Valid to tell an archived (reversible) owner apart.
SELECT ci.workspace_id, ci.agent_id, a.archived_at AS agent_archived_at
FROM channel_installation ci
JOIN agent a ON a.id = ci.agent_id
WHERE ci.channel_type = sqlc.arg('channel_type')
AND ci.config ->> 'app_id' = sqlc.arg('app_id')::text;
-- name: LockChannelInstallationAppIDSlot :exec
-- Serializes everything an install does to one (channel_type, config->>'app_id')
-- routing slot: read the current owner, decide, reclaim, upsert. Taken as the
-- first statement of the install transaction and released by COMMIT/ROLLBACK,
-- so the owner read below cannot go stale under a concurrent install or
-- reconnect — a plain read-then-write leaves a TOCTOU window in which two
-- callers both see "no live owner" and both go on to touch the slot.
--
-- Two-key form: the first key namespaces by channel so a feishu app_id and a
-- wecom bot id that hash alike do not serialize against each other. hashtext
-- collisions inside one channel only cost extra serialization, never
-- correctness. pg_advisory_xact_lock (not pg_try_) so a second caller waits
-- its turn rather than failing.
SELECT pg_advisory_xact_lock(
hashtext(sqlc.arg('channel_type')::text),
hashtext(sqlc.arg('app_id')::text)
);
-- name: GetChannelInstallationSlotOwnerByAppID :one
-- Everything the install path needs to classify the current holder of a
-- (channel_type, config->>'app_id') slot BEFORE it acts on it, in one read.
-- Distinct from GetChannelInstallationOwnerByAppID, which is the after-the-fact
-- "name the conflict" read and INNER JOINs the agent away.
--
-- Here the joins are LEFT so an ORPHAN row survives the read: with no FKs
-- (MUL-3515 §4) an installation outlives a deleted workspace or agent, and the
-- caller has to tell "orphan, reclaimable" apart from "live owner, refuse".
-- workspace_exists / agent_exists carry that; status and agent_archived_at
-- carry the rest of ReclaimDeadChannelInstallationByAppID's own definition of
-- dead, so the caller can predict what the reclaim would do without running it.
-- pgx.ErrNoRows means the slot is free.
SELECT ci.id, ci.workspace_id, ci.agent_id, ci.status,
a.archived_at AS agent_archived_at,
(a.id IS NOT NULL)::boolean AS agent_exists,
(w.id IS NOT NULL)::boolean AS workspace_exists
FROM channel_installation ci
LEFT JOIN agent a ON a.id = ci.agent_id
LEFT JOIN workspace w ON w.id = ci.workspace_id
WHERE ci.channel_type = sqlc.arg('channel_type')
AND ci.config ->> 'app_id' = sqlc.arg('app_id')::text;
-- name: ReclaimDeadChannelInstallationByAppID :one
-- Rebind cleanup gate. Frees the (channel_type, config->>'app_id') routing slot
-- so a valid new agent can (re)bind a bot whose previous owner is DEAD, and, in
-- the same statement, clears every application-owned dependent row of the removed
-- installation (channel_* has no FK/cascade, MUL-3515 §4). Returns the removed id
-- (pgx.ErrNoRows when nothing was dead — a no-op the caller treats as success).
--
-- "Dead" is exactly one of:
-- 1. a REVOKED placeholder held by ANY agent OTHER than the caller's own
-- (workspace, agent) pair. Disconnect only flips status to 'revoked' — no
-- product path ever hard-deletes the row — so a revoked row would otherwise
-- pin the bot's app_id slot forever with no self-serve recovery, even across
-- workspaces (workspace A disconnects; workspace B, which proves control by
-- holding the same app credentials, rebinds). Revoke is the owner's explicit
-- "I'm done with this bot", so any revoked row is reclaimable — only the
-- caller's OWN revoked row is spared (reactivated in place; see below).
-- 2. an ORPHAN whose owning workspace OR agent row no longer exists — the
-- workspace was deleted, or the agent was hard-deleted on runtime teardown.
-- With no FK the installation outlives its owner and keeps occupying the
-- app_id slot: the "ghost binding" that made a bot un-rebindable (#4810).
--
-- Deliberately NOT dead (the caller refuses these with an accurate conflict):
-- - the SAME agent's own revoked row (agent_id = @agent_id): the upsert
-- reactivates it in place, preserving its installation_id and every binding;
-- - a live ACTIVE owner whose agent still exists — INCLUDING an ARCHIVED agent:
-- archive is reversible, so its bot stays owned rather than being silently
-- stolen. Only a hard delete frees the slot.
--
-- The guard lives in the DELETE predicate (not a prior SELECT) so under READ
-- COMMITTED the row is re-checked at execution (EvalPlanQual): a concurrent
-- same-agent reconnect that flips the revoked row back to 'active' first makes
-- the predicate re-check fail, this deletes nothing, and no dependents are
-- touched — closing the read-then-delete TOCTOU. Dependent cleanup keys off the
-- actually-deleted id (the `dead` CTE), so it runs ONLY for a row this statement
-- removed. The (channel_type, app_id) unique index guarantees at most one match.
WITH dead AS (
DELETE FROM channel_installation ci
WHERE ci.channel_type = sqlc.arg('channel_type')
AND ci.config ->> 'app_id' = sqlc.arg('app_id')::text
AND (
(ci.status = 'revoked'
AND NOT (ci.workspace_id = sqlc.arg('workspace_id')
AND ci.agent_id = sqlc.arg('agent_id')))
OR NOT EXISTS (SELECT 1 FROM workspace w WHERE w.id = ci.workspace_id)
OR NOT EXISTS (SELECT 1 FROM agent a WHERE a.id = ci.agent_id)
)
RETURNING ci.id
),
cleared_dingtalk_group_presence AS (
DELETE FROM dingtalk_group_presence WHERE installation_id IN (SELECT id FROM dead)
),
cleared_dingtalk_bot_identity AS (
DELETE FROM dingtalk_bot_identity WHERE installation_id IN (SELECT id FROM dead)
),
cleared_dingtalk_group_routes AS (
DELETE FROM dingtalk_group_route WHERE installation_id IN (SELECT id FROM dead)
),
cleared_task_deliveries AS (
DELETE FROM channel_task_delivery WHERE installation_id IN (SELECT id FROM dead)
),
cleared_outbound_messages AS (
DELETE FROM channel_outbound_message WHERE installation_id IN (SELECT id FROM dead)
),
cleared_chat_sessions AS (
DELETE FROM channel_chat_session_binding
WHERE installation_id IN (SELECT id FROM dead)
RETURNING chat_session_id
),
cleared_chat_contexts AS (
-- A revoked installation can still own preserved Chat history and an
-- in-flight task snapshot. Remove only generations whose Chat was already
-- cascade-deleted by an earlier orphan teardown.
DELETE FROM channel_chat_context_generation AS generation
WHERE generation.chat_session_id IN (SELECT chat_session_id FROM cleared_chat_sessions)
AND NOT EXISTS (
SELECT 1 FROM chat_session AS session
WHERE session.id = generation.chat_session_id
)
),
cleared_outbound_cards AS (
-- channel_outbound_card_message is keyed by chat_session_id (no installation_id,
-- no FK), so it is reached through the just-removed chat-session bindings. On an
-- orphan reclaim the chat_session row itself is already cascade-gone, but its
-- binding survived and still carries the id — the only reliable link back.
DELETE FROM channel_outbound_card_message
WHERE chat_session_id IN (SELECT chat_session_id FROM cleared_chat_sessions)
),
cleared_binding_tokens AS (
DELETE FROM channel_binding_token
WHERE installation_id IN (SELECT id FROM dead)
),
cleared_user_bindings AS (
DELETE FROM channel_user_binding
WHERE installation_id IN (SELECT id FROM dead)
),
cleared_inbound_dedup AS (
DELETE FROM channel_inbound_message_dedup
WHERE installation_id IN (SELECT id FROM dead)
),
detached_audit AS (
-- Reclaim keeps the DETACH semantics: the workspace still exists, so a
-- NULL-installation audit row stays meaningful for operator triage. The hard-
-- delete paths (DeleteWorkspace / runtime teardown) purge audit outright.
UPDATE channel_inbound_audit SET installation_id = NULL
WHERE installation_id IN (SELECT id FROM dead)
)
SELECT id FROM dead;
-- name: DeleteChannelInstallationsBySystemRuntimeAgents :exec
-- Application-layer replacement for the (deliberately absent, MUL-3515 §4)
-- workspace/agent ON DELETE CASCADE: on runtime teardown, before the system
-- agents are hard-deleted, remove every channel installation they own — plus all
-- of each installation's dependent rows — so no orphaned installation keeps
-- occupying its bot's (channel_type, app_id) routing slot after its agent is gone
-- (#4810). MUST run in the same tx as, and BEFORE, DeleteSystemAgentsByRuntime.
-- Mirrors the agent hard-delete predicate (runtime_id, kind = 'system') exactly.
--
-- Scoped to kind = 'system' since MUL-5559: a user agent now survives its
-- runtime's deletion as an unbound agent, so tearing down its installations
-- here would take a working bot away from an agent that is still there.
WITH system_agents AS (
SELECT system_agent.id FROM agent AS system_agent
WHERE system_agent.runtime_id = sqlc.arg('runtime_id') AND system_agent.kind = 'system'
),
doomed_sessions AS (
SELECT id FROM chat_session
WHERE agent_id IN (SELECT id FROM system_agents)
),
doomed AS (
SELECT id FROM channel_installation
WHERE agent_id IN (SELECT id FROM system_agents)
),
cleared_dingtalk_group_presence AS (
DELETE FROM dingtalk_group_presence WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_dingtalk_bot_identity AS (
DELETE FROM dingtalk_bot_identity WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_dingtalk_group_routes AS (
DELETE FROM dingtalk_group_route WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_task_deliveries AS (
DELETE FROM channel_task_delivery WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_outbound_messages AS (
DELETE FROM channel_outbound_message WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_chat_sessions AS (
DELETE FROM channel_chat_session_binding WHERE installation_id IN (SELECT id FROM doomed)
RETURNING chat_session_id
),
cleared_chat_contexts AS (
DELETE FROM channel_chat_context_generation
WHERE chat_session_id IN (SELECT chat_session_id FROM cleared_chat_sessions)
OR chat_session_id IN (SELECT id FROM doomed_sessions)
),
cleared_outbound_cards AS (
-- Reach channel_outbound_card_message (keyed by chat_session_id, no FK)
-- through the just-removed chat-session bindings, same as the reclaim path.
DELETE FROM channel_outbound_card_message
WHERE chat_session_id IN (SELECT chat_session_id FROM cleared_chat_sessions)
),
cleared_binding_tokens AS (
DELETE FROM channel_binding_token WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_user_bindings AS (
DELETE FROM channel_user_binding WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_inbound_dedup AS (
DELETE FROM channel_inbound_message_dedup WHERE installation_id IN (SELECT id FROM doomed)
),
cleared_audit AS (
-- Hard delete: purge audit rows rather than detaching them into permanently
-- unattributable NULL rows (channel_inbound_audit has no workspace_id / reaper).
DELETE FROM channel_inbound_audit WHERE installation_id IN (SELECT id FROM doomed)
)
DELETE FROM channel_installation WHERE id IN (SELECT id FROM doomed);
-- name: ListChannelInstallationsByWorkspace :many
-- Scoped by channel_type so a per-channel management surface (e.g. the Lark
-- installation list) only ever sees its own platform's installations.
SELECT * FROM channel_installation
WHERE workspace_id = sqlc.arg('workspace_id')
AND channel_type = sqlc.arg('channel_type')
ORDER BY created_at ASC;
-- name: ListActiveChannelInstallations :many
-- Boot path for a per-channel-type inbound hub: every active installation of
-- the given channel_type, so a hub claims leases and opens connections only
-- for its own platform and never supervises another channel's installation.
--
-- The JOINs require the owning workspace and agent rows to still exist.
-- channel_installation has no FK (MUL-3515 §4), so unlike the old
-- lark_installation (which cascaded away on workspace/agent deletion) an
-- installation can be orphaned when its workspace is deleted or its agent is
-- hard-deleted (e.g. runtime teardown). Without this guard the hub would keep
-- opening a WebSocket for a bot whose workspace/agent is gone. The JOIN matches
-- the old ON DELETE CASCADE semantics: it filters on row existence, not agent
-- archival, so an archived-but-present agent's installation is still listed.
SELECT ci.* FROM channel_installation ci
JOIN workspace w ON w.id = ci.workspace_id
JOIN agent a ON a.id = ci.agent_id
WHERE ci.status = 'active'
AND ci.channel_type = sqlc.arg('channel_type')
ORDER BY ci.created_at ASC;
-- name: ListAllActiveChannelInstallations :many
-- Boot path for the channel-agnostic engine Supervisor (MUL-3620): every
-- active installation across ALL channel types, so one Supervisor drives every
-- platform's connections rather than a per-platform hub. This is the de-
-- hardcoded counterpart of ListActiveChannelInstallations — the Supervisor
-- routes each row to its registered channel.Factory by channel_type, so it
-- never needs to know which platforms exist. Same orphan guard as the per-type
-- query: the workspace + agent JOINs drop installations whose owning rows are
-- gone (channel_installation has no FK, MUL-3515 §4), matching the old ON
-- DELETE CASCADE semantics (row existence, not agent archival).
SELECT ci.* FROM channel_installation ci
JOIN workspace w ON w.id = ci.workspace_id
JOIN agent a ON a.id = ci.agent_id
WHERE ci.status = 'active'
ORDER BY ci.created_at ASC;
-- name: SetChannelInstallationStatus :exec
UPDATE channel_installation
SET status = $2, updated_at = now()
WHERE id = $1;
-- name: SetChannelInstallationConfig :exec
-- Replaces the whole config blob for one installation. Used by the
-- operator backfills (e.g. setting a freshly-fetched bot_union_id) that
-- read-modify-write the JSON in Go and persist it back atomically by id.
UPDATE channel_installation
SET config = $2, updated_at = now()
WHERE id = $1;
-- name: BackfillChannelInstallationRegionToFeishuLark :execrows
-- Operator repair, feishu-only: flip every feishu installation still
-- carrying region='feishu' to 'lark'. Called only on deployments whose
-- legacy global base-URL override pointed at Lark international. Idempotent.
UPDATE channel_installation
SET config = jsonb_set(config, '{region}', '"lark"'),
updated_at = now()
WHERE channel_type = 'feishu'
AND config ->> 'region' = 'feishu';
-- name: AcquireChannelWSLease :one
-- Atomically claims the WebSocket lease. CAS predicate accepts when no
-- holder exists, the holder expired, or the holder is us (renewal).
UPDATE channel_installation
SET ws_lease_token = sqlc.arg('new_token'),
ws_lease_expires_at = sqlc.arg('new_expires_at'),
updated_at = now()
WHERE id = sqlc.arg('id')
AND status = 'active'
AND (
ws_lease_token IS NULL
OR ws_lease_expires_at < now()
OR ws_lease_token = sqlc.arg('new_token')
)
RETURNING *;
-- name: ReleaseChannelWSLease :exec
-- Drops the lease iff we are still the holder.
UPDATE channel_installation
SET ws_lease_token = NULL,
ws_lease_expires_at = NULL,
updated_at = now()
WHERE id = $1
AND ws_lease_token = sqlc.arg('current_token');
-- =====================
-- channel_user_binding
-- =====================
-- name: CreateChannelUserBinding :one
-- Records that a platform user id (per-installation; Feishu open_id) maps
-- to a Multica user. The old composite member-FK is gone, so this no
-- longer fails when the redeemer is not a workspace member — the caller
-- (BindingTokenService.RedeemAndBind) validates membership explicitly
-- before calling. ON CONFLICT DO UPDATE is still gated on multica_user_id
-- matching, so a second redeemer cannot steal an already-bound user id;
-- a cross-user conflict updates zero rows and the caller maps that to
-- ErrBindingAlreadyAssigned. config carries secondary identity (union_id).
INSERT INTO channel_user_binding (
workspace_id, multica_user_id, installation_id,
channel_type, channel_user_id, config
) VALUES (
$1, $2, $3, $4, $5, $6
)
ON CONFLICT (installation_id, channel_user_id) DO UPDATE SET
-- jsonb_strip_nulls(EXCLUDED.config) preserves the old lark semantics
-- `union_id = COALESCE(EXCLUDED.union_id, lark_user_binding.union_id)`:
-- a re-bind that carries `{"union_id": null}` (or omits the key) must NOT
-- erase a union_id we already captured. Only non-null incoming keys win.
config = channel_user_binding.config || jsonb_strip_nulls(EXCLUDED.config),
bound_at = now()
WHERE channel_user_binding.multica_user_id = EXCLUDED.multica_user_id
RETURNING *;
-- name: GetChannelUserBindingByUserID :one
-- The inbound identity lookup: does this platform user id map to a Multica
-- user for this installation? With the member-FK removed, a row's
-- existence no longer proves current workspace membership — the dispatcher
-- re-checks membership after this lookup.
SELECT * FROM channel_user_binding
WHERE installation_id = $1 AND channel_user_id = $2;
-- name: FindChannelBindingForMember :one
-- Outbound notification lookup: given a Multica member and a channel_type,
-- return the (installation, channel_user_id) that outbound push should
-- target. The wecom smart-bot inbox-notification path uses this to decide
-- whether to deliver via the bot at all — no row means "unbound member,
-- fall back to the legacy path (TOF/RTX)".
--
-- If a member has bound multiple installations of the same channel_type in
-- one workspace (multi-bot org), the most-recently-bound wins — matches
-- FindReusableChannelUserBinding's tiebreak so the two lookups agree.
SELECT b.* FROM channel_user_binding b
JOIN channel_installation ci ON ci.id = b.installation_id
WHERE b.workspace_id = sqlc.arg('workspace_id')
AND b.multica_user_id = sqlc.arg('multica_user_id')
AND b.channel_type = sqlc.arg('channel_type')
AND ci.status = 'active'
ORDER BY b.bound_at DESC
LIMIT 1;
-- name: FindReusableChannelUserBinding :one
-- Cross-installation account-link reuse (MUL-3911). When a platform user
-- messages an installation they have NOT linked, but the SAME user id is already
-- bound to ANOTHER installation in the SAME Multica workspace + SAME Slack team,
-- the inbound identity step reuses that link instead of re-prompting. Slack user
-- ids are stable within a team, so an identical channel_user_id denotes the same
-- human across that team's apps. The match is fenced to one workspace AND one
-- team (installation config->>'team_id'): a Slack team can be connected to two
-- different Multica workspaces, and a user may hold different Multica accounts in
-- each, so reuse must cross neither boundary. Most-recently-bound wins. The
-- caller re-checks membership and materializes a fresh per-installation binding.
--
-- team_id is pinned ::text so sqlc types the arg as a string instead of
-- attributing the bare param to the JSONB config column (mirrors
-- GetChannelInstallationByAppID's app_id cast).
SELECT b.* FROM channel_user_binding b
JOIN channel_installation ci ON ci.id = b.installation_id
WHERE b.workspace_id = sqlc.arg('workspace_id')
AND b.channel_type = sqlc.arg('channel_type')
AND b.channel_user_id = sqlc.arg('channel_user_id')
AND ci.config ->> 'team_id' = sqlc.arg('team_id')::text
ORDER BY b.bound_at DESC
LIMIT 1;
-- name: DeleteChannelUserBindingsByWorkspaceMember :exec
-- Application-layer integrity (replaces the old member-FK ON DELETE
-- CASCADE): prune every binding for a user who has been removed from a
-- workspace, across all installations in that workspace.
DELETE FROM channel_user_binding
WHERE workspace_id = $1 AND multica_user_id = $2;
-- name: DeleteChannelUserBindingsByInstallation :exec
-- Application-layer integrity (schema has no FK/cascade, MUL-3515 §4): drop
-- every member account link for an installation that is being hard-deleted.
-- Rebinding a Feishu bot to a DIFFERENT agent starts a fresh installation, so
-- old links do not follow — a different agent is a distinct connection and
-- members re-establish their link on first contact. The rows could never be
-- reused anyway (every Feishu identity lookup is installation_id-scoped, and
-- FindReusableChannelUserBinding is Slack-only), so removing them just keeps
-- dead rows from accumulating.
DELETE FROM channel_user_binding
WHERE installation_id = $1;
-- =====================
-- channel_chat_session_binding
-- =====================
-- name: CreateChannelChatSessionBinding :one
-- channel_chat_id is the session-isolation key (one chat_session per
-- (installation_id, channel_chat_id)): Feishu passes the chat id; Slack passes
-- a stable key that, for channels, includes the thread root so each @bot thread
-- is its own session. config carries any platform-specific outbound routing the
-- key alone does not (e.g. Slack's real channel_id when the key is composite);
-- it is opaque to the shared session service.
WITH next_route AS (
SELECT COALESCE(MAX(route_revision) + 1, 1)::bigint AS route_revision
FROM channel_chat_session_binding AS existing
WHERE existing.installation_id = $2 AND existing.channel_chat_id = $4
), binding AS (
INSERT INTO channel_chat_session_binding (
chat_session_id, installation_id, channel_type, channel_chat_id, chat_type,
config, route_revision
)
SELECT $1, $2, $3, $4, $5, $6, next_route.route_revision
FROM next_route
RETURNING *
), generation AS (
INSERT INTO channel_chat_context_generation (chat_session_id, revision)
SELECT chat_session_id, context_revision FROM binding
)
SELECT * FROM binding;
-- name: CreateChannelChatSessionBindingGeneration :one
WITH next_route AS (
SELECT GREATEST(
@route_revision::bigint,
COALESCE(MAX(route_revision) + 1, 1)::bigint
) AS route_revision
FROM channel_chat_session_binding AS existing
WHERE existing.installation_id = @installation_id
AND existing.channel_chat_id = @channel_chat_id
), binding AS (
INSERT INTO channel_chat_session_binding (
chat_session_id, installation_id, channel_type, channel_chat_id, chat_type,
config, route_revision, history_start_message_id, history_boundary_pending
) SELECT
@chat_session_id, @installation_id, @channel_type, @channel_chat_id, @chat_type,
@config, next_route.route_revision, sqlc.narg('history_start_message_id'), @history_boundary_pending
FROM next_route
RETURNING *
), generation AS (
INSERT INTO channel_chat_context_generation (
chat_session_id, revision, history_start_message_id, history_boundary_pending
)
SELECT chat_session_id, context_revision, history_start_message_id, history_boundary_pending
FROM binding
)
SELECT * FROM binding;
-- name: GetChannelChatSessionBinding :one
-- Lookup-by-channel-chat: the inbound dispatcher finds the existing
-- chat_session before deciding whether to create one.
SELECT * FROM channel_chat_session_binding
WHERE installation_id = $1 AND channel_chat_id = $2 AND retired_at IS NULL;
-- name: LockCurrentChannelChatSessionBinding :one
SELECT * FROM channel_chat_session_binding
WHERE installation_id = $1 AND channel_chat_id = $2 AND retired_at IS NULL
FOR UPDATE;
-- name: LockCurrentChannelChatSessionBindingBySession :one
-- Shared append fence. A message may have resolved a Chat immediately before
-- /new retired it; locking only chat_session would then let the stale append
-- land after the route switch. Requiring the binding to remain current makes
-- every adapter retry against the winning generation.
SELECT * FROM channel_chat_session_binding
WHERE chat_session_id = $1 AND retired_at IS NULL
FOR UPDATE;
-- name: RetireChannelChatSessionBinding :one
UPDATE channel_chat_session_binding
SET retired_at = now(),
history_end_message_id = sqlc.narg('history_end_message_id')
WHERE id = @id AND retired_at IS NULL
RETURNING *;
-- name: GetChannelChatSessionBindingBySession :one
-- Reverse lookup for the outbound patcher: given a chat_session_id, find
-- its channel binding to know which (installation, chat_id) to send to.
-- Scoped by channel_type so a future non-Feishu binding on the same
-- chat_session is never treated as a Feishu reply target.
SELECT * FROM channel_chat_session_binding
WHERE chat_session_id = sqlc.arg('chat_session_id')
AND channel_type = sqlc.arg('channel_type');
-- name: GetChannelChatSessionBindingBySessionAny :one
-- Channel-agnostic reverse lookup: which channel, if any, is behind this
-- chat_session? UNIQUE (chat_session_id) guarantees at most one row, so a
-- caller that only needs to READ the binding never has to name the channel it
-- is hoping for — and therefore cannot go blind on a channel added later.
-- The channel_type-scoped variant above stays for the outbound senders, which
-- are per-platform by construction and must not deliver into a foreign one.
SELECT * FROM channel_chat_session_binding
WHERE chat_session_id = $1;
-- name: ListChannelChatSessionBindingsBySessions :many
-- Batch projection for Chat list responses. Historical bindings are retained,
-- so retired Chats keep their channel source while only the active generation
-- reports itself as the current route.
SELECT * FROM channel_chat_session_binding
WHERE chat_session_id = ANY(@chat_session_ids::uuid[]);
-- name: UpdateChannelChatSessionBindingReplyTarget :exec
-- Records the most recent inbound trigger message + thread so the decoupled
-- outbound patcher can thread its reply back into the originating topic.
WITH current_route AS (
SELECT current_binding.*
FROM channel_chat_session_binding AS current_binding
WHERE current_binding.chat_session_id = sqlc.arg('reply_chat_session_id')
FOR UPDATE OF current_binding
), closed_previous AS (
UPDATE channel_chat_session_binding AS previous
SET history_end_message_id = sqlc.narg('last_message_id')
FROM current_route AS current
WHERE current.history_boundary_pending
AND sqlc.narg('last_message_id')::text IS NOT NULL
AND previous.installation_id = current.installation_id
AND previous.channel_chat_id = current.channel_chat_id
-- Multiple native Slack /new commands can rotate through empty
-- generations before another public message supplies a platform cursor.
-- That cursor closes every still-open retired generation.
AND previous.route_revision < current.route_revision
AND previous.retired_at IS NOT NULL
AND previous.history_end_message_id IS NULL
)
UPDATE channel_chat_session_binding AS binding
SET last_message_id = sqlc.narg('last_message_id'),
last_thread_id = sqlc.narg('last_thread_id'),
history_start_message_id = CASE
WHEN binding.history_boundary_pending
AND sqlc.narg('last_message_id')::text IS NOT NULL
THEN sqlc.narg('last_message_id')
ELSE binding.history_start_message_id
END,
history_boundary_pending = CASE
WHEN sqlc.narg('last_message_id')::text IS NOT NULL THEN FALSE
ELSE binding.history_boundary_pending
END
FROM current_route
WHERE binding.id = current_route.id;
-- name: MarkChannelChatSessionPendingFresh :one
-- Persists a channel `/clear` intent until the next chat task is successfully
-- created. RETURNING makes a missing binding an error instead of silently
-- acknowledging a fresh start that was never stored.
UPDATE channel_chat_session_binding
SET pending_fresh = TRUE
WHERE chat_session_id = $1 AND retired_at IS NULL
RETURNING pending_fresh;
-- name: LockChannelChatSessionPendingFresh :one
-- EnqueueChatTask reads this under the same row lock and transaction that
-- creates the task. A concurrent `/clear` therefore lands either before this
-- task and is consumed by it, or after this task and remains for the next one.
SELECT pending_fresh FROM channel_chat_session_binding
WHERE chat_session_id = $1
FOR UPDATE;
-- name: LockChannelChatSessionBindingForContext :one
-- Context mutations acquire this row after chat_session and before any
-- channel_chat_context_generation row. Keeping the statements separate makes
-- the lock order explicit and identical for append, /clear, and task enqueue.
SELECT * FROM channel_chat_session_binding
WHERE chat_session_id = $1
FOR UPDATE;
-- name: ClearChannelChatSessionPendingFresh :exec
UPDATE channel_chat_session_binding
SET pending_fresh = FALSE
WHERE chat_session_id = $1;
-- name: ClearChannelChatSessionPendingFreshForRevision :exec
UPDATE channel_chat_session_binding
SET pending_fresh = FALSE
WHERE chat_session_id = @chat_session_id
AND context_revision = @revision;
-- name: AdvanceChannelChatContextGeneration :one
-- Opens a new agent-visible context while retaining the same Multica Chat.
-- The triggering platform message is the exclusive end of the old generation
-- and, when it has a body, the inclusive start of the new one.
WITH closed AS (
UPDATE channel_chat_context_generation AS generation
SET history_end_message_id = sqlc.narg('history_boundary_message_id')
WHERE generation.chat_session_id = @chat_session_id
AND generation.revision = @current_revision
), advanced AS (
UPDATE channel_chat_session_binding AS binding
SET context_revision = binding.context_revision + 1,
pending_fresh = TRUE
WHERE binding.chat_session_id = @chat_session_id
AND binding.context_revision = @current_revision
RETURNING binding.*
), opened AS (
INSERT INTO channel_chat_context_generation (
chat_session_id, revision, history_start_message_id,
history_boundary_pending, pending_fresh
)
SELECT chat_session_id, context_revision,
CASE WHEN @has_message_body::boolean THEN sqlc.narg('history_boundary_message_id') END,
NOT @has_message_body::boolean,
TRUE
FROM advanced
RETURNING *
)
SELECT * FROM opened;
-- name: LockChannelChatContextGenerationByRevision :one
SELECT * FROM channel_chat_context_generation
WHERE chat_session_id = @chat_session_id
AND revision = @revision
FOR UPDATE;
-- name: ResolveChannelChatContextHistoryStart :exec
UPDATE channel_chat_context_generation
SET history_start_message_id = @history_start_message_id,
history_boundary_pending = FALSE
WHERE chat_session_id = @chat_session_id
AND revision = @revision
AND history_boundary_pending;
-- name: SetChannelChatContextInitiator :one
-- Snapshots the latest authenticated sender whose durable input belongs to a
-- generation. Crash recovery must use this identity rather than the sender of
-- a later generation that happened to re-arm the lost debounce timer.
UPDATE channel_chat_context_generation
SET initiator_user_id = @initiator_user_id
WHERE chat_session_id = @chat_session_id
AND revision = @revision
RETURNING initiator_user_id;
-- name: ClearChannelChatContextPendingFresh :exec
UPDATE channel_chat_context_generation
SET pending_fresh = FALSE
WHERE chat_session_id = @chat_session_id
AND revision = @revision;
-- name: GetChannelChatContextGeneration :one
SELECT * FROM channel_chat_context_generation
WHERE chat_session_id = @chat_session_id
AND revision = @revision;
-- name: DeleteChannelChatSessionBindingBySession :exec
-- Application-layer integrity (replaces the old chat_session-FK ON DELETE
-- CASCADE): drop the binding, its immutable delivery/history dependents,
-- and the Chat's context generations.
WITH target AS (
SELECT binding.id FROM channel_chat_session_binding AS binding
WHERE binding.chat_session_id = sqlc.arg('chat_session_id')
), cleared_deliveries AS (
DELETE FROM channel_task_delivery AS delivery WHERE delivery.binding_id IN (SELECT id FROM target)
), cleared_outbound AS (
DELETE FROM channel_outbound_message AS outbound WHERE outbound.binding_id IN (SELECT id FROM target)
), deleted_binding AS (
DELETE FROM channel_chat_session_binding AS binding WHERE binding.id IN (SELECT id FROM target)
)
DELETE FROM channel_chat_context_generation AS generation
WHERE generation.chat_session_id = sqlc.arg('chat_session_id');
-- name: DeleteChannelChatSessionBindingsByInstallation :exec
-- Retire every chat-session binding for an installation. Used when an
-- installation is re-pointed to a different agent (Slack re-connect): each
-- existing chat_session is permanently tied to the agent it was created under,
-- so reusing it would keep routing the conversation to the OLD agent. Dropping
-- the bindings forces the next inbound message to create a fresh session under
-- the new agent. The chat_session rows are preserved for history; only the
-- channel binding is removed. Context generations belong to the preserved Chat,
-- not to the retired binding: in-flight task history snapshots may still read
-- them after this statement commits.
WITH cleared_deliveries AS (
DELETE FROM channel_task_delivery AS delivery
WHERE delivery.installation_id = sqlc.arg('installation_id')
AND delivery.channel_type = sqlc.arg('channel_type')
), cleared_outbound AS (
DELETE FROM channel_outbound_message AS outbound
WHERE outbound.installation_id = sqlc.arg('installation_id')
AND outbound.channel_type = sqlc.arg('channel_type')
)
DELETE FROM channel_chat_session_binding AS binding
WHERE binding.installation_id = sqlc.arg('installation_id')
AND binding.channel_type = sqlc.arg('channel_type');
-- =====================
-- channel_task_delivery
-- =====================
-- name: CreateChannelTaskDeliveryFromSession :one
INSERT INTO channel_task_delivery (
task_id, binding_id, installation_id, channel_type, channel_chat_id, chat_type,
channel_message_id, channel_thread_id, route_revision, config
)
SELECT
@task_id, binding.id, binding.installation_id, binding.channel_type,
binding.channel_chat_id, binding.chat_type, binding.last_message_id, binding.last_thread_id,
binding.route_revision, binding.config
FROM channel_chat_session_binding AS binding
WHERE binding.chat_session_id = @chat_session_id
RETURNING *;
-- name: GetChannelTaskDelivery :one
SELECT * FROM channel_task_delivery WHERE task_id = $1;
-- name: CopyChannelTaskDelivery :exec
INSERT INTO channel_task_delivery (
task_id, binding_id, installation_id, channel_type, channel_chat_id, chat_type,
channel_message_id, channel_thread_id, route_revision, config
)
SELECT
@child_task_id, delivery.binding_id, delivery.installation_id, delivery.channel_type, delivery.channel_chat_id, delivery.chat_type,
delivery.channel_message_id, delivery.channel_thread_id, delivery.route_revision, delivery.config
FROM channel_task_delivery AS delivery
WHERE delivery.task_id = @parent_task_id;
-- =====================
-- channel_outbound_message
-- =====================
-- name: RecordChannelOutboundMessage :exec
INSERT INTO channel_outbound_message (
installation_id, channel_type, channel_message_id, binding_id,
route_revision, task_id, outbound_kind
) VALUES (
sqlc.arg('outbound_installation_id'), sqlc.arg('outbound_channel_type'),
sqlc.arg('outbound_message_id'), sqlc.arg('outbound_binding_id'),
sqlc.arg('outbound_route_revision'), sqlc.narg('outbound_task_id'),
sqlc.arg('outbound_kind')
)
ON CONFLICT (installation_id, channel_message_id) DO NOTHING;
-- name: ListChannelOutboundMessageIDsForBinding :many
SELECT outbound.channel_message_id FROM channel_outbound_message AS outbound
WHERE outbound.binding_id = $1
ORDER BY outbound.created_at ASC;
-- name: ListChannelOutboundMessagesByIDs :many
SELECT outbound.* FROM channel_outbound_message AS outbound
WHERE outbound.installation_id = @installation_id
AND outbound.channel_message_id = ANY(@channel_message_ids::text[]);
-- =====================
-- channel_inbound_message_dedup
-- =====================
-- name: ClaimChannelInboundDedup :one
-- Two-phase idempotency gate with owner fencing. Returns the row when a
-- claim is acquired (fresh insert, or stale-reclaim of an in-flight claim
-- older than 60s); returns no rows when terminal (processed) or actively
-- in-flight. Every claim mints a fresh claim_token; Mark/Release are
-- fenced on it. See the table comment in migration 124 / the lark
-- predecessor for the full invariant set.
INSERT INTO channel_inbound_message_dedup (installation_id, message_id, claim_token)
VALUES ($1, $2, gen_random_uuid())
ON CONFLICT (installation_id, message_id) DO UPDATE
SET received_at = now(),
claim_token = gen_random_uuid()
WHERE channel_inbound_message_dedup.processed_at IS NULL
AND channel_inbound_message_dedup.received_at < now() - INTERVAL '60 seconds'
RETURNING installation_id, message_id, received_at, processed_at, claim_token;
-- name: MarkChannelInboundDedupProcessed :execrows
-- Locks a claim in as permanently processed after a durable outcome.
-- Invoked inside the chat_message tx (via qtx) on the ingest path so the
-- durable write and the Mark commit atomically. Token mismatch returns
-- zero rows (a reclaim happened); the caller rolls back its in-tx write.
UPDATE channel_inbound_message_dedup
SET processed_at = now()
WHERE installation_id = $1
AND message_id = $2
AND claim_token = $3
AND processed_at IS NULL;
-- name: ReleaseChannelInboundDedup :execrows
-- Releases an in-flight claim when an infra error occurred before any
-- durable side effect, so a retry can re-acquire immediately. Fenced on
-- processed_at IS NULL and claim_token.
DELETE FROM channel_inbound_message_dedup
WHERE installation_id = $1
AND message_id = $2
AND claim_token = $3
AND processed_at IS NULL;
-- name: PurgeChannelInboundDedup :exec
-- Vacuum job: remove dedup rows older than the supplied cutoff (e.g. 24h).
DELETE FROM channel_inbound_message_dedup
WHERE received_at < $1;
-- =====================
-- channel_inbound_audit
-- =====================
-- name: RecordChannelInboundDrop :exec
-- The only write path for dropped events. Deliberately carries no body
-- column — only routing / identity / drop_reason / timestamp.
INSERT INTO channel_inbound_audit (
installation_id, channel_type, channel_chat_id, event_type,
channel_event_id, channel_message_id, drop_reason, id
) VALUES (
sqlc.narg('installation_id'),
$1,
sqlc.narg('channel_chat_id'),
$2,
sqlc.narg('channel_event_id'),
sqlc.narg('channel_message_id'),
$3,
COALESCE(sqlc.narg('id')::uuid, gen_random_uuid())
);
-- name: ListChannelInboundAuditByInstallation :many
SELECT * FROM channel_inbound_audit
WHERE installation_id = $1
ORDER BY received_at DESC
LIMIT $2 OFFSET $3;
-- name: NullChannelInboundAuditInstallationID :exec
-- Application-layer stand-in for the old ON DELETE SET NULL (MUL-3515 §4,
-- migration 124 keeps installation_id nullable for exactly this): before an
-- installation row is hard-deleted, detach its inbound-audit rows by NULLing
-- installation_id. The drop-audit history is preserved (channel_type,
-- chat/message ids, drop_reason stay) without a dangling reference to a
-- removed installation.
UPDATE channel_inbound_audit
SET installation_id = NULL
WHERE installation_id = $1;
-- =====================
-- channel_outbound_card_message
-- =====================
-- name: CreateChannelOutboundCardMessage :one
INSERT INTO channel_outbound_card_message (
chat_session_id, task_id, channel_type, channel_chat_id,
channel_card_message_id, status
) VALUES (
$1, sqlc.narg('task_id'), $2, $3, $4, $5
)
RETURNING *;
-- name: GetChannelOutboundCardByTask :one
-- The partial unique index on (task_id) WHERE task_id IS NOT NULL
-- guarantees at most one row. Scoped by channel_type so a future non-Feishu
-- card for the same task is not patched as a Feishu card.
SELECT * FROM channel_outbound_card_message
WHERE task_id = sqlc.arg('task_id')
AND channel_type = sqlc.arg('channel_type');
-- name: UpdateChannelOutboundCardStatus :exec
UPDATE channel_outbound_card_message
SET status = $2,
last_patched_at = now()
WHERE id = $1;
-- name: DeleteChannelOutboundCardMessagesBySession :exec
-- Application-layer integrity (channel_* has no FK/cascade, MUL-3515 §4): drop the
-- outbound card-message rows for a chat_session being deleted. They are keyed by
-- chat_session_id with no FK and no reaper, so the standalone chat-session delete
-- path must prune them here alongside DeleteChannelChatSessionBindingBySession —
-- otherwise deleting a chat session leaves them as permanent orphans (Elon's
-- follow-up on #4810; the workspace/agent/reclaim sweeps already cover their
-- paths). A card that survived its session could only mis-route a later patch.
DELETE FROM channel_outbound_card_message
WHERE chat_session_id = $1;
-- =====================
-- channel_binding_token
-- =====================
-- name: CreateChannelBindingToken :one
-- Mints a single-use binding token for an unbound platform user. TTL cap
-- (15 min) enforced by the table CHECK in lockstep with
-- channel.BindingTokenTTL. Clamp against the database clock so small clock
-- skew between an app node and Postgres cannot reject an otherwise valid
-- 15-minute token. The HASH is stored, never the raw token.
INSERT INTO channel_binding_token (
token_hash, workspace_id, installation_id, channel_type,
channel_user_id, expires_at
) VALUES (
$1, $2, $3, $4, $5,
LEAST(sqlc.arg('expires_at')::timestamptz, now() + INTERVAL '15 minutes')
)
RETURNING *;
-- name: FindLiveChannelBindingToken :one
-- Mint guard: the newest token for this platform user that is still
-- unconsumed, unexpired, and recent enough that the link already sitting in
-- their chat is the one to point back at. Without it every message from an
-- unbound user mints another row, so a user who keeps typing at a bot they
-- have not linked yet writes one row per message. This narrows that to
-- roughly one row per window; it is not a hard guarantee, since the caller
-- runs this and the insert as two statements.
--
-- `mint_interval` is the caller's throttle window (see
-- wecom.BindingTokenMintInterval). It is subtracted from now() rather than
-- passed in as an absolute cutoff so the whole window is measured on the
-- database clock: created_at is stamped by the column default, and comparing
-- it against an application-side timestamp would let clock skew between the
-- two stretch or shrink the window. The consumed_at / expires_at predicates
-- keep an already-redeemed or stale token from suppressing a mint the user
-- actually needs.
--
-- idx_channel_binding_token_installation covers the installation_id prefix;
-- the rest is a filter over that installation's live tokens, which is a small
-- set because nothing here outlives the 15-minute TTL.
SELECT * FROM channel_binding_token
WHERE installation_id = $1
AND channel_type = $2
AND channel_user_id = $3
AND consumed_at IS NULL
AND expires_at > now()
AND created_at >= now() - sqlc.arg('mint_interval')::interval
ORDER BY created_at DESC
LIMIT 1;
-- name: ConsumeChannelBindingToken :one
-- Atomic redemption: returns the row only if the hash exists, is
-- unconsumed, and unexpired. Two simultaneous redemptions cannot both win.
UPDATE channel_binding_token
SET consumed_at = now()
WHERE token_hash = $1
AND consumed_at IS NULL
AND expires_at > now()
RETURNING *;
-- name: PurgeExpiredChannelBindingTokens :exec
DELETE FROM channel_binding_token
WHERE expires_at < $1;
-- name: DeleteChannelBindingTokensByInstallation :exec
-- Application-layer integrity (schema has no FK/cascade, MUL-3515 §4): drop
-- every pending binding token for an installation that is being hard-deleted.
-- A token stays redeemable for up to 15 min; without this a user who clicks a
-- still-unexpired bind link right after the bot was rebound to another agent
-- would consume the token and get a "bound" result written against a deleted
-- installation — a link that never actually reaches the live bot.
DELETE FROM channel_binding_token
WHERE installation_id = $1;
-- =====================
-- channel_media_pending_object (media intent ledger)
-- =====================
-- name: RecordChannelMediaPendingObject :one
-- Records upload intent BEFORE the PUT. A redelivered attempt refreshes the
-- settle window, but only while the row is still 'pending' — a key the
-- reconciler owns ('deleting') must never be resurrected — and only within
-- the SAME workspace: a cross-workspace key collision (impossible via the
-- derived key, but tenancy must never trust the key string) updates nothing
-- and returns no row, so the caller skips the upload entirely.
INSERT INTO channel_media_pending_object (
storage_key, workspace_id, chat_message_id, storage_url, installation_id
)
VALUES ($1, $2, $3, $4, sqlc.narg(installation_id))
ON CONFLICT (storage_key) DO UPDATE
SET created_at = now(), next_attempt_at = now(),
chat_message_id = EXCLUDED.chat_message_id,
storage_url = EXCLUDED.storage_url
WHERE channel_media_pending_object.state = 'pending'
AND channel_media_pending_object.workspace_id = EXCLUDED.workspace_id
RETURNING storage_key;
-- name: ClaimChannelMediaPendingObjectsForBind :many
-- Runs inside the attachment-insert transaction: commit landed ⇔ the intents
-- are gone, atomically, so an ambiguous COMMIT never needs adjudication. Only
-- 'pending' rows can be claimed — a key the reconciler moved to 'deleting'
-- is NOT returned, and the caller must skip attaching that object (the
-- placeholder stays; the reconciler will delete the object).
DELETE FROM channel_media_pending_object
WHERE storage_key = ANY(@storage_keys::text[])
AND workspace_id = @workspace_id
AND state = 'pending'
RETURNING storage_key;
-- name: ClaimNextChannelMediaPendingObjectForReconcile :one
-- Short-transaction claim of ONE due row, taken immediately before that row is
-- settled. Claiming a whole batch up front made the claim a promise the sweep
-- might not keep: a tail row sat in 'deleting' with attempt already bumped
-- while earlier rows ran, and could expire and be reclaimed before its own
-- DELETE was ever tried, inflating attempt/backoff for work that never
-- happened. One row per claim means attempt counts attempts.
--
-- Due means (a) 'pending' rows older than the settle delay — an operational
-- buffer only; correctness comes from the state flip, after which a bind can
-- never succeed on the key — or (b) 'deleting' rows whose lease expired (a
-- crashed or failed worker) — or (c) tombstones: the object was deleted, but a
-- PUT the client abandoned may still materialize it afterwards, so each due
-- tombstone gets another idempotent delete before the row is finally dropped.
-- FOR UPDATE SKIP LOCKED keeps replicas off each other's row; the
-- object-storage DELETE happens outside any transaction, gated by the lease
-- token. No row (ErrNoRows) means nothing is due — the sweep is done.
UPDATE channel_media_pending_object AS obj
SET state = CASE WHEN obj.state = 'tombstoned' THEN 'tombstoned' ELSE 'deleting' END,
lease_token = @lease_token,
lease_expires_at = now() + @lease::interval,
attempt = obj.attempt + 1
FROM (
SELECT cand.storage_key FROM channel_media_pending_object AS cand
WHERE cand.next_attempt_at <= now()
AND (
(cand.state = 'pending' AND cand.created_at <= now() - @settle_delay::interval)
OR (cand.state = 'deleting' AND (cand.lease_expires_at IS NULL OR cand.lease_expires_at <= now()))
OR (cand.state = 'tombstoned' AND (cand.lease_expires_at IS NULL OR cand.lease_expires_at <= now()))
)
ORDER BY cand.next_attempt_at
LIMIT 1
FOR UPDATE SKIP LOCKED
) AS due
WHERE obj.storage_key = due.storage_key
RETURNING obj.*;
-- name: ReleaseChannelMediaPendingObject :exec
-- Object-storage DELETE failed: keep the row in 'deleting' (bind must still
-- never attach it), release the lease, and back off the next attempt.
-- workspace_id is redundant with the storage_key PK but explicit per the
-- tenancy rule: every query constrains the workspace column, never trusting
-- the key string.
UPDATE channel_media_pending_object
SET lease_token = NULL,
lease_expires_at = NULL,
next_attempt_at = now() + @backoff::interval,
last_error = @last_error
WHERE storage_key = @storage_key
AND workspace_id = @workspace_id
AND lease_token = @lease_token;
-- name: TombstoneChannelMediaPendingObject :execrows
-- The object was deleted, but the row is KEPT as a tombstone: a PUT the client
-- abandoned before the delete may still materialize the object afterwards, and
-- no DELETE can be ordered against it. Each due tombstone re-runs the
-- reference check and, only if still unreferenced, triggers another idempotent
-- delete, so a late materialization is reclaimed by a later pass while an
-- object something durably reads is never removed;
-- only after the re-delete schedule is exhausted is the row dropped
-- (DeleteChannelMediaPendingObject). Lease-token guarded like every other
-- settle write; workspace_id explicit per the tenancy rule.
UPDATE channel_media_pending_object
SET state = 'tombstoned',
lease_token = NULL,
lease_expires_at = NULL,
next_attempt_at = now() + @redelete_delay::interval,
-- The pass index lives in its own column: a failed re-delete writes
-- last_error, so carrying the schedule position there would reset the
-- walk on every failure and a flaky store could keep the row alive
-- indefinitely. The delete that got here succeeded, so any previous
-- failure text is stale.
tombstone_pass = @tombstone_pass,
last_error = NULL
WHERE storage_key = @storage_key
AND workspace_id = @workspace_id
AND lease_token = @lease_token;
-- name: DeleteChannelMediaPendingObject :execrows
-- Drops a claimed row for good: a durable attachment reference was found, or
-- the tombstone's re-delete schedule is exhausted. Lease-token guarded so an
-- expired-lease reclaim by another
-- replica cannot be clobbered; workspace_id explicit per the tenancy rule.
DELETE FROM channel_media_pending_object
WHERE storage_key = @storage_key
AND workspace_id = @workspace_id
AND lease_token = @lease_token;
-- name: ChannelMediaObjectIsReferenced :one
-- The post-claim reference check: an attachment row carrying this object's
-- URL on the intended message. Only meaningful AFTER the claim flipped the
-- row to 'deleting' — from that point a bind can no longer succeed on the
-- key, so a negative answer is terminal, not a snapshot race. Re-run on every
-- tombstone pass as well: a positive answer there is an invariant violation,
-- and the object is kept and reported rather than deleted.
SELECT EXISTS (
SELECT 1 FROM attachment
WHERE chat_message_id = @chat_message_id
AND workspace_id = @workspace_id
AND url = @storage_url
) AS referenced;