fix: boost table score for exact match when searching table (pinterest#1097)

Author: jczhong84

querybook/server/lib/elasticsearch/search_table.py

@@ -6,6 +6,18 @@
 )
 
 
+def _get_potential_exact_schema_table_name(keywords):
+    """Get the schema and table name from a full table name.
+
+    E.g. "default.table_a", will return (default, table_a)
+    """
+    dot_index = keywords.find(".")
+    if dot_index == -1:
+        return None, keywords
+
+    return keywords[:dot_index], keywords[dot_index + 1 :]
+
+
 def _match_table_word_fields(fields):
     search_fields = []
     for field in fields:
@@ -44,6 +56,18 @@ def construct_tables_query(
 ):
     keywords_query = {}
     if keywords:
+        should_clause = _match_table_phrase_queries(fields, keywords)
+
+        table_schema, table_name = _get_potential_exact_schema_table_name(keywords)
+        if table_schema:
+            filters.append(["schema", table_schema])
+
+        # boost score for table name exact match
+        if table_name:
+            should_clause.append(
+                {"term": {"name": {"value": table_name, "boost": 100}}},
+            )
+
         keywords_query = {
             "bool": {
                 "must": {
@@ -54,7 +78,7 @@ def construct_tables_query(
                         "operator": "and",
                     },
                 },
-                "should": _match_table_phrase_queries(fields, keywords),
+                "should": should_clause,
             }
         }
     else:

querybook/webapp/redux/dataTableSearch/action.ts

@@ -28,11 +28,6 @@ function mapStateToSearch(state: IDataTableSearchState) {
         ([_, filterValue]) => filterValue != null
     );
 
-    const matchSchemaName = searchString.match(/(\w+)\.(\w*)/);
-    if (matchSchemaName) {
-        filters.push(['schema', matchSchemaName[1]]);
-    }
-
     const searchParam = {
         metastore_id: state.metastoreId,
         keywords: searchString,