When updating query info, return existing final info if already set

Author: dain

core/trino-main/src/main/java/io/trino/execution/QueryStateMachine.java

@@ -1387,7 +1387,10 @@ public QueryInfo updateQueryInfo(Optional<StageInfo> stageInfo)
     {
         QueryInfo queryInfo = getQueryInfo(stageInfo);
         if (queryInfo.isFinalQueryInfo()) {
-            finalQueryInfo.compareAndSet(Optional.empty(), Optional.of(queryInfo));
+            if (!finalQueryInfo.compareAndSet(Optional.empty(), Optional.of(queryInfo))) {
+                // use the final query info if it is already set
+                queryInfo = finalQueryInfo.get().orElseThrow();
+            }
         }
         return queryInfo;
     }