Filter queries by trace token in the web UI

Author: ms1111

core/trino-main/src/main/java/io/trino/server/ui/TrimmedBasicQueryInfo.java

@@ -55,6 +55,7 @@ public class TrimmedBasicQueryInfo
     private final Optional<QueryType> queryType;
     private final RetryPolicy retryPolicy;
     private final Optional<Set<String>> clientTags;
+    private final Optional<String> traceToken;
 
     public TrimmedBasicQueryInfo(BasicQueryInfo queryInfo)
     {
@@ -82,6 +83,7 @@ public TrimmedBasicQueryInfo(BasicQueryInfo queryInfo)
         this.queryType = requireNonNull(queryInfo.getQueryType(), "queryType is null");
         this.retryPolicy = requireNonNull(queryInfo.getRetryPolicy(), "retryPolicy is null");
         this.clientTags = Optional.ofNullable(queryInfo.getSession().getClientTags());
+        this.traceToken = requireNonNull(queryInfo.getSession().getTraceToken(), "traceToken is null");
     }
 
     @JsonProperty
@@ -192,6 +194,12 @@ public Optional<Set<String>> getClientTags()
         return clientTags;
     }
 
+    @JsonProperty
+    public Optional<String> getTraceToken()
+    {
+        return traceToken;
+    }
+
     @Override
     public String toString()
     {

core/trino-web-ui/src/main/resources/webapp-preview/src/api/webapp/api.ts

@@ -175,6 +175,7 @@ export interface QueryInfo extends QueryInfoBase {
     sessionSource: string
     sessionUser: string
     queryDataEncoding: string
+    traceToken: string
 }
 
 export interface Session {

core/trino-web-ui/src/main/resources/webapp-preview/src/components/QueryList.tsx

@@ -171,6 +171,7 @@ export const QueryList = () => {
                       query.resourceGroupId?.join('.'),
                       query.errorCode?.name,
                       ...(query.clientTags || []),
+                      query.traceToken,
                   ].some((value) => value?.toLowerCase().includes(term))
               })
     }

core/trino-web-ui/src/main/resources/webapp-preview/src/constant.ts

@@ -68,7 +68,7 @@ export const Texts = {
         Filter: {
             Search: 'Search',
             SearchPlaceholder:
-                'User, source, query ID, query state, resource group, error name, query text or client tags',
+                'User, source, query ID, state, resource group, error name, query text, client tags or trace token',
             State: 'State',
             Type: {
                 RUNNING: 'Running',

core/trino-web-ui/src/main/resources/webapp/src/components/QueryList.jsx

@@ -477,6 +477,10 @@ export class QueryList extends React.Component {
                 ) {
                     return true
                 }
+
+                if (query.traceToken && query.traceToken.toLowerCase().indexOf(term) !== -1) {
+                    return true
+                }
             }, this)
         }
     }
@@ -817,7 +821,7 @@ export class QueryList extends React.Component {
                             <input
                                 type="text"
                                 className="form-control form-control-small search-bar"
-                                placeholder="User, source, query ID, query state, resource group, error name, query text or client tags"
+                                placeholder="User, source, query ID, state, resource group, error name, query text, client tags or trace token"
                                 onChange={this.handleSearchStringChange}
                                 value={this.state.searchString}
                             />