feat: add more h3 scalar functions (#4707)

* feat: add more h3 scalar functions * chore: comment up
GreptimeTeam · Sep 20, 2024 · 75c6fad · 75c6fad
1 parent e12ffbe
commit 75c6fad
Show file tree

Hide file tree

Showing 6 changed files with 867 additions and 125 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/common/function/Cargo.toml b/src/common/function/Cargo.toml
@@ -27,6 +27,7 @@ common-time.workspace = true
 common-version.workspace = true
 datafusion.workspace = true
 datatypes.workspace = true
+derive_more = { version = "1", default-features = false, features = ["display"] }
 geohash = { version = "0.13", optional = true }
 h3o = { version = "0.6", optional = true }
 jsonb.workspace = true

diff --git a/src/common/function/src/scalars/geo.rs b/src/common/function/src/scalars/geo.rs
@@ -17,15 +17,27 @@ mod geohash;
 mod h3;
 
 use geohash::GeohashFunction;
-use h3::H3Function;
 
 use crate::function_registry::FunctionRegistry;
 
 pub(crate) struct GeoFunctions;
 
 impl GeoFunctions {
     pub fn register(registry: &FunctionRegistry) {
+        // geohash
         registry.register(Arc::new(GeohashFunction));
-        registry.register(Arc::new(H3Function));
+        // h3 family
+        registry.register(Arc::new(h3::H3LatLngToCell));
+        registry.register(Arc::new(h3::H3LatLngToCellString));
+        registry.register(Arc::new(h3::H3CellBase));
+        registry.register(Arc::new(h3::H3CellCenterChild));
+        registry.register(Arc::new(h3::H3CellCenterLat));
+        registry.register(Arc::new(h3::H3CellCenterLng));
+        registry.register(Arc::new(h3::H3CellIsPentagon));
+        registry.register(Arc::new(h3::H3CellParent));
+        registry.register(Arc::new(h3::H3CellResolution));
+        registry.register(Arc::new(h3::H3CellToString));
+        registry.register(Arc::new(h3::H3IsNeighbour));
+        registry.register(Arc::new(h3::H3StringToCell));
     }
 }