Method ArchiveChat

coderd/x/chatd/chatd.go:2200–2277

ArchiveChat archives a chat family and broadcasts deleted events for each affected chat so watching clients converge without a full refetch. If the target chat is pending or running, it first transitions the chat back to waiting so active processing stops before the archive is broadcast.

(ctx context.Context, chat database.Chat)

// target chat is pending or running, it first transitions the chat back to
// waiting so active processing stops before the archive is broadcast.
func (p *Server) ArchiveChat(ctx context.Context, chat database.Chat) error {
	if chat.ID == uuid.Nil {
		return xerrors.New("chat_id is required")
	}

	var (
		archivedChats    []database.Chat
		interruptedChats []database.Chat
	)
	if err := p.db.InTx(func(tx database.Store) error {
		if _, err := tx.GetChatByIDForUpdate(ctx, chat.ID); err != nil {
			return xerrors.Errorf("lock chat for archive: %w", err)
		}

		var err error
		archivedChats, err = tx.ArchiveChatByID(ctx, chat.ID)
		if err != nil {
			return xerrors.Errorf("archive chat: %w", err)
		}

		for i, archivedChat := range archivedChats {
			if archivedChat.Status != database.ChatStatusPending &&
				archivedChat.Status != database.ChatStatusRunning {
				continue
			}

			updatedChat, updateErr := tx.UpdateChatStatus(ctx, database.UpdateChatStatusParams{
				ID:          archivedChat.ID,
				Status:      database.ChatStatusWaiting,
				WorkerID:    uuid.NullUUID{},
				StartedAt:   sql.NullTime{},
				HeartbeatAt: sql.NullTime{},
				LastError:   pqtype.NullRawMessage{},
			})
			if updateErr != nil {
				return xerrors.Errorf("set archived chat waiting before cleanup: %w", updateErr)
			}
			archivedChats[i] = updatedChat
			interruptedChats = append(interruptedChats, updatedChat)
		}
		return nil
	}, nil); err != nil {
		return err
	}

	for _, interruptedChat := range interruptedChats {
		p.publishStatus(interruptedChat.ID, interruptedChat.Status, interruptedChat.WorkerID)
		p.publishChatPubsubEvent(interruptedChat, codersdk.ChatWatchEventKindStatusChange, nil)
	}

	// Archiving can race with an interrupted worker still flushing its
	// final debug writes. Retry a few times so orphaned rows are
	// removed quickly instead of waiting for the stale sweeper. Source
	// archiveCutoff from the DB-stamped updated_at returned by
	// ArchiveChatByID so the filter uses the same clock that stamps
	// replacement-turn debug rows; subtract debugCleanupClockSkew so
	// replica clock drift cannot let the retry delete a replacement's
	// debug rows if an unarchive races ahead (see the constant for the