feat: Show bigquery datasource table and query on UI (feast-dev#3600)

Signed-off-by: Hai Nguyen <[email protected]>
zerafachris · Apr 21, 2023 · 58d63f7 · 58d63f7
1 parent 01a98f0
commit 58d63f7
Show file tree

Hide file tree

Showing 3 changed files with 364 additions and 6 deletions.
diff --git a/ui/package.json b/ui/package.json
@@ -43,6 +43,7 @@
     "prop-types": "^15.8.1",
     "protobufjs": "^7.1.1",
     "query-string": "^7.1.1",
+    "react-code-blocks": "^0.0.9-0",
     "react-query": "^3.34.12",
     "react-router-dom": "6",
     "react-scripts": "^5.0.0",

diff --git a/ui/src/pages/data-sources/BatchSourcePropertiesView.tsx b/ui/src/pages/data-sources/BatchSourcePropertiesView.tsx
@@ -6,6 +6,7 @@ import {
   EuiFlexGroup,
   EuiFlexItem,
 } from "@elastic/eui";
+import { CopyBlock, atomOneDark } from "react-code-blocks";
 import { feast } from "../../protos";
 import { toDate } from "../../utils/timestamp";
 
@@ -61,6 +62,24 @@ const BatchSourcePropertiesView = (props: BatchSourcePropertiesViewProps) => {
                 </EuiDescriptionListDescription>
               </React.Fragment>
             )}
+            {batchSource.bigqueryOptions && (
+              <React.Fragment>
+                <EuiDescriptionListTitle>Source {batchSource.bigqueryOptions.table ? "Table" : "Query"}</EuiDescriptionListTitle>
+                {batchSource.bigqueryOptions.table ? (
+                    <EuiDescriptionListDescription>
+                        {batchSource.bigqueryOptions.table}
+                    </EuiDescriptionListDescription>
+                ) : <CopyBlock
+                        text={batchSource.bigqueryOptions.query}
+                        language="sql"
+                        showLineNumbers={false}
+                        theme={atomOneDark}
+                        wrapLines
+                    />
+                }
+
+              </React.Fragment>
+            )}
             {batchSource.meta?.latestEventTimestamp && (
               <React.Fragment>
                 <EuiDescriptionListTitle>Latest Event</EuiDescriptionListTitle>