Added initial support for BQ HLL functions

Signed-off-by: Jason Smart <jlsmart@meta.com>

Author: anagrd-ai

packages/malloy/src/dialect/standardsql/dialect_functions.ts

@@ -10,8 +10,12 @@ import {
   DefinitionBlueprint,
   DefinitionBlueprintMap,
   OverloadedDefinitionBlueprint,
+  TypeDescBlueprint,
 } from '../functions/util';
 
+// Cute shortcut So you can write things like: {array: T} and {dimension: T}
+const T: TypeDescBlueprint = {generic: 'T'};
+
 const date_from_unix_date: DefinitionBlueprint = {
   takes: {'unix_date': 'number'},
   returns: 'date',
@@ -69,6 +73,38 @@ export const STANDARDSQL_DIALECT_FUNCTIONS: DefinitionBlueprintMap = {
   date_from_unix_date,
   string_agg,
   string_agg_distinct,
+  hll_accumulate: {
+    default: {
+      takes: {'value': {dimension: T}},
+      returns: {measure: {sql_native: 'bytes'}},
+      generic: {
+        'T': ['string', 'number'],
+      },
+      isSymmetric: true,
+      impl: {function: 'HLL_COUNT.INIT'},
+    },
+  },
+  hll_combine: {
+    takes: {'value': {sql_native: 'bytes'}},
+    returns: {measure: {sql_native: 'bytes'}},
+    impl: {function: 'HLL_COUNT.MERGE_PARTIAL'},
+    isSymmetric: true,
+  },
+  hll_estimate: {
+    takes: {'value': {sql_native: 'bytes'}},
+    returns: {dimension: 'number'},
+    impl: {function: 'HLL_COUNT.EXTRACT'},
+  },
+  hll_export: {
+    takes: {'value': {sql_native: 'bytes'}},
+    returns: {dimension: {sql_native: 'bytes'}},
+    impl: {sql: 'CAST(${value} AS BYTES)'},
+  },
+  hll_import: {
+    takes: {'value': {sql_native: 'bytes'}},
+    returns: {dimension: {sql_native: 'bytes'}},
+    impl: {sql: 'CAST(${value} AS BYTES)'},
+  },
   ...def('repeat', {'str': 'string', 'n': 'number'}, 'string'),
   ...def('reverse', {'str': 'string'}, 'string'),
 };

packages/malloy/src/dialect/standardsql/standardsql.ts

@@ -126,6 +126,7 @@ export class StandardSQLDialect extends Dialect {
   cantPartitionWindowFunctionsOnExpressions = true;
   hasModOperator = false;
   nestedArrays = false; // Can't have an array of arrays for some reason
+  supportsHyperLogLog = true;
 
   quoteTablePath(tablePath: string): string {
     return `\`${tablePath}\``;