Merge branch 'y-scope:main' into graceful-shutdown

Author: AVMatthews

components/webui/client/src/components/QueryBox/InputWithCaseSensitive/index.tsx

@@ -1,6 +1,7 @@
 import {
     Input,
     InputProps,
+    InputRef,
 } from "antd";
 
 import CaseSensitiveToggle, {CaseSensitiveToggleProps} from "./CaseSenstiveToggle";
@@ -9,7 +10,8 @@ import CaseSensitiveToggle, {CaseSensitiveToggleProps} from "./CaseSenstiveToggl
 /**
  * Antd Input props and case sensitive toggle props with suffix omitted since set by component.
  */
-type InputWithCaseSensitiveProps = Omit<InputProps, "suffix"> & CaseSensitiveToggleProps;
+type InputWithCaseSensitiveProps =
+    Omit<InputProps, "suffix"> & CaseSensitiveToggleProps & React.RefAttributes<InputRef>;
 
 /**
  * Antd Input with a built-in case sensitivity toggle.

components/webui/client/src/pages/SearchPage/SearchControls/QueryInput/index.tsx

@@ -6,6 +6,7 @@ import {
     useState,
 } from "react";
 
+import type {InputRef} from "antd";
 import {Nullable} from "src/typings/common";
 
 import QueryBox from "../../../../components/QueryBox";
@@ -28,6 +29,7 @@ const QueryInput = () => {
     const searchUiState = useSearchStore((state) => state.searchUiState);
     const [pseudoProgress, setPseudoProgress] = useState<Nullable<number>>(null);
     const intervalIdRef = useRef<number>(0);
+    const inputRef = useRef<InputRef>(null);
 
     const handleCaseSensitiveChange = useCallback((newValue: boolean) => {
         const {updateQueryIsCaseSensitive} = useSearchStore.getState();
@@ -67,11 +69,21 @@ const QueryInput = () => {
         clearInterval(intervalIdRef.current);
     }, []);
 
+    useEffect(() => {
+        if (
+            searchUiState === SEARCH_UI_STATE.DEFAULT ||
+            searchUiState === SEARCH_UI_STATE.DONE
+        ) {
+            inputRef.current?.focus();
+        }
+    }, [searchUiState]);
+
     return (
         <QueryBox
             isCaseSensitive={queryIsCaseSensitive}
             placeholder={"Enter your query"}
             progress={pseudoProgress}
+            ref={inputRef}
             size={"large"}
             value={queryString}
             disabled={

components/webui/server/src/routes/api/presto-search/index.ts

@@ -6,21 +6,28 @@ import {
     PrestoQueryJobCreationSchema,
     PrestoQueryJobSchema,
 } from "../../../schemas/presto-search.js";
+import {insertPrestoRowsToMongo} from "./utils.js";
 
 
 /**
  * Presto search API routes.
  *
  * @param fastify
  */
+// eslint-disable-next-line max-lines-per-function
 const plugin: FastifyPluginAsyncTypebox = async (fastify) => {
-    const {Presto} = fastify;
+    const {Presto, mongo} = fastify;
+    const mongoDb = mongo.db;
 
     if ("undefined" === typeof Presto) {
         // If Presto client is not available, skip the plugin registration.
         return;
     }
 
+    if ("undefined" === typeof mongoDb) {
+        throw new Error("MongoDB database not found");
+    }
+
     /**
      * Submits a search query.
      */
@@ -36,7 +43,7 @@ const plugin: FastifyPluginAsyncTypebox = async (fastify) => {
                 tags: ["Presto Search"],
             },
         },
-
+        // eslint-disable-next-line max-lines-per-function
         async (request, reply) => {
             const {queryString} = request.body;
 
@@ -45,13 +52,39 @@ const plugin: FastifyPluginAsyncTypebox = async (fastify) => {
             try {
                 searchJobId = await new Promise<string>((resolve, reject) => {
                     let isResolved = false;
-
                     Presto.client.execute({
                         // eslint-disable-next-line no-warning-comments
-                        // TODO: Data, error, and success handlers are dummy implementations
+                        // TODO: Error, and success handlers are dummy implementations
                         // and will be replaced with proper implementations.
                         data: (_, data, columns) => {
-                            request.log.info({columns, data}, "Presto data");
+                            request.log.info(
+                                `Received ${data.length} rows from Presto query`
+                            );
+
+                            if (false === isResolved) {
+                                request.log.error(
+                                    "Presto data received before searchJobId was resolved; " +
+                                    "skipping insert."
+                                );
+
+                                return;
+                            }
+
+                            if (0 === data.length) {
+                                return;
+                            }
+
+                            insertPrestoRowsToMongo(
+                                data,
+                                columns,
+                                searchJobId,
+                                mongoDb
+                            ).catch((err: unknown) => {
+                                request.log.error(
+                                    err,
+                                    "Failed to insert Presto results into MongoDB"
+                                );
+                            });
                         },
                         error: (error) => {
                             request.log.info(error, "Presto search failed");

components/webui/server/src/routes/api/presto-search/utils.ts

@@ -0,0 +1,47 @@
+import type {
+    Db,
+    InsertManyResult,
+} from "mongodb";
+
+
+/**
+ * Converts a Presto result row (array of values) into an object, using the provided column
+ * definitions to assign property names.
+ *
+ * @param row Array of values representing a single Presto result row.
+ * @param columns Array of column definitions, each containing a `name` property.
+ * @return An object mapping each column name to its corresponding value from the row.
+ */
+const prestoRowToObject = (
+    row: unknown[],
+    columns: {name: string}[]
+): Record<string, unknown> => {
+    const obj: Record<string, unknown> = {};
+    columns.forEach((col, idx) => {
+        obj[col.name] = row[idx];
+    });
+
+    return obj;
+};
+
+/**
+ * Inserts Presto rows into a MongoDB collection for a given search job.
+ *
+ * @param data Array of Presto result rows
+ * @param columns Array of column definitions
+ * @param searchJobId
+ * @param mongoDb
+ * @return Promise that resolves when the insertion is complete
+ */
+const insertPrestoRowsToMongo = (
+    data: unknown[][],
+    columns: {name: string}[],
+    searchJobId: string,
+    mongoDb: Db
+): Promise<InsertManyResult<Document>> => {
+    const collection = mongoDb.collection(searchJobId);
+    const resultDocs = data.map((row) => prestoRowToObject(row, columns));
+    return collection.insertMany(resultDocs);
+};
+
+export {insertPrestoRowsToMongo};