Move the check.

Valeriy Zainullin · Valeriy Zainullin · commit a7346108ae6e · 2025-06-23T17:56:20.000+03:00
diff --git a/pg_wait_sampling.c b/pg_wait_sampling.c
@@ -816,6 +816,15 @@ pg_wait_sampling_reset_profile(PG_FUNCTION_ARGS)
 	 */
 	check_shmem();
 
+	pgws_init_lock_tag(&queueTag, PGWS_QUEUE_LOCK);
+
+	LockAcquire(&queueTag, ExclusiveLock, false, false);
+
+	pgws_init_lock_tag(&collectorTag, PGWS_COLLECTOR_LOCK);
+	LockAcquire(&collectorTag, ExclusiveLock, false, false);
+	pgws_collector_hdr->request = PROFILE_RESET;
+	LockRelease(&collectorTag, ExclusiveLock, false);
+
         /* The collector may have not started yet (it'd be a
          * race condition with background worker startup to
          * use latch without a check here).
@@ -828,15 +837,6 @@ pg_wait_sampling_reset_profile(PG_FUNCTION_ARGS)
 		ereport(ERROR, (errcode(ERRCODE_INTERNAL_ERROR),
 						errmsg("pg_wait_sampling collector wasn't started")));
 
-	pgws_init_lock_tag(&queueTag, PGWS_QUEUE_LOCK);
-
-	LockAcquire(&queueTag, ExclusiveLock, false, false);
-
-	pgws_init_lock_tag(&collectorTag, PGWS_COLLECTOR_LOCK);
-	LockAcquire(&collectorTag, ExclusiveLock, false, false);
-	pgws_collector_hdr->request = PROFILE_RESET;
-	LockRelease(&collectorTag, ExclusiveLock, false);
-
 	SetLatch(pgws_collector_hdr->latch);
 
 	LockRelease(&queueTag, ExclusiveLock, false);
