doc-gen: make user_doc to work with predefined consts

ding-young · Jan 15, 2025 · 0590e5f · 0590e5f
1 parent c7c200a
commit 0590e5f
Show file tree

Hide file tree

Showing 4 changed files with 89 additions and 67 deletions.
diff --git a/datafusion-cli/Cargo.lock b/datafusion-cli/Cargo.lock
diff --git a/datafusion/expr/src/udf.rs b/datafusion/expr/src/udf.rs
@@ -881,6 +881,22 @@ pub mod scalar_doc_sections {
         ]
     }
 
+    pub const fn doc_sections_const() -> &'static [DocSection] {
+        &[
+            DOC_SECTION_MATH,
+            DOC_SECTION_CONDITIONAL,
+            DOC_SECTION_STRING,
+            DOC_SECTION_BINARY_STRING,
+            DOC_SECTION_REGEX,
+            DOC_SECTION_DATETIME,
+            DOC_SECTION_ARRAY,
+            DOC_SECTION_STRUCT,
+            DOC_SECTION_MAP,
+            DOC_SECTION_HASHING,
+            DOC_SECTION_OTHER,
+        ]
+    }
+
     pub const DOC_SECTION_MATH: DocSection = DocSection {
         include: true,
         label: "Math Functions",

diff --git a/datafusion/macros/Cargo.toml b/datafusion/macros/Cargo.toml
@@ -37,5 +37,6 @@ path = "src/user_doc.rs"
 proc-macro = true
 
 [dependencies]
+datafusion-expr = { workspace = true }
 quote = "1.0.37"
 syn = { version = "2.0.79", features = ["full"] }