feat(webui): Add support for submitting Presto SQL queries.

components/webui/client/src/api/presto-search.ts

@@ -0,0 +1,43 @@
+import {
+    Static,
+    Type,
+} from "@sinclair/typebox";
+import {Value} from "@sinclair/typebox/value";
+import axios from "axios";
+
+
+// eslint-disable-next-line no-warning-comments
+// TODO: Replace with shared type from the `@common` directory once refactoring is completed.
+// Currently, server schema types require typebox dependency so they cannot be moved to the
+// `@common` directory with current implementation.
+const StringSchema = Type.String({
+    minLength: 1,
+});
+
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+const PrestoSearchJobCreationSchema = Type.Object({
+    queryString: StringSchema,
+});
+
+type PrestoSearchJobCreation = Static<typeof PrestoSearchJobCreationSchema>;
+
+const PrestoJobSchema = Type.Object(
+    {
+        searchJobId: StringSchema,
+    }
+);
+
+type PrestoJob = Static<typeof PrestoJobSchema>;
+
+/**
+ * Sends post request to server tosubmit presto query.
+ *
+ * @param payload
+ * @return
+ */
+const submitQuery = async (payload: PrestoSearchJobCreation): Promise<PrestoJob> => {
+    const ret = await axios.post<PrestoJob>("/api/presto-search/query", payload);
+    return Value.Parse(PrestoJobSchema, ret.data);
+};
+
+export {submitQuery};

components/webui/client/src/pages/SearchPage/SearchControls/Presto/RunButton/index.tsx

@@ -1,9 +1,12 @@
+import {useCallback} from "react";
+
 import {CaretRightOutlined} from "@ant-design/icons";
 import {
     Button,
     Tooltip,
 } from "antd";
 
+import {submitQuery} from "../../../../../api/presto-search";
 import useSearchStore from "../../../SearchState/index";
 
 
@@ -20,6 +23,17 @@ const RunButton = () => {
         "Enter SQL query to run" :
         "";
 
+    const handleClick = useCallback(() => {
+        submitQuery({queryString})
+            .then(({searchJobId}) => {
+                const {updateSearchJobId} = useSearchStore.getState();
+                updateSearchJobId(searchJobId);
+            })
+            .catch((err: unknown) => {
+                console.error("Failed to submit query:", err);
+            });
+    }, [queryString]);
+
     return (
         <Tooltip title={tooltipTitle}>
             <Button
@@ -28,6 +42,7 @@ const RunButton = () => {
                 icon={<CaretRightOutlined/>}
                 size={"large"}
                 variant={"solid"}
+                onClick={handleClick}
             >
                 Run
             </Button>

components/webui/server/package.json

@@ -22,10 +22,10 @@
     "@aws-sdk/s3-request-presigner": "^3.758.0",
     "@fastify/autoload": "^6.3.0",
     "@fastify/env": "^5.0.2",
+    "@fastify/http-proxy": "^11.3.0",
     "@fastify/mongodb": "^9.0.2",
     "@fastify/mysql": "^5.0.2",
     "@fastify/rate-limit": "^10.2.2",
-    "@fastify/http-proxy": "^11.3.0",
     "@fastify/sensible": "^6.0.3",
     "@fastify/static": "^8.1.1",
     "@fastify/type-provider-typebox": "^5.1.0",
@@ -38,10 +38,12 @@
     "fastify-plugin": "^5.0.1",
     "http-status-codes": "^2.3.0",
     "pino-pretty": "^13.0.0",
+    "presto-client": "^1.1.0",
     "socket.io": "^4.8.1",
     "typescript": "~5.7.3"
   },
   "devDependencies": {
+    "@types/presto-client": "^1.0.2",
     "concurrently": "^9.1.2",
     "eslint-config-yscope": "latest",
     "fastify-cli": "^7.4.0",

components/webui/server/settings.json

@@ -16,5 +16,9 @@
     "StreamTargetUncompressedSize": 134217728,
     "StreamFilesS3Region": null,
     "StreamFilesS3PathPrefix": null,
-    "StreamFilesS3Profile": null
+    "StreamFilesS3Profile": null,
+
+    "EnablePresto": true,
+    "PrestoHost": "localhost",
+    "PrestoPort": 8080
 }

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

@@ -0,0 +1,89 @@
+import {FastifyPluginAsyncTypebox} from "@fastify/type-provider-typebox";
+import {StatusCodes} from "http-status-codes";
+import {Client} from "presto-client";
+
+import settings from "../../../../settings.json" with {type: "json"};
+import {ErrorSchema} from "../../../schemas/error.js";
+import {
+    PrestoJobSchema,
+    PrestoSearchJobCreationSchema,
+} from "../../../schemas/presto-search.js";
+import {Nullable} from "../../../typings/common.js";
+
+
+/**
+ * Presto search API routes.
+ *
+ * @param fastify
+ */
+const plugin: FastifyPluginAsyncTypebox = async (fastify) => {
+    if (false === settings.EnablePresto) {
+        return;
+    }
+
+    const client = new Client({host: settings.PrestoHost, port: settings.PrestoPort});
+
+    /**
+     * Submits a search query and initiates the search process.
+     */
+    fastify.post(
+        "/query",
+        {
+            schema: {
+                body: PrestoSearchJobCreationSchema,
+                response: {
+                    [StatusCodes.CREATED]: PrestoJobSchema,
+                    [StatusCodes.INTERNAL_SERVER_ERROR]: ErrorSchema,
+                },
+                tags: ["Search"],
+            },
+        },
+
+        async (request, reply) => {
+            const {queryString} = request.body;
+
+            let searchJobId: Nullable<string> = null;
+
+            searchJobId = await new Promise((resolve) => {
+                client.execute({
+                    data: (error, data, columns) => {
+                        if (null === searchJobId) {
+                            request.log.error("Data arrived from Presto before the search job id.");
+                        }
+                        request.log.info({error, searchJobId, columns, data}, "Presto data");
+                    },
+                    error: (error) => {
+                        request.log.info(error, "Presto search failed");
+
+                        // The `/query` endpoint will catch this error if `resolve` hasn't
+                        // been called.
+                        throw new Error("Presto search failed.");
+                    },
+                    query: queryString,
+                    state: (error, newSearchJobId, stats) => {
+                        request.log.info({
+                            error: error,
+                            searchJobId: newSearchJobId,
+                            state: stats.state,
+                        }, "Presto search state updated");
+
+                        resolve(newSearchJobId);
+                    },
+                    success: () => {
+                        request.log.info("Presto search succeeded");
+                    },
+                });
+            });
+
+            reply.code(StatusCodes.CREATED);
+
+            if (null === searchJobId) {
+                throw new Error("searchJobId is null");
+            }
+
+            return {searchJobId};
+        }
+    );
+};
+
+export default plugin;

components/webui/server/src/schemas/presto-search.ts

@@ -0,0 +1,20 @@
+import {Type} from "@sinclair/typebox";
+
+import {StringSchema} from "./common.js";
+
+
+/**
+ * Schema for request to create a new query job.
+ */
+const PrestoSearchJobCreationSchema = Type.Object({
+    queryString: StringSchema,
+});
+
+const PrestoJobSchema = Type.Object({
+    searchJobId: StringSchema,
+});
+
+export {
+    PrestoJobSchema,
+    PrestoSearchJobCreationSchema,
+};