mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-05-23 16:30:39 +00:00
Deploying to gh-pages from @ GreptimeTeam/greptimedb@0a2400cbb9 🚀
This commit is contained in:
github-merge-queue[bot]
365
src/common_macro/admin_fn.rs.html
Normal file
365
src/common_macro/admin_fn.rs.html
Normal file
@@ -0,0 +1,365 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/admin_fn.rs`."><title>admin_fn.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>admin_fn.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro::TokenStream;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::{
|
||||
<a href=#19 id=19 data-nosnippet>19</a> Attribute, Ident, ItemFn, Path, Signature, Type, TypePath, TypeReference, Visibility,
|
||||
<a href=#20 id=20 data-nosnippet>20</a> parse_macro_input,
|
||||
<a href=#21 id=21 data-nosnippet>21</a>};
|
||||
<a href=#22 id=22 data-nosnippet>22</a>
|
||||
<a href=#23 id=23 data-nosnippet>23</a><span class="kw">use </span><span class="kw">crate</span>::utils::extract_input_types;
|
||||
<a href=#24 id=24 data-nosnippet>24</a>
|
||||
<a href=#25 id=25 data-nosnippet>25</a><span class="doccomment">/// Internal util macro to early return on error.
|
||||
<a href=#26 id=26 data-nosnippet>26</a></span><span class="macro">macro_rules!</span> ok {
|
||||
<a href=#27 id=27 data-nosnippet>27</a> (<span class="macro-nonterminal">$item</span>:expr) => {
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="kw">match </span><span class="macro-nonterminal">$item </span>{
|
||||
<a href=#29 id=29 data-nosnippet>29</a> <span class="prelude-val">Ok</span>(item) => item,
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="prelude-val">Err</span>(e) => <span class="kw">return </span>e.into_compile_error().into(),
|
||||
<a href=#31 id=31 data-nosnippet>31</a> }
|
||||
<a href=#32 id=32 data-nosnippet>32</a> };
|
||||
<a href=#33 id=33 data-nosnippet>33</a>}
|
||||
<a href=#34 id=34 data-nosnippet>34</a>
|
||||
<a href=#35 id=35 data-nosnippet>35</a><span class="doccomment">/// Internal util macro to create an error.
|
||||
<a href=#36 id=36 data-nosnippet>36</a></span><span class="macro">macro_rules!</span> error {
|
||||
<a href=#37 id=37 data-nosnippet>37</a> (<span class="macro-nonterminal">$span</span>:expr, <span class="macro-nonterminal">$msg</span>: expr) => {
|
||||
<a href=#38 id=38 data-nosnippet>38</a> <span class="prelude-val">Err</span>(syn::Error::new(<span class="macro-nonterminal">$span</span>, <span class="macro-nonterminal">$msg</span>))
|
||||
<a href=#39 id=39 data-nosnippet>39</a> };
|
||||
<a href=#40 id=40 data-nosnippet>40</a>}
|
||||
<a href=#41 id=41 data-nosnippet>41</a>
|
||||
<a href=#42 id=42 data-nosnippet>42</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>process_admin_fn(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#43 id=43 data-nosnippet>43</a> <span class="kw">let </span><span class="kw-2">mut </span>name: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#44 id=44 data-nosnippet>44</a> <span class="kw">let </span><span class="kw-2">mut </span>display_name: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#45 id=45 data-nosnippet>45</a> <span class="kw">let </span><span class="kw-2">mut </span>sig_fn: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#46 id=46 data-nosnippet>46</a> <span class="kw">let </span><span class="kw-2">mut </span>ret: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#47 id=47 data-nosnippet>47</a> <span class="kw">let </span><span class="kw-2">mut </span>user_path: <span class="prelude-ty">Option</span><Path> = <span class="prelude-val">None</span>;
|
||||
<a href=#48 id=48 data-nosnippet>48</a>
|
||||
<a href=#49 id=49 data-nosnippet>49</a> <span class="kw">let </span>parser = syn::meta::parser(|meta| {
|
||||
<a href=#50 id=50 data-nosnippet>50</a> <span class="kw">if </span>meta.path.is_ident(<span class="string">"name"</span>) {
|
||||
<a href=#51 id=51 data-nosnippet>51</a> name = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#52 id=52 data-nosnippet>52</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#53 id=53 data-nosnippet>53</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"display_name"</span>) {
|
||||
<a href=#54 id=54 data-nosnippet>54</a> display_name = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#55 id=55 data-nosnippet>55</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#56 id=56 data-nosnippet>56</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"sig_fn"</span>) {
|
||||
<a href=#57 id=57 data-nosnippet>57</a> sig_fn = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#58 id=58 data-nosnippet>58</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#59 id=59 data-nosnippet>59</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"ret"</span>) {
|
||||
<a href=#60 id=60 data-nosnippet>60</a> ret = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#61 id=61 data-nosnippet>61</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#62 id=62 data-nosnippet>62</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"user_path"</span>) {
|
||||
<a href=#63 id=63 data-nosnippet>63</a> user_path = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#65 id=65 data-nosnippet>65</a> } <span class="kw">else </span>{
|
||||
<a href=#66 id=66 data-nosnippet>66</a> <span class="prelude-val">Err</span>(meta.error(<span class="string">"unsupported property"</span>))
|
||||
<a href=#67 id=67 data-nosnippet>67</a> }
|
||||
<a href=#68 id=68 data-nosnippet>68</a> });
|
||||
<a href=#69 id=69 data-nosnippet>69</a>
|
||||
<a href=#70 id=70 data-nosnippet>70</a> <span class="comment">// extract arg map
|
||||
<a href=#71 id=71 data-nosnippet>71</a> </span><span class="macro">parse_macro_input!</span>(args with parser);
|
||||
<a href=#72 id=72 data-nosnippet>72</a>
|
||||
<a href=#73 id=73 data-nosnippet>73</a> <span class="kw">if </span>user_path.is_none() {
|
||||
<a href=#74 id=74 data-nosnippet>74</a> user_path = <span class="prelude-val">Some</span>(syn::parse_str(<span class="string">"crate"</span>).expect(<span class="string">"failed to parse user path"</span>));
|
||||
<a href=#75 id=75 data-nosnippet>75</a> }
|
||||
<a href=#76 id=76 data-nosnippet>76</a>
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="comment">// decompose the fn block
|
||||
<a href=#78 id=78 data-nosnippet>78</a> </span><span class="kw">let </span>compute_fn = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>ItemFn);
|
||||
<a href=#79 id=79 data-nosnippet>79</a> <span class="kw">let </span>ItemFn {
|
||||
<a href=#80 id=80 data-nosnippet>80</a> attrs,
|
||||
<a href=#81 id=81 data-nosnippet>81</a> vis,
|
||||
<a href=#82 id=82 data-nosnippet>82</a> sig,
|
||||
<a href=#83 id=83 data-nosnippet>83</a> block,
|
||||
<a href=#84 id=84 data-nosnippet>84</a> } = compute_fn;
|
||||
<a href=#85 id=85 data-nosnippet>85</a>
|
||||
<a href=#86 id=86 data-nosnippet>86</a> <span class="comment">// extract fn arg list
|
||||
<a href=#87 id=87 data-nosnippet>87</a> </span><span class="kw">let </span>Signature {
|
||||
<a href=#88 id=88 data-nosnippet>88</a> inputs,
|
||||
<a href=#89 id=89 data-nosnippet>89</a> ident: fn_name,
|
||||
<a href=#90 id=90 data-nosnippet>90</a> ..
|
||||
<a href=#91 id=91 data-nosnippet>91</a> } = <span class="kw-2">&</span>sig;
|
||||
<a href=#92 id=92 data-nosnippet>92</a>
|
||||
<a href=#93 id=93 data-nosnippet>93</a> <span class="kw">let </span>arg_types = <span class="macro">ok!</span>(extract_input_types(inputs));
|
||||
<a href=#94 id=94 data-nosnippet>94</a> <span class="kw">if </span>arg_types.len() < <span class="number">2 </span>{
|
||||
<a href=#95 id=95 data-nosnippet>95</a> <span class="macro">ok!</span>(<span class="macro">error!</span>(
|
||||
<a href=#96 id=96 data-nosnippet>96</a> sig.span(),
|
||||
<a href=#97 id=97 data-nosnippet>97</a> <span class="string">"Expect at least two argument for admin fn: (handler, query_ctx)"
|
||||
<a href=#98 id=98 data-nosnippet>98</a> </span>));
|
||||
<a href=#99 id=99 data-nosnippet>99</a> }
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="kw">let </span>handler_type = <span class="macro">ok!</span>(extract_handler_type(<span class="kw-2">&</span>arg_types));
|
||||
<a href=#101 id=101 data-nosnippet>101</a>
|
||||
<a href=#102 id=102 data-nosnippet>102</a> <span class="kw">let </span><span class="kw-2">mut </span>result = TokenStream::new();
|
||||
<a href=#103 id=103 data-nosnippet>103</a> <span class="comment">// build the struct and its impl block
|
||||
<a href=#104 id=104 data-nosnippet>104</a> // only do this when `display_name` is specified
|
||||
<a href=#105 id=105 data-nosnippet>105</a> </span><span class="kw">if let </span><span class="prelude-val">Some</span>(display_name) = display_name {
|
||||
<a href=#106 id=106 data-nosnippet>106</a> <span class="kw">let </span>struct_code = build_struct(
|
||||
<a href=#107 id=107 data-nosnippet>107</a> attrs,
|
||||
<a href=#108 id=108 data-nosnippet>108</a> vis,
|
||||
<a href=#109 id=109 data-nosnippet>109</a> fn_name,
|
||||
<a href=#110 id=110 data-nosnippet>110</a> name.expect(<span class="string">"name required"</span>),
|
||||
<a href=#111 id=111 data-nosnippet>111</a> sig_fn.expect(<span class="string">"sig_fn required"</span>),
|
||||
<a href=#112 id=112 data-nosnippet>112</a> ret.expect(<span class="string">"ret required"</span>),
|
||||
<a href=#113 id=113 data-nosnippet>113</a> handler_type,
|
||||
<a href=#114 id=114 data-nosnippet>114</a> display_name,
|
||||
<a href=#115 id=115 data-nosnippet>115</a> user_path.expect(<span class="string">"user_path required"</span>),
|
||||
<a href=#116 id=116 data-nosnippet>116</a> );
|
||||
<a href=#117 id=117 data-nosnippet>117</a> result.extend(struct_code);
|
||||
<a href=#118 id=118 data-nosnippet>118</a> }
|
||||
<a href=#119 id=119 data-nosnippet>119</a>
|
||||
<a href=#120 id=120 data-nosnippet>120</a> <span class="comment">// preserve this fn
|
||||
<a href=#121 id=121 data-nosnippet>121</a> </span><span class="kw">let </span>input_fn_code: TokenStream = <span class="macro">quote!</span> {
|
||||
<a href=#122 id=122 data-nosnippet>122</a> #sig { #block }
|
||||
<a href=#123 id=123 data-nosnippet>123</a> }
|
||||
<a href=#124 id=124 data-nosnippet>124</a> .into();
|
||||
<a href=#125 id=125 data-nosnippet>125</a>
|
||||
<a href=#126 id=126 data-nosnippet>126</a> result.extend(input_fn_code);
|
||||
<a href=#127 id=127 data-nosnippet>127</a> result
|
||||
<a href=#128 id=128 data-nosnippet>128</a>}
|
||||
<a href=#129 id=129 data-nosnippet>129</a>
|
||||
<a href=#130 id=130 data-nosnippet>130</a><span class="doccomment">/// Retrieve the handler type, `ProcedureServiceHandlerRef` or `TableMutationHandlerRef`.
|
||||
<a href=#131 id=131 data-nosnippet>131</a></span><span class="kw">fn </span>extract_handler_type(arg_types: <span class="kw-2">&</span>[Type]) -> <span class="prelude-ty">Result</span><<span class="kw-2">&</span>Ident, syn::Error> {
|
||||
<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">match </span><span class="kw-2">&</span>arg_types[<span class="number">0</span>] {
|
||||
<a href=#133 id=133 data-nosnippet>133</a> Type::Reference(TypeReference { elem, .. }) => <span class="kw">match </span><span class="kw-2">&**</span>elem {
|
||||
<a href=#134 id=134 data-nosnippet>134</a> Type::Path(TypePath { path, .. }) => <span class="prelude-val">Ok</span>(<span class="kw-2">&</span>path
|
||||
<a href=#135 id=135 data-nosnippet>135</a> .segments
|
||||
<a href=#136 id=136 data-nosnippet>136</a> .first()
|
||||
<a href=#137 id=137 data-nosnippet>137</a> .expect(<span class="string">"Expected a reference of handler"</span>)
|
||||
<a href=#138 id=138 data-nosnippet>138</a> .ident),
|
||||
<a href=#139 id=139 data-nosnippet>139</a> other => {
|
||||
<a href=#140 id=140 data-nosnippet>140</a> <span class="macro">error!</span>(other.span(), <span class="string">"Expected a reference of handler"</span>)
|
||||
<a href=#141 id=141 data-nosnippet>141</a> }
|
||||
<a href=#142 id=142 data-nosnippet>142</a> },
|
||||
<a href=#143 id=143 data-nosnippet>143</a> other => {
|
||||
<a href=#144 id=144 data-nosnippet>144</a> <span class="macro">error!</span>(other.span(), <span class="string">"Expected a reference of handler"</span>)
|
||||
<a href=#145 id=145 data-nosnippet>145</a> }
|
||||
<a href=#146 id=146 data-nosnippet>146</a> }
|
||||
<a href=#147 id=147 data-nosnippet>147</a>}
|
||||
<a href=#148 id=148 data-nosnippet>148</a>
|
||||
<a href=#149 id=149 data-nosnippet>149</a><span class="doccomment">/// Build the function struct
|
||||
<a href=#150 id=150 data-nosnippet>150</a></span><span class="attr">#[allow(clippy::too_many_arguments)]
|
||||
<a href=#151 id=151 data-nosnippet>151</a></span><span class="kw">fn </span>build_struct(
|
||||
<a href=#152 id=152 data-nosnippet>152</a> attrs: Vec<Attribute>,
|
||||
<a href=#153 id=153 data-nosnippet>153</a> vis: Visibility,
|
||||
<a href=#154 id=154 data-nosnippet>154</a> fn_name: <span class="kw-2">&</span>Ident,
|
||||
<a href=#155 id=155 data-nosnippet>155</a> name: Ident,
|
||||
<a href=#156 id=156 data-nosnippet>156</a> sig_fn: Ident,
|
||||
<a href=#157 id=157 data-nosnippet>157</a> ret: Ident,
|
||||
<a href=#158 id=158 data-nosnippet>158</a> handler_type: <span class="kw-2">&</span>Ident,
|
||||
<a href=#159 id=159 data-nosnippet>159</a> display_name_ident: Ident,
|
||||
<a href=#160 id=160 data-nosnippet>160</a> user_path: Path,
|
||||
<a href=#161 id=161 data-nosnippet>161</a>) -> TokenStream {
|
||||
<a href=#162 id=162 data-nosnippet>162</a> <span class="kw">let </span>display_name = display_name_ident.to_string();
|
||||
<a href=#163 id=163 data-nosnippet>163</a> <span class="kw">let </span>ret = Ident::new(<span class="kw-2">&</span><span class="macro">format!</span>(<span class="string">"{ret}_datatype"</span>), ret.span());
|
||||
<a href=#164 id=164 data-nosnippet>164</a> <span class="kw">let </span>uppcase_display_name = display_name.to_uppercase();
|
||||
<a href=#165 id=165 data-nosnippet>165</a> <span class="comment">// Get the handler name in function state by the argument ident
|
||||
<a href=#166 id=166 data-nosnippet>166</a> // TODO(discord9): consider simple depend injection if more handlers are needed
|
||||
<a href=#167 id=167 data-nosnippet>167</a> </span><span class="kw">let </span>(handler, snafu_type) = <span class="kw">match </span>handler_type.to_string().as_str() {
|
||||
<a href=#168 id=168 data-nosnippet>168</a> <span class="string">"ProcedureServiceHandlerRef" </span>=> (
|
||||
<a href=#169 id=169 data-nosnippet>169</a> Ident::new(<span class="string">"procedure_service_handler"</span>, handler_type.span()),
|
||||
<a href=#170 id=170 data-nosnippet>170</a> Ident::new(<span class="string">"MissingProcedureServiceHandlerSnafu"</span>, handler_type.span()),
|
||||
<a href=#171 id=171 data-nosnippet>171</a> ),
|
||||
<a href=#172 id=172 data-nosnippet>172</a>
|
||||
<a href=#173 id=173 data-nosnippet>173</a> <span class="string">"TableMutationHandlerRef" </span>=> (
|
||||
<a href=#174 id=174 data-nosnippet>174</a> Ident::new(<span class="string">"table_mutation_handler"</span>, handler_type.span()),
|
||||
<a href=#175 id=175 data-nosnippet>175</a> Ident::new(<span class="string">"MissingTableMutationHandlerSnafu"</span>, handler_type.span()),
|
||||
<a href=#176 id=176 data-nosnippet>176</a> ),
|
||||
<a href=#177 id=177 data-nosnippet>177</a>
|
||||
<a href=#178 id=178 data-nosnippet>178</a> <span class="string">"FlowServiceHandlerRef" </span>=> (
|
||||
<a href=#179 id=179 data-nosnippet>179</a> Ident::new(<span class="string">"flow_service_handler"</span>, handler_type.span()),
|
||||
<a href=#180 id=180 data-nosnippet>180</a> Ident::new(<span class="string">"MissingFlowServiceHandlerSnafu"</span>, handler_type.span()),
|
||||
<a href=#181 id=181 data-nosnippet>181</a> ),
|
||||
<a href=#182 id=182 data-nosnippet>182</a> handler => <span class="macro">ok!</span>(<span class="macro">error!</span>(
|
||||
<a href=#183 id=183 data-nosnippet>183</a> handler_type.span(),
|
||||
<a href=#184 id=184 data-nosnippet>184</a> <span class="macro">format!</span>(<span class="string">"Unknown handler type: {handler}"</span>)
|
||||
<a href=#185 id=185 data-nosnippet>185</a> )),
|
||||
<a href=#186 id=186 data-nosnippet>186</a> };
|
||||
<a href=#187 id=187 data-nosnippet>187</a>
|
||||
<a href=#188 id=188 data-nosnippet>188</a> <span class="macro">quote!</span> {
|
||||
<a href=#189 id=189 data-nosnippet>189</a> #(#attrs)*
|
||||
<a href=#190 id=190 data-nosnippet>190</a> #vis <span class="kw">struct </span>#name {
|
||||
<a href=#191 id=191 data-nosnippet>191</a> signature: datafusion_expr::Signature,
|
||||
<a href=#192 id=192 data-nosnippet>192</a> func_ctx: #user_path::function::FunctionContext,
|
||||
<a href=#193 id=193 data-nosnippet>193</a> }
|
||||
<a href=#194 id=194 data-nosnippet>194</a>
|
||||
<a href=#195 id=195 data-nosnippet>195</a> <span class="kw">impl </span>#name {
|
||||
<a href=#196 id=196 data-nosnippet>196</a> <span class="doccomment">/// Creates a new instance of the function with function context.
|
||||
<a href=#197 id=197 data-nosnippet>197</a> </span><span class="kw">fn </span>create(signature: datafusion_expr::Signature, func_ctx: #user_path::function::FunctionContext) -> <span class="self">Self </span>{
|
||||
<a href=#198 id=198 data-nosnippet>198</a> <span class="self">Self </span>{
|
||||
<a href=#199 id=199 data-nosnippet>199</a> signature,
|
||||
<a href=#200 id=200 data-nosnippet>200</a> func_ctx,
|
||||
<a href=#201 id=201 data-nosnippet>201</a> }
|
||||
<a href=#202 id=202 data-nosnippet>202</a> }
|
||||
<a href=#203 id=203 data-nosnippet>203</a>
|
||||
<a href=#204 id=204 data-nosnippet>204</a> <span class="doccomment">/// Returns the [`ScalarFunctionFactory`] of the function.
|
||||
<a href=#205 id=205 data-nosnippet>205</a> </span><span class="kw">pub fn </span>factory() -> <span class="kw">impl </span>Into< #user_path::function_factory::ScalarFunctionFactory> {
|
||||
<a href=#206 id=206 data-nosnippet>206</a> <span class="self">Self </span>{
|
||||
<a href=#207 id=207 data-nosnippet>207</a> signature: #sig_fn().into(),
|
||||
<a href=#208 id=208 data-nosnippet>208</a> func_ctx: #user_path::function::FunctionContext::default(),
|
||||
<a href=#209 id=209 data-nosnippet>209</a> }
|
||||
<a href=#210 id=210 data-nosnippet>210</a> }
|
||||
<a href=#211 id=211 data-nosnippet>211</a> }
|
||||
<a href=#212 id=212 data-nosnippet>212</a>
|
||||
<a href=#213 id=213 data-nosnippet>213</a> <span class="kw">impl </span>std::fmt::Display <span class="kw">for </span>#name {
|
||||
<a href=#214 id=214 data-nosnippet>214</a> <span class="kw">fn </span>fmt(<span class="kw-2">&</span><span class="self">self</span>, f: <span class="kw-2">&mut </span>std::fmt::Formatter) -> std::fmt::Result {
|
||||
<a href=#215 id=215 data-nosnippet>215</a> <span class="macro">write!</span>(f, #uppcase_display_name)
|
||||
<a href=#216 id=216 data-nosnippet>216</a> }
|
||||
<a href=#217 id=217 data-nosnippet>217</a> }
|
||||
<a href=#218 id=218 data-nosnippet>218</a>
|
||||
<a href=#219 id=219 data-nosnippet>219</a> <span class="kw">impl </span>std::fmt::Debug <span class="kw">for </span>#name {
|
||||
<a href=#220 id=220 data-nosnippet>220</a> <span class="kw">fn </span>fmt(<span class="kw-2">&</span><span class="self">self</span>, f: <span class="kw-2">&mut </span>std::fmt::Formatter) -> std::fmt::Result {
|
||||
<a href=#221 id=221 data-nosnippet>221</a> <span class="macro">write!</span>(f, <span class="string">"{}({})"</span>, #uppcase_display_name, <span class="self">self</span>.func_ctx)
|
||||
<a href=#222 id=222 data-nosnippet>222</a> }
|
||||
<a href=#223 id=223 data-nosnippet>223</a> }
|
||||
<a href=#224 id=224 data-nosnippet>224</a>
|
||||
<a href=#225 id=225 data-nosnippet>225</a> <span class="comment">// Implement DataFusion's ScalarUDFImpl trait
|
||||
<a href=#226 id=226 data-nosnippet>226</a> </span><span class="kw">impl </span>datafusion::logical_expr::ScalarUDFImpl <span class="kw">for </span>#name {
|
||||
<a href=#227 id=227 data-nosnippet>227</a> <span class="kw">fn </span>as_any(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="kw-2">&</span><span class="kw">dyn </span>std::any::Any {
|
||||
<a href=#228 id=228 data-nosnippet>228</a> <span class="self">self
|
||||
<a href=#229 id=229 data-nosnippet>229</a> </span>}
|
||||
<a href=#230 id=230 data-nosnippet>230</a>
|
||||
<a href=#231 id=231 data-nosnippet>231</a> <span class="kw">fn </span>name(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="kw-2">&</span>str {
|
||||
<a href=#232 id=232 data-nosnippet>232</a> #display_name
|
||||
<a href=#233 id=233 data-nosnippet>233</a> }
|
||||
<a href=#234 id=234 data-nosnippet>234</a>
|
||||
<a href=#235 id=235 data-nosnippet>235</a> <span class="kw">fn </span>signature(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="kw-2">&</span>datafusion_expr::Signature {
|
||||
<a href=#236 id=236 data-nosnippet>236</a> <span class="kw-2">&</span><span class="self">self</span>.signature
|
||||
<a href=#237 id=237 data-nosnippet>237</a> }
|
||||
<a href=#238 id=238 data-nosnippet>238</a>
|
||||
<a href=#239 id=239 data-nosnippet>239</a> <span class="kw">fn </span>return_type(<span class="kw-2">&</span><span class="self">self</span>, _arg_types: <span class="kw-2">&</span>[datafusion::arrow::datatypes::DataType]) -> datafusion_common::Result<datafusion::arrow::datatypes::DataType> {
|
||||
<a href=#240 id=240 data-nosnippet>240</a> <span class="kw">use </span>datatypes::data_type::DataType;
|
||||
<a href=#241 id=241 data-nosnippet>241</a> <span class="prelude-val">Ok</span>(store_api::storage::ConcreteDataType::#ret().as_arrow_type())
|
||||
<a href=#242 id=242 data-nosnippet>242</a> }
|
||||
<a href=#243 id=243 data-nosnippet>243</a>
|
||||
<a href=#244 id=244 data-nosnippet>244</a> <span class="kw">fn </span>invoke_with_args(
|
||||
<a href=#245 id=245 data-nosnippet>245</a> <span class="kw-2">&</span><span class="self">self</span>,
|
||||
<a href=#246 id=246 data-nosnippet>246</a> _args: datafusion::logical_expr::ScalarFunctionArgs,
|
||||
<a href=#247 id=247 data-nosnippet>247</a> ) -> datafusion_common::Result<datafusion_expr::ColumnarValue> {
|
||||
<a href=#248 id=248 data-nosnippet>248</a> <span class="prelude-val">Err</span>(datafusion_common::DataFusionError::NotImplemented(
|
||||
<a href=#249 id=249 data-nosnippet>249</a> <span class="macro">format!</span>(<span class="string">"{} can only be called from async contexts"</span>, #display_name)
|
||||
<a href=#250 id=250 data-nosnippet>250</a> ))
|
||||
<a href=#251 id=251 data-nosnippet>251</a> }
|
||||
<a href=#252 id=252 data-nosnippet>252</a> }
|
||||
<a href=#253 id=253 data-nosnippet>253</a>
|
||||
<a href=#254 id=254 data-nosnippet>254</a> <span class="doccomment">/// Implement From trait for ScalarFunctionFactory
|
||||
<a href=#255 id=255 data-nosnippet>255</a> </span><span class="kw">impl </span>From<#name> <span class="kw">for </span>#user_path::function_factory::ScalarFunctionFactory {
|
||||
<a href=#256 id=256 data-nosnippet>256</a> <span class="kw">fn </span>from(func: #name) -> <span class="self">Self </span>{
|
||||
<a href=#257 id=257 data-nosnippet>257</a> <span class="kw">use </span>std::sync::Arc;
|
||||
<a href=#258 id=258 data-nosnippet>258</a> <span class="kw">use </span>datafusion_expr::ScalarUDFImpl;
|
||||
<a href=#259 id=259 data-nosnippet>259</a> <span class="kw">use </span>datafusion_expr::async_udf::AsyncScalarUDF;
|
||||
<a href=#260 id=260 data-nosnippet>260</a>
|
||||
<a href=#261 id=261 data-nosnippet>261</a> <span class="kw">let </span>name = func.name().to_string();
|
||||
<a href=#262 id=262 data-nosnippet>262</a>
|
||||
<a href=#263 id=263 data-nosnippet>263</a> <span class="kw">let </span>func = Arc::new(<span class="kw">move </span>|ctx: #user_path::function::FunctionContext| {
|
||||
<a href=#264 id=264 data-nosnippet>264</a> <span class="comment">// create the UDF dynamically with function context
|
||||
<a href=#265 id=265 data-nosnippet>265</a> </span><span class="kw">let </span>udf_impl = #name::create(func.signature.clone(), ctx);
|
||||
<a href=#266 id=266 data-nosnippet>266</a> <span class="kw">let </span>async_udf = AsyncScalarUDF::new(Arc::new(udf_impl));
|
||||
<a href=#267 id=267 data-nosnippet>267</a> async_udf.into_scalar_udf()
|
||||
<a href=#268 id=268 data-nosnippet>268</a> });
|
||||
<a href=#269 id=269 data-nosnippet>269</a> <span class="self">Self </span>{
|
||||
<a href=#270 id=270 data-nosnippet>270</a> name,
|
||||
<a href=#271 id=271 data-nosnippet>271</a> factory: func,
|
||||
<a href=#272 id=272 data-nosnippet>272</a> }
|
||||
<a href=#273 id=273 data-nosnippet>273</a> }
|
||||
<a href=#274 id=274 data-nosnippet>274</a> }
|
||||
<a href=#275 id=275 data-nosnippet>275</a>
|
||||
<a href=#276 id=276 data-nosnippet>276</a> <span class="comment">// Implement DataFusion's AsyncScalarUDFImpl trait
|
||||
<a href=#277 id=277 data-nosnippet>277</a> </span><span class="attr">#[async_trait::async_trait]
|
||||
<a href=#278 id=278 data-nosnippet>278</a> </span><span class="kw">impl </span>datafusion_expr::async_udf::AsyncScalarUDFImpl <span class="kw">for </span>#name {
|
||||
<a href=#279 id=279 data-nosnippet>279</a> <span class="kw">async fn </span>invoke_async_with_args(
|
||||
<a href=#280 id=280 data-nosnippet>280</a> <span class="kw-2">&</span><span class="self">self</span>,
|
||||
<a href=#281 id=281 data-nosnippet>281</a> args: datafusion::logical_expr::ScalarFunctionArgs,
|
||||
<a href=#282 id=282 data-nosnippet>282</a> ) -> datafusion_common::Result<datafusion_expr::ColumnarValue> {
|
||||
<a href=#283 id=283 data-nosnippet>283</a> <span class="kw">use </span>common_error::ext::ErrorExt;
|
||||
<a href=#284 id=284 data-nosnippet>284</a>
|
||||
<a href=#285 id=285 data-nosnippet>285</a> <span class="kw">let </span>columns = args.args
|
||||
<a href=#286 id=286 data-nosnippet>286</a> .iter()
|
||||
<a href=#287 id=287 data-nosnippet>287</a> .map(|arg| {
|
||||
<a href=#288 id=288 data-nosnippet>288</a> common_query::prelude::ColumnarValue::try_from(arg)
|
||||
<a href=#289 id=289 data-nosnippet>289</a> .and_then(|cv| <span class="kw">match </span>cv {
|
||||
<a href=#290 id=290 data-nosnippet>290</a> common_query::prelude::ColumnarValue::Vector(v) => <span class="prelude-val">Ok</span>(v),
|
||||
<a href=#291 id=291 data-nosnippet>291</a> common_query::prelude::ColumnarValue::Scalar(s) => {
|
||||
<a href=#292 id=292 data-nosnippet>292</a> datatypes::vectors::Helper::try_from_scalar_value(s, args.number_rows)
|
||||
<a href=#293 id=293 data-nosnippet>293</a> .context(common_query::error::FromScalarValueSnafu)
|
||||
<a href=#294 id=294 data-nosnippet>294</a> }
|
||||
<a href=#295 id=295 data-nosnippet>295</a> })
|
||||
<a href=#296 id=296 data-nosnippet>296</a> })
|
||||
<a href=#297 id=297 data-nosnippet>297</a> .collect::<common_query::error::Result<Vec<<span class="kw">_</span>>>>()
|
||||
<a href=#298 id=298 data-nosnippet>298</a> .map_err(|e| datafusion_common::DataFusionError::Execution(<span class="macro">format!</span>(<span class="string">"Column conversion error: {}"</span>, e.output_msg())))<span class="question-mark">?</span>;
|
||||
<a href=#299 id=299 data-nosnippet>299</a>
|
||||
<a href=#300 id=300 data-nosnippet>300</a> <span class="comment">// Safety check: Ensure under the `greptime` catalog for security
|
||||
<a href=#301 id=301 data-nosnippet>301</a> </span>#<span class="macro">user_path::ensure_greptime!</span>(<span class="self">self</span>.func_ctx);
|
||||
<a href=#302 id=302 data-nosnippet>302</a>
|
||||
<a href=#303 id=303 data-nosnippet>303</a> <span class="kw">let </span>columns_num = columns.len();
|
||||
<a href=#304 id=304 data-nosnippet>304</a> <span class="kw">let </span>rows_num = <span class="kw">if </span>columns.is_empty() {
|
||||
<a href=#305 id=305 data-nosnippet>305</a> <span class="number">1
|
||||
<a href=#306 id=306 data-nosnippet>306</a> </span>} <span class="kw">else </span>{
|
||||
<a href=#307 id=307 data-nosnippet>307</a> columns[<span class="number">0</span>].len()
|
||||
<a href=#308 id=308 data-nosnippet>308</a> };
|
||||
<a href=#309 id=309 data-nosnippet>309</a>
|
||||
<a href=#310 id=310 data-nosnippet>310</a> <span class="kw">use </span>snafu::{OptionExt, ResultExt};
|
||||
<a href=#311 id=311 data-nosnippet>311</a> <span class="kw">use </span>datatypes::data_type::DataType;
|
||||
<a href=#312 id=312 data-nosnippet>312</a>
|
||||
<a href=#313 id=313 data-nosnippet>313</a> <span class="kw">let </span>query_ctx = <span class="kw-2">&</span><span class="self">self</span>.func_ctx.query_ctx;
|
||||
<a href=#314 id=314 data-nosnippet>314</a> <span class="kw">let </span>handler = <span class="self">self</span>.func_ctx
|
||||
<a href=#315 id=315 data-nosnippet>315</a> .state
|
||||
<a href=#316 id=316 data-nosnippet>316</a> .#handler
|
||||
<a href=#317 id=317 data-nosnippet>317</a> .as_ref()
|
||||
<a href=#318 id=318 data-nosnippet>318</a> .context(#snafu_type)
|
||||
<a href=#319 id=319 data-nosnippet>319</a> .map_err(|e| datafusion_common::DataFusionError::Execution(e.output_msg()))<span class="question-mark">?</span>;
|
||||
<a href=#320 id=320 data-nosnippet>320</a>
|
||||
<a href=#321 id=321 data-nosnippet>321</a> <span class="kw">let </span><span class="kw-2">mut </span>builder = store_api::storage::ConcreteDataType::#ret()
|
||||
<a href=#322 id=322 data-nosnippet>322</a> .create_mutable_vector(rows_num);
|
||||
<a href=#323 id=323 data-nosnippet>323</a>
|
||||
<a href=#324 id=324 data-nosnippet>324</a> <span class="kw">if </span>columns_num == <span class="number">0 </span>{
|
||||
<a href=#325 id=325 data-nosnippet>325</a> <span class="kw">let </span>result = #fn_name(handler, query_ctx, <span class="kw-2">&</span>[]).<span class="kw">await
|
||||
<a href=#326 id=326 data-nosnippet>326</a> </span>.map_err(|e| datafusion_common::DataFusionError::Execution(e.output_msg()))<span class="question-mark">?</span>;
|
||||
<a href=#327 id=327 data-nosnippet>327</a>
|
||||
<a href=#328 id=328 data-nosnippet>328</a> builder.push_value_ref(<span class="kw-2">&</span>result.as_value_ref());
|
||||
<a href=#329 id=329 data-nosnippet>329</a> } <span class="kw">else </span>{
|
||||
<a href=#330 id=330 data-nosnippet>330</a> <span class="kw">for </span>i <span class="kw">in </span><span class="number">0</span>..rows_num {
|
||||
<a href=#331 id=331 data-nosnippet>331</a> <span class="kw">let </span>args: Vec<<span class="kw">_</span>> = columns.iter()
|
||||
<a href=#332 id=332 data-nosnippet>332</a> .map(|vector| vector.get_ref(i))
|
||||
<a href=#333 id=333 data-nosnippet>333</a> .collect();
|
||||
<a href=#334 id=334 data-nosnippet>334</a>
|
||||
<a href=#335 id=335 data-nosnippet>335</a> <span class="kw">let </span>result = #fn_name(handler, query_ctx, <span class="kw-2">&</span>args).<span class="kw">await
|
||||
<a href=#336 id=336 data-nosnippet>336</a> </span>.map_err(|e| datafusion_common::DataFusionError::Execution(e.output_msg()))<span class="question-mark">?</span>;
|
||||
<a href=#337 id=337 data-nosnippet>337</a>
|
||||
<a href=#338 id=338 data-nosnippet>338</a> builder.push_value_ref(<span class="kw-2">&</span>result.as_value_ref());
|
||||
<a href=#339 id=339 data-nosnippet>339</a> }
|
||||
<a href=#340 id=340 data-nosnippet>340</a> }
|
||||
<a href=#341 id=341 data-nosnippet>341</a>
|
||||
<a href=#342 id=342 data-nosnippet>342</a> <span class="kw">let </span>result_vector = builder.to_vector();
|
||||
<a href=#343 id=343 data-nosnippet>343</a>
|
||||
<a href=#344 id=344 data-nosnippet>344</a> <span class="comment">// Convert result back to DataFusion ColumnarValue
|
||||
<a href=#345 id=345 data-nosnippet>345</a> </span><span class="prelude-val">Ok</span>(datafusion_expr::ColumnarValue::Array(result_vector.to_arrow_array()))
|
||||
<a href=#346 id=346 data-nosnippet>346</a> }
|
||||
<a href=#347 id=347 data-nosnippet>347</a> }
|
||||
<a href=#348 id=348 data-nosnippet>348</a>
|
||||
<a href=#349 id=349 data-nosnippet>349</a> <span class="kw">impl </span>PartialEq <span class="kw">for </span>#name {
|
||||
<a href=#350 id=350 data-nosnippet>350</a> <span class="kw">fn </span>eq(<span class="kw-2">&</span><span class="self">self</span>, other: <span class="kw-2">&</span><span class="self">Self</span>) -> bool {
|
||||
<a href=#351 id=351 data-nosnippet>351</a> <span class="self">self</span>.signature == other.signature
|
||||
<a href=#352 id=352 data-nosnippet>352</a> }
|
||||
<a href=#353 id=353 data-nosnippet>353</a> }
|
||||
<a href=#354 id=354 data-nosnippet>354</a>
|
||||
<a href=#355 id=355 data-nosnippet>355</a> <span class="kw">impl </span>Eq <span class="kw">for </span>#name {}
|
||||
<a href=#356 id=356 data-nosnippet>356</a>
|
||||
<a href=#357 id=357 data-nosnippet>357</a> <span class="kw">impl </span>std::hash::Hash <span class="kw">for </span>#name {
|
||||
<a href=#358 id=358 data-nosnippet>358</a> <span class="kw">fn </span>hash<H: std::hash::Hasher>(<span class="kw-2">&</span><span class="self">self</span>, state: <span class="kw-2">&mut </span>H) {
|
||||
<a href=#359 id=359 data-nosnippet>359</a> <span class="self">self</span>.signature.hash(state)
|
||||
<a href=#360 id=360 data-nosnippet>360</a> }
|
||||
<a href=#361 id=361 data-nosnippet>361</a> }
|
||||
<a href=#362 id=362 data-nosnippet>362</a> }
|
||||
<a href=#363 id=363 data-nosnippet>363</a> .into()
|
||||
<a href=#364 id=364 data-nosnippet>364</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
69
src/common_macro/aggr_func.rs.html
Normal file
69
src/common_macro/aggr_func.rs.html
Normal file
@@ -0,0 +1,69 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/aggr_func.rs`."><title>aggr_func.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>aggr_func.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro::TokenStream;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::{quote, quote_spanned};
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::parse::Parser;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">use </span>syn::{DeriveInput, ItemStruct, parse_macro_input};
|
||||
<a href=#20 id=20 data-nosnippet>20</a>
|
||||
<a href=#21 id=21 data-nosnippet>21</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>impl_aggr_func_type_store(ast: <span class="kw-2">&</span>DeriveInput) -> TokenStream {
|
||||
<a href=#22 id=22 data-nosnippet>22</a> <span class="kw">let </span>name = <span class="kw-2">&</span>ast.ident;
|
||||
<a href=#23 id=23 data-nosnippet>23</a> <span class="kw">let </span>generated = <span class="macro">quote!</span> {
|
||||
<a href=#24 id=24 data-nosnippet>24</a> <span class="kw">impl </span>common_query::logical_plan::accumulator::AggrFuncTypeStore <span class="kw">for </span>#name {
|
||||
<a href=#25 id=25 data-nosnippet>25</a> <span class="kw">fn </span>input_types(<span class="kw-2">&</span><span class="self">self</span>) -> std::result::Result<Vec<datatypes::prelude::ConcreteDataType>, common_query::error::Error> {
|
||||
<a href=#26 id=26 data-nosnippet>26</a> <span class="kw">let </span>input_types = <span class="self">self</span>.input_types.load();
|
||||
<a href=#27 id=27 data-nosnippet>27</a> <span class="macro">snafu::ensure!</span>(input_types.is_some(), common_query::error::InvalidInputStateSnafu);
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="prelude-val">Ok</span>(input_types.as_ref().unwrap().as_ref().clone())
|
||||
<a href=#29 id=29 data-nosnippet>29</a> }
|
||||
<a href=#30 id=30 data-nosnippet>30</a>
|
||||
<a href=#31 id=31 data-nosnippet>31</a> <span class="kw">fn </span>set_input_types(<span class="kw-2">&</span><span class="self">self</span>, input_types: Vec<datatypes::prelude::ConcreteDataType>) -> std::result::Result<(), common_query::error::Error> {
|
||||
<a href=#32 id=32 data-nosnippet>32</a> <span class="kw">let </span>old = <span class="self">self</span>.input_types.swap(<span class="prelude-val">Some</span>(std::sync::Arc::new(input_types.clone())));
|
||||
<a href=#33 id=33 data-nosnippet>33</a> <span class="kw">if let </span><span class="prelude-val">Some</span>(old) = old {
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="macro">snafu::ensure!</span>(old.len() == input_types.len(), common_query::error::InvalidInputStateSnafu);
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">for </span>(x, y) <span class="kw">in </span>old.iter().zip(input_types.iter()) {
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="macro">snafu::ensure!</span>(x == y, common_query::error::InvalidInputStateSnafu);
|
||||
<a href=#37 id=37 data-nosnippet>37</a> }
|
||||
<a href=#38 id=38 data-nosnippet>38</a> }
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#40 id=40 data-nosnippet>40</a> }
|
||||
<a href=#41 id=41 data-nosnippet>41</a> }
|
||||
<a href=#42 id=42 data-nosnippet>42</a> };
|
||||
<a href=#43 id=43 data-nosnippet>43</a> generated.into()
|
||||
<a href=#44 id=44 data-nosnippet>44</a>}
|
||||
<a href=#45 id=45 data-nosnippet>45</a>
|
||||
<a href=#46 id=46 data-nosnippet>46</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>impl_as_aggr_func_creator(_args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#47 id=47 data-nosnippet>47</a> <span class="kw">let </span><span class="kw-2">mut </span>item_struct = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>ItemStruct);
|
||||
<a href=#48 id=48 data-nosnippet>48</a> <span class="kw">if let </span>syn::Fields::Named(<span class="kw-2">ref mut </span>fields) = item_struct.fields {
|
||||
<a href=#49 id=49 data-nosnippet>49</a> <span class="kw">let </span>result = syn::Field::parse_named.parse2(<span class="macro">quote!</span> {
|
||||
<a href=#50 id=50 data-nosnippet>50</a> input_types: arc_swap::ArcSwapOption<Vec<datatypes::prelude::ConcreteDataType>>
|
||||
<a href=#51 id=51 data-nosnippet>51</a> });
|
||||
<a href=#52 id=52 data-nosnippet>52</a> <span class="kw">match </span>result {
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="prelude-val">Ok</span>(field) => fields.named.push(field),
|
||||
<a href=#54 id=54 data-nosnippet>54</a> <span class="prelude-val">Err</span>(e) => <span class="kw">return </span>e.into_compile_error().into(),
|
||||
<a href=#55 id=55 data-nosnippet>55</a> }
|
||||
<a href=#56 id=56 data-nosnippet>56</a> } <span class="kw">else </span>{
|
||||
<a href=#57 id=57 data-nosnippet>57</a> <span class="kw">return </span><span class="macro">quote_spanned!</span>(
|
||||
<a href=#58 id=58 data-nosnippet>58</a> item_struct.fields.span() => <span class="macro">compile_error!</span>(
|
||||
<a href=#59 id=59 data-nosnippet>59</a> <span class="string">"This attribute macro needs to add fields to the its annotated struct, \
|
||||
<a href=#60 id=60 data-nosnippet>60</a> so the struct must have \"{}\"."</span>)
|
||||
<a href=#61 id=61 data-nosnippet>61</a> )
|
||||
<a href=#62 id=62 data-nosnippet>62</a> .into();
|
||||
<a href=#63 id=63 data-nosnippet>63</a> }
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="macro">quote!</span> {
|
||||
<a href=#65 id=65 data-nosnippet>65</a> #item_struct
|
||||
<a href=#66 id=66 data-nosnippet>66</a> }
|
||||
<a href=#67 id=67 data-nosnippet>67</a> .into()
|
||||
<a href=#68 id=68 data-nosnippet>68</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
307
src/common_macro/lib.rs.html
Normal file
307
src/common_macro/lib.rs.html
Normal file
@@ -0,0 +1,307 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/lib.rs`."><title>lib.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>lib.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">mod </span>admin_fn;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">mod </span>aggr_func;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">mod </span>print_caller;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">mod </span>range_fn;
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">mod </span>row;
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">mod </span>stack_trace_debug;
|
||||
<a href=#21 id=21 data-nosnippet>21</a><span class="kw">mod </span>utils;
|
||||
<a href=#22 id=22 data-nosnippet>22</a>
|
||||
<a href=#23 id=23 data-nosnippet>23</a><span class="kw">use </span>aggr_func::{impl_aggr_func_type_store, impl_as_aggr_func_creator};
|
||||
<a href=#24 id=24 data-nosnippet>24</a><span class="kw">use </span>print_caller::process_print_caller;
|
||||
<a href=#25 id=25 data-nosnippet>25</a><span class="kw">use </span>proc_macro::TokenStream;
|
||||
<a href=#26 id=26 data-nosnippet>26</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#27 id=27 data-nosnippet>27</a><span class="kw">use </span>range_fn::process_range_fn;
|
||||
<a href=#28 id=28 data-nosnippet>28</a><span class="kw">use </span>syn::{Data, DeriveInput, Fields, parse_macro_input};
|
||||
<a href=#29 id=29 data-nosnippet>29</a>
|
||||
<a href=#30 id=30 data-nosnippet>30</a><span class="kw">use </span><span class="kw">crate</span>::admin_fn::process_admin_fn;
|
||||
<a href=#31 id=31 data-nosnippet>31</a><span class="kw">use </span><span class="kw">crate</span>::row::into_row::derive_into_row_impl;
|
||||
<a href=#32 id=32 data-nosnippet>32</a><span class="kw">use </span><span class="kw">crate</span>::row::schema::derive_schema_impl;
|
||||
<a href=#33 id=33 data-nosnippet>33</a><span class="kw">use </span><span class="kw">crate</span>::row::to_row::derive_to_row_impl;
|
||||
<a href=#34 id=34 data-nosnippet>34</a>
|
||||
<a href=#35 id=35 data-nosnippet>35</a><span class="doccomment">/// Make struct implemented trait [AggrFuncTypeStore], which is necessary when writing UDAF.
|
||||
<a href=#36 id=36 data-nosnippet>36</a>/// This derive macro is expect to be used along with attribute macro [macro@as_aggr_func_creator].
|
||||
<a href=#37 id=37 data-nosnippet>37</a></span><span class="attr">#[proc_macro_derive(AggrFuncTypeStore)]
|
||||
<a href=#38 id=38 data-nosnippet>38</a></span><span class="kw">pub fn </span>aggr_func_type_store_derive(input: TokenStream) -> TokenStream {
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="kw">let </span>ast = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>DeriveInput);
|
||||
<a href=#40 id=40 data-nosnippet>40</a> impl_aggr_func_type_store(<span class="kw-2">&</span>ast)
|
||||
<a href=#41 id=41 data-nosnippet>41</a>}
|
||||
<a href=#42 id=42 data-nosnippet>42</a>
|
||||
<a href=#43 id=43 data-nosnippet>43</a><span class="doccomment">/// A struct can be used as a creator for aggregate function if it has been annotated with this
|
||||
<a href=#44 id=44 data-nosnippet>44</a>/// attribute first.
|
||||
<a href=#45 id=45 data-nosnippet>45</a>///
|
||||
<a href=#46 id=46 data-nosnippet>46</a>/// This attribute add a necessary field which is intended to store the input
|
||||
<a href=#47 id=47 data-nosnippet>47</a>/// data's types to the struct.
|
||||
<a href=#48 id=48 data-nosnippet>48</a>/// This attribute is expected to be used along with derive macro [AggrFuncTypeStore].
|
||||
<a href=#49 id=49 data-nosnippet>49</a></span><span class="attr">#[proc_macro_attribute]
|
||||
<a href=#50 id=50 data-nosnippet>50</a></span><span class="kw">pub fn </span>as_aggr_func_creator(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#51 id=51 data-nosnippet>51</a> impl_as_aggr_func_creator(args, input)
|
||||
<a href=#52 id=52 data-nosnippet>52</a>}
|
||||
<a href=#53 id=53 data-nosnippet>53</a>
|
||||
<a href=#54 id=54 data-nosnippet>54</a><span class="doccomment">/// Attribute macro to convert an arithimetic function to a range function. The annotated function
|
||||
<a href=#55 id=55 data-nosnippet>55</a>/// should accept servaral arrays as input and return a single value as output.
|
||||
<a href=#56 id=56 data-nosnippet>56</a>///
|
||||
<a href=#57 id=57 data-nosnippet>57</a>/// This procedure macro can works on any number of input parameters. Return type can be either
|
||||
<a href=#58 id=58 data-nosnippet>58</a>/// primitive type or wrapped in `Option`.
|
||||
<a href=#59 id=59 data-nosnippet>59</a>///
|
||||
<a href=#60 id=60 data-nosnippet>60</a>/// # Example
|
||||
<a href=#61 id=61 data-nosnippet>61</a>/// Take `count_over_time()` in PromQL as an example:
|
||||
<a href=#62 id=62 data-nosnippet>62</a>/// ```rust, ignore
|
||||
<a href=#63 id=63 data-nosnippet>63</a>/// /// The count of all values in the specified interval.
|
||||
<a href=#64 id=64 data-nosnippet>64</a>/// #[range_fn(
|
||||
<a href=#65 id=65 data-nosnippet>65</a>/// name = "CountOverTime",
|
||||
<a href=#66 id=66 data-nosnippet>66</a>/// ret = "Float64Array",
|
||||
<a href=#67 id=67 data-nosnippet>67</a>/// display_name = "prom_count_over_time"
|
||||
<a href=#68 id=68 data-nosnippet>68</a>/// )]
|
||||
<a href=#69 id=69 data-nosnippet>69</a>/// pub fn count_over_time(_: &TimestampMillisecondArray, values: &Float64Array) -> f64 {
|
||||
<a href=#70 id=70 data-nosnippet>70</a>/// values.len() as f64
|
||||
<a href=#71 id=71 data-nosnippet>71</a>/// }
|
||||
<a href=#72 id=72 data-nosnippet>72</a>/// ```
|
||||
<a href=#73 id=73 data-nosnippet>73</a>///
|
||||
<a href=#74 id=74 data-nosnippet>74</a>/// # Arguments
|
||||
<a href=#75 id=75 data-nosnippet>75</a>/// - `name`: The name of the generated `ScalarUDF` struct.
|
||||
<a href=#76 id=76 data-nosnippet>76</a>/// - `ret`: The return type of the generated UDF function.
|
||||
<a href=#77 id=77 data-nosnippet>77</a>/// - `display_name`: The display name of the generated UDF function.
|
||||
<a href=#78 id=78 data-nosnippet>78</a></span><span class="attr">#[proc_macro_attribute]
|
||||
<a href=#79 id=79 data-nosnippet>79</a></span><span class="kw">pub fn </span>range_fn(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#80 id=80 data-nosnippet>80</a> process_range_fn(args, input)
|
||||
<a href=#81 id=81 data-nosnippet>81</a>}
|
||||
<a href=#82 id=82 data-nosnippet>82</a>
|
||||
<a href=#83 id=83 data-nosnippet>83</a><span class="doccomment">/// Attribute macro to convert a normal function to SQL administration function. The annotated function
|
||||
<a href=#84 id=84 data-nosnippet>84</a>/// should accept:
|
||||
<a href=#85 id=85 data-nosnippet>85</a>/// - `&ProcedureServiceHandlerRef` or `&TableMutationHandlerRef` or `FlowServiceHandlerRef` as the first argument,
|
||||
<a href=#86 id=86 data-nosnippet>86</a>/// - `&QueryContextRef` as the second argument, and
|
||||
<a href=#87 id=87 data-nosnippet>87</a>/// - `&[ValueRef<'_>]` as the third argument which is SQL function input values in each row.
|
||||
<a href=#88 id=88 data-nosnippet>88</a>///
|
||||
<a href=#89 id=89 data-nosnippet>89</a>/// Return type must be `common_query::error::Result<Value>`.
|
||||
<a href=#90 id=90 data-nosnippet>90</a>///
|
||||
<a href=#91 id=91 data-nosnippet>91</a>/// # Example see `common/function/src/system/procedure_state.rs`.
|
||||
<a href=#92 id=92 data-nosnippet>92</a>///
|
||||
<a href=#93 id=93 data-nosnippet>93</a>/// # Arguments
|
||||
<a href=#94 id=94 data-nosnippet>94</a>/// - `name`: The name of the generated `Function` implementation.
|
||||
<a href=#95 id=95 data-nosnippet>95</a>/// - `ret`: The return type of the generated SQL function, it will be transformed into `ConcreteDataType::{ret}_datatype()` result.
|
||||
<a href=#96 id=96 data-nosnippet>96</a>/// - `display_name`: The display name of the generated SQL function.
|
||||
<a href=#97 id=97 data-nosnippet>97</a>/// - `sig_fn`: the function to returns `Signature` of generated `Function`.
|
||||
<a href=#98 id=98 data-nosnippet>98</a>/// - `user_path`: Optional path to the trait and context (e.g., `crate`);
|
||||
<a href=#99 id=99 data-nosnippet>99</a>/// defaults to `crate` if not provided.
|
||||
<a href=#100 id=100 data-nosnippet>100</a></span><span class="attr">#[proc_macro_attribute]
|
||||
<a href=#101 id=101 data-nosnippet>101</a></span><span class="kw">pub fn </span>admin_fn(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#102 id=102 data-nosnippet>102</a> process_admin_fn(args, input)
|
||||
<a href=#103 id=103 data-nosnippet>103</a>}
|
||||
<a href=#104 id=104 data-nosnippet>104</a>
|
||||
<a href=#105 id=105 data-nosnippet>105</a><span class="doccomment">/// Attribute macro to print the caller to the annotated function.
|
||||
<a href=#106 id=106 data-nosnippet>106</a>/// The caller is printed as its filename and the call site line number.
|
||||
<a href=#107 id=107 data-nosnippet>107</a>///
|
||||
<a href=#108 id=108 data-nosnippet>108</a>/// This macro works like this: inject the tracking codes as the first statement to the annotated
|
||||
<a href=#109 id=109 data-nosnippet>109</a>/// function body. The tracking codes use [backtrace-rs](https://crates.io/crates/backtrace) to get
|
||||
<a href=#110 id=110 data-nosnippet>110</a>/// the callers. So you must dependent on the `backtrace-rs` crate.
|
||||
<a href=#111 id=111 data-nosnippet>111</a>///
|
||||
<a href=#112 id=112 data-nosnippet>112</a>/// # Arguments
|
||||
<a href=#113 id=113 data-nosnippet>113</a>/// - `depth`: The max depth of call stack to print. Optional, defaults to 1.
|
||||
<a href=#114 id=114 data-nosnippet>114</a>///
|
||||
<a href=#115 id=115 data-nosnippet>115</a>/// # Example
|
||||
<a href=#116 id=116 data-nosnippet>116</a>/// ```rust, ignore
|
||||
<a href=#117 id=117 data-nosnippet>117</a>///
|
||||
<a href=#118 id=118 data-nosnippet>118</a>/// #[print_caller(depth = 3)]
|
||||
<a href=#119 id=119 data-nosnippet>119</a>/// fn foo() {}
|
||||
<a href=#120 id=120 data-nosnippet>120</a>/// ```
|
||||
<a href=#121 id=121 data-nosnippet>121</a></span><span class="attr">#[proc_macro_attribute]
|
||||
<a href=#122 id=122 data-nosnippet>122</a></span><span class="kw">pub fn </span>print_caller(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#123 id=123 data-nosnippet>123</a> process_print_caller(args, input)
|
||||
<a href=#124 id=124 data-nosnippet>124</a>}
|
||||
<a href=#125 id=125 data-nosnippet>125</a>
|
||||
<a href=#126 id=126 data-nosnippet>126</a><span class="doccomment">/// Attribute macro to derive [std::fmt::Debug] for the annotated `Error` type.
|
||||
<a href=#127 id=127 data-nosnippet>127</a>///
|
||||
<a href=#128 id=128 data-nosnippet>128</a>/// The generated `Debug` implementation will print the error in a stack trace style. E.g.:
|
||||
<a href=#129 id=129 data-nosnippet>129</a>/// ```plaintext
|
||||
<a href=#130 id=130 data-nosnippet>130</a>/// 0: Foo error, at src/common/catalog/src/error.rs:80:10
|
||||
<a href=#131 id=131 data-nosnippet>131</a>/// 1: Bar error, at src/common/function/src/error.rs:90:10
|
||||
<a href=#132 id=132 data-nosnippet>132</a>/// 2: Root cause, invalid table name, at src/common/catalog/src/error.rs:100:10
|
||||
<a href=#133 id=133 data-nosnippet>133</a>/// ```
|
||||
<a href=#134 id=134 data-nosnippet>134</a>///
|
||||
<a href=#135 id=135 data-nosnippet>135</a>/// Notes on using this macro:
|
||||
<a href=#136 id=136 data-nosnippet>136</a>/// - `#[snafu(display)]` must present on each enum variants,
|
||||
<a href=#137 id=137 data-nosnippet>137</a>/// and should not include `location` and `source`.
|
||||
<a href=#138 id=138 data-nosnippet>138</a>/// - Only our internal error can be named `source`.
|
||||
<a href=#139 id=139 data-nosnippet>139</a>/// All external error should be `error` with an `#[snafu(source)]` annotation.
|
||||
<a href=#140 id=140 data-nosnippet>140</a>/// - `common_error` crate must be accessible.
|
||||
<a href=#141 id=141 data-nosnippet>141</a></span><span class="attr">#[proc_macro_attribute]
|
||||
<a href=#142 id=142 data-nosnippet>142</a></span><span class="kw">pub fn </span>stack_trace_debug(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#143 id=143 data-nosnippet>143</a> stack_trace_debug::stack_trace_style_impl(args.into(), input.into()).into()
|
||||
<a href=#144 id=144 data-nosnippet>144</a>}
|
||||
<a href=#145 id=145 data-nosnippet>145</a>
|
||||
<a href=#146 id=146 data-nosnippet>146</a><span class="doccomment">/// Generates implementation for `From<&TableMeta> for TableMetaBuilder`
|
||||
<a href=#147 id=147 data-nosnippet>147</a></span><span class="attr">#[proc_macro_derive(ToMetaBuilder)]
|
||||
<a href=#148 id=148 data-nosnippet>148</a></span><span class="kw">pub fn </span>derive_meta_builder(input: TokenStream) -> TokenStream {
|
||||
<a href=#149 id=149 data-nosnippet>149</a> <span class="kw">let </span>input = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>DeriveInput);
|
||||
<a href=#150 id=150 data-nosnippet>150</a>
|
||||
<a href=#151 id=151 data-nosnippet>151</a> <span class="kw">let </span>Data::Struct(data_struct) = input.data <span class="kw">else </span>{
|
||||
<a href=#152 id=152 data-nosnippet>152</a> <span class="macro">panic!</span>(<span class="string">"ToMetaBuilder can only be derived for structs"</span>);
|
||||
<a href=#153 id=153 data-nosnippet>153</a> };
|
||||
<a href=#154 id=154 data-nosnippet>154</a>
|
||||
<a href=#155 id=155 data-nosnippet>155</a> <span class="kw">let </span>Fields::Named(fields) = data_struct.fields <span class="kw">else </span>{
|
||||
<a href=#156 id=156 data-nosnippet>156</a> <span class="macro">panic!</span>(<span class="string">"ToMetaBuilder can only be derived for structs with named fields"</span>);
|
||||
<a href=#157 id=157 data-nosnippet>157</a> };
|
||||
<a href=#158 id=158 data-nosnippet>158</a>
|
||||
<a href=#159 id=159 data-nosnippet>159</a> <span class="comment">// Check that this is being applied to TableMeta struct
|
||||
<a href=#160 id=160 data-nosnippet>160</a> </span><span class="kw">if </span>input.ident != <span class="string">"TableMeta" </span>{
|
||||
<a href=#161 id=161 data-nosnippet>161</a> <span class="macro">panic!</span>(<span class="string">"ToMetaBuilder can only be derived for TableMeta struct"</span>);
|
||||
<a href=#162 id=162 data-nosnippet>162</a> }
|
||||
<a href=#163 id=163 data-nosnippet>163</a>
|
||||
<a href=#164 id=164 data-nosnippet>164</a> <span class="kw">let </span>field_init = fields.named.iter().map(|field| {
|
||||
<a href=#165 id=165 data-nosnippet>165</a> <span class="kw">let </span>field_name = field.ident.as_ref().unwrap();
|
||||
<a href=#166 id=166 data-nosnippet>166</a> <span class="macro">quote!</span> {
|
||||
<a href=#167 id=167 data-nosnippet>167</a> #field_name: Default::default(),
|
||||
<a href=#168 id=168 data-nosnippet>168</a> }
|
||||
<a href=#169 id=169 data-nosnippet>169</a> });
|
||||
<a href=#170 id=170 data-nosnippet>170</a>
|
||||
<a href=#171 id=171 data-nosnippet>171</a> <span class="kw">let </span>field_assignments = fields.named.iter().map(|field| {
|
||||
<a href=#172 id=172 data-nosnippet>172</a> <span class="kw">let </span>field_name = field.ident.as_ref().unwrap();
|
||||
<a href=#173 id=173 data-nosnippet>173</a> <span class="macro">quote!</span> {
|
||||
<a href=#174 id=174 data-nosnippet>174</a> builder.#field_name(meta.#field_name.clone());
|
||||
<a href=#175 id=175 data-nosnippet>175</a> }
|
||||
<a href=#176 id=176 data-nosnippet>176</a> });
|
||||
<a href=#177 id=177 data-nosnippet>177</a>
|
||||
<a href=#178 id=178 data-nosnippet>178</a> <span class="kw">let </span>generated = <span class="macro">quote!</span> {
|
||||
<a href=#179 id=179 data-nosnippet>179</a> <span class="kw">impl </span>From<<span class="kw-2">&</span>TableMeta> <span class="kw">for </span>TableMetaBuilder {
|
||||
<a href=#180 id=180 data-nosnippet>180</a> <span class="kw">fn </span>from(meta: <span class="kw-2">&</span>TableMeta) -> <span class="self">Self </span>{
|
||||
<a href=#181 id=181 data-nosnippet>181</a> <span class="kw">let </span><span class="kw-2">mut </span>builder = <span class="self">Self </span>{
|
||||
<a href=#182 id=182 data-nosnippet>182</a> #(#field_init)*
|
||||
<a href=#183 id=183 data-nosnippet>183</a> };
|
||||
<a href=#184 id=184 data-nosnippet>184</a>
|
||||
<a href=#185 id=185 data-nosnippet>185</a> #(#field_assignments)*
|
||||
<a href=#186 id=186 data-nosnippet>186</a> builder
|
||||
<a href=#187 id=187 data-nosnippet>187</a> }
|
||||
<a href=#188 id=188 data-nosnippet>188</a> }
|
||||
<a href=#189 id=189 data-nosnippet>189</a> };
|
||||
<a href=#190 id=190 data-nosnippet>190</a>
|
||||
<a href=#191 id=191 data-nosnippet>191</a> generated.into()
|
||||
<a href=#192 id=192 data-nosnippet>192</a>}
|
||||
<a href=#193 id=193 data-nosnippet>193</a>
|
||||
<a href=#194 id=194 data-nosnippet>194</a><span class="doccomment">/// Derive macro to convert a struct to a row.
|
||||
<a href=#195 id=195 data-nosnippet>195</a>///
|
||||
<a href=#196 id=196 data-nosnippet>196</a>/// # Example
|
||||
<a href=#197 id=197 data-nosnippet>197</a>/// ```rust, ignore
|
||||
<a href=#198 id=198 data-nosnippet>198</a>/// use api::v1::Row;
|
||||
<a href=#199 id=199 data-nosnippet>199</a>/// use api::v1::value::ValueData;
|
||||
<a href=#200 id=200 data-nosnippet>200</a>/// use api::v1::Value;
|
||||
<a href=#201 id=201 data-nosnippet>201</a>///
|
||||
<a href=#202 id=202 data-nosnippet>202</a>/// #[derive(ToRow)]
|
||||
<a href=#203 id=203 data-nosnippet>203</a>/// struct ToRowTest {
|
||||
<a href=#204 id=204 data-nosnippet>204</a>/// my_value: i32,
|
||||
<a href=#205 id=205 data-nosnippet>205</a>/// #[col(name = "string_value", datatype = "string", semantic = "tag")]
|
||||
<a href=#206 id=206 data-nosnippet>206</a>/// my_string: String,
|
||||
<a href=#207 id=207 data-nosnippet>207</a>/// my_bool: bool,
|
||||
<a href=#208 id=208 data-nosnippet>208</a>/// my_float: f32,
|
||||
<a href=#209 id=209 data-nosnippet>209</a>/// #[col(
|
||||
<a href=#210 id=210 data-nosnippet>210</a>/// name = "timestamp_value",
|
||||
<a href=#211 id=211 data-nosnippet>211</a>/// semantic = "Timestamp",
|
||||
<a href=#212 id=212 data-nosnippet>212</a>/// datatype = "TimestampMillisecond"
|
||||
<a href=#213 id=213 data-nosnippet>213</a>/// )]
|
||||
<a href=#214 id=214 data-nosnippet>214</a>/// my_timestamp: i64,
|
||||
<a href=#215 id=215 data-nosnippet>215</a>/// #[col(skip)]
|
||||
<a href=#216 id=216 data-nosnippet>216</a>/// my_skip: i32,
|
||||
<a href=#217 id=217 data-nosnippet>217</a>/// }
|
||||
<a href=#218 id=218 data-nosnippet>218</a>///
|
||||
<a href=#219 id=219 data-nosnippet>219</a>/// let row = ToRowTest {
|
||||
<a href=#220 id=220 data-nosnippet>220</a>/// my_value: 1,
|
||||
<a href=#221 id=221 data-nosnippet>221</a>/// my_string: "test".to_string(),
|
||||
<a href=#222 id=222 data-nosnippet>222</a>/// my_bool: true,
|
||||
<a href=#223 id=223 data-nosnippet>223</a>/// my_float: 1.0,
|
||||
<a href=#224 id=224 data-nosnippet>224</a>/// my_timestamp: 1718563200000,
|
||||
<a href=#225 id=225 data-nosnippet>225</a>/// my_skip: 1,
|
||||
<a href=#226 id=226 data-nosnippet>226</a>/// }.to_row();
|
||||
<a href=#227 id=227 data-nosnippet>227</a>/// ```
|
||||
<a href=#228 id=228 data-nosnippet>228</a></span><span class="attr">#[proc_macro_derive(ToRow, attributes(col))]
|
||||
<a href=#229 id=229 data-nosnippet>229</a></span><span class="kw">pub fn </span>derive_to_row(input: TokenStream) -> TokenStream {
|
||||
<a href=#230 id=230 data-nosnippet>230</a> <span class="kw">let </span>input = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>DeriveInput);
|
||||
<a href=#231 id=231 data-nosnippet>231</a> <span class="kw">let </span>output = derive_to_row_impl(input);
|
||||
<a href=#232 id=232 data-nosnippet>232</a> output.unwrap_or_else(|e| e.to_compile_error()).into()
|
||||
<a href=#233 id=233 data-nosnippet>233</a>}
|
||||
<a href=#234 id=234 data-nosnippet>234</a>
|
||||
<a href=#235 id=235 data-nosnippet>235</a><span class="doccomment">/// Derive macro to convert a struct to a row with move semantics.
|
||||
<a href=#236 id=236 data-nosnippet>236</a>///
|
||||
<a href=#237 id=237 data-nosnippet>237</a>/// # Example
|
||||
<a href=#238 id=238 data-nosnippet>238</a>/// ```rust, ignore
|
||||
<a href=#239 id=239 data-nosnippet>239</a>/// use api::v1::Row;
|
||||
<a href=#240 id=240 data-nosnippet>240</a>/// use api::v1::value::ValueData;
|
||||
<a href=#241 id=241 data-nosnippet>241</a>/// use api::v1::Value;
|
||||
<a href=#242 id=242 data-nosnippet>242</a>///
|
||||
<a href=#243 id=243 data-nosnippet>243</a>/// #[derive(IntoRow)]
|
||||
<a href=#244 id=244 data-nosnippet>244</a>/// struct IntoRowTest {
|
||||
<a href=#245 id=245 data-nosnippet>245</a>/// my_value: i32,
|
||||
<a href=#246 id=246 data-nosnippet>246</a>/// #[col(name = "string_value", datatype = "string", semantic = "tag")]
|
||||
<a href=#247 id=247 data-nosnippet>247</a>/// my_string: String,
|
||||
<a href=#248 id=248 data-nosnippet>248</a>/// my_bool: bool,
|
||||
<a href=#249 id=249 data-nosnippet>249</a>/// my_float: f32,
|
||||
<a href=#250 id=250 data-nosnippet>250</a>/// #[col(
|
||||
<a href=#251 id=251 data-nosnippet>251</a>/// name = "timestamp_value",
|
||||
<a href=#252 id=252 data-nosnippet>252</a>/// semantic = "Timestamp",
|
||||
<a href=#253 id=253 data-nosnippet>253</a>/// datatype = "TimestampMillisecond"
|
||||
<a href=#254 id=254 data-nosnippet>254</a>/// )]
|
||||
<a href=#255 id=255 data-nosnippet>255</a>/// my_timestamp: i64,
|
||||
<a href=#256 id=256 data-nosnippet>256</a>/// #[col(skip)]
|
||||
<a href=#257 id=257 data-nosnippet>257</a>/// my_skip: i32,
|
||||
<a href=#258 id=258 data-nosnippet>258</a>/// }
|
||||
<a href=#259 id=259 data-nosnippet>259</a>///
|
||||
<a href=#260 id=260 data-nosnippet>260</a>/// let row = IntoRowTest {
|
||||
<a href=#261 id=261 data-nosnippet>261</a>/// my_value: 1,
|
||||
<a href=#262 id=262 data-nosnippet>262</a>/// my_string: "test".to_string(),
|
||||
<a href=#263 id=263 data-nosnippet>263</a>/// my_bool: true,
|
||||
<a href=#264 id=264 data-nosnippet>264</a>/// my_float: 1.0,
|
||||
<a href=#265 id=265 data-nosnippet>265</a>/// my_timestamp: 1718563200000,
|
||||
<a href=#266 id=266 data-nosnippet>266</a>/// my_skip: 1,
|
||||
<a href=#267 id=267 data-nosnippet>267</a>/// }.into_row();
|
||||
<a href=#268 id=268 data-nosnippet>268</a>/// ```
|
||||
<a href=#269 id=269 data-nosnippet>269</a></span><span class="attr">#[proc_macro_derive(IntoRow, attributes(col))]
|
||||
<a href=#270 id=270 data-nosnippet>270</a></span><span class="kw">pub fn </span>derive_into_row(input: TokenStream) -> TokenStream {
|
||||
<a href=#271 id=271 data-nosnippet>271</a> <span class="kw">let </span>input = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>DeriveInput);
|
||||
<a href=#272 id=272 data-nosnippet>272</a> <span class="kw">let </span>output = derive_into_row_impl(input);
|
||||
<a href=#273 id=273 data-nosnippet>273</a> output.unwrap_or_else(|e| e.to_compile_error()).into()
|
||||
<a href=#274 id=274 data-nosnippet>274</a>}
|
||||
<a href=#275 id=275 data-nosnippet>275</a>
|
||||
<a href=#276 id=276 data-nosnippet>276</a><span class="doccomment">/// Derive macro to convert a struct to a schema.
|
||||
<a href=#277 id=277 data-nosnippet>277</a>///
|
||||
<a href=#278 id=278 data-nosnippet>278</a>/// # Example
|
||||
<a href=#279 id=279 data-nosnippet>279</a>/// ```rust, ignore
|
||||
<a href=#280 id=280 data-nosnippet>280</a>/// use api::v1::ColumnSchema;
|
||||
<a href=#281 id=281 data-nosnippet>281</a>///
|
||||
<a href=#282 id=282 data-nosnippet>282</a>/// #[derive(Schema)]
|
||||
<a href=#283 id=283 data-nosnippet>283</a>/// struct SchemaTest {
|
||||
<a href=#284 id=284 data-nosnippet>284</a>/// my_value: i32,
|
||||
<a href=#285 id=285 data-nosnippet>285</a>/// #[col(name = "string_value", datatype = "string", semantic = "tag")]
|
||||
<a href=#286 id=286 data-nosnippet>286</a>/// my_string: String,
|
||||
<a href=#287 id=287 data-nosnippet>287</a>/// my_bool: bool,
|
||||
<a href=#288 id=288 data-nosnippet>288</a>/// my_float: f32,
|
||||
<a href=#289 id=289 data-nosnippet>289</a>/// #[col(
|
||||
<a href=#290 id=290 data-nosnippet>290</a>/// name = "timestamp_value",
|
||||
<a href=#291 id=291 data-nosnippet>291</a>/// semantic = "Timestamp",
|
||||
<a href=#292 id=292 data-nosnippet>292</a>/// datatype = "TimestampMillisecond"
|
||||
<a href=#293 id=293 data-nosnippet>293</a>/// )]
|
||||
<a href=#294 id=294 data-nosnippet>294</a>/// my_timestamp: i64,
|
||||
<a href=#295 id=295 data-nosnippet>295</a>/// #[col(skip)]
|
||||
<a href=#296 id=296 data-nosnippet>296</a>/// my_skip: i32,
|
||||
<a href=#297 id=297 data-nosnippet>297</a>/// }
|
||||
<a href=#298 id=298 data-nosnippet>298</a>///
|
||||
<a href=#299 id=299 data-nosnippet>299</a>/// let schema = SchemaTest::schema();
|
||||
<a href=#300 id=300 data-nosnippet>300</a>/// ```
|
||||
<a href=#301 id=301 data-nosnippet>301</a></span><span class="attr">#[proc_macro_derive(Schema, attributes(col))]
|
||||
<a href=#302 id=302 data-nosnippet>302</a></span><span class="kw">pub fn </span>derive_schema(input: TokenStream) -> TokenStream {
|
||||
<a href=#303 id=303 data-nosnippet>303</a> <span class="kw">let </span>input = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>DeriveInput);
|
||||
<a href=#304 id=304 data-nosnippet>304</a> <span class="kw">let </span>output = derive_schema_impl(input);
|
||||
<a href=#305 id=305 data-nosnippet>305</a> output.unwrap_or_else(|e| e.to_compile_error()).into()
|
||||
<a href=#306 id=306 data-nosnippet>306</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
105
src/common_macro/print_caller.rs.html
Normal file
105
src/common_macro/print_caller.rs.html
Normal file
@@ -0,0 +1,105 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/print_caller.rs`."><title>print_caller.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>print_caller.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro::TokenStream;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::{ToTokens, quote};
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::{ItemFn, LitInt, parse_macro_input};
|
||||
<a href=#18 id=18 data-nosnippet>18</a>
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>process_print_caller(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#20 id=20 data-nosnippet>20</a> <span class="kw">let </span><span class="kw-2">mut </span>depth = <span class="number">1</span>;
|
||||
<a href=#21 id=21 data-nosnippet>21</a> <span class="kw">let </span>parser = syn::meta::parser(|meta| {
|
||||
<a href=#22 id=22 data-nosnippet>22</a> <span class="kw">if </span>meta.path.is_ident(<span class="string">"depth"</span>) {
|
||||
<a href=#23 id=23 data-nosnippet>23</a> depth = meta
|
||||
<a href=#24 id=24 data-nosnippet>24</a> .value()<span class="question-mark">?
|
||||
<a href=#25 id=25 data-nosnippet>25</a> </span>.parse::<LitInt>()
|
||||
<a href=#26 id=26 data-nosnippet>26</a> .and_then(|v| v.base10_parse::<usize>())
|
||||
<a href=#27 id=27 data-nosnippet>27</a> .expect(<span class="string">"Invalid 'depth' value"</span>);
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#29 id=29 data-nosnippet>29</a> } <span class="kw">else </span>{
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="prelude-val">Err</span>(meta.error(<span class="string">"unsupported property"</span>))
|
||||
<a href=#31 id=31 data-nosnippet>31</a> }
|
||||
<a href=#32 id=32 data-nosnippet>32</a> });
|
||||
<a href=#33 id=33 data-nosnippet>33</a>
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="macro">parse_macro_input!</span>(args with parser);
|
||||
<a href=#35 id=35 data-nosnippet>35</a>
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="kw">let </span>tokens: TokenStream = <span class="macro">quote!</span> {
|
||||
<a href=#37 id=37 data-nosnippet>37</a> {
|
||||
<a href=#38 id=38 data-nosnippet>38</a> <span class="kw">let </span>curr_file = <span class="macro">file!</span>();
|
||||
<a href=#39 id=39 data-nosnippet>39</a>
|
||||
<a href=#40 id=40 data-nosnippet>40</a> <span class="kw">let </span>bt = backtrace::Backtrace::new();
|
||||
<a href=#41 id=41 data-nosnippet>41</a> <span class="kw">let </span>call_stack = bt
|
||||
<a href=#42 id=42 data-nosnippet>42</a> .frames()
|
||||
<a href=#43 id=43 data-nosnippet>43</a> .iter()
|
||||
<a href=#44 id=44 data-nosnippet>44</a> .skip_while(|f| {
|
||||
<a href=#45 id=45 data-nosnippet>45</a> !f.symbols().iter().any(|s| {
|
||||
<a href=#46 id=46 data-nosnippet>46</a> s.filename()
|
||||
<a href=#47 id=47 data-nosnippet>47</a> .map(|p| p.ends_with(curr_file))
|
||||
<a href=#48 id=48 data-nosnippet>48</a> .unwrap_or(<span class="bool-val">false</span>)
|
||||
<a href=#49 id=49 data-nosnippet>49</a> })
|
||||
<a href=#50 id=50 data-nosnippet>50</a> })
|
||||
<a href=#51 id=51 data-nosnippet>51</a> .skip(<span class="number">1</span>)
|
||||
<a href=#52 id=52 data-nosnippet>52</a> .take(#depth);
|
||||
<a href=#53 id=53 data-nosnippet>53</a>
|
||||
<a href=#54 id=54 data-nosnippet>54</a> <span class="kw">let </span>call_stack = call_stack
|
||||
<a href=#55 id=55 data-nosnippet>55</a> .map(|f| {
|
||||
<a href=#56 id=56 data-nosnippet>56</a> f.symbols()
|
||||
<a href=#57 id=57 data-nosnippet>57</a> .iter()
|
||||
<a href=#58 id=58 data-nosnippet>58</a> .map(|s| {
|
||||
<a href=#59 id=59 data-nosnippet>59</a> <span class="kw">let </span>filename = s
|
||||
<a href=#60 id=60 data-nosnippet>60</a> .filename()
|
||||
<a href=#61 id=61 data-nosnippet>61</a> .map(|p| <span class="macro">format!</span>(<span class="string">"{:?}"</span>, p))
|
||||
<a href=#62 id=62 data-nosnippet>62</a> .unwrap_or_else(|| <span class="string">"unknown"</span>.to_string());
|
||||
<a href=#63 id=63 data-nosnippet>63</a>
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="kw">let </span>lineno = s
|
||||
<a href=#65 id=65 data-nosnippet>65</a> .lineno()
|
||||
<a href=#66 id=66 data-nosnippet>66</a> .map(|l| <span class="macro">format!</span>(<span class="string">"{}"</span>, l))
|
||||
<a href=#67 id=67 data-nosnippet>67</a> .unwrap_or_else(|| <span class="string">"unknown"</span>.to_string());
|
||||
<a href=#68 id=68 data-nosnippet>68</a>
|
||||
<a href=#69 id=69 data-nosnippet>69</a> <span class="macro">format!</span>(<span class="string">"filename: {}, lineno: {}"</span>, filename, lineno)
|
||||
<a href=#70 id=70 data-nosnippet>70</a> })
|
||||
<a href=#71 id=71 data-nosnippet>71</a> .collect::<Vec<String>>()
|
||||
<a href=#72 id=72 data-nosnippet>72</a> .join(<span class="string">", "</span>)
|
||||
<a href=#73 id=73 data-nosnippet>73</a> })
|
||||
<a href=#74 id=74 data-nosnippet>74</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#75 id=75 data-nosnippet>75</a>
|
||||
<a href=#76 id=76 data-nosnippet>76</a> <span class="kw">match </span>call_stack.len() {
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="number">0 </span>=> <span class="macro">common_telemetry::info!</span>(<span class="string">"unable to find call stack"</span>),
|
||||
<a href=#78 id=78 data-nosnippet>78</a> <span class="number">1 </span>=> <span class="macro">common_telemetry::info!</span>(<span class="string">"caller: {}"</span>, call_stack[<span class="number">0</span>]),
|
||||
<a href=#79 id=79 data-nosnippet>79</a> <span class="kw">_ </span>=> {
|
||||
<a href=#80 id=80 data-nosnippet>80</a> <span class="kw">let </span><span class="kw-2">mut </span>s = String::new();
|
||||
<a href=#81 id=81 data-nosnippet>81</a> s.push_str(<span class="string">"[\n"</span>);
|
||||
<a href=#82 id=82 data-nosnippet>82</a> <span class="kw">for </span>e <span class="kw">in </span>call_stack {
|
||||
<a href=#83 id=83 data-nosnippet>83</a> s.push_str(<span class="string">"\t"</span>);
|
||||
<a href=#84 id=84 data-nosnippet>84</a> s.push_str(<span class="kw-2">&</span>e);
|
||||
<a href=#85 id=85 data-nosnippet>85</a> s.push_str(<span class="string">"\n"</span>);
|
||||
<a href=#86 id=86 data-nosnippet>86</a> }
|
||||
<a href=#87 id=87 data-nosnippet>87</a> s.push_str(<span class="string">"]"</span>);
|
||||
<a href=#88 id=88 data-nosnippet>88</a> <span class="macro">common_telemetry::info!</span>(<span class="string">"call stack: {}"</span>, s)
|
||||
<a href=#89 id=89 data-nosnippet>89</a> }
|
||||
<a href=#90 id=90 data-nosnippet>90</a> }
|
||||
<a href=#91 id=91 data-nosnippet>91</a> }
|
||||
<a href=#92 id=92 data-nosnippet>92</a> }
|
||||
<a href=#93 id=93 data-nosnippet>93</a> .into();
|
||||
<a href=#94 id=94 data-nosnippet>94</a>
|
||||
<a href=#95 id=95 data-nosnippet>95</a> <span class="kw">let </span>stmt = <span class="kw">match </span>syn::parse(tokens) {
|
||||
<a href=#96 id=96 data-nosnippet>96</a> <span class="prelude-val">Ok</span>(stmt) => stmt,
|
||||
<a href=#97 id=97 data-nosnippet>97</a> <span class="prelude-val">Err</span>(e) => <span class="kw">return </span>e.into_compile_error().into(),
|
||||
<a href=#98 id=98 data-nosnippet>98</a> };
|
||||
<a href=#99 id=99 data-nosnippet>99</a>
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="kw">let </span><span class="kw-2">mut </span>item = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>ItemFn);
|
||||
<a href=#101 id=101 data-nosnippet>101</a> item.block.stmts.insert(<span class="number">0</span>, stmt);
|
||||
<a href=#102 id=102 data-nosnippet>102</a>
|
||||
<a href=#103 id=103 data-nosnippet>103</a> item.into_token_stream().into()
|
||||
<a href=#104 id=104 data-nosnippet>104</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
231
src/common_macro/range_fn.rs.html
Normal file
231
src/common_macro/range_fn.rs.html
Normal file
@@ -0,0 +1,231 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/range_fn.rs`."><title>range_fn.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>range_fn.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro::TokenStream;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::{
|
||||
<a href=#19 id=19 data-nosnippet>19</a> Attribute, Ident, ItemFn, Signature, Type, TypeReference, Visibility, parse_macro_input,
|
||||
<a href=#20 id=20 data-nosnippet>20</a>};
|
||||
<a href=#21 id=21 data-nosnippet>21</a>
|
||||
<a href=#22 id=22 data-nosnippet>22</a><span class="kw">use </span><span class="kw">crate</span>::utils::extract_input_types;
|
||||
<a href=#23 id=23 data-nosnippet>23</a>
|
||||
<a href=#24 id=24 data-nosnippet>24</a><span class="macro">macro_rules!</span> ok {
|
||||
<a href=#25 id=25 data-nosnippet>25</a> (<span class="macro-nonterminal">$item</span>:expr) => {
|
||||
<a href=#26 id=26 data-nosnippet>26</a> <span class="kw">match </span><span class="macro-nonterminal">$item </span>{
|
||||
<a href=#27 id=27 data-nosnippet>27</a> <span class="prelude-val">Ok</span>(item) => item,
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="prelude-val">Err</span>(e) => <span class="kw">return </span>e.into_compile_error().into(),
|
||||
<a href=#29 id=29 data-nosnippet>29</a> }
|
||||
<a href=#30 id=30 data-nosnippet>30</a> };
|
||||
<a href=#31 id=31 data-nosnippet>31</a>}
|
||||
<a href=#32 id=32 data-nosnippet>32</a>
|
||||
<a href=#33 id=33 data-nosnippet>33</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>process_range_fn(args: TokenStream, input: TokenStream) -> TokenStream {
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="kw">let </span><span class="kw-2">mut </span>name: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">let </span><span class="kw-2">mut </span>display_name: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="kw">let </span><span class="kw-2">mut </span>ret: <span class="prelude-ty">Option</span><Ident> = <span class="prelude-val">None</span>;
|
||||
<a href=#37 id=37 data-nosnippet>37</a>
|
||||
<a href=#38 id=38 data-nosnippet>38</a> <span class="kw">let </span>parser = syn::meta::parser(|meta| {
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="kw">if </span>meta.path.is_ident(<span class="string">"name"</span>) {
|
||||
<a href=#40 id=40 data-nosnippet>40</a> name = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#41 id=41 data-nosnippet>41</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#42 id=42 data-nosnippet>42</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"display_name"</span>) {
|
||||
<a href=#43 id=43 data-nosnippet>43</a> display_name = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#44 id=44 data-nosnippet>44</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#45 id=45 data-nosnippet>45</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"ret"</span>) {
|
||||
<a href=#46 id=46 data-nosnippet>46</a> ret = <span class="prelude-val">Some</span>(meta.value()<span class="question-mark">?</span>.parse()<span class="question-mark">?</span>);
|
||||
<a href=#47 id=47 data-nosnippet>47</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#48 id=48 data-nosnippet>48</a> } <span class="kw">else </span>{
|
||||
<a href=#49 id=49 data-nosnippet>49</a> <span class="prelude-val">Err</span>(meta.error(<span class="string">"unsupported property"</span>))
|
||||
<a href=#50 id=50 data-nosnippet>50</a> }
|
||||
<a href=#51 id=51 data-nosnippet>51</a> });
|
||||
<a href=#52 id=52 data-nosnippet>52</a>
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="comment">// extract arg map
|
||||
<a href=#54 id=54 data-nosnippet>54</a> </span><span class="macro">parse_macro_input!</span>(args with parser);
|
||||
<a href=#55 id=55 data-nosnippet>55</a>
|
||||
<a href=#56 id=56 data-nosnippet>56</a> <span class="comment">// decompose the fn block
|
||||
<a href=#57 id=57 data-nosnippet>57</a> </span><span class="kw">let </span>compute_fn = <span class="macro">parse_macro_input!</span>(input <span class="kw">as </span>ItemFn);
|
||||
<a href=#58 id=58 data-nosnippet>58</a> <span class="kw">let </span>ItemFn {
|
||||
<a href=#59 id=59 data-nosnippet>59</a> attrs,
|
||||
<a href=#60 id=60 data-nosnippet>60</a> vis,
|
||||
<a href=#61 id=61 data-nosnippet>61</a> sig,
|
||||
<a href=#62 id=62 data-nosnippet>62</a> block,
|
||||
<a href=#63 id=63 data-nosnippet>63</a> } = compute_fn;
|
||||
<a href=#64 id=64 data-nosnippet>64</a>
|
||||
<a href=#65 id=65 data-nosnippet>65</a> <span class="comment">// extract fn arg list
|
||||
<a href=#66 id=66 data-nosnippet>66</a> </span><span class="kw">let </span>Signature {
|
||||
<a href=#67 id=67 data-nosnippet>67</a> inputs,
|
||||
<a href=#68 id=68 data-nosnippet>68</a> ident: fn_name,
|
||||
<a href=#69 id=69 data-nosnippet>69</a> ..
|
||||
<a href=#70 id=70 data-nosnippet>70</a> } = <span class="kw-2">&</span>sig;
|
||||
<a href=#71 id=71 data-nosnippet>71</a> <span class="kw">let </span>arg_types = <span class="macro">ok!</span>(extract_input_types(inputs));
|
||||
<a href=#72 id=72 data-nosnippet>72</a>
|
||||
<a href=#73 id=73 data-nosnippet>73</a> <span class="comment">// with format like Float64Array
|
||||
<a href=#74 id=74 data-nosnippet>74</a> </span><span class="kw">let </span>array_types = arg_types
|
||||
<a href=#75 id=75 data-nosnippet>75</a> .iter()
|
||||
<a href=#76 id=76 data-nosnippet>76</a> .map(|ty| {
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="kw">if let </span>Type::Reference(TypeReference { elem, .. }) = ty {
|
||||
<a href=#78 id=78 data-nosnippet>78</a> elem.as_ref().clone()
|
||||
<a href=#79 id=79 data-nosnippet>79</a> } <span class="kw">else </span>{
|
||||
<a href=#80 id=80 data-nosnippet>80</a> ty.clone()
|
||||
<a href=#81 id=81 data-nosnippet>81</a> }
|
||||
<a href=#82 id=82 data-nosnippet>82</a> })
|
||||
<a href=#83 id=83 data-nosnippet>83</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#84 id=84 data-nosnippet>84</a>
|
||||
<a href=#85 id=85 data-nosnippet>85</a> <span class="kw">let </span><span class="kw-2">mut </span>result = TokenStream::new();
|
||||
<a href=#86 id=86 data-nosnippet>86</a>
|
||||
<a href=#87 id=87 data-nosnippet>87</a> <span class="comment">// build the struct and its impl block
|
||||
<a href=#88 id=88 data-nosnippet>88</a> // only do this when `display_name` is specified
|
||||
<a href=#89 id=89 data-nosnippet>89</a> </span><span class="kw">if let </span><span class="prelude-val">Some</span>(display_name) = display_name {
|
||||
<a href=#90 id=90 data-nosnippet>90</a> <span class="kw">let </span>struct_code = build_struct(
|
||||
<a href=#91 id=91 data-nosnippet>91</a> attrs,
|
||||
<a href=#92 id=92 data-nosnippet>92</a> vis,
|
||||
<a href=#93 id=93 data-nosnippet>93</a> name.clone().expect(<span class="string">"name required"</span>),
|
||||
<a href=#94 id=94 data-nosnippet>94</a> display_name,
|
||||
<a href=#95 id=95 data-nosnippet>95</a> array_types,
|
||||
<a href=#96 id=96 data-nosnippet>96</a> ret.clone().expect(<span class="string">"ret required"</span>),
|
||||
<a href=#97 id=97 data-nosnippet>97</a> );
|
||||
<a href=#98 id=98 data-nosnippet>98</a> result.extend(struct_code);
|
||||
<a href=#99 id=99 data-nosnippet>99</a> }
|
||||
<a href=#100 id=100 data-nosnippet>100</a>
|
||||
<a href=#101 id=101 data-nosnippet>101</a> <span class="kw">let </span>calc_fn_code = build_calc_fn(
|
||||
<a href=#102 id=102 data-nosnippet>102</a> name.expect(<span class="string">"name required"</span>),
|
||||
<a href=#103 id=103 data-nosnippet>103</a> arg_types,
|
||||
<a href=#104 id=104 data-nosnippet>104</a> fn_name.clone(),
|
||||
<a href=#105 id=105 data-nosnippet>105</a> ret.expect(<span class="string">"ret required"</span>),
|
||||
<a href=#106 id=106 data-nosnippet>106</a> );
|
||||
<a href=#107 id=107 data-nosnippet>107</a> <span class="comment">// preserve this fn, but remove its `pub` modifier
|
||||
<a href=#108 id=108 data-nosnippet>108</a> </span><span class="kw">let </span>input_fn_code: TokenStream = <span class="macro">quote!</span> {
|
||||
<a href=#109 id=109 data-nosnippet>109</a> #sig { #block }
|
||||
<a href=#110 id=110 data-nosnippet>110</a> }
|
||||
<a href=#111 id=111 data-nosnippet>111</a> .into();
|
||||
<a href=#112 id=112 data-nosnippet>112</a>
|
||||
<a href=#113 id=113 data-nosnippet>113</a> result.extend(calc_fn_code);
|
||||
<a href=#114 id=114 data-nosnippet>114</a> result.extend(input_fn_code);
|
||||
<a href=#115 id=115 data-nosnippet>115</a> result
|
||||
<a href=#116 id=116 data-nosnippet>116</a>}
|
||||
<a href=#117 id=117 data-nosnippet>117</a>
|
||||
<a href=#118 id=118 data-nosnippet>118</a><span class="kw">fn </span>build_struct(
|
||||
<a href=#119 id=119 data-nosnippet>119</a> attrs: Vec<Attribute>,
|
||||
<a href=#120 id=120 data-nosnippet>120</a> vis: Visibility,
|
||||
<a href=#121 id=121 data-nosnippet>121</a> name: Ident,
|
||||
<a href=#122 id=122 data-nosnippet>122</a> display_name_ident: Ident,
|
||||
<a href=#123 id=123 data-nosnippet>123</a> array_types: Vec<Type>,
|
||||
<a href=#124 id=124 data-nosnippet>124</a> return_array_type: Ident,
|
||||
<a href=#125 id=125 data-nosnippet>125</a>) -> TokenStream {
|
||||
<a href=#126 id=126 data-nosnippet>126</a> <span class="kw">let </span>display_name = display_name_ident.to_string();
|
||||
<a href=#127 id=127 data-nosnippet>127</a> <span class="macro">quote!</span> {
|
||||
<a href=#128 id=128 data-nosnippet>128</a> #(#attrs)*
|
||||
<a href=#129 id=129 data-nosnippet>129</a> <span class="attr">#[derive(Debug)]
|
||||
<a href=#130 id=130 data-nosnippet>130</a> </span>#vis <span class="kw">struct </span>#name {}
|
||||
<a href=#131 id=131 data-nosnippet>131</a>
|
||||
<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">impl </span>#name {
|
||||
<a href=#133 id=133 data-nosnippet>133</a> <span class="kw">pub const fn </span>name() -> <span class="kw-2">&</span><span class="lifetime">'static </span>str {
|
||||
<a href=#134 id=134 data-nosnippet>134</a> #display_name
|
||||
<a href=#135 id=135 data-nosnippet>135</a> }
|
||||
<a href=#136 id=136 data-nosnippet>136</a>
|
||||
<a href=#137 id=137 data-nosnippet>137</a> <span class="kw">pub fn </span>scalar_udf() -> ScalarUDF {
|
||||
<a href=#138 id=138 data-nosnippet>138</a> datafusion_expr::create_udf(
|
||||
<a href=#139 id=139 data-nosnippet>139</a> <span class="self">Self</span>::name(),
|
||||
<a href=#140 id=140 data-nosnippet>140</a> <span class="self">Self</span>::input_type(),
|
||||
<a href=#141 id=141 data-nosnippet>141</a> <span class="self">Self</span>::return_type(),
|
||||
<a href=#142 id=142 data-nosnippet>142</a> Volatility::Volatile,
|
||||
<a href=#143 id=143 data-nosnippet>143</a> Arc::new(<span class="self">Self</span>::calc) <span class="kw">as _</span>,
|
||||
<a href=#144 id=144 data-nosnippet>144</a> )
|
||||
<a href=#145 id=145 data-nosnippet>145</a> }
|
||||
<a href=#146 id=146 data-nosnippet>146</a>
|
||||
<a href=#147 id=147 data-nosnippet>147</a> <span class="kw">fn </span>input_type() -> Vec<DataType> {
|
||||
<a href=#148 id=148 data-nosnippet>148</a> <span class="macro">vec!</span>[#( RangeArray::convert_data_type(#array_types::new_null(<span class="number">0</span>).data_type().clone()), )<span class="kw-2">*</span>]
|
||||
<a href=#149 id=149 data-nosnippet>149</a> }
|
||||
<a href=#150 id=150 data-nosnippet>150</a>
|
||||
<a href=#151 id=151 data-nosnippet>151</a> <span class="kw">fn </span>return_type() -> DataType {
|
||||
<a href=#152 id=152 data-nosnippet>152</a> #return_array_type::new_null(<span class="number">0</span>).data_type().clone()
|
||||
<a href=#153 id=153 data-nosnippet>153</a> }
|
||||
<a href=#154 id=154 data-nosnippet>154</a> }
|
||||
<a href=#155 id=155 data-nosnippet>155</a> }
|
||||
<a href=#156 id=156 data-nosnippet>156</a> .into()
|
||||
<a href=#157 id=157 data-nosnippet>157</a>}
|
||||
<a href=#158 id=158 data-nosnippet>158</a>
|
||||
<a href=#159 id=159 data-nosnippet>159</a><span class="kw">fn </span>build_calc_fn(
|
||||
<a href=#160 id=160 data-nosnippet>160</a> name: Ident,
|
||||
<a href=#161 id=161 data-nosnippet>161</a> param_types: Vec<Type>,
|
||||
<a href=#162 id=162 data-nosnippet>162</a> fn_name: Ident,
|
||||
<a href=#163 id=163 data-nosnippet>163</a> ret_type: Ident,
|
||||
<a href=#164 id=164 data-nosnippet>164</a>) -> TokenStream {
|
||||
<a href=#165 id=165 data-nosnippet>165</a> <span class="kw">let </span>param_names = param_types
|
||||
<a href=#166 id=166 data-nosnippet>166</a> .iter()
|
||||
<a href=#167 id=167 data-nosnippet>167</a> .enumerate()
|
||||
<a href=#168 id=168 data-nosnippet>168</a> .map(|(i, ty)| Ident::new(<span class="kw-2">&</span><span class="macro">format!</span>(<span class="string">"param_{}"</span>, i), ty.span()))
|
||||
<a href=#169 id=169 data-nosnippet>169</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#170 id=170 data-nosnippet>170</a> <span class="kw">let </span>unref_param_types = param_types
|
||||
<a href=#171 id=171 data-nosnippet>171</a> .iter()
|
||||
<a href=#172 id=172 data-nosnippet>172</a> .map(|ty| {
|
||||
<a href=#173 id=173 data-nosnippet>173</a> <span class="kw">if let </span>Type::Reference(TypeReference { elem, .. }) = ty {
|
||||
<a href=#174 id=174 data-nosnippet>174</a> elem.as_ref().clone()
|
||||
<a href=#175 id=175 data-nosnippet>175</a> } <span class="kw">else </span>{
|
||||
<a href=#176 id=176 data-nosnippet>176</a> ty.clone()
|
||||
<a href=#177 id=177 data-nosnippet>177</a> }
|
||||
<a href=#178 id=178 data-nosnippet>178</a> })
|
||||
<a href=#179 id=179 data-nosnippet>179</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#180 id=180 data-nosnippet>180</a> <span class="kw">let </span>num_params = param_types.len();
|
||||
<a href=#181 id=181 data-nosnippet>181</a> <span class="kw">let </span>param_numbers = (<span class="number">0</span>..num_params).collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#182 id=182 data-nosnippet>182</a> <span class="kw">let </span>range_array_names = param_names
|
||||
<a href=#183 id=183 data-nosnippet>183</a> .iter()
|
||||
<a href=#184 id=184 data-nosnippet>184</a> .map(|name| Ident::new(<span class="kw-2">&</span><span class="macro">format!</span>(<span class="string">"{}_range_array"</span>, name), name.span()))
|
||||
<a href=#185 id=185 data-nosnippet>185</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#186 id=186 data-nosnippet>186</a> <span class="kw">let </span>first_range_array_name = range_array_names.first().unwrap().clone();
|
||||
<a href=#187 id=187 data-nosnippet>187</a> <span class="kw">let </span>first_param_name = param_names.first().unwrap().clone();
|
||||
<a href=#188 id=188 data-nosnippet>188</a>
|
||||
<a href=#189 id=189 data-nosnippet>189</a> <span class="macro">quote!</span> {
|
||||
<a href=#190 id=190 data-nosnippet>190</a> <span class="kw">impl </span>#name {
|
||||
<a href=#191 id=191 data-nosnippet>191</a> <span class="kw">fn </span>calc(input: <span class="kw-2">&</span>[ColumnarValue]) -> <span class="prelude-ty">Result</span><ColumnarValue, DataFusionError> {
|
||||
<a href=#192 id=192 data-nosnippet>192</a> <span class="macro">assert_eq!</span>(input.len(), #num_params);
|
||||
<a href=#193 id=193 data-nosnippet>193</a>
|
||||
<a href=#194 id=194 data-nosnippet>194</a> #( <span class="kw">let </span>#range_array_names = RangeArray::try_new(extract_array(<span class="kw-2">&</span>input[#param_numbers])<span class="question-mark">?</span>.to_data().into())<span class="question-mark">?</span>; )*
|
||||
<a href=#195 id=195 data-nosnippet>195</a>
|
||||
<a href=#196 id=196 data-nosnippet>196</a> <span class="comment">// check arrays len
|
||||
<a href=#197 id=197 data-nosnippet>197</a> </span>{
|
||||
<a href=#198 id=198 data-nosnippet>198</a> <span class="kw">let </span>len_first = #first_range_array_name.len();
|
||||
<a href=#199 id=199 data-nosnippet>199</a> #(
|
||||
<a href=#200 id=200 data-nosnippet>200</a> <span class="kw">if </span>len_first != #range_array_names.len() {
|
||||
<a href=#201 id=201 data-nosnippet>201</a> <span class="kw">return </span><span class="prelude-val">Err</span>(DataFusionError::Execution(<span class="macro">format!</span>(<span class="string">"RangeArray have different lengths in PromQL function {}: array1={}, array2={}"</span>, #name::name(), len_first, #range_array_names.len())));
|
||||
<a href=#202 id=202 data-nosnippet>202</a> }
|
||||
<a href=#203 id=203 data-nosnippet>203</a> )*
|
||||
<a href=#204 id=204 data-nosnippet>204</a> }
|
||||
<a href=#205 id=205 data-nosnippet>205</a>
|
||||
<a href=#206 id=206 data-nosnippet>206</a> <span class="kw">let </span><span class="kw-2">mut </span>result_array = Vec::new();
|
||||
<a href=#207 id=207 data-nosnippet>207</a> <span class="kw">for </span>index <span class="kw">in </span><span class="number">0</span>..#first_range_array_name.len(){
|
||||
<a href=#208 id=208 data-nosnippet>208</a> #( <span class="kw">let </span>#param_names = #range_array_names.get(index).unwrap().as_any().downcast_ref::<#unref_param_types>().unwrap().clone(); )*
|
||||
<a href=#209 id=209 data-nosnippet>209</a>
|
||||
<a href=#210 id=210 data-nosnippet>210</a> <span class="comment">// check element len
|
||||
<a href=#211 id=211 data-nosnippet>211</a> </span>{
|
||||
<a href=#212 id=212 data-nosnippet>212</a> <span class="kw">let </span>len_first = #first_param_name.len();
|
||||
<a href=#213 id=213 data-nosnippet>213</a> #(
|
||||
<a href=#214 id=214 data-nosnippet>214</a> <span class="kw">if </span>len_first != #param_names.len() {
|
||||
<a href=#215 id=215 data-nosnippet>215</a> <span class="kw">return </span><span class="prelude-val">Err</span>(DataFusionError::Execution(<span class="macro">format!</span>(<span class="string">"RangeArray's element {} have different lengths in PromQL function {}: array1={}, array2={}"</span>, index, #name::name(), len_first, #param_names.len())));
|
||||
<a href=#216 id=216 data-nosnippet>216</a> }
|
||||
<a href=#217 id=217 data-nosnippet>217</a> )*
|
||||
<a href=#218 id=218 data-nosnippet>218</a> }
|
||||
<a href=#219 id=219 data-nosnippet>219</a>
|
||||
<a href=#220 id=220 data-nosnippet>220</a> <span class="kw">let </span>result = #fn_name(#( <span class="kw-2">&</span>#param_names, )<span class="kw-2">*</span>);
|
||||
<a href=#221 id=221 data-nosnippet>221</a> result_array.push(result);
|
||||
<a href=#222 id=222 data-nosnippet>222</a> }
|
||||
<a href=#223 id=223 data-nosnippet>223</a>
|
||||
<a href=#224 id=224 data-nosnippet>224</a> <span class="kw">let </span>result = ColumnarValue::Array(Arc::new(#ret_type::from_iter(result_array)));
|
||||
<a href=#225 id=225 data-nosnippet>225</a> <span class="prelude-val">Ok</span>(result)
|
||||
<a href=#226 id=226 data-nosnippet>226</a> }
|
||||
<a href=#227 id=227 data-nosnippet>227</a> }
|
||||
<a href=#228 id=228 data-nosnippet>228</a> }
|
||||
<a href=#229 id=229 data-nosnippet>229</a> .into()
|
||||
<a href=#230 id=230 data-nosnippet>230</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
26
src/common_macro/row.rs.html
Normal file
26
src/common_macro/row.rs.html
Normal file
@@ -0,0 +1,26 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row.rs`."><title>row.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>row.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">mod </span>attribute;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">mod </span>into_row;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">mod </span>schema;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">mod </span>to_row;
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">mod </span>utils;
|
||||
<a href=#20 id=20 data-nosnippet>20</a>
|
||||
<a href=#21 id=21 data-nosnippet>21</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">const </span>META_KEY_COL: <span class="kw-2">&</span>str = <span class="string">"col"</span>;
|
||||
<a href=#22 id=22 data-nosnippet>22</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">const </span>META_KEY_NAME: <span class="kw-2">&</span>str = <span class="string">"name"</span>;
|
||||
<a href=#23 id=23 data-nosnippet>23</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">const </span>META_KEY_DATATYPE: <span class="kw-2">&</span>str = <span class="string">"datatype"</span>;
|
||||
<a href=#24 id=24 data-nosnippet>24</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">const </span>META_KEY_SEMANTIC: <span class="kw-2">&</span>str = <span class="string">"semantic"</span>;
|
||||
<a href=#25 id=25 data-nosnippet>25</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">const </span>META_KEY_SKIP: <span class="kw-2">&</span>str = <span class="string">"skip"</span>;
|
||||
</code></pre></div></section></main></body></html>
|
||||
127
src/common_macro/row/attribute.rs.html
Normal file
127
src/common_macro/row/attribute.rs.html
Normal file
@@ -0,0 +1,127 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row/attribute.rs`."><title>attribute.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../../static.files/storage-41dd4d93.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/row/</div>attribute.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>std::collections::HashMap;
|
||||
<a href=#16 id=16 data-nosnippet>16</a>
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>once_cell::sync::Lazy;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::meta::ParseNestedMeta;
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span>syn::{Attribute, LitStr, Meta, <span class="prelude-ty">Result</span>};
|
||||
<a href=#21 id=21 data-nosnippet>21</a>
|
||||
<a href=#22 id=22 data-nosnippet>22</a><span class="kw">use </span><span class="kw">crate</span>::row::utils::{
|
||||
<a href=#23 id=23 data-nosnippet>23</a> ColumnDataTypeWithExtension, SemanticType, column_data_type_from_str, semantic_type_from_str,
|
||||
<a href=#24 id=24 data-nosnippet>24</a>};
|
||||
<a href=#25 id=25 data-nosnippet>25</a><span class="kw">use </span><span class="kw">crate</span>::row::{
|
||||
<a href=#26 id=26 data-nosnippet>26</a> META_KEY_COL, META_KEY_DATATYPE, META_KEY_NAME, META_KEY_SEMANTIC, META_KEY_SKIP,
|
||||
<a href=#27 id=27 data-nosnippet>27</a>};
|
||||
<a href=#28 id=28 data-nosnippet>28</a>
|
||||
<a href=#29 id=29 data-nosnippet>29</a><span class="doccomment">/// Column attribute.
|
||||
<a href=#30 id=30 data-nosnippet>30</a></span><span class="attr">#[derive(Default)]
|
||||
<a href=#31 id=31 data-nosnippet>31</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">struct </span>ColumnAttribute {
|
||||
<a href=#32 id=32 data-nosnippet>32</a> <span class="doccomment">/// User-defined name of the column.,
|
||||
<a href=#33 id=33 data-nosnippet>33</a> </span><span class="kw">pub</span>(<span class="kw">crate</span>) name: <span class="prelude-ty">Option</span><String>,
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="doccomment">/// Data type of the column.
|
||||
<a href=#35 id=35 data-nosnippet>35</a> </span><span class="kw">pub</span>(<span class="kw">crate</span>) datatype: <span class="prelude-ty">Option</span><ColumnDataTypeWithExtension>,
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="doccomment">/// Semantic type of the column.
|
||||
<a href=#37 id=37 data-nosnippet>37</a> </span><span class="kw">pub</span>(<span class="kw">crate</span>) semantic_type: SemanticType,
|
||||
<a href=#38 id=38 data-nosnippet>38</a> <span class="doccomment">/// Whether to skip the column.
|
||||
<a href=#39 id=39 data-nosnippet>39</a> </span><span class="kw">pub</span>(<span class="kw">crate</span>) skip: bool,
|
||||
<a href=#40 id=40 data-nosnippet>40</a>}
|
||||
<a href=#41 id=41 data-nosnippet>41</a>
|
||||
<a href=#42 id=42 data-nosnippet>42</a><span class="doccomment">/// Find the column attribute in the attributes.
|
||||
<a href=#43 id=43 data-nosnippet>43</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>find_column_attribute(attrs: <span class="kw-2">&</span>[Attribute]) -> <span class="prelude-ty">Option</span><<span class="kw-2">&</span>Attribute> {
|
||||
<a href=#44 id=44 data-nosnippet>44</a> attrs
|
||||
<a href=#45 id=45 data-nosnippet>45</a> .iter()
|
||||
<a href=#46 id=46 data-nosnippet>46</a> .find(|attr| <span class="macro">matches!</span>(<span class="kw-2">&</span>attr.meta, Meta::List(list) <span class="kw">if </span>list.path.is_ident(META_KEY_COL)))
|
||||
<a href=#47 id=47 data-nosnippet>47</a>}
|
||||
<a href=#48 id=48 data-nosnippet>48</a>
|
||||
<a href=#49 id=49 data-nosnippet>49</a><span class="doccomment">/// Parse the column attribute.
|
||||
<a href=#50 id=50 data-nosnippet>50</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>parse_column_attribute(attr: <span class="kw-2">&</span>Attribute) -> <span class="prelude-ty">Result</span><ColumnAttribute> {
|
||||
<a href=#51 id=51 data-nosnippet>51</a> <span class="kw">match </span><span class="kw-2">&</span>attr.meta {
|
||||
<a href=#52 id=52 data-nosnippet>52</a> Meta::List(list) <span class="kw">if </span>list.path.is_ident(META_KEY_COL) => {
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="kw">let </span><span class="kw-2">mut </span>attribute = ColumnAttribute::default();
|
||||
<a href=#54 id=54 data-nosnippet>54</a> list.parse_nested_meta(|meta| parse_column_attribute_field(<span class="kw-2">&</span>meta, <span class="kw-2">&mut </span>attribute))<span class="question-mark">?</span>;
|
||||
<a href=#55 id=55 data-nosnippet>55</a> <span class="prelude-val">Ok</span>(attribute)
|
||||
<a href=#56 id=56 data-nosnippet>56</a> }
|
||||
<a href=#57 id=57 data-nosnippet>57</a> <span class="kw">_ </span>=> <span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#58 id=58 data-nosnippet>58</a> attr.span(),
|
||||
<a href=#59 id=59 data-nosnippet>59</a> <span class="macro">format!</span>(
|
||||
<a href=#60 id=60 data-nosnippet>60</a> <span class="string">"expected `{META_KEY_COL}({META_KEY_NAME} = \"...\", {META_KEY_DATATYPE} = \"...\", {META_KEY_SEMANTIC} = \"...\")`"
|
||||
<a href=#61 id=61 data-nosnippet>61</a> </span>),
|
||||
<a href=#62 id=62 data-nosnippet>62</a> )),
|
||||
<a href=#63 id=63 data-nosnippet>63</a> }
|
||||
<a href=#64 id=64 data-nosnippet>64</a>}
|
||||
<a href=#65 id=65 data-nosnippet>65</a>
|
||||
<a href=#66 id=66 data-nosnippet>66</a><span class="kw">type </span>ParseColumnAttributeField = <span class="kw">fn</span>(<span class="kw-2">&</span>ParseNestedMeta, <span class="kw-2">&mut </span>ColumnAttribute) -> <span class="prelude-ty">Result</span><()>;
|
||||
<a href=#67 id=67 data-nosnippet>67</a>
|
||||
<a href=#68 id=68 data-nosnippet>68</a><span class="kw">static </span>PARSE_COLUMN_ATTRIBUTE_FIELDS: Lazy<HashMap<<span class="kw-2">&</span>str, ParseColumnAttributeField>> =
|
||||
<a href=#69 id=69 data-nosnippet>69</a> Lazy::new(|| {
|
||||
<a href=#70 id=70 data-nosnippet>70</a> HashMap::from([
|
||||
<a href=#71 id=71 data-nosnippet>71</a> (META_KEY_NAME, parse_name_field <span class="kw">as _</span>),
|
||||
<a href=#72 id=72 data-nosnippet>72</a> (META_KEY_DATATYPE, parse_datatype_field <span class="kw">as _</span>),
|
||||
<a href=#73 id=73 data-nosnippet>73</a> (META_KEY_SEMANTIC, parse_semantic_field <span class="kw">as _</span>),
|
||||
<a href=#74 id=74 data-nosnippet>74</a> (META_KEY_SKIP, parse_skip_field <span class="kw">as _</span>),
|
||||
<a href=#75 id=75 data-nosnippet>75</a> ])
|
||||
<a href=#76 id=76 data-nosnippet>76</a> });
|
||||
<a href=#77 id=77 data-nosnippet>77</a>
|
||||
<a href=#78 id=78 data-nosnippet>78</a><span class="kw">fn </span>parse_name_field(meta: <span class="kw-2">&</span>ParseNestedMeta<<span class="lifetime">'_</span>>, attribute: <span class="kw-2">&mut </span>ColumnAttribute) -> <span class="prelude-ty">Result</span><()> {
|
||||
<a href=#79 id=79 data-nosnippet>79</a> <span class="kw">let </span>value = meta.value()<span class="question-mark">?</span>;
|
||||
<a href=#80 id=80 data-nosnippet>80</a> <span class="kw">let </span>s: LitStr = value.parse()<span class="question-mark">?</span>;
|
||||
<a href=#81 id=81 data-nosnippet>81</a> attribute.name = <span class="prelude-val">Some</span>(s.value());
|
||||
<a href=#82 id=82 data-nosnippet>82</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#83 id=83 data-nosnippet>83</a>}
|
||||
<a href=#84 id=84 data-nosnippet>84</a>
|
||||
<a href=#85 id=85 data-nosnippet>85</a><span class="kw">fn </span>parse_datatype_field(meta: <span class="kw-2">&</span>ParseNestedMeta<<span class="lifetime">'_</span>>, attribute: <span class="kw-2">&mut </span>ColumnAttribute) -> <span class="prelude-ty">Result</span><()> {
|
||||
<a href=#86 id=86 data-nosnippet>86</a> <span class="kw">let </span>value = meta.value()<span class="question-mark">?</span>;
|
||||
<a href=#87 id=87 data-nosnippet>87</a> <span class="kw">let </span>s: LitStr = value.parse()<span class="question-mark">?</span>;
|
||||
<a href=#88 id=88 data-nosnippet>88</a> <span class="kw">let </span>ident = s.value();
|
||||
<a href=#89 id=89 data-nosnippet>89</a> <span class="kw">let </span><span class="prelude-val">Some</span>(value) = column_data_type_from_str(<span class="kw-2">&</span>ident) <span class="kw">else </span>{
|
||||
<a href=#90 id=90 data-nosnippet>90</a> <span class="kw">return </span><span class="prelude-val">Err</span>(meta.error(<span class="macro">format!</span>(<span class="string">"unexpected {META_KEY_DATATYPE}: {ident}"</span>)));
|
||||
<a href=#91 id=91 data-nosnippet>91</a> };
|
||||
<a href=#92 id=92 data-nosnippet>92</a> attribute.datatype = <span class="prelude-val">Some</span>(value);
|
||||
<a href=#93 id=93 data-nosnippet>93</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#94 id=94 data-nosnippet>94</a>}
|
||||
<a href=#95 id=95 data-nosnippet>95</a>
|
||||
<a href=#96 id=96 data-nosnippet>96</a><span class="kw">fn </span>parse_semantic_field(meta: <span class="kw-2">&</span>ParseNestedMeta<<span class="lifetime">'_</span>>, attribute: <span class="kw-2">&mut </span>ColumnAttribute) -> <span class="prelude-ty">Result</span><()> {
|
||||
<a href=#97 id=97 data-nosnippet>97</a> <span class="kw">let </span>value = meta.value()<span class="question-mark">?</span>;
|
||||
<a href=#98 id=98 data-nosnippet>98</a> <span class="kw">let </span>s: LitStr = value.parse()<span class="question-mark">?</span>;
|
||||
<a href=#99 id=99 data-nosnippet>99</a> <span class="kw">let </span>ident = s.value();
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="kw">let </span><span class="prelude-val">Some</span>(value) = semantic_type_from_str(<span class="kw-2">&</span>ident) <span class="kw">else </span>{
|
||||
<a href=#101 id=101 data-nosnippet>101</a> <span class="kw">return </span><span class="prelude-val">Err</span>(meta.error(<span class="macro">format!</span>(<span class="string">"unexpected {META_KEY_SEMANTIC}: {ident}"</span>)));
|
||||
<a href=#102 id=102 data-nosnippet>102</a> };
|
||||
<a href=#103 id=103 data-nosnippet>103</a> attribute.semantic_type = value;
|
||||
<a href=#104 id=104 data-nosnippet>104</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#105 id=105 data-nosnippet>105</a>}
|
||||
<a href=#106 id=106 data-nosnippet>106</a>
|
||||
<a href=#107 id=107 data-nosnippet>107</a><span class="kw">fn </span>parse_skip_field(<span class="kw">_</span>: <span class="kw-2">&</span>ParseNestedMeta<<span class="lifetime">'_</span>>, attribute: <span class="kw-2">&mut </span>ColumnAttribute) -> <span class="prelude-ty">Result</span><()> {
|
||||
<a href=#108 id=108 data-nosnippet>108</a> attribute.skip = <span class="bool-val">true</span>;
|
||||
<a href=#109 id=109 data-nosnippet>109</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#110 id=110 data-nosnippet>110</a>}
|
||||
<a href=#111 id=111 data-nosnippet>111</a>
|
||||
<a href=#112 id=112 data-nosnippet>112</a><span class="kw">fn </span>parse_column_attribute_field(
|
||||
<a href=#113 id=113 data-nosnippet>113</a> meta: <span class="kw-2">&</span>ParseNestedMeta<<span class="lifetime">'_</span>>,
|
||||
<a href=#114 id=114 data-nosnippet>114</a> attribute: <span class="kw-2">&mut </span>ColumnAttribute,
|
||||
<a href=#115 id=115 data-nosnippet>115</a>) -> <span class="prelude-ty">Result</span><()> {
|
||||
<a href=#116 id=116 data-nosnippet>116</a> <span class="kw">let </span><span class="prelude-val">Some</span>(ident) = meta.path.get_ident() <span class="kw">else </span>{
|
||||
<a href=#117 id=117 data-nosnippet>117</a> <span class="kw">return </span><span class="prelude-val">Err</span>(meta.error(<span class="macro">format!</span>(<span class="string">"expected `{META_KEY_COL}({META_KEY_NAME} = \"...\", {META_KEY_DATATYPE} = \"...\", {META_KEY_SEMANTIC} = \"...\")`"</span>)));
|
||||
<a href=#118 id=118 data-nosnippet>118</a> };
|
||||
<a href=#119 id=119 data-nosnippet>119</a> <span class="kw">let </span><span class="prelude-val">Some</span>(parse_column_attribute) =
|
||||
<a href=#120 id=120 data-nosnippet>120</a> PARSE_COLUMN_ATTRIBUTE_FIELDS.get(ident.to_string().as_str())
|
||||
<a href=#121 id=121 data-nosnippet>121</a> <span class="kw">else </span>{
|
||||
<a href=#122 id=122 data-nosnippet>122</a> <span class="kw">return </span><span class="prelude-val">Err</span>(meta.error(<span class="macro">format!</span>(<span class="string">"unexpected attribute: {ident}"</span>)));
|
||||
<a href=#123 id=123 data-nosnippet>123</a> };
|
||||
<a href=#124 id=124 data-nosnippet>124</a>
|
||||
<a href=#125 id=125 data-nosnippet>125</a> parse_column_attribute(meta, attribute)
|
||||
<a href=#126 id=126 data-nosnippet>126</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
89
src/common_macro/row/into_row.rs.html
Normal file
89
src/common_macro/row/into_row.rs.html
Normal file
@@ -0,0 +1,89 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row/into_row.rs`."><title>into_row.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../../static.files/storage-41dd4d93.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/row/</div>into_row.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro2::TokenStream <span class="kw">as </span>TokenStream2;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::{DeriveInput, <span class="prelude-ty">Result</span>};
|
||||
<a href=#19 id=19 data-nosnippet>19</a>
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span><span class="kw">crate</span>::row::utils::{
|
||||
<a href=#21 id=21 data-nosnippet>21</a> ParsedField, convert_column_data_type_to_value_data_ident, extract_struct_fields,
|
||||
<a href=#22 id=22 data-nosnippet>22</a> get_column_data_type, parse_fields_from_fields_named,
|
||||
<a href=#23 id=23 data-nosnippet>23</a>};
|
||||
<a href=#24 id=24 data-nosnippet>24</a><span class="kw">use </span><span class="kw">crate</span>::row::{META_KEY_COL, META_KEY_DATATYPE};
|
||||
<a href=#25 id=25 data-nosnippet>25</a>
|
||||
<a href=#26 id=26 data-nosnippet>26</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>derive_into_row_impl(input: DeriveInput) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#27 id=27 data-nosnippet>27</a> <span class="kw">let </span><span class="prelude-val">Some</span>(fields) = extract_struct_fields(<span class="kw-2">&</span>input.data) <span class="kw">else </span>{
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#29 id=29 data-nosnippet>29</a> input.span(),
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="string">"IntoRow can only be derived for structs"</span>,
|
||||
<a href=#31 id=31 data-nosnippet>31</a> ));
|
||||
<a href=#32 id=32 data-nosnippet>32</a> };
|
||||
<a href=#33 id=33 data-nosnippet>33</a> <span class="kw">let </span>ident = input.ident;
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="kw">let </span>(impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">let </span>fields = parse_fields_from_fields_named(fields)<span class="question-mark">?</span>;
|
||||
<a href=#36 id=36 data-nosnippet>36</a>
|
||||
<a href=#37 id=37 data-nosnippet>37</a> <span class="comment">// Implement `into_row` method.
|
||||
<a href=#38 id=38 data-nosnippet>38</a> </span><span class="kw">let </span>impl_to_row_method = impl_into_row_method_combined(<span class="kw-2">&</span>fields)<span class="question-mark">?</span>;
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#40 id=40 data-nosnippet>40</a> <span class="kw">impl </span>#impl_generics #ident #ty_generics #where_clause {
|
||||
<a href=#41 id=41 data-nosnippet>41</a> #impl_to_row_method
|
||||
<a href=#42 id=42 data-nosnippet>42</a> }
|
||||
<a href=#43 id=43 data-nosnippet>43</a> })
|
||||
<a href=#44 id=44 data-nosnippet>44</a>}
|
||||
<a href=#45 id=45 data-nosnippet>45</a>
|
||||
<a href=#46 id=46 data-nosnippet>46</a><span class="kw">fn </span>impl_into_row_method_combined(fields: <span class="kw-2">&</span>[ParsedField<<span class="lifetime">'_</span>>]) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#47 id=47 data-nosnippet>47</a> <span class="kw">let </span>value_exprs = fields
|
||||
<a href=#48 id=48 data-nosnippet>48</a> .iter()
|
||||
<a href=#49 id=49 data-nosnippet>49</a> .map(|field| {
|
||||
<a href=#50 id=50 data-nosnippet>50</a> <span class="kw">let </span>ParsedField {ident, field_type, column_data_type, column_attribute} = field;
|
||||
<a href=#51 id=51 data-nosnippet>51</a> <span class="kw">let </span><span class="prelude-val">Some</span>(column_data_type) = get_column_data_type(column_data_type, column_attribute)
|
||||
<a href=#52 id=52 data-nosnippet>52</a> <span class="kw">else </span>{
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#54 id=54 data-nosnippet>54</a> ident.span(),
|
||||
<a href=#55 id=55 data-nosnippet>55</a> <span class="macro">format!</span>(
|
||||
<a href=#56 id=56 data-nosnippet>56</a> <span class="string">"expected to set data type explicitly via [({META_KEY_COL}({META_KEY_DATATYPE} = \"...\"))]"
|
||||
<a href=#57 id=57 data-nosnippet>57</a> </span>),
|
||||
<a href=#58 id=58 data-nosnippet>58</a> ));
|
||||
<a href=#59 id=59 data-nosnippet>59</a> };
|
||||
<a href=#60 id=60 data-nosnippet>60</a> <span class="kw">let </span>value_data = convert_column_data_type_to_value_data_ident(<span class="kw-2">&</span>column_data_type.data_type);
|
||||
<a href=#61 id=61 data-nosnippet>61</a> <span class="kw">let </span>expr = <span class="kw">if </span>field_type.is_optional() {
|
||||
<a href=#62 id=62 data-nosnippet>62</a> <span class="macro">quote!</span> {
|
||||
<a href=#63 id=63 data-nosnippet>63</a> <span class="kw">match </span><span class="self">self</span>.#ident {
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="prelude-val">Some</span>(v) => Value {
|
||||
<a href=#65 id=65 data-nosnippet>65</a> value_data: <span class="prelude-val">Some</span>(ValueData::#value_data(v.into())),
|
||||
<a href=#66 id=66 data-nosnippet>66</a> },
|
||||
<a href=#67 id=67 data-nosnippet>67</a> <span class="prelude-val">None </span>=> Value { value_data: <span class="prelude-val">None </span>},
|
||||
<a href=#68 id=68 data-nosnippet>68</a> }
|
||||
<a href=#69 id=69 data-nosnippet>69</a> }
|
||||
<a href=#70 id=70 data-nosnippet>70</a> } <span class="kw">else </span>{
|
||||
<a href=#71 id=71 data-nosnippet>71</a> <span class="macro">quote!</span> {
|
||||
<a href=#72 id=72 data-nosnippet>72</a> Value {
|
||||
<a href=#73 id=73 data-nosnippet>73</a> value_data: <span class="prelude-val">Some</span>(ValueData::#value_data(<span class="self">self</span>.#ident.into())),
|
||||
<a href=#74 id=74 data-nosnippet>74</a> }
|
||||
<a href=#75 id=75 data-nosnippet>75</a> }
|
||||
<a href=#76 id=76 data-nosnippet>76</a> };
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="prelude-val">Ok</span>(expr)
|
||||
<a href=#78 id=78 data-nosnippet>78</a> })
|
||||
<a href=#79 id=79 data-nosnippet>79</a> .collect::<<span class="prelude-ty">Result</span><Vec<<span class="kw">_</span>>>>()<span class="question-mark">?</span>;
|
||||
<a href=#80 id=80 data-nosnippet>80</a>
|
||||
<a href=#81 id=81 data-nosnippet>81</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#82 id=82 data-nosnippet>82</a> <span class="kw">pub fn </span>into_row(<span class="self">self</span>) -> Row {
|
||||
<a href=#83 id=83 data-nosnippet>83</a> Row {
|
||||
<a href=#84 id=84 data-nosnippet>84</a> values: <span class="macro">vec!</span>[ #( #value_exprs ),* ]
|
||||
<a href=#85 id=85 data-nosnippet>85</a> }
|
||||
<a href=#86 id=86 data-nosnippet>86</a> }
|
||||
<a href=#87 id=87 data-nosnippet>87</a> })
|
||||
<a href=#88 id=88 data-nosnippet>88</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
217
src/common_macro/row/schema.rs.html
Normal file
217
src/common_macro/row/schema.rs.html
Normal file
@@ -0,0 +1,217 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row/schema.rs`."><title>schema.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../../static.files/storage-41dd4d93.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/row/</div>schema.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>greptime_proto::v1::ColumnDataTypeExtension;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>greptime_proto::v1::column_data_type_extension::TypeExt;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>proc_macro2::{Span, TokenStream <span class="kw">as </span>TokenStream2};
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span>syn::{DeriveInput, <span class="prelude-ty">Result</span>};
|
||||
<a href=#21 id=21 data-nosnippet>21</a>
|
||||
<a href=#22 id=22 data-nosnippet>22</a><span class="kw">use </span><span class="kw">crate</span>::row::utils::{
|
||||
<a href=#23 id=23 data-nosnippet>23</a> ColumnDataTypeWithExtension, ParsedField, convert_semantic_type_to_proto_semantic_type,
|
||||
<a href=#24 id=24 data-nosnippet>24</a> extract_struct_fields, get_column_data_type, parse_fields_from_fields_named,
|
||||
<a href=#25 id=25 data-nosnippet>25</a>};
|
||||
<a href=#26 id=26 data-nosnippet>26</a><span class="kw">use </span><span class="kw">crate</span>::row::{META_KEY_COL, META_KEY_DATATYPE};
|
||||
<a href=#27 id=27 data-nosnippet>27</a>
|
||||
<a href=#28 id=28 data-nosnippet>28</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>derive_schema_impl(input: DeriveInput) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#29 id=29 data-nosnippet>29</a> <span class="kw">let </span><span class="prelude-val">Some</span>(fields) = extract_struct_fields(<span class="kw-2">&</span>input.data) <span class="kw">else </span>{
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#31 id=31 data-nosnippet>31</a> input.span(),
|
||||
<a href=#32 id=32 data-nosnippet>32</a> <span class="string">"Schema can only be derived for structs"</span>,
|
||||
<a href=#33 id=33 data-nosnippet>33</a> ));
|
||||
<a href=#34 id=34 data-nosnippet>34</a> };
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">let </span>ident = input.ident;
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="kw">let </span>(impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
|
||||
<a href=#37 id=37 data-nosnippet>37</a> <span class="kw">let </span>fields = parse_fields_from_fields_named(fields)<span class="question-mark">?</span>;
|
||||
<a href=#38 id=38 data-nosnippet>38</a>
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="comment">// Implement `schema` method.
|
||||
<a href=#40 id=40 data-nosnippet>40</a> </span><span class="kw">let </span>impl_schema_method = impl_schema_method(<span class="kw-2">&</span>fields)<span class="question-mark">?</span>;
|
||||
<a href=#41 id=41 data-nosnippet>41</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#42 id=42 data-nosnippet>42</a> <span class="kw">impl </span>#impl_generics #ident #ty_generics #where_clause {
|
||||
<a href=#43 id=43 data-nosnippet>43</a> #impl_schema_method
|
||||
<a href=#44 id=44 data-nosnippet>44</a> }
|
||||
<a href=#45 id=45 data-nosnippet>45</a> })
|
||||
<a href=#46 id=46 data-nosnippet>46</a>}
|
||||
<a href=#47 id=47 data-nosnippet>47</a>
|
||||
<a href=#48 id=48 data-nosnippet>48</a><span class="kw">fn </span>impl_schema_method(fields: <span class="kw-2">&</span>[ParsedField<<span class="lifetime">'_</span>>]) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#49 id=49 data-nosnippet>49</a> <span class="kw">let </span>schemas: Vec<TokenStream2> = fields
|
||||
<a href=#50 id=50 data-nosnippet>50</a> .iter()
|
||||
<a href=#51 id=51 data-nosnippet>51</a> .map(|field| {
|
||||
<a href=#52 id=52 data-nosnippet>52</a> <span class="kw">let </span>ParsedField{ ident, column_data_type, column_attribute, ..} = field;
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="kw">let </span><span class="prelude-val">Some</span>(ColumnDataTypeWithExtension{data_type, extension}) = get_column_data_type(column_data_type, column_attribute)
|
||||
<a href=#54 id=54 data-nosnippet>54</a> <span class="kw">else </span>{
|
||||
<a href=#55 id=55 data-nosnippet>55</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#56 id=56 data-nosnippet>56</a> ident.span(),
|
||||
<a href=#57 id=57 data-nosnippet>57</a> <span class="macro">format!</span>(
|
||||
<a href=#58 id=58 data-nosnippet>58</a> <span class="string">"expected to set data type explicitly via [({META_KEY_COL}({META_KEY_DATATYPE} = \"...\"))]"
|
||||
<a href=#59 id=59 data-nosnippet>59</a> </span>),
|
||||
<a href=#60 id=60 data-nosnippet>60</a> ));
|
||||
<a href=#61 id=61 data-nosnippet>61</a> };
|
||||
<a href=#62 id=62 data-nosnippet>62</a> <span class="comment">// Uses user explicit name or field name as column name.
|
||||
<a href=#63 id=63 data-nosnippet>63</a> </span><span class="kw">let </span>name = column_attribute
|
||||
<a href=#64 id=64 data-nosnippet>64</a> .name
|
||||
<a href=#65 id=65 data-nosnippet>65</a> .clone()
|
||||
<a href=#66 id=66 data-nosnippet>66</a> .unwrap_or_else(|| ident.to_string());
|
||||
<a href=#67 id=67 data-nosnippet>67</a> <span class="kw">let </span>name = syn::LitStr::new(<span class="kw-2">&</span>name, ident.span());
|
||||
<a href=#68 id=68 data-nosnippet>68</a> <span class="kw">let </span>column_data_type =
|
||||
<a href=#69 id=69 data-nosnippet>69</a> syn::LitInt::new(<span class="kw-2">&</span>(data_type <span class="kw">as </span>i32).to_string(), ident.span());
|
||||
<a href=#70 id=70 data-nosnippet>70</a> <span class="kw">let </span>semantic_type_val = convert_semantic_type_to_proto_semantic_type(column_attribute.semantic_type) <span class="kw">as </span>i32;
|
||||
<a href=#71 id=71 data-nosnippet>71</a> <span class="kw">let </span>semantic_type = syn::LitInt::new(<span class="kw-2">&</span>semantic_type_val.to_string(), ident.span());
|
||||
<a href=#72 id=72 data-nosnippet>72</a> <span class="kw">let </span>extension = <span class="kw">match </span>extension {
|
||||
<a href=#73 id=73 data-nosnippet>73</a> <span class="prelude-val">Some</span>(ext) => column_data_type_extension_to_tokens(<span class="kw-2">&</span>ext, ident.span()),
|
||||
<a href=#74 id=74 data-nosnippet>74</a> <span class="prelude-val">None </span>=> <span class="macro">quote!</span> { <span class="prelude-val">None </span>},
|
||||
<a href=#75 id=75 data-nosnippet>75</a> };
|
||||
<a href=#76 id=76 data-nosnippet>76</a>
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#78 id=78 data-nosnippet>78</a> ColumnSchema {
|
||||
<a href=#79 id=79 data-nosnippet>79</a> column_name: #name.to_string(),
|
||||
<a href=#80 id=80 data-nosnippet>80</a> datatype: #column_data_type,
|
||||
<a href=#81 id=81 data-nosnippet>81</a> datatype_extension: #extension,
|
||||
<a href=#82 id=82 data-nosnippet>82</a> options: <span class="prelude-val">None</span>,
|
||||
<a href=#83 id=83 data-nosnippet>83</a> semantic_type: #semantic_type,
|
||||
<a href=#84 id=84 data-nosnippet>84</a> }
|
||||
<a href=#85 id=85 data-nosnippet>85</a> })
|
||||
<a href=#86 id=86 data-nosnippet>86</a> })
|
||||
<a href=#87 id=87 data-nosnippet>87</a> .collect::<<span class="prelude-ty">Result</span><<span class="kw">_</span>>>()<span class="question-mark">?</span>;
|
||||
<a href=#88 id=88 data-nosnippet>88</a>
|
||||
<a href=#89 id=89 data-nosnippet>89</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#90 id=90 data-nosnippet>90</a> <span class="kw">pub fn </span>schema() -> Vec<ColumnSchema> {
|
||||
<a href=#91 id=91 data-nosnippet>91</a> <span class="macro">vec!</span>[ #(#schemas),* ]
|
||||
<a href=#92 id=92 data-nosnippet>92</a> }
|
||||
<a href=#93 id=93 data-nosnippet>93</a> })
|
||||
<a href=#94 id=94 data-nosnippet>94</a>}
|
||||
<a href=#95 id=95 data-nosnippet>95</a>
|
||||
<a href=#96 id=96 data-nosnippet>96</a><span class="kw">fn </span>column_data_type_extension_to_tokens(
|
||||
<a href=#97 id=97 data-nosnippet>97</a> extension: <span class="kw-2">&</span>ColumnDataTypeExtension,
|
||||
<a href=#98 id=98 data-nosnippet>98</a> span: Span,
|
||||
<a href=#99 id=99 data-nosnippet>99</a>) -> TokenStream2 {
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="kw">match </span>extension.type_ext.as_ref() {
|
||||
<a href=#101 id=101 data-nosnippet>101</a> <span class="prelude-val">Some</span>(TypeExt::DecimalType(ext)) => {
|
||||
<a href=#102 id=102 data-nosnippet>102</a> <span class="kw">let </span>precision = syn::LitInt::new(<span class="kw-2">&</span>ext.precision.to_string(), span);
|
||||
<a href=#103 id=103 data-nosnippet>103</a> <span class="kw">let </span>scale = syn::LitInt::new(<span class="kw-2">&</span>ext.scale.to_string(), span);
|
||||
<a href=#104 id=104 data-nosnippet>104</a> <span class="macro">quote!</span> {
|
||||
<a href=#105 id=105 data-nosnippet>105</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#106 id=106 data-nosnippet>106</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::DecimalType(DecimalTypeExtension {
|
||||
<a href=#107 id=107 data-nosnippet>107</a> precision: #precision,
|
||||
<a href=#108 id=108 data-nosnippet>108</a> scale: #scale,
|
||||
<a href=#109 id=109 data-nosnippet>109</a> })),
|
||||
<a href=#110 id=110 data-nosnippet>110</a> })
|
||||
<a href=#111 id=111 data-nosnippet>111</a> }
|
||||
<a href=#112 id=112 data-nosnippet>112</a> }
|
||||
<a href=#113 id=113 data-nosnippet>113</a> <span class="prelude-val">Some</span>(TypeExt::JsonType(ext)) => {
|
||||
<a href=#114 id=114 data-nosnippet>114</a> <span class="kw">let </span>json_type = syn::LitInt::new(<span class="kw-2">&</span>ext.to_string(), span);
|
||||
<a href=#115 id=115 data-nosnippet>115</a> <span class="macro">quote!</span> {
|
||||
<a href=#116 id=116 data-nosnippet>116</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#117 id=117 data-nosnippet>117</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::JsonType(#json_type)),
|
||||
<a href=#118 id=118 data-nosnippet>118</a> })
|
||||
<a href=#119 id=119 data-nosnippet>119</a> }
|
||||
<a href=#120 id=120 data-nosnippet>120</a> }
|
||||
<a href=#121 id=121 data-nosnippet>121</a> <span class="prelude-val">Some</span>(TypeExt::VectorType(ext)) => {
|
||||
<a href=#122 id=122 data-nosnippet>122</a> <span class="kw">let </span>dim = syn::LitInt::new(<span class="kw-2">&</span>ext.dim.to_string(), span);
|
||||
<a href=#123 id=123 data-nosnippet>123</a> <span class="macro">quote!</span> {
|
||||
<a href=#124 id=124 data-nosnippet>124</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#125 id=125 data-nosnippet>125</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::VectorType(VectorTypeExtension { dim: #dim })),
|
||||
<a href=#126 id=126 data-nosnippet>126</a> })
|
||||
<a href=#127 id=127 data-nosnippet>127</a> }
|
||||
<a href=#128 id=128 data-nosnippet>128</a> }
|
||||
<a href=#129 id=129 data-nosnippet>129</a> <span class="prelude-val">Some</span>(TypeExt::ListType(ext)) => {
|
||||
<a href=#130 id=130 data-nosnippet>130</a> <span class="kw">let </span>datatype = syn::LitInt::new(<span class="kw-2">&</span>ext.datatype.to_string(), span);
|
||||
<a href=#131 id=131 data-nosnippet>131</a> <span class="kw">let </span>datatype_extension = ext
|
||||
<a href=#132 id=132 data-nosnippet>132</a> .datatype_extension
|
||||
<a href=#133 id=133 data-nosnippet>133</a> .as_deref()
|
||||
<a href=#134 id=134 data-nosnippet>134</a> .map(|ext| column_data_type_extension_to_tokens(ext, span))
|
||||
<a href=#135 id=135 data-nosnippet>135</a> .unwrap_or_else(|| <span class="macro">quote!</span> { <span class="prelude-val">None </span>});
|
||||
<a href=#136 id=136 data-nosnippet>136</a> <span class="macro">quote!</span> {
|
||||
<a href=#137 id=137 data-nosnippet>137</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#138 id=138 data-nosnippet>138</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::ListType(Box::new(ListTypeExtension {
|
||||
<a href=#139 id=139 data-nosnippet>139</a> datatype: #datatype,
|
||||
<a href=#140 id=140 data-nosnippet>140</a> datatype_extension: #datatype_extension,
|
||||
<a href=#141 id=141 data-nosnippet>141</a> }))),
|
||||
<a href=#142 id=142 data-nosnippet>142</a> })
|
||||
<a href=#143 id=143 data-nosnippet>143</a> }
|
||||
<a href=#144 id=144 data-nosnippet>144</a> }
|
||||
<a href=#145 id=145 data-nosnippet>145</a> <span class="prelude-val">Some</span>(TypeExt::StructType(ext)) => {
|
||||
<a href=#146 id=146 data-nosnippet>146</a> <span class="kw">let </span>fields = ext.fields.iter().map(|field| {
|
||||
<a href=#147 id=147 data-nosnippet>147</a> <span class="kw">let </span>field_name = <span class="kw-2">&</span>field.name;
|
||||
<a href=#148 id=148 data-nosnippet>148</a> <span class="kw">let </span>datatype = syn::LitInt::new(<span class="kw-2">&</span>field.datatype.to_string(), span);
|
||||
<a href=#149 id=149 data-nosnippet>149</a> <span class="kw">let </span>datatype_extension = field
|
||||
<a href=#150 id=150 data-nosnippet>150</a> .datatype_extension
|
||||
<a href=#151 id=151 data-nosnippet>151</a> .as_ref()
|
||||
<a href=#152 id=152 data-nosnippet>152</a> .map(|ext| column_data_type_extension_to_tokens(ext, span))
|
||||
<a href=#153 id=153 data-nosnippet>153</a> .unwrap_or_else(|| <span class="macro">quote!</span> { <span class="prelude-val">None </span>});
|
||||
<a href=#154 id=154 data-nosnippet>154</a> <span class="macro">quote!</span> {
|
||||
<a href=#155 id=155 data-nosnippet>155</a> greptime_proto::v1::StructField {
|
||||
<a href=#156 id=156 data-nosnippet>156</a> name: #field_name.to_string(),
|
||||
<a href=#157 id=157 data-nosnippet>157</a> datatype: #datatype,
|
||||
<a href=#158 id=158 data-nosnippet>158</a> datatype_extension: #datatype_extension,
|
||||
<a href=#159 id=159 data-nosnippet>159</a> }
|
||||
<a href=#160 id=160 data-nosnippet>160</a> }
|
||||
<a href=#161 id=161 data-nosnippet>161</a> });
|
||||
<a href=#162 id=162 data-nosnippet>162</a> <span class="macro">quote!</span> {
|
||||
<a href=#163 id=163 data-nosnippet>163</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#164 id=164 data-nosnippet>164</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::StructType(StructTypeExtension {
|
||||
<a href=#165 id=165 data-nosnippet>165</a> fields: <span class="macro">vec!</span>[#(#fields),<span class="kw-2">*</span>],
|
||||
<a href=#166 id=166 data-nosnippet>166</a> })),
|
||||
<a href=#167 id=167 data-nosnippet>167</a> })
|
||||
<a href=#168 id=168 data-nosnippet>168</a> }
|
||||
<a href=#169 id=169 data-nosnippet>169</a> }
|
||||
<a href=#170 id=170 data-nosnippet>170</a> <span class="prelude-val">Some</span>(TypeExt::JsonNativeType(ext)) => {
|
||||
<a href=#171 id=171 data-nosnippet>171</a> <span class="kw">let </span>inner = syn::LitInt::new(<span class="kw-2">&</span>ext.datatype.to_string(), span);
|
||||
<a href=#172 id=172 data-nosnippet>172</a> <span class="kw">let </span>datatype_extension = ext
|
||||
<a href=#173 id=173 data-nosnippet>173</a> .datatype_extension
|
||||
<a href=#174 id=174 data-nosnippet>174</a> .as_deref()
|
||||
<a href=#175 id=175 data-nosnippet>175</a> .map(|ext| column_data_type_extension_to_tokens(ext, span))
|
||||
<a href=#176 id=176 data-nosnippet>176</a> .unwrap_or_else(|| <span class="macro">quote!</span> { <span class="prelude-val">None </span>});
|
||||
<a href=#177 id=177 data-nosnippet>177</a> <span class="macro">quote!</span> {
|
||||
<a href=#178 id=178 data-nosnippet>178</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#179 id=179 data-nosnippet>179</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::JsonNativeType(Box::new(
|
||||
<a href=#180 id=180 data-nosnippet>180</a> JsonNativeTypeExtension {
|
||||
<a href=#181 id=181 data-nosnippet>181</a> datatype: #inner,
|
||||
<a href=#182 id=182 data-nosnippet>182</a> datatype_extension: #datatype_extension,
|
||||
<a href=#183 id=183 data-nosnippet>183</a> },
|
||||
<a href=#184 id=184 data-nosnippet>184</a> ))),
|
||||
<a href=#185 id=185 data-nosnippet>185</a> })
|
||||
<a href=#186 id=186 data-nosnippet>186</a> }
|
||||
<a href=#187 id=187 data-nosnippet>187</a> }
|
||||
<a href=#188 id=188 data-nosnippet>188</a> <span class="prelude-val">Some</span>(TypeExt::DictionaryType(ext)) => {
|
||||
<a href=#189 id=189 data-nosnippet>189</a> <span class="kw">let </span>key_datatype = syn::LitInt::new(<span class="kw-2">&</span>ext.key_datatype.to_string(), span);
|
||||
<a href=#190 id=190 data-nosnippet>190</a> <span class="kw">let </span>value_datatype = syn::LitInt::new(<span class="kw-2">&</span>ext.value_datatype.to_string(), span);
|
||||
<a href=#191 id=191 data-nosnippet>191</a> <span class="kw">let </span>key_datatype_extension = ext
|
||||
<a href=#192 id=192 data-nosnippet>192</a> .key_datatype_extension
|
||||
<a href=#193 id=193 data-nosnippet>193</a> .as_deref()
|
||||
<a href=#194 id=194 data-nosnippet>194</a> .map(|ext| column_data_type_extension_to_tokens(ext, span))
|
||||
<a href=#195 id=195 data-nosnippet>195</a> .unwrap_or_else(|| <span class="macro">quote!</span> { <span class="prelude-val">None </span>});
|
||||
<a href=#196 id=196 data-nosnippet>196</a> <span class="kw">let </span>value_datatype_extension = ext
|
||||
<a href=#197 id=197 data-nosnippet>197</a> .value_datatype_extension
|
||||
<a href=#198 id=198 data-nosnippet>198</a> .as_deref()
|
||||
<a href=#199 id=199 data-nosnippet>199</a> .map(|ext| column_data_type_extension_to_tokens(ext, span))
|
||||
<a href=#200 id=200 data-nosnippet>200</a> .unwrap_or_else(|| <span class="macro">quote!</span> { <span class="prelude-val">None </span>});
|
||||
<a href=#201 id=201 data-nosnippet>201</a> <span class="macro">quote!</span> {
|
||||
<a href=#202 id=202 data-nosnippet>202</a> <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#203 id=203 data-nosnippet>203</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::DictionaryType(Box::new(
|
||||
<a href=#204 id=204 data-nosnippet>204</a> DictionaryTypeExtension {
|
||||
<a href=#205 id=205 data-nosnippet>205</a> key_datatype: #key_datatype,
|
||||
<a href=#206 id=206 data-nosnippet>206</a> key_datatype_extension: #key_datatype_extension,
|
||||
<a href=#207 id=207 data-nosnippet>207</a> value_datatype: #value_datatype,
|
||||
<a href=#208 id=208 data-nosnippet>208</a> value_datatype_extension: #value_datatype_extension,
|
||||
<a href=#209 id=209 data-nosnippet>209</a> },
|
||||
<a href=#210 id=210 data-nosnippet>210</a> ))),
|
||||
<a href=#211 id=211 data-nosnippet>211</a> })
|
||||
<a href=#212 id=212 data-nosnippet>212</a> }
|
||||
<a href=#213 id=213 data-nosnippet>213</a> }
|
||||
<a href=#214 id=214 data-nosnippet>214</a> <span class="prelude-val">None </span>=> <span class="macro">quote!</span> { <span class="prelude-val">None </span>},
|
||||
<a href=#215 id=215 data-nosnippet>215</a> }
|
||||
<a href=#216 id=216 data-nosnippet>216</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
89
src/common_macro/row/to_row.rs.html
Normal file
89
src/common_macro/row/to_row.rs.html
Normal file
@@ -0,0 +1,89 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row/to_row.rs`."><title>to_row.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../../static.files/storage-41dd4d93.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/row/</div>to_row.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>proc_macro2::TokenStream <span class="kw">as </span>TokenStream2;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>quote::quote;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::{DeriveInput, <span class="prelude-ty">Result</span>};
|
||||
<a href=#19 id=19 data-nosnippet>19</a>
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span><span class="kw">crate</span>::row::utils::{
|
||||
<a href=#21 id=21 data-nosnippet>21</a> ParsedField, convert_column_data_type_to_value_data_ident, extract_struct_fields,
|
||||
<a href=#22 id=22 data-nosnippet>22</a> get_column_data_type, parse_fields_from_fields_named,
|
||||
<a href=#23 id=23 data-nosnippet>23</a>};
|
||||
<a href=#24 id=24 data-nosnippet>24</a><span class="kw">use </span><span class="kw">crate</span>::row::{META_KEY_COL, META_KEY_DATATYPE};
|
||||
<a href=#25 id=25 data-nosnippet>25</a>
|
||||
<a href=#26 id=26 data-nosnippet>26</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>derive_to_row_impl(input: DeriveInput) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#27 id=27 data-nosnippet>27</a> <span class="kw">let </span><span class="prelude-val">Some</span>(fields) = extract_struct_fields(<span class="kw-2">&</span>input.data) <span class="kw">else </span>{
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#29 id=29 data-nosnippet>29</a> input.span(),
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="string">"ToRow can only be derived for structs"</span>,
|
||||
<a href=#31 id=31 data-nosnippet>31</a> ));
|
||||
<a href=#32 id=32 data-nosnippet>32</a> };
|
||||
<a href=#33 id=33 data-nosnippet>33</a> <span class="kw">let </span>ident = input.ident;
|
||||
<a href=#34 id=34 data-nosnippet>34</a> <span class="kw">let </span>(impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">let </span>fields = parse_fields_from_fields_named(fields)<span class="question-mark">?</span>;
|
||||
<a href=#36 id=36 data-nosnippet>36</a>
|
||||
<a href=#37 id=37 data-nosnippet>37</a> <span class="comment">// Implement `to_row` method.
|
||||
<a href=#38 id=38 data-nosnippet>38</a> </span><span class="kw">let </span>impl_to_row_method = impl_to_row_method_combined(<span class="kw-2">&</span>fields)<span class="question-mark">?</span>;
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#40 id=40 data-nosnippet>40</a> <span class="kw">impl </span>#impl_generics #ident #ty_generics #where_clause {
|
||||
<a href=#41 id=41 data-nosnippet>41</a> #impl_to_row_method
|
||||
<a href=#42 id=42 data-nosnippet>42</a> }
|
||||
<a href=#43 id=43 data-nosnippet>43</a> })
|
||||
<a href=#44 id=44 data-nosnippet>44</a>}
|
||||
<a href=#45 id=45 data-nosnippet>45</a>
|
||||
<a href=#46 id=46 data-nosnippet>46</a><span class="kw">fn </span>impl_to_row_method_combined(fields: <span class="kw-2">&</span>[ParsedField<<span class="lifetime">'_</span>>]) -> <span class="prelude-ty">Result</span><TokenStream2> {
|
||||
<a href=#47 id=47 data-nosnippet>47</a> <span class="kw">let </span>value_exprs = fields
|
||||
<a href=#48 id=48 data-nosnippet>48</a> .iter()
|
||||
<a href=#49 id=49 data-nosnippet>49</a> .map(|field| {
|
||||
<a href=#50 id=50 data-nosnippet>50</a> <span class="kw">let </span>ParsedField {ident, field_type, column_data_type, column_attribute} = field;
|
||||
<a href=#51 id=51 data-nosnippet>51</a> <span class="kw">let </span><span class="prelude-val">Some</span>(column_data_type) = get_column_data_type(column_data_type, column_attribute)
|
||||
<a href=#52 id=52 data-nosnippet>52</a> <span class="kw">else </span>{
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="kw">return </span><span class="prelude-val">Err</span>(syn::Error::new(
|
||||
<a href=#54 id=54 data-nosnippet>54</a> ident.span(),
|
||||
<a href=#55 id=55 data-nosnippet>55</a> <span class="macro">format!</span>(
|
||||
<a href=#56 id=56 data-nosnippet>56</a> <span class="string">"expected to set data type explicitly via [({META_KEY_COL}({META_KEY_DATATYPE} = \"...\"))]"
|
||||
<a href=#57 id=57 data-nosnippet>57</a> </span>),
|
||||
<a href=#58 id=58 data-nosnippet>58</a> ));
|
||||
<a href=#59 id=59 data-nosnippet>59</a> };
|
||||
<a href=#60 id=60 data-nosnippet>60</a> <span class="kw">let </span>value_data = convert_column_data_type_to_value_data_ident(<span class="kw-2">&</span>column_data_type.data_type);
|
||||
<a href=#61 id=61 data-nosnippet>61</a> <span class="kw">let </span>expr = <span class="kw">if </span>field_type.is_optional() {
|
||||
<a href=#62 id=62 data-nosnippet>62</a> <span class="macro">quote!</span> {
|
||||
<a href=#63 id=63 data-nosnippet>63</a> <span class="kw">match </span><span class="kw-2">&</span><span class="self">self</span>.#ident {
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="prelude-val">Some</span>(v) => Value {
|
||||
<a href=#65 id=65 data-nosnippet>65</a> value_data: <span class="prelude-val">Some</span>(ValueData::#value_data(v.clone().into())),
|
||||
<a href=#66 id=66 data-nosnippet>66</a> },
|
||||
<a href=#67 id=67 data-nosnippet>67</a> <span class="prelude-val">None </span>=> Value { value_data: <span class="prelude-val">None </span>},
|
||||
<a href=#68 id=68 data-nosnippet>68</a> }
|
||||
<a href=#69 id=69 data-nosnippet>69</a> }
|
||||
<a href=#70 id=70 data-nosnippet>70</a> } <span class="kw">else </span>{
|
||||
<a href=#71 id=71 data-nosnippet>71</a> <span class="macro">quote!</span> {
|
||||
<a href=#72 id=72 data-nosnippet>72</a> Value {
|
||||
<a href=#73 id=73 data-nosnippet>73</a> value_data: <span class="prelude-val">Some</span>(ValueData::#value_data(<span class="self">self</span>.#ident.clone().into())),
|
||||
<a href=#74 id=74 data-nosnippet>74</a> }
|
||||
<a href=#75 id=75 data-nosnippet>75</a> }
|
||||
<a href=#76 id=76 data-nosnippet>76</a> };
|
||||
<a href=#77 id=77 data-nosnippet>77</a> <span class="prelude-val">Ok</span>(expr)
|
||||
<a href=#78 id=78 data-nosnippet>78</a> })
|
||||
<a href=#79 id=79 data-nosnippet>79</a> .collect::<<span class="prelude-ty">Result</span><Vec<<span class="kw">_</span>>>>()<span class="question-mark">?</span>;
|
||||
<a href=#80 id=80 data-nosnippet>80</a>
|
||||
<a href=#81 id=81 data-nosnippet>81</a> <span class="prelude-val">Ok</span>(<span class="macro">quote!</span> {
|
||||
<a href=#82 id=82 data-nosnippet>82</a> <span class="kw">pub fn </span>to_row(<span class="kw-2">&</span><span class="self">self</span>) -> Row {
|
||||
<a href=#83 id=83 data-nosnippet>83</a> Row {
|
||||
<a href=#84 id=84 data-nosnippet>84</a> values: <span class="macro">vec!</span>[ #( #value_exprs ),* ]
|
||||
<a href=#85 id=85 data-nosnippet>85</a> }
|
||||
<a href=#86 id=86 data-nosnippet>86</a> }
|
||||
<a href=#87 id=87 data-nosnippet>87</a> })
|
||||
<a href=#88 id=88 data-nosnippet>88</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
317
src/common_macro/row/utils.rs.html
Normal file
317
src/common_macro/row/utils.rs.html
Normal file
@@ -0,0 +1,317 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/row/utils.rs`."><title>utils.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../../static.files/storage-41dd4d93.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/row/</div>utils.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>std::collections::HashMap;
|
||||
<a href=#16 id=16 data-nosnippet>16</a>
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>greptime_proto::v1::column_data_type_extension::TypeExt;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>greptime_proto::v1::{ColumnDataType, ColumnDataTypeExtension, JsonTypeExtension};
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">use </span>once_cell::sync::Lazy;
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span>quote::format_ident;
|
||||
<a href=#21 id=21 data-nosnippet>21</a><span class="kw">use </span>syn::{
|
||||
<a href=#22 id=22 data-nosnippet>22</a> AngleBracketedGenericArguments, Data, DataStruct, Fields, FieldsNamed, GenericArgument, Ident,
|
||||
<a href=#23 id=23 data-nosnippet>23</a> Path, PathArguments, PathSegment, <span class="prelude-ty">Result</span>, Type, TypePath, TypeReference,
|
||||
<a href=#24 id=24 data-nosnippet>24</a>};
|
||||
<a href=#25 id=25 data-nosnippet>25</a>
|
||||
<a href=#26 id=26 data-nosnippet>26</a><span class="kw">use </span><span class="kw">crate</span>::row::attribute::{ColumnAttribute, find_column_attribute, parse_column_attribute};
|
||||
<a href=#27 id=27 data-nosnippet>27</a>
|
||||
<a href=#28 id=28 data-nosnippet>28</a><span class="kw">static </span>SEMANTIC_TYPES: Lazy<HashMap<<span class="kw-2">&</span><span class="lifetime">'static </span>str, SemanticType>> = Lazy::new(|| {
|
||||
<a href=#29 id=29 data-nosnippet>29</a> HashMap::from([
|
||||
<a href=#30 id=30 data-nosnippet>30</a> (<span class="string">"field"</span>, SemanticType::Field),
|
||||
<a href=#31 id=31 data-nosnippet>31</a> (<span class="string">"tag"</span>, SemanticType::Tag),
|
||||
<a href=#32 id=32 data-nosnippet>32</a> (<span class="string">"timestamp"</span>, SemanticType::Timestamp),
|
||||
<a href=#33 id=33 data-nosnippet>33</a> ])
|
||||
<a href=#34 id=34 data-nosnippet>34</a>});
|
||||
<a href=#35 id=35 data-nosnippet>35</a>
|
||||
<a href=#36 id=36 data-nosnippet>36</a><span class="kw">static </span>DATATYPE_TO_COLUMN_DATA_TYPE: Lazy<HashMap<<span class="kw-2">&</span><span class="lifetime">'static </span>str, ColumnDataTypeWithExtension>> =
|
||||
<a href=#37 id=37 data-nosnippet>37</a> Lazy::new(|| {
|
||||
<a href=#38 id=38 data-nosnippet>38</a> HashMap::from([
|
||||
<a href=#39 id=39 data-nosnippet>39</a> <span class="comment">// Timestamp
|
||||
<a href=#40 id=40 data-nosnippet>40</a> </span>(<span class="string">"timestampsecond"</span>, ColumnDataType::TimestampSecond.into()),
|
||||
<a href=#41 id=41 data-nosnippet>41</a> (
|
||||
<a href=#42 id=42 data-nosnippet>42</a> <span class="string">"timestampmillisecond"</span>,
|
||||
<a href=#43 id=43 data-nosnippet>43</a> ColumnDataType::TimestampMillisecond.into(),
|
||||
<a href=#44 id=44 data-nosnippet>44</a> ),
|
||||
<a href=#45 id=45 data-nosnippet>45</a> (
|
||||
<a href=#46 id=46 data-nosnippet>46</a> <span class="string">"timestampmicrosecond"</span>,
|
||||
<a href=#47 id=47 data-nosnippet>47</a> ColumnDataType::TimestampMicrosecond.into(),
|
||||
<a href=#48 id=48 data-nosnippet>48</a> ),
|
||||
<a href=#49 id=49 data-nosnippet>49</a> (
|
||||
<a href=#50 id=50 data-nosnippet>50</a> <span class="string">"timestampnanosecond"</span>,
|
||||
<a href=#51 id=51 data-nosnippet>51</a> ColumnDataType::TimestampNanosecond.into(),
|
||||
<a href=#52 id=52 data-nosnippet>52</a> ),
|
||||
<a href=#53 id=53 data-nosnippet>53</a> <span class="comment">// Date
|
||||
<a href=#54 id=54 data-nosnippet>54</a> </span>(<span class="string">"date"</span>, ColumnDataType::Date.into()),
|
||||
<a href=#55 id=55 data-nosnippet>55</a> (<span class="string">"datetime"</span>, ColumnDataType::Datetime.into()),
|
||||
<a href=#56 id=56 data-nosnippet>56</a> <span class="comment">// Time
|
||||
<a href=#57 id=57 data-nosnippet>57</a> </span>(<span class="string">"timesecond"</span>, ColumnDataType::TimeSecond.into()),
|
||||
<a href=#58 id=58 data-nosnippet>58</a> (<span class="string">"timemillisecond"</span>, ColumnDataType::TimeMillisecond.into()),
|
||||
<a href=#59 id=59 data-nosnippet>59</a> (<span class="string">"timemicrosecond"</span>, ColumnDataType::TimeMicrosecond.into()),
|
||||
<a href=#60 id=60 data-nosnippet>60</a> (<span class="string">"timenanosecond"</span>, ColumnDataType::TimeNanosecond.into()),
|
||||
<a href=#61 id=61 data-nosnippet>61</a> <span class="comment">// Others
|
||||
<a href=#62 id=62 data-nosnippet>62</a> </span>(<span class="string">"string"</span>, ColumnDataType::String.into()),
|
||||
<a href=#63 id=63 data-nosnippet>63</a> (<span class="string">"json"</span>, ColumnDataTypeWithExtension::json()),
|
||||
<a href=#64 id=64 data-nosnippet>64</a> <span class="comment">// TODO(weny): support vector and decimal128.
|
||||
<a href=#65 id=65 data-nosnippet>65</a> </span>])
|
||||
<a href=#66 id=66 data-nosnippet>66</a> });
|
||||
<a href=#67 id=67 data-nosnippet>67</a>
|
||||
<a href=#68 id=68 data-nosnippet>68</a><span class="kw">static </span>PRIMITIVE_TYPE_TO_COLUMN_DATA_TYPE: Lazy<HashMap<<span class="kw-2">&</span><span class="lifetime">'static </span>str, ColumnDataType>> =
|
||||
<a href=#69 id=69 data-nosnippet>69</a> Lazy::new(|| {
|
||||
<a href=#70 id=70 data-nosnippet>70</a> HashMap::from([
|
||||
<a href=#71 id=71 data-nosnippet>71</a> (<span class="string">"i8"</span>, ColumnDataType::Int8),
|
||||
<a href=#72 id=72 data-nosnippet>72</a> (<span class="string">"i16"</span>, ColumnDataType::Int16),
|
||||
<a href=#73 id=73 data-nosnippet>73</a> (<span class="string">"i32"</span>, ColumnDataType::Int32),
|
||||
<a href=#74 id=74 data-nosnippet>74</a> (<span class="string">"i64"</span>, ColumnDataType::Int64),
|
||||
<a href=#75 id=75 data-nosnippet>75</a> (<span class="string">"u8"</span>, ColumnDataType::Uint8),
|
||||
<a href=#76 id=76 data-nosnippet>76</a> (<span class="string">"u16"</span>, ColumnDataType::Uint16),
|
||||
<a href=#77 id=77 data-nosnippet>77</a> (<span class="string">"u32"</span>, ColumnDataType::Uint32),
|
||||
<a href=#78 id=78 data-nosnippet>78</a> (<span class="string">"u64"</span>, ColumnDataType::Uint64),
|
||||
<a href=#79 id=79 data-nosnippet>79</a> (<span class="string">"f32"</span>, ColumnDataType::Float32),
|
||||
<a href=#80 id=80 data-nosnippet>80</a> (<span class="string">"f64"</span>, ColumnDataType::Float64),
|
||||
<a href=#81 id=81 data-nosnippet>81</a> (<span class="string">"bool"</span>, ColumnDataType::Boolean),
|
||||
<a href=#82 id=82 data-nosnippet>82</a> ])
|
||||
<a href=#83 id=83 data-nosnippet>83</a> });
|
||||
<a href=#84 id=84 data-nosnippet>84</a>
|
||||
<a href=#85 id=85 data-nosnippet>85</a><span class="doccomment">/// Extract the fields of a struct.
|
||||
<a href=#86 id=86 data-nosnippet>86</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>extract_struct_fields(data: <span class="kw-2">&</span>Data) -> <span class="prelude-ty">Option</span><<span class="kw-2">&</span>FieldsNamed> {
|
||||
<a href=#87 id=87 data-nosnippet>87</a> <span class="kw">let </span>Data::Struct(DataStruct {
|
||||
<a href=#88 id=88 data-nosnippet>88</a> fields: Fields::Named(named),
|
||||
<a href=#89 id=89 data-nosnippet>89</a> ..
|
||||
<a href=#90 id=90 data-nosnippet>90</a> }) = <span class="kw-2">&</span>data
|
||||
<a href=#91 id=91 data-nosnippet>91</a> <span class="kw">else </span>{
|
||||
<a href=#92 id=92 data-nosnippet>92</a> <span class="kw">return </span><span class="prelude-val">None</span>;
|
||||
<a href=#93 id=93 data-nosnippet>93</a> };
|
||||
<a href=#94 id=94 data-nosnippet>94</a>
|
||||
<a href=#95 id=95 data-nosnippet>95</a> <span class="prelude-val">Some</span>(named)
|
||||
<a href=#96 id=96 data-nosnippet>96</a>}
|
||||
<a href=#97 id=97 data-nosnippet>97</a>
|
||||
<a href=#98 id=98 data-nosnippet>98</a><span class="doccomment">/// Convert an identifier to a semantic type.
|
||||
<a href=#99 id=99 data-nosnippet>99</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>semantic_type_from_str(ident: <span class="kw-2">&</span>str) -> <span class="prelude-ty">Option</span><SemanticType> {
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="comment">// Ignores the case of the identifier.
|
||||
<a href=#101 id=101 data-nosnippet>101</a> </span><span class="kw">let </span>lowercase = ident.to_lowercase();
|
||||
<a href=#102 id=102 data-nosnippet>102</a> <span class="kw">let </span>lowercase_str = lowercase.as_str();
|
||||
<a href=#103 id=103 data-nosnippet>103</a> SEMANTIC_TYPES.get(lowercase_str).cloned()
|
||||
<a href=#104 id=104 data-nosnippet>104</a>}
|
||||
<a href=#105 id=105 data-nosnippet>105</a>
|
||||
<a href=#106 id=106 data-nosnippet>106</a><span class="doccomment">/// Convert a field type to a column data type.
|
||||
<a href=#107 id=107 data-nosnippet>107</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>column_data_type_from_str(ident: <span class="kw-2">&</span>str) -> <span class="prelude-ty">Option</span><ColumnDataTypeWithExtension> {
|
||||
<a href=#108 id=108 data-nosnippet>108</a> <span class="comment">// Ignores the case of the identifier.
|
||||
<a href=#109 id=109 data-nosnippet>109</a> </span><span class="kw">let </span>lowercase = ident.to_lowercase();
|
||||
<a href=#110 id=110 data-nosnippet>110</a> <span class="kw">let </span>lowercase_str = lowercase.as_str();
|
||||
<a href=#111 id=111 data-nosnippet>111</a> DATATYPE_TO_COLUMN_DATA_TYPE.get(lowercase_str).cloned()
|
||||
<a href=#112 id=112 data-nosnippet>112</a>}
|
||||
<a href=#113 id=113 data-nosnippet>113</a>
|
||||
<a href=#114 id=114 data-nosnippet>114</a><span class="attr">#[derive(Default, Clone, Copy)]
|
||||
<a href=#115 id=115 data-nosnippet>115</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">enum </span>SemanticType {
|
||||
<a href=#116 id=116 data-nosnippet>116</a> <span class="attr">#[default]
|
||||
<a href=#117 id=117 data-nosnippet>117</a> </span>Field,
|
||||
<a href=#118 id=118 data-nosnippet>118</a> Tag,
|
||||
<a href=#119 id=119 data-nosnippet>119</a> Timestamp,
|
||||
<a href=#120 id=120 data-nosnippet>120</a>}
|
||||
<a href=#121 id=121 data-nosnippet>121</a>
|
||||
<a href=#122 id=122 data-nosnippet>122</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">enum </span>FieldType<<span class="lifetime">'a</span>> {
|
||||
<a href=#123 id=123 data-nosnippet>123</a> Required(<span class="kw-2">&</span><span class="lifetime">'a </span>Type),
|
||||
<a href=#124 id=124 data-nosnippet>124</a> Optional(<span class="kw-2">&</span><span class="lifetime">'a </span>Type),
|
||||
<a href=#125 id=125 data-nosnippet>125</a>}
|
||||
<a href=#126 id=126 data-nosnippet>126</a>
|
||||
<a href=#127 id=127 data-nosnippet>127</a><span class="kw">impl </span>FieldType<<span class="lifetime">'_</span>> {
|
||||
<a href=#128 id=128 data-nosnippet>128</a> <span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>is_optional(<span class="kw-2">&</span><span class="self">self</span>) -> bool {
|
||||
<a href=#129 id=129 data-nosnippet>129</a> <span class="macro">matches!</span>(<span class="self">self</span>, FieldType::Optional(<span class="kw">_</span>))
|
||||
<a href=#130 id=130 data-nosnippet>130</a> }
|
||||
<a href=#131 id=131 data-nosnippet>131</a>
|
||||
<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>extract_ident(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="prelude-ty">Option</span><<span class="kw-2">&</span>Ident> {
|
||||
<a href=#133 id=133 data-nosnippet>133</a> <span class="kw">match </span><span class="self">self </span>{
|
||||
<a href=#134 id=134 data-nosnippet>134</a> FieldType::Required(ty) => extract_ident_from_type(ty),
|
||||
<a href=#135 id=135 data-nosnippet>135</a> FieldType::Optional(ty) => extract_ident_from_type(ty),
|
||||
<a href=#136 id=136 data-nosnippet>136</a> }
|
||||
<a href=#137 id=137 data-nosnippet>137</a> }
|
||||
<a href=#138 id=138 data-nosnippet>138</a>}
|
||||
<a href=#139 id=139 data-nosnippet>139</a>
|
||||
<a href=#140 id=140 data-nosnippet>140</a><span class="kw">fn </span>field_type(ty: <span class="kw-2">&</span>Type) -> FieldType<<span class="lifetime">'_</span>> {
|
||||
<a href=#141 id=141 data-nosnippet>141</a> <span class="kw">if let </span>Type::Reference(TypeReference { elem, .. }) = ty {
|
||||
<a href=#142 id=142 data-nosnippet>142</a> <span class="kw">return </span>field_type(elem);
|
||||
<a href=#143 id=143 data-nosnippet>143</a> }
|
||||
<a href=#144 id=144 data-nosnippet>144</a>
|
||||
<a href=#145 id=145 data-nosnippet>145</a> <span class="kw">if let </span>Type::Path(TypePath {
|
||||
<a href=#146 id=146 data-nosnippet>146</a> qself: <span class="kw">_</span>,
|
||||
<a href=#147 id=147 data-nosnippet>147</a> path: Path {
|
||||
<a href=#148 id=148 data-nosnippet>148</a> leading_colon,
|
||||
<a href=#149 id=149 data-nosnippet>149</a> segments,
|
||||
<a href=#150 id=150 data-nosnippet>150</a> },
|
||||
<a href=#151 id=151 data-nosnippet>151</a> }) = ty
|
||||
<a href=#152 id=152 data-nosnippet>152</a> && leading_colon.is_none()
|
||||
<a href=#153 id=153 data-nosnippet>153</a> && segments.len() == <span class="number">1
|
||||
<a href=#154 id=154 data-nosnippet>154</a> </span>&& <span class="kw">let </span><span class="prelude-val">Some</span>(PathSegment {
|
||||
<a href=#155 id=155 data-nosnippet>155</a> ident,
|
||||
<a href=#156 id=156 data-nosnippet>156</a> arguments: PathArguments::AngleBracketed(AngleBracketedGenericArguments { args, .. }),
|
||||
<a href=#157 id=157 data-nosnippet>157</a> }) = segments.first()
|
||||
<a href=#158 id=158 data-nosnippet>158</a> && <span class="kw">let </span>(<span class="number">1</span>, <span class="prelude-val">Some</span>(GenericArgument::Type(t))) = (args.len(), args.first())
|
||||
<a href=#159 id=159 data-nosnippet>159</a> && ident == <span class="string">"Option"
|
||||
<a href=#160 id=160 data-nosnippet>160</a> </span>{
|
||||
<a href=#161 id=161 data-nosnippet>161</a> <span class="kw">return </span>FieldType::Optional(t);
|
||||
<a href=#162 id=162 data-nosnippet>162</a> }
|
||||
<a href=#163 id=163 data-nosnippet>163</a>
|
||||
<a href=#164 id=164 data-nosnippet>164</a> FieldType::Required(ty)
|
||||
<a href=#165 id=165 data-nosnippet>165</a>}
|
||||
<a href=#166 id=166 data-nosnippet>166</a>
|
||||
<a href=#167 id=167 data-nosnippet>167</a><span class="kw">fn </span>extract_ident_from_type(ty: <span class="kw-2">&</span>Type) -> <span class="prelude-ty">Option</span><<span class="kw-2">&</span>Ident> {
|
||||
<a href=#168 id=168 data-nosnippet>168</a> <span class="kw">match </span>ty {
|
||||
<a href=#169 id=169 data-nosnippet>169</a> Type::Path(TypePath { qself: <span class="prelude-val">None</span>, path }) => path.get_ident(),
|
||||
<a href=#170 id=170 data-nosnippet>170</a> Type::Reference(type_ref) => extract_ident_from_type(<span class="kw-2">&</span>type_ref.elem),
|
||||
<a href=#171 id=171 data-nosnippet>171</a> Type::Group(type_group) => extract_ident_from_type(<span class="kw-2">&</span>type_group.elem),
|
||||
<a href=#172 id=172 data-nosnippet>172</a> <span class="kw">_ </span>=> <span class="prelude-val">None</span>,
|
||||
<a href=#173 id=173 data-nosnippet>173</a> }
|
||||
<a href=#174 id=174 data-nosnippet>174</a>}
|
||||
<a href=#175 id=175 data-nosnippet>175</a>
|
||||
<a href=#176 id=176 data-nosnippet>176</a><span class="doccomment">/// Convert a semantic type to a proto semantic type.
|
||||
<a href=#177 id=177 data-nosnippet>177</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>convert_semantic_type_to_proto_semantic_type(
|
||||
<a href=#178 id=178 data-nosnippet>178</a> semantic_type: SemanticType,
|
||||
<a href=#179 id=179 data-nosnippet>179</a>) -> greptime_proto::v1::SemanticType {
|
||||
<a href=#180 id=180 data-nosnippet>180</a> <span class="kw">match </span>semantic_type {
|
||||
<a href=#181 id=181 data-nosnippet>181</a> SemanticType::Field => greptime_proto::v1::SemanticType::Field,
|
||||
<a href=#182 id=182 data-nosnippet>182</a> SemanticType::Tag => greptime_proto::v1::SemanticType::Tag,
|
||||
<a href=#183 id=183 data-nosnippet>183</a> SemanticType::Timestamp => greptime_proto::v1::SemanticType::Timestamp,
|
||||
<a href=#184 id=184 data-nosnippet>184</a> }
|
||||
<a href=#185 id=185 data-nosnippet>185</a>}
|
||||
<a href=#186 id=186 data-nosnippet>186</a>
|
||||
<a href=#187 id=187 data-nosnippet>187</a><span class="attr">#[derive(Debug, Clone)]
|
||||
<a href=#188 id=188 data-nosnippet>188</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">struct </span>ColumnDataTypeWithExtension {
|
||||
<a href=#189 id=189 data-nosnippet>189</a> <span class="kw">pub</span>(<span class="kw">crate</span>) data_type: ColumnDataType,
|
||||
<a href=#190 id=190 data-nosnippet>190</a> <span class="kw">pub</span>(<span class="kw">crate</span>) extension: <span class="prelude-ty">Option</span><ColumnDataTypeExtension>,
|
||||
<a href=#191 id=191 data-nosnippet>191</a>}
|
||||
<a href=#192 id=192 data-nosnippet>192</a>
|
||||
<a href=#193 id=193 data-nosnippet>193</a><span class="kw">impl </span>ColumnDataTypeWithExtension {
|
||||
<a href=#194 id=194 data-nosnippet>194</a> <span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>json() -> <span class="self">Self </span>{
|
||||
<a href=#195 id=195 data-nosnippet>195</a> <span class="self">Self </span>{
|
||||
<a href=#196 id=196 data-nosnippet>196</a> data_type: ColumnDataType::Json,
|
||||
<a href=#197 id=197 data-nosnippet>197</a> extension: <span class="prelude-val">Some</span>(ColumnDataTypeExtension {
|
||||
<a href=#198 id=198 data-nosnippet>198</a> type_ext: <span class="prelude-val">Some</span>(TypeExt::JsonType(JsonTypeExtension::JsonBinary.into())),
|
||||
<a href=#199 id=199 data-nosnippet>199</a> }),
|
||||
<a href=#200 id=200 data-nosnippet>200</a> }
|
||||
<a href=#201 id=201 data-nosnippet>201</a> }
|
||||
<a href=#202 id=202 data-nosnippet>202</a>}
|
||||
<a href=#203 id=203 data-nosnippet>203</a>
|
||||
<a href=#204 id=204 data-nosnippet>204</a><span class="kw">impl </span>From<ColumnDataType> <span class="kw">for </span>ColumnDataTypeWithExtension {
|
||||
<a href=#205 id=205 data-nosnippet>205</a> <span class="kw">fn </span>from(data_type: ColumnDataType) -> <span class="self">Self </span>{
|
||||
<a href=#206 id=206 data-nosnippet>206</a> <span class="self">Self </span>{
|
||||
<a href=#207 id=207 data-nosnippet>207</a> data_type,
|
||||
<a href=#208 id=208 data-nosnippet>208</a> extension: <span class="prelude-val">None</span>,
|
||||
<a href=#209 id=209 data-nosnippet>209</a> }
|
||||
<a href=#210 id=210 data-nosnippet>210</a> }
|
||||
<a href=#211 id=211 data-nosnippet>211</a>}
|
||||
<a href=#212 id=212 data-nosnippet>212</a>
|
||||
<a href=#213 id=213 data-nosnippet>213</a><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">struct </span>ParsedField<<span class="lifetime">'a</span>> {
|
||||
<a href=#214 id=214 data-nosnippet>214</a> <span class="kw">pub</span>(<span class="kw">crate</span>) ident: <span class="kw-2">&</span><span class="lifetime">'a </span>Ident,
|
||||
<a href=#215 id=215 data-nosnippet>215</a> <span class="kw">pub</span>(<span class="kw">crate</span>) field_type: FieldType<<span class="lifetime">'a</span>>,
|
||||
<a href=#216 id=216 data-nosnippet>216</a> <span class="kw">pub</span>(<span class="kw">crate</span>) column_data_type: <span class="prelude-ty">Option</span><ColumnDataTypeWithExtension>,
|
||||
<a href=#217 id=217 data-nosnippet>217</a> <span class="kw">pub</span>(<span class="kw">crate</span>) column_attribute: ColumnAttribute,
|
||||
<a href=#218 id=218 data-nosnippet>218</a>}
|
||||
<a href=#219 id=219 data-nosnippet>219</a>
|
||||
<a href=#220 id=220 data-nosnippet>220</a><span class="doccomment">/// Parse fields from fields named.
|
||||
<a href=#221 id=221 data-nosnippet>221</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>parse_fields_from_fields_named(named: <span class="kw-2">&</span>FieldsNamed) -> <span class="prelude-ty">Result</span><Vec<ParsedField<<span class="lifetime">'_</span>>>> {
|
||||
<a href=#222 id=222 data-nosnippet>222</a> <span class="prelude-val">Ok</span>(named
|
||||
<a href=#223 id=223 data-nosnippet>223</a> .named
|
||||
<a href=#224 id=224 data-nosnippet>224</a> .iter()
|
||||
<a href=#225 id=225 data-nosnippet>225</a> .map(|field| {
|
||||
<a href=#226 id=226 data-nosnippet>226</a> <span class="kw">let </span>ident = field.ident.as_ref().expect(<span class="string">"field must have an ident"</span>);
|
||||
<a href=#227 id=227 data-nosnippet>227</a> <span class="kw">let </span>field_type = field_type(<span class="kw-2">&</span>field.ty);
|
||||
<a href=#228 id=228 data-nosnippet>228</a> <span class="kw">let </span>column_data_type = field_type
|
||||
<a href=#229 id=229 data-nosnippet>229</a> .extract_ident()
|
||||
<a href=#230 id=230 data-nosnippet>230</a> .and_then(convert_primitive_type_to_column_data_type);
|
||||
<a href=#231 id=231 data-nosnippet>231</a> <span class="kw">let </span>column_attribute = find_column_attribute(<span class="kw-2">&</span>field.attrs)
|
||||
<a href=#232 id=232 data-nosnippet>232</a> .map(parse_column_attribute)
|
||||
<a href=#233 id=233 data-nosnippet>233</a> .transpose()<span class="question-mark">?
|
||||
<a href=#234 id=234 data-nosnippet>234</a> </span>.unwrap_or_default();
|
||||
<a href=#235 id=235 data-nosnippet>235</a>
|
||||
<a href=#236 id=236 data-nosnippet>236</a> <span class="prelude-val">Ok</span>(ParsedField {
|
||||
<a href=#237 id=237 data-nosnippet>237</a> ident,
|
||||
<a href=#238 id=238 data-nosnippet>238</a> field_type,
|
||||
<a href=#239 id=239 data-nosnippet>239</a> column_data_type,
|
||||
<a href=#240 id=240 data-nosnippet>240</a> column_attribute,
|
||||
<a href=#241 id=241 data-nosnippet>241</a> })
|
||||
<a href=#242 id=242 data-nosnippet>242</a> })
|
||||
<a href=#243 id=243 data-nosnippet>243</a> .collect::<<span class="prelude-ty">Result</span><Vec<ParsedField<<span class="lifetime">'_</span>>>>>()<span class="question-mark">?
|
||||
<a href=#244 id=244 data-nosnippet>244</a> </span>.into_iter()
|
||||
<a href=#245 id=245 data-nosnippet>245</a> .filter(|field| !field.column_attribute.skip)
|
||||
<a href=#246 id=246 data-nosnippet>246</a> .collect::<Vec<<span class="kw">_</span>>>())
|
||||
<a href=#247 id=247 data-nosnippet>247</a>}
|
||||
<a href=#248 id=248 data-nosnippet>248</a>
|
||||
<a href=#249 id=249 data-nosnippet>249</a><span class="kw">fn </span>convert_primitive_type_to_column_data_type(
|
||||
<a href=#250 id=250 data-nosnippet>250</a> ident: <span class="kw-2">&</span>Ident,
|
||||
<a href=#251 id=251 data-nosnippet>251</a>) -> <span class="prelude-ty">Option</span><ColumnDataTypeWithExtension> {
|
||||
<a href=#252 id=252 data-nosnippet>252</a> PRIMITIVE_TYPE_TO_COLUMN_DATA_TYPE
|
||||
<a href=#253 id=253 data-nosnippet>253</a> .get(ident.to_string().as_str())
|
||||
<a href=#254 id=254 data-nosnippet>254</a> .cloned()
|
||||
<a href=#255 id=255 data-nosnippet>255</a> .map(ColumnDataTypeWithExtension::from)
|
||||
<a href=#256 id=256 data-nosnippet>256</a>}
|
||||
<a href=#257 id=257 data-nosnippet>257</a>
|
||||
<a href=#258 id=258 data-nosnippet>258</a><span class="doccomment">/// Get the column data type from the attribute or the inferred column data type.
|
||||
<a href=#259 id=259 data-nosnippet>259</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>get_column_data_type(
|
||||
<a href=#260 id=260 data-nosnippet>260</a> infer_column_data_type: <span class="kw-2">&</span><span class="prelude-ty">Option</span><ColumnDataTypeWithExtension>,
|
||||
<a href=#261 id=261 data-nosnippet>261</a> attribute: <span class="kw-2">&</span>ColumnAttribute,
|
||||
<a href=#262 id=262 data-nosnippet>262</a>) -> <span class="prelude-ty">Option</span><ColumnDataTypeWithExtension> {
|
||||
<a href=#263 id=263 data-nosnippet>263</a> attribute
|
||||
<a href=#264 id=264 data-nosnippet>264</a> .datatype
|
||||
<a href=#265 id=265 data-nosnippet>265</a> .clone()
|
||||
<a href=#266 id=266 data-nosnippet>266</a> .or_else(|| infer_column_data_type.clone())
|
||||
<a href=#267 id=267 data-nosnippet>267</a>}
|
||||
<a href=#268 id=268 data-nosnippet>268</a>
|
||||
<a href=#269 id=269 data-nosnippet>269</a><span class="doccomment">/// Convert a column data type to a value data ident.
|
||||
<a href=#270 id=270 data-nosnippet>270</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>convert_column_data_type_to_value_data_ident(
|
||||
<a href=#271 id=271 data-nosnippet>271</a> column_data_type: <span class="kw-2">&</span>ColumnDataType,
|
||||
<a href=#272 id=272 data-nosnippet>272</a>) -> Ident {
|
||||
<a href=#273 id=273 data-nosnippet>273</a> <span class="kw">match </span>column_data_type {
|
||||
<a href=#274 id=274 data-nosnippet>274</a> ColumnDataType::Boolean => <span class="macro">format_ident!</span>(<span class="string">"BoolValue"</span>),
|
||||
<a href=#275 id=275 data-nosnippet>275</a> ColumnDataType::Int8 => <span class="macro">format_ident!</span>(<span class="string">"I8Value"</span>),
|
||||
<a href=#276 id=276 data-nosnippet>276</a> ColumnDataType::Int16 => <span class="macro">format_ident!</span>(<span class="string">"I16Value"</span>),
|
||||
<a href=#277 id=277 data-nosnippet>277</a> ColumnDataType::Int32 => <span class="macro">format_ident!</span>(<span class="string">"I32Value"</span>),
|
||||
<a href=#278 id=278 data-nosnippet>278</a> ColumnDataType::Int64 => <span class="macro">format_ident!</span>(<span class="string">"I64Value"</span>),
|
||||
<a href=#279 id=279 data-nosnippet>279</a> ColumnDataType::Uint8 => <span class="macro">format_ident!</span>(<span class="string">"U8Value"</span>),
|
||||
<a href=#280 id=280 data-nosnippet>280</a> ColumnDataType::Uint16 => <span class="macro">format_ident!</span>(<span class="string">"U16Value"</span>),
|
||||
<a href=#281 id=281 data-nosnippet>281</a> ColumnDataType::Uint32 => <span class="macro">format_ident!</span>(<span class="string">"U32Value"</span>),
|
||||
<a href=#282 id=282 data-nosnippet>282</a> ColumnDataType::Uint64 => <span class="macro">format_ident!</span>(<span class="string">"U64Value"</span>),
|
||||
<a href=#283 id=283 data-nosnippet>283</a> ColumnDataType::Float32 => <span class="macro">format_ident!</span>(<span class="string">"F32Value"</span>),
|
||||
<a href=#284 id=284 data-nosnippet>284</a> ColumnDataType::Float64 => <span class="macro">format_ident!</span>(<span class="string">"F64Value"</span>),
|
||||
<a href=#285 id=285 data-nosnippet>285</a> ColumnDataType::Binary => <span class="macro">format_ident!</span>(<span class="string">"BinaryValue"</span>),
|
||||
<a href=#286 id=286 data-nosnippet>286</a> ColumnDataType::String => <span class="macro">format_ident!</span>(<span class="string">"StringValue"</span>),
|
||||
<a href=#287 id=287 data-nosnippet>287</a> ColumnDataType::Date => <span class="macro">format_ident!</span>(<span class="string">"DateValue"</span>),
|
||||
<a href=#288 id=288 data-nosnippet>288</a> ColumnDataType::Datetime => <span class="macro">format_ident!</span>(<span class="string">"DatetimeValue"</span>),
|
||||
<a href=#289 id=289 data-nosnippet>289</a> ColumnDataType::TimestampSecond => <span class="macro">format_ident!</span>(<span class="string">"TimestampSecondValue"</span>),
|
||||
<a href=#290 id=290 data-nosnippet>290</a> ColumnDataType::TimestampMillisecond => {
|
||||
<a href=#291 id=291 data-nosnippet>291</a> <span class="macro">format_ident!</span>(<span class="string">"TimestampMillisecondValue"</span>)
|
||||
<a href=#292 id=292 data-nosnippet>292</a> }
|
||||
<a href=#293 id=293 data-nosnippet>293</a> ColumnDataType::TimestampMicrosecond => {
|
||||
<a href=#294 id=294 data-nosnippet>294</a> <span class="macro">format_ident!</span>(<span class="string">"TimestampMicrosecondValue"</span>)
|
||||
<a href=#295 id=295 data-nosnippet>295</a> }
|
||||
<a href=#296 id=296 data-nosnippet>296</a> ColumnDataType::TimestampNanosecond => <span class="macro">format_ident!</span>(<span class="string">"TimestampNanosecondValue"</span>),
|
||||
<a href=#297 id=297 data-nosnippet>297</a> ColumnDataType::TimeSecond => <span class="macro">format_ident!</span>(<span class="string">"TimeSecondValue"</span>),
|
||||
<a href=#298 id=298 data-nosnippet>298</a> ColumnDataType::TimeMillisecond => <span class="macro">format_ident!</span>(<span class="string">"TimeMillisecondValue"</span>),
|
||||
<a href=#299 id=299 data-nosnippet>299</a> ColumnDataType::TimeMicrosecond => <span class="macro">format_ident!</span>(<span class="string">"TimeMicrosecondValue"</span>),
|
||||
<a href=#300 id=300 data-nosnippet>300</a> ColumnDataType::TimeNanosecond => <span class="macro">format_ident!</span>(<span class="string">"TimeNanosecondValue"</span>),
|
||||
<a href=#301 id=301 data-nosnippet>301</a> ColumnDataType::IntervalYearMonth => <span class="macro">format_ident!</span>(<span class="string">"IntervalYearMonthValue"</span>),
|
||||
<a href=#302 id=302 data-nosnippet>302</a> ColumnDataType::IntervalDayTime => <span class="macro">format_ident!</span>(<span class="string">"IntervalDayTimeValue"</span>),
|
||||
<a href=#303 id=303 data-nosnippet>303</a> ColumnDataType::IntervalMonthDayNano => {
|
||||
<a href=#304 id=304 data-nosnippet>304</a> <span class="macro">format_ident!</span>(<span class="string">"IntervalMonthDayNanoValue"</span>)
|
||||
<a href=#305 id=305 data-nosnippet>305</a> }
|
||||
<a href=#306 id=306 data-nosnippet>306</a> ColumnDataType::Decimal128 => <span class="macro">format_ident!</span>(<span class="string">"Decimal128Value"</span>),
|
||||
<a href=#307 id=307 data-nosnippet>307</a> <span class="comment">// Json is a special case, it is actually a string column.
|
||||
<a href=#308 id=308 data-nosnippet>308</a> </span>ColumnDataType::Json => <span class="macro">format_ident!</span>(<span class="string">"StringValue"</span>),
|
||||
<a href=#309 id=309 data-nosnippet>309</a> ColumnDataType::Vector => <span class="macro">format_ident!</span>(<span class="string">"VectorValue"</span>),
|
||||
<a href=#310 id=310 data-nosnippet>310</a> ColumnDataType::List => <span class="macro">format_ident!</span>(<span class="string">"ListValue"</span>),
|
||||
<a href=#311 id=311 data-nosnippet>311</a> ColumnDataType::Struct => <span class="macro">format_ident!</span>(<span class="string">"StructValue"</span>),
|
||||
<a href=#312 id=312 data-nosnippet>312</a> ColumnDataType::Dictionary => {
|
||||
<a href=#313 id=313 data-nosnippet>313</a> <span class="macro">panic!</span>(<span class="string">"Dictionary data type is not supported in row macros yet"</span>)
|
||||
<a href=#314 id=314 data-nosnippet>314</a> }
|
||||
<a href=#315 id=315 data-nosnippet>315</a> }
|
||||
<a href=#316 id=316 data-nosnippet>316</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
336
src/common_macro/stack_trace_debug.rs.html
Normal file
336
src/common_macro/stack_trace_debug.rs.html
Normal file
@@ -0,0 +1,336 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/stack_trace_debug.rs`."><title>stack_trace_debug.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>stack_trace_debug.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-3"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="doccomment">//! implement `::common_error::ext::StackError`
|
||||
<a href=#16 id=16 data-nosnippet>16</a>
|
||||
<a href=#17 id=17 data-nosnippet>17</a></span><span class="kw">use </span>proc_macro2::{Literal, Span, TokenStream <span class="kw">as </span>TokenStream2, TokenTree};
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>quote::{quote, quote_spanned};
|
||||
<a href=#19 id=19 data-nosnippet>19</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="kw">use </span>syn::{Attribute, Ident, ItemEnum, Variant, parenthesized};
|
||||
<a href=#21 id=21 data-nosnippet>21</a>
|
||||
<a href=#22 id=22 data-nosnippet>22</a><span class="kw">pub fn </span>stack_trace_style_impl(args: TokenStream2, input: TokenStream2) -> TokenStream2 {
|
||||
<a href=#23 id=23 data-nosnippet>23</a> <span class="kw">let </span>input_cloned: TokenStream2 = input.clone();
|
||||
<a href=#24 id=24 data-nosnippet>24</a>
|
||||
<a href=#25 id=25 data-nosnippet>25</a> <span class="kw">let </span>error_enum_definition: ItemEnum = syn::parse2(input_cloned).unwrap();
|
||||
<a href=#26 id=26 data-nosnippet>26</a> <span class="kw">let </span>enum_name = error_enum_definition.ident;
|
||||
<a href=#27 id=27 data-nosnippet>27</a>
|
||||
<a href=#28 id=28 data-nosnippet>28</a> <span class="kw">let </span><span class="kw-2">mut </span>variants = <span class="macro">vec!</span>[];
|
||||
<a href=#29 id=29 data-nosnippet>29</a>
|
||||
<a href=#30 id=30 data-nosnippet>30</a> <span class="kw">for </span>error_variant <span class="kw">in </span>error_enum_definition.variants {
|
||||
<a href=#31 id=31 data-nosnippet>31</a> <span class="kw">let </span>variant = ErrorVariant::from_enum_variant(error_variant);
|
||||
<a href=#32 id=32 data-nosnippet>32</a> variants.push(variant);
|
||||
<a href=#33 id=33 data-nosnippet>33</a> }
|
||||
<a href=#34 id=34 data-nosnippet>34</a>
|
||||
<a href=#35 id=35 data-nosnippet>35</a> <span class="kw">let </span>transparent_fn = build_transparent_fn(enum_name.clone(), <span class="kw-2">&</span>variants);
|
||||
<a href=#36 id=36 data-nosnippet>36</a> <span class="kw">let </span>debug_fmt_fn = build_debug_fmt_impl(enum_name.clone(), variants.clone());
|
||||
<a href=#37 id=37 data-nosnippet>37</a> <span class="kw">let </span>next_fn = build_next_impl(enum_name.clone(), variants);
|
||||
<a href=#38 id=38 data-nosnippet>38</a> <span class="kw">let </span>debug_impl = build_debug_impl(enum_name.clone());
|
||||
<a href=#39 id=39 data-nosnippet>39</a>
|
||||
<a href=#40 id=40 data-nosnippet>40</a> <span class="macro">quote!</span> {
|
||||
<a href=#41 id=41 data-nosnippet>41</a> #args
|
||||
<a href=#42 id=42 data-nosnippet>42</a> #input
|
||||
<a href=#43 id=43 data-nosnippet>43</a>
|
||||
<a href=#44 id=44 data-nosnippet>44</a> <span class="kw">impl </span>::common_error::ext::StackError <span class="kw">for </span>#enum_name {
|
||||
<a href=#45 id=45 data-nosnippet>45</a> #debug_fmt_fn
|
||||
<a href=#46 id=46 data-nosnippet>46</a> #next_fn
|
||||
<a href=#47 id=47 data-nosnippet>47</a> #transparent_fn
|
||||
<a href=#48 id=48 data-nosnippet>48</a> }
|
||||
<a href=#49 id=49 data-nosnippet>49</a>
|
||||
<a href=#50 id=50 data-nosnippet>50</a> #debug_impl
|
||||
<a href=#51 id=51 data-nosnippet>51</a> }
|
||||
<a href=#52 id=52 data-nosnippet>52</a>}
|
||||
<a href=#53 id=53 data-nosnippet>53</a>
|
||||
<a href=#54 id=54 data-nosnippet>54</a><span class="doccomment">/// Generate `debug_fmt` fn.
|
||||
<a href=#55 id=55 data-nosnippet>55</a>///
|
||||
<a href=#56 id=56 data-nosnippet>56</a>/// The generated fn will be like:
|
||||
<a href=#57 id=57 data-nosnippet>57</a>/// ```rust, ignore
|
||||
<a href=#58 id=58 data-nosnippet>58</a>/// fn debug_fmt(&self, layer: usize, buf: &mut Vec<String>);
|
||||
<a href=#59 id=59 data-nosnippet>59</a>/// ```
|
||||
<a href=#60 id=60 data-nosnippet>60</a></span><span class="kw">fn </span>build_debug_fmt_impl(enum_name: Ident, variants: Vec<ErrorVariant>) -> TokenStream2 {
|
||||
<a href=#61 id=61 data-nosnippet>61</a> <span class="kw">let </span>match_arms = variants
|
||||
<a href=#62 id=62 data-nosnippet>62</a> .iter()
|
||||
<a href=#63 id=63 data-nosnippet>63</a> .map(|v| v.to_debug_match_arm())
|
||||
<a href=#64 id=64 data-nosnippet>64</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#65 id=65 data-nosnippet>65</a>
|
||||
<a href=#66 id=66 data-nosnippet>66</a> <span class="macro">quote!</span> {
|
||||
<a href=#67 id=67 data-nosnippet>67</a> <span class="kw">fn </span>debug_fmt(<span class="kw-2">&</span><span class="self">self</span>, layer: usize, buf: <span class="kw-2">&mut </span>Vec<String>) {
|
||||
<a href=#68 id=68 data-nosnippet>68</a> <span class="kw">use </span>#enum_name::<span class="kw-2">*</span>;
|
||||
<a href=#69 id=69 data-nosnippet>69</a> <span class="kw">match </span><span class="self">self </span>{
|
||||
<a href=#70 id=70 data-nosnippet>70</a> #(#match_arms)*
|
||||
<a href=#71 id=71 data-nosnippet>71</a> }
|
||||
<a href=#72 id=72 data-nosnippet>72</a> }
|
||||
<a href=#73 id=73 data-nosnippet>73</a> }
|
||||
<a href=#74 id=74 data-nosnippet>74</a>}
|
||||
<a href=#75 id=75 data-nosnippet>75</a>
|
||||
<a href=#76 id=76 data-nosnippet>76</a><span class="doccomment">/// Generate `next` fn.
|
||||
<a href=#77 id=77 data-nosnippet>77</a>///
|
||||
<a href=#78 id=78 data-nosnippet>78</a>/// The generated fn will be like:
|
||||
<a href=#79 id=79 data-nosnippet>79</a>/// ```rust, ignore
|
||||
<a href=#80 id=80 data-nosnippet>80</a>/// fn next(&self) -> Option<&dyn ::common_error::ext::StackError>;
|
||||
<a href=#81 id=81 data-nosnippet>81</a>/// ```
|
||||
<a href=#82 id=82 data-nosnippet>82</a></span><span class="kw">fn </span>build_next_impl(enum_name: Ident, variants: Vec<ErrorVariant>) -> TokenStream2 {
|
||||
<a href=#83 id=83 data-nosnippet>83</a> <span class="kw">let </span>match_arms = variants
|
||||
<a href=#84 id=84 data-nosnippet>84</a> .iter()
|
||||
<a href=#85 id=85 data-nosnippet>85</a> .map(|v| v.to_next_match_arm())
|
||||
<a href=#86 id=86 data-nosnippet>86</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#87 id=87 data-nosnippet>87</a>
|
||||
<a href=#88 id=88 data-nosnippet>88</a> <span class="macro">quote!</span> {
|
||||
<a href=#89 id=89 data-nosnippet>89</a> <span class="kw">fn </span>next(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="prelude-ty">Option</span><<span class="kw-2">&</span><span class="kw">dyn </span>::common_error::ext::StackError> {
|
||||
<a href=#90 id=90 data-nosnippet>90</a> <span class="kw">use </span>#enum_name::<span class="kw-2">*</span>;
|
||||
<a href=#91 id=91 data-nosnippet>91</a> <span class="kw">match </span><span class="self">self </span>{
|
||||
<a href=#92 id=92 data-nosnippet>92</a> #(#match_arms)*
|
||||
<a href=#93 id=93 data-nosnippet>93</a> }
|
||||
<a href=#94 id=94 data-nosnippet>94</a> }
|
||||
<a href=#95 id=95 data-nosnippet>95</a> }
|
||||
<a href=#96 id=96 data-nosnippet>96</a>}
|
||||
<a href=#97 id=97 data-nosnippet>97</a>
|
||||
<a href=#98 id=98 data-nosnippet>98</a><span class="doccomment">/// Implement [std::fmt::Debug] via `debug_fmt`
|
||||
<a href=#99 id=99 data-nosnippet>99</a></span><span class="kw">fn </span>build_debug_impl(enum_name: Ident) -> TokenStream2 {
|
||||
<a href=#100 id=100 data-nosnippet>100</a> <span class="macro">quote!</span> {
|
||||
<a href=#101 id=101 data-nosnippet>101</a> <span class="kw">impl </span>std::fmt::Debug <span class="kw">for </span>#enum_name {
|
||||
<a href=#102 id=102 data-nosnippet>102</a> <span class="kw">fn </span>fmt(<span class="kw-2">&</span><span class="self">self</span>, f: <span class="kw-2">&mut </span>std::fmt::Formatter<<span class="lifetime">'_</span>>) -> std::fmt::Result {
|
||||
<a href=#103 id=103 data-nosnippet>103</a> <span class="kw">use </span>::common_error::ext::StackError;
|
||||
<a href=#104 id=104 data-nosnippet>104</a> <span class="kw">let </span><span class="kw-2">mut </span>buf = <span class="macro">vec!</span>[];
|
||||
<a href=#105 id=105 data-nosnippet>105</a> <span class="self">self</span>.debug_fmt(<span class="number">0</span>, <span class="kw-2">&mut </span>buf);
|
||||
<a href=#106 id=106 data-nosnippet>106</a> <span class="macro">write!</span>(f, <span class="string">"{}"</span>, buf.join(<span class="string">"\n"</span>))
|
||||
<a href=#107 id=107 data-nosnippet>107</a> }
|
||||
<a href=#108 id=108 data-nosnippet>108</a> }
|
||||
<a href=#109 id=109 data-nosnippet>109</a> }
|
||||
<a href=#110 id=110 data-nosnippet>110</a>}
|
||||
<a href=#111 id=111 data-nosnippet>111</a>
|
||||
<a href=#112 id=112 data-nosnippet>112</a><span class="attr">#[derive(Clone, Debug)]
|
||||
<a href=#113 id=113 data-nosnippet>113</a></span><span class="kw">struct </span>ErrorVariant {
|
||||
<a href=#114 id=114 data-nosnippet>114</a> name: Ident,
|
||||
<a href=#115 id=115 data-nosnippet>115</a> fields: Vec<Ident>,
|
||||
<a href=#116 id=116 data-nosnippet>116</a> has_location: bool,
|
||||
<a href=#117 id=117 data-nosnippet>117</a> has_source: bool,
|
||||
<a href=#118 id=118 data-nosnippet>118</a> has_external_cause: bool,
|
||||
<a href=#119 id=119 data-nosnippet>119</a> display: TokenStream2,
|
||||
<a href=#120 id=120 data-nosnippet>120</a> transparent: bool,
|
||||
<a href=#121 id=121 data-nosnippet>121</a> span: Span,
|
||||
<a href=#122 id=122 data-nosnippet>122</a> cfg_attr: <span class="prelude-ty">Option</span><Attribute>,
|
||||
<a href=#123 id=123 data-nosnippet>123</a>}
|
||||
<a href=#124 id=124 data-nosnippet>124</a>
|
||||
<a href=#125 id=125 data-nosnippet>125</a><span class="kw">impl </span>ErrorVariant {
|
||||
<a href=#126 id=126 data-nosnippet>126</a> <span class="doccomment">/// Construct self from [Variant]
|
||||
<a href=#127 id=127 data-nosnippet>127</a> </span><span class="kw">fn </span>from_enum_variant(variant: Variant) -> <span class="self">Self </span>{
|
||||
<a href=#128 id=128 data-nosnippet>128</a> <span class="kw">let </span>span = variant.span();
|
||||
<a href=#129 id=129 data-nosnippet>129</a> <span class="kw">let </span><span class="kw-2">mut </span>has_location = <span class="bool-val">false</span>;
|
||||
<a href=#130 id=130 data-nosnippet>130</a> <span class="kw">let </span><span class="kw-2">mut </span>has_source = <span class="bool-val">false</span>;
|
||||
<a href=#131 id=131 data-nosnippet>131</a> <span class="kw">let </span><span class="kw-2">mut </span>has_external_cause = <span class="bool-val">false</span>;
|
||||
<a href=#132 id=132 data-nosnippet>132</a>
|
||||
<a href=#133 id=133 data-nosnippet>133</a> <span class="kw">for </span>field <span class="kw">in </span><span class="kw-2">&</span>variant.fields {
|
||||
<a href=#134 id=134 data-nosnippet>134</a> <span class="kw">if let </span><span class="prelude-val">Some</span>(ident) = <span class="kw-2">&</span>field.ident {
|
||||
<a href=#135 id=135 data-nosnippet>135</a> <span class="kw">if </span>ident == <span class="string">"location" </span>{
|
||||
<a href=#136 id=136 data-nosnippet>136</a> has_location = <span class="bool-val">true</span>;
|
||||
<a href=#137 id=137 data-nosnippet>137</a> } <span class="kw">else if </span>ident == <span class="string">"source" </span>{
|
||||
<a href=#138 id=138 data-nosnippet>138</a> has_source = <span class="bool-val">true</span>;
|
||||
<a href=#139 id=139 data-nosnippet>139</a> } <span class="kw">else if </span>ident == <span class="string">"error" </span>{
|
||||
<a href=#140 id=140 data-nosnippet>140</a> has_external_cause = <span class="bool-val">true</span>;
|
||||
<a href=#141 id=141 data-nosnippet>141</a> }
|
||||
<a href=#142 id=142 data-nosnippet>142</a> }
|
||||
<a href=#143 id=143 data-nosnippet>143</a> }
|
||||
<a href=#144 id=144 data-nosnippet>144</a>
|
||||
<a href=#145 id=145 data-nosnippet>145</a> <span class="kw">let </span><span class="kw-2">mut </span>display = <span class="prelude-val">None</span>;
|
||||
<a href=#146 id=146 data-nosnippet>146</a> <span class="kw">let </span><span class="kw-2">mut </span>transparent = <span class="bool-val">false</span>;
|
||||
<a href=#147 id=147 data-nosnippet>147</a> <span class="kw">let </span><span class="kw-2">mut </span>cfg_attr = <span class="prelude-val">None</span>;
|
||||
<a href=#148 id=148 data-nosnippet>148</a> <span class="kw">for </span>attr <span class="kw">in </span>variant.attrs {
|
||||
<a href=#149 id=149 data-nosnippet>149</a> <span class="kw">if </span>attr.path().is_ident(<span class="string">"snafu"</span>) {
|
||||
<a href=#150 id=150 data-nosnippet>150</a> attr.parse_nested_meta(|meta| {
|
||||
<a href=#151 id=151 data-nosnippet>151</a> <span class="kw">if </span>meta.path.is_ident(<span class="string">"display"</span>) {
|
||||
<a href=#152 id=152 data-nosnippet>152</a> <span class="kw">let </span>content;
|
||||
<a href=#153 id=153 data-nosnippet>153</a> <span class="macro">parenthesized!</span>(content <span class="kw">in </span>meta.input);
|
||||
<a href=#154 id=154 data-nosnippet>154</a> <span class="kw">let </span>display_ts: TokenStream2 = content.parse()<span class="question-mark">?</span>;
|
||||
<a href=#155 id=155 data-nosnippet>155</a> display = <span class="prelude-val">Some</span>(display_ts);
|
||||
<a href=#156 id=156 data-nosnippet>156</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#157 id=157 data-nosnippet>157</a> } <span class="kw">else if </span>meta.path.is_ident(<span class="string">"transparent"</span>) {
|
||||
<a href=#158 id=158 data-nosnippet>158</a> display = <span class="prelude-val">Some</span>(TokenStream2::from(TokenTree::Literal(Literal::string(
|
||||
<a href=#159 id=159 data-nosnippet>159</a> <span class="string">"<transparent>"</span>,
|
||||
<a href=#160 id=160 data-nosnippet>160</a> ))));
|
||||
<a href=#161 id=161 data-nosnippet>161</a> transparent = <span class="bool-val">true</span>;
|
||||
<a href=#162 id=162 data-nosnippet>162</a> <span class="prelude-val">Ok</span>(())
|
||||
<a href=#163 id=163 data-nosnippet>163</a> } <span class="kw">else </span>{
|
||||
<a href=#164 id=164 data-nosnippet>164</a> <span class="prelude-val">Err</span>(meta.error(<span class="string">"unrecognized repr"</span>))
|
||||
<a href=#165 id=165 data-nosnippet>165</a> }
|
||||
<a href=#166 id=166 data-nosnippet>166</a> })
|
||||
<a href=#167 id=167 data-nosnippet>167</a> .unwrap_or_else(|e| <span class="macro">panic!</span>(<span class="string">"{e}"</span>));
|
||||
<a href=#168 id=168 data-nosnippet>168</a> }
|
||||
<a href=#169 id=169 data-nosnippet>169</a>
|
||||
<a href=#170 id=170 data-nosnippet>170</a> <span class="kw">if </span>attr.path().is_ident(<span class="string">"cfg"</span>) {
|
||||
<a href=#171 id=171 data-nosnippet>171</a> cfg_attr = <span class="prelude-val">Some</span>(attr);
|
||||
<a href=#172 id=172 data-nosnippet>172</a> }
|
||||
<a href=#173 id=173 data-nosnippet>173</a> }
|
||||
<a href=#174 id=174 data-nosnippet>174</a> <span class="kw">let </span>display = display.unwrap_or_else(|| {
|
||||
<a href=#175 id=175 data-nosnippet>175</a> <span class="macro">panic!</span>(
|
||||
<a href=#176 id=176 data-nosnippet>176</a> <span class="string">r#"Error "{}" must be annotated with attribute "display" or "transparent"."#</span>,
|
||||
<a href=#177 id=177 data-nosnippet>177</a> variant.ident,
|
||||
<a href=#178 id=178 data-nosnippet>178</a> )
|
||||
<a href=#179 id=179 data-nosnippet>179</a> });
|
||||
<a href=#180 id=180 data-nosnippet>180</a>
|
||||
<a href=#181 id=181 data-nosnippet>181</a> <span class="kw">let </span>field_ident = variant
|
||||
<a href=#182 id=182 data-nosnippet>182</a> .fields
|
||||
<a href=#183 id=183 data-nosnippet>183</a> .iter()
|
||||
<a href=#184 id=184 data-nosnippet>184</a> .map(|f| f.ident.clone().unwrap_or_else(|| Ident::new(<span class="string">"_"</span>, f.span())))
|
||||
<a href=#185 id=185 data-nosnippet>185</a> .collect();
|
||||
<a href=#186 id=186 data-nosnippet>186</a>
|
||||
<a href=#187 id=187 data-nosnippet>187</a> <span class="self">Self </span>{
|
||||
<a href=#188 id=188 data-nosnippet>188</a> name: variant.ident,
|
||||
<a href=#189 id=189 data-nosnippet>189</a> fields: field_ident,
|
||||
<a href=#190 id=190 data-nosnippet>190</a> has_location,
|
||||
<a href=#191 id=191 data-nosnippet>191</a> has_source,
|
||||
<a href=#192 id=192 data-nosnippet>192</a> has_external_cause,
|
||||
<a href=#193 id=193 data-nosnippet>193</a> display,
|
||||
<a href=#194 id=194 data-nosnippet>194</a> transparent,
|
||||
<a href=#195 id=195 data-nosnippet>195</a> span,
|
||||
<a href=#196 id=196 data-nosnippet>196</a> cfg_attr,
|
||||
<a href=#197 id=197 data-nosnippet>197</a> }
|
||||
<a href=#198 id=198 data-nosnippet>198</a> }
|
||||
<a href=#199 id=199 data-nosnippet>199</a>
|
||||
<a href=#200 id=200 data-nosnippet>200</a> <span class="doccomment">/// Convert self into an match arm that will be used in [build_debug_impl].
|
||||
<a href=#201 id=201 data-nosnippet>201</a> ///
|
||||
<a href=#202 id=202 data-nosnippet>202</a> /// The generated match arm will be like:
|
||||
<a href=#203 id=203 data-nosnippet>203</a> /// ```rust, ignore
|
||||
<a href=#204 id=204 data-nosnippet>204</a> /// ErrorKindWithSource { source, .. } => {
|
||||
<a href=#205 id=205 data-nosnippet>205</a> /// debug_fmt(source, layer + 1, buf);
|
||||
<a href=#206 id=206 data-nosnippet>206</a> /// },
|
||||
<a href=#207 id=207 data-nosnippet>207</a> /// ErrorKindWithoutSource { .. } => {
|
||||
<a href=#208 id=208 data-nosnippet>208</a> /// buf.push(format!("{layer}: {}, at {}", format!(#display), location)));
|
||||
<a href=#209 id=209 data-nosnippet>209</a> /// }
|
||||
<a href=#210 id=210 data-nosnippet>210</a> /// ```
|
||||
<a href=#211 id=211 data-nosnippet>211</a> ///
|
||||
<a href=#212 id=212 data-nosnippet>212</a> /// The generated code assumes fn `debug_fmt`, var `layer`, var `buf` are in scope.
|
||||
<a href=#213 id=213 data-nosnippet>213</a> </span><span class="kw">fn </span>to_debug_match_arm(<span class="kw-2">&</span><span class="self">self</span>) -> TokenStream2 {
|
||||
<a href=#214 id=214 data-nosnippet>214</a> <span class="kw">let </span>name = <span class="kw-2">&</span><span class="self">self</span>.name;
|
||||
<a href=#215 id=215 data-nosnippet>215</a> <span class="kw">let </span>fields = <span class="kw-2">&</span><span class="self">self</span>.fields;
|
||||
<a href=#216 id=216 data-nosnippet>216</a> <span class="kw">let </span>display = <span class="kw-2">&</span><span class="self">self</span>.display;
|
||||
<a href=#217 id=217 data-nosnippet>217</a> <span class="kw">let </span>cfg = <span class="kw">if let </span><span class="prelude-val">Some</span>(cfg) = <span class="kw-2">&</span><span class="self">self</span>.cfg_attr {
|
||||
<a href=#218 id=218 data-nosnippet>218</a> <span class="macro">quote_spanned!</span>(cfg.span() => #cfg)
|
||||
<a href=#219 id=219 data-nosnippet>219</a> } <span class="kw">else </span>{
|
||||
<a href=#220 id=220 data-nosnippet>220</a> <span class="macro">quote!</span> {}
|
||||
<a href=#221 id=221 data-nosnippet>221</a> };
|
||||
<a href=#222 id=222 data-nosnippet>222</a>
|
||||
<a href=#223 id=223 data-nosnippet>223</a> <span class="kw">match </span>(<span class="self">self</span>.has_location, <span class="self">self</span>.has_source, <span class="self">self</span>.has_external_cause) {
|
||||
<a href=#224 id=224 data-nosnippet>224</a> (<span class="bool-val">true</span>, <span class="bool-val">true</span>, <span class="kw">_</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#225 id=225 data-nosnippet>225</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),<span class="kw-2">*</span>, } => {
|
||||
<a href=#226 id=226 data-nosnippet>226</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}, at {}"</span>, <span class="macro">format!</span>(#display), location));
|
||||
<a href=#227 id=227 data-nosnippet>227</a> source.debug_fmt(layer + <span class="number">1</span>, buf);
|
||||
<a href=#228 id=228 data-nosnippet>228</a> },
|
||||
<a href=#229 id=229 data-nosnippet>229</a> },
|
||||
<a href=#230 id=230 data-nosnippet>230</a> (<span class="bool-val">true</span>, <span class="bool-val">false</span>, <span class="bool-val">true</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#231 id=231 data-nosnippet>231</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#232 id=232 data-nosnippet>232</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}, at {}"</span>, <span class="macro">format!</span>(#display), location));
|
||||
<a href=#233 id=233 data-nosnippet>233</a> buf.push(<span class="macro">format!</span>(<span class="string">"{}: {:?}"</span>, layer + <span class="number">1</span>, error));
|
||||
<a href=#234 id=234 data-nosnippet>234</a> },
|
||||
<a href=#235 id=235 data-nosnippet>235</a> },
|
||||
<a href=#236 id=236 data-nosnippet>236</a> (<span class="bool-val">true</span>, <span class="bool-val">false</span>, <span class="bool-val">false</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#237 id=237 data-nosnippet>237</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#238 id=238 data-nosnippet>238</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}, at {}"</span>, <span class="macro">format!</span>(#display), location));
|
||||
<a href=#239 id=239 data-nosnippet>239</a> },
|
||||
<a href=#240 id=240 data-nosnippet>240</a> },
|
||||
<a href=#241 id=241 data-nosnippet>241</a> (<span class="bool-val">false</span>, <span class="bool-val">true</span>, <span class="kw">_</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#242 id=242 data-nosnippet>242</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#243 id=243 data-nosnippet>243</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}"</span>, <span class="macro">format!</span>(#display)));
|
||||
<a href=#244 id=244 data-nosnippet>244</a> source.debug_fmt(layer + <span class="number">1</span>, buf);
|
||||
<a href=#245 id=245 data-nosnippet>245</a> },
|
||||
<a href=#246 id=246 data-nosnippet>246</a> },
|
||||
<a href=#247 id=247 data-nosnippet>247</a> (<span class="bool-val">false</span>, <span class="bool-val">false</span>, <span class="bool-val">true</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#248 id=248 data-nosnippet>248</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#249 id=249 data-nosnippet>249</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}"</span>, <span class="macro">format!</span>(#display)));
|
||||
<a href=#250 id=250 data-nosnippet>250</a> buf.push(<span class="macro">format!</span>(<span class="string">"{}: {:?}"</span>, layer + <span class="number">1</span>, error));
|
||||
<a href=#251 id=251 data-nosnippet>251</a> },
|
||||
<a href=#252 id=252 data-nosnippet>252</a> },
|
||||
<a href=#253 id=253 data-nosnippet>253</a> (<span class="bool-val">false</span>, <span class="bool-val">false</span>, <span class="bool-val">false</span>) => <span class="macro">quote_spanned!</span> {
|
||||
<a href=#254 id=254 data-nosnippet>254</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#255 id=255 data-nosnippet>255</a> buf.push(<span class="macro">format!</span>(<span class="string">"{layer}: {}"</span>, <span class="macro">format!</span>(#display)));
|
||||
<a href=#256 id=256 data-nosnippet>256</a> },
|
||||
<a href=#257 id=257 data-nosnippet>257</a> },
|
||||
<a href=#258 id=258 data-nosnippet>258</a> }
|
||||
<a href=#259 id=259 data-nosnippet>259</a> }
|
||||
<a href=#260 id=260 data-nosnippet>260</a>
|
||||
<a href=#261 id=261 data-nosnippet>261</a> <span class="doccomment">/// Convert self into an match arm that will be used in [build_next_impl].
|
||||
<a href=#262 id=262 data-nosnippet>262</a> ///
|
||||
<a href=#263 id=263 data-nosnippet>263</a> /// The generated match arm will be like:
|
||||
<a href=#264 id=264 data-nosnippet>264</a> /// ```rust, ignore
|
||||
<a href=#265 id=265 data-nosnippet>265</a> /// ErrorKindWithSource { source, .. } => {
|
||||
<a href=#266 id=266 data-nosnippet>266</a> /// Some(source)
|
||||
<a href=#267 id=267 data-nosnippet>267</a> /// },
|
||||
<a href=#268 id=268 data-nosnippet>268</a> /// ErrorKindWithoutSource { .. } => {
|
||||
<a href=#269 id=269 data-nosnippet>269</a> /// None
|
||||
<a href=#270 id=270 data-nosnippet>270</a> /// }
|
||||
<a href=#271 id=271 data-nosnippet>271</a> /// ```
|
||||
<a href=#272 id=272 data-nosnippet>272</a> </span><span class="kw">fn </span>to_next_match_arm(<span class="kw-2">&</span><span class="self">self</span>) -> TokenStream2 {
|
||||
<a href=#273 id=273 data-nosnippet>273</a> <span class="kw">let </span>name = <span class="kw-2">&</span><span class="self">self</span>.name;
|
||||
<a href=#274 id=274 data-nosnippet>274</a> <span class="kw">let </span>fields = <span class="kw-2">&</span><span class="self">self</span>.fields;
|
||||
<a href=#275 id=275 data-nosnippet>275</a> <span class="kw">let </span>cfg = <span class="kw">if let </span><span class="prelude-val">Some</span>(cfg) = <span class="kw-2">&</span><span class="self">self</span>.cfg_attr {
|
||||
<a href=#276 id=276 data-nosnippet>276</a> <span class="macro">quote_spanned!</span>(cfg.span() => #cfg)
|
||||
<a href=#277 id=277 data-nosnippet>277</a> } <span class="kw">else </span>{
|
||||
<a href=#278 id=278 data-nosnippet>278</a> <span class="macro">quote!</span> {}
|
||||
<a href=#279 id=279 data-nosnippet>279</a> };
|
||||
<a href=#280 id=280 data-nosnippet>280</a>
|
||||
<a href=#281 id=281 data-nosnippet>281</a> <span class="kw">if </span><span class="self">self</span>.has_source {
|
||||
<a href=#282 id=282 data-nosnippet>282</a> <span class="macro">quote_spanned!</span> {
|
||||
<a href=#283 id=283 data-nosnippet>283</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#284 id=284 data-nosnippet>284</a> <span class="prelude-val">Some</span>(source)
|
||||
<a href=#285 id=285 data-nosnippet>285</a> },
|
||||
<a href=#286 id=286 data-nosnippet>286</a> }
|
||||
<a href=#287 id=287 data-nosnippet>287</a> } <span class="kw">else </span>{
|
||||
<a href=#288 id=288 data-nosnippet>288</a> <span class="macro">quote_spanned!</span> {
|
||||
<a href=#289 id=289 data-nosnippet>289</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } =>{
|
||||
<a href=#290 id=290 data-nosnippet>290</a> <span class="prelude-val">None
|
||||
<a href=#291 id=291 data-nosnippet>291</a> </span>}
|
||||
<a href=#292 id=292 data-nosnippet>292</a> }
|
||||
<a href=#293 id=293 data-nosnippet>293</a> }
|
||||
<a href=#294 id=294 data-nosnippet>294</a> }
|
||||
<a href=#295 id=295 data-nosnippet>295</a>
|
||||
<a href=#296 id=296 data-nosnippet>296</a> <span class="kw">fn </span>build_transparent_match_arm(<span class="kw-2">&</span><span class="self">self</span>) -> TokenStream2 {
|
||||
<a href=#297 id=297 data-nosnippet>297</a> <span class="kw">let </span>cfg = <span class="kw">if let </span><span class="prelude-val">Some</span>(cfg) = <span class="kw-2">&</span><span class="self">self</span>.cfg_attr {
|
||||
<a href=#298 id=298 data-nosnippet>298</a> <span class="macro">quote_spanned!</span>(cfg.span() => #cfg)
|
||||
<a href=#299 id=299 data-nosnippet>299</a> } <span class="kw">else </span>{
|
||||
<a href=#300 id=300 data-nosnippet>300</a> <span class="macro">quote!</span> {}
|
||||
<a href=#301 id=301 data-nosnippet>301</a> };
|
||||
<a href=#302 id=302 data-nosnippet>302</a> <span class="kw">let </span>name = <span class="kw-2">&</span><span class="self">self</span>.name;
|
||||
<a href=#303 id=303 data-nosnippet>303</a> <span class="kw">let </span>fields = <span class="kw-2">&</span><span class="self">self</span>.fields;
|
||||
<a href=#304 id=304 data-nosnippet>304</a>
|
||||
<a href=#305 id=305 data-nosnippet>305</a> <span class="kw">if </span><span class="self">self</span>.transparent {
|
||||
<a href=#306 id=306 data-nosnippet>306</a> <span class="macro">quote_spanned!</span> {
|
||||
<a href=#307 id=307 data-nosnippet>307</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } => {
|
||||
<a href=#308 id=308 data-nosnippet>308</a> <span class="bool-val">true
|
||||
<a href=#309 id=309 data-nosnippet>309</a> </span>},
|
||||
<a href=#310 id=310 data-nosnippet>310</a> }
|
||||
<a href=#311 id=311 data-nosnippet>311</a> } <span class="kw">else </span>{
|
||||
<a href=#312 id=312 data-nosnippet>312</a> <span class="macro">quote_spanned!</span> {
|
||||
<a href=#313 id=313 data-nosnippet>313</a> <span class="self">self</span>.span => #cfg <span class="attr">#[allow(unused_variables)] </span>#name { #(#fields),* } =>{
|
||||
<a href=#314 id=314 data-nosnippet>314</a> <span class="bool-val">false
|
||||
<a href=#315 id=315 data-nosnippet>315</a> </span>}
|
||||
<a href=#316 id=316 data-nosnippet>316</a> }
|
||||
<a href=#317 id=317 data-nosnippet>317</a> }
|
||||
<a href=#318 id=318 data-nosnippet>318</a> }
|
||||
<a href=#319 id=319 data-nosnippet>319</a>}
|
||||
<a href=#320 id=320 data-nosnippet>320</a>
|
||||
<a href=#321 id=321 data-nosnippet>321</a><span class="kw">fn </span>build_transparent_fn(enum_name: Ident, variants: <span class="kw-2">&</span>[ErrorVariant]) -> TokenStream2 {
|
||||
<a href=#322 id=322 data-nosnippet>322</a> <span class="kw">let </span>match_arms = variants
|
||||
<a href=#323 id=323 data-nosnippet>323</a> .iter()
|
||||
<a href=#324 id=324 data-nosnippet>324</a> .map(|v| v.build_transparent_match_arm())
|
||||
<a href=#325 id=325 data-nosnippet>325</a> .collect::<Vec<<span class="kw">_</span>>>();
|
||||
<a href=#326 id=326 data-nosnippet>326</a>
|
||||
<a href=#327 id=327 data-nosnippet>327</a> <span class="macro">quote!</span> {
|
||||
<a href=#328 id=328 data-nosnippet>328</a> <span class="kw">fn </span>transparent(<span class="kw-2">&</span><span class="self">self</span>) -> bool {
|
||||
<a href=#329 id=329 data-nosnippet>329</a> <span class="kw">use </span>#enum_name::<span class="kw-2">*</span>;
|
||||
<a href=#330 id=330 data-nosnippet>330</a> <span class="kw">match </span><span class="self">self </span>{
|
||||
<a href=#331 id=331 data-nosnippet>331</a> #(#match_arms)*
|
||||
<a href=#332 id=332 data-nosnippet>332</a> }
|
||||
<a href=#333 id=333 data-nosnippet>333</a> }
|
||||
<a href=#334 id=334 data-nosnippet>334</a> }
|
||||
<a href=#335 id=335 data-nosnippet>335</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
32
src/common_macro/utils.rs.html
Normal file
32
src/common_macro/utils.rs.html
Normal file
@@ -0,0 +1,32 @@
|
||||
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/common/macro/src/utils.rs`."><title>utils.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../static.files/rustdoc-17e0aaed.css"><meta name="rustdoc-vars" data-root-path="../../" data-static-root-path="../../static.files/" data-current-crate="common_macro" data-themes="" data-resource-suffix="" data-rustdoc-version="1.96.0-nightly (ac7f9ec7d 2026-03-20)" data-channel="nightly" data-search-js="search-63369b7b.js" data-stringdex-js="stringdex-2da4960a.js" data-settings-js="settings-170eb4bf.js" ><script src="../../static.files/storage-41dd4d93.js"></script><script defer src="../../static.files/src-script-813739b1.js"></script><script defer src="../../src-files.js"></script><script defer src="../../static.files/main-5013f961.js"></script><noscript><link rel="stylesheet" href="../../static.files/noscript-f7c3ffd8.css"></noscript><link rel="alternate icon" type="image/png" href="../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><a class="skip-main-content" href="#main-content">Skip to main content</a><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content" tabindex="-1"><div class="main-heading"><h1><div class="sub-heading">common_macro/</div>utils.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Copyright 2023 Greptime Team
|
||||
<a href=#2 id=2 data-nosnippet>2</a>//
|
||||
<a href=#3 id=3 data-nosnippet>3</a>// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
<a href=#4 id=4 data-nosnippet>4</a>// you may not use this file except in compliance with the License.
|
||||
<a href=#5 id=5 data-nosnippet>5</a>// You may obtain a copy of the License at
|
||||
<a href=#6 id=6 data-nosnippet>6</a>//
|
||||
<a href=#7 id=7 data-nosnippet>7</a>// http://www.apache.org/licenses/LICENSE-2.0
|
||||
<a href=#8 id=8 data-nosnippet>8</a>//
|
||||
<a href=#9 id=9 data-nosnippet>9</a>// Unless required by applicable law or agreed to in writing, software
|
||||
<a href=#10 id=10 data-nosnippet>10</a>// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
<a href=#11 id=11 data-nosnippet>11</a>// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
<a href=#12 id=12 data-nosnippet>12</a>// See the License for the specific language governing permissions and
|
||||
<a href=#13 id=13 data-nosnippet>13</a>// limitations under the License.
|
||||
<a href=#14 id=14 data-nosnippet>14</a>
|
||||
<a href=#15 id=15 data-nosnippet>15</a></span><span class="kw">use </span>syn::punctuated::Punctuated;
|
||||
<a href=#16 id=16 data-nosnippet>16</a><span class="kw">use </span>syn::spanned::Spanned;
|
||||
<a href=#17 id=17 data-nosnippet>17</a><span class="kw">use </span>syn::token::Comma;
|
||||
<a href=#18 id=18 data-nosnippet>18</a><span class="kw">use </span>syn::{FnArg, Type};
|
||||
<a href=#19 id=19 data-nosnippet>19</a>
|
||||
<a href=#20 id=20 data-nosnippet>20</a><span class="doccomment">/// Extract the argument list from the annotated function.
|
||||
<a href=#21 id=21 data-nosnippet>21</a></span><span class="kw">pub</span>(<span class="kw">crate</span>) <span class="kw">fn </span>extract_input_types(
|
||||
<a href=#22 id=22 data-nosnippet>22</a> inputs: <span class="kw-2">&</span>Punctuated<FnArg, Comma>,
|
||||
<a href=#23 id=23 data-nosnippet>23</a>) -> <span class="prelude-ty">Result</span><Vec<Type>, syn::Error> {
|
||||
<a href=#24 id=24 data-nosnippet>24</a> inputs
|
||||
<a href=#25 id=25 data-nosnippet>25</a> .iter()
|
||||
<a href=#26 id=26 data-nosnippet>26</a> .map(|arg| <span class="kw">match </span>arg {
|
||||
<a href=#27 id=27 data-nosnippet>27</a> FnArg::Receiver(receiver) => <span class="prelude-val">Err</span>(syn::Error::new(receiver.span(), <span class="string">"expected bool"</span>)),
|
||||
<a href=#28 id=28 data-nosnippet>28</a> FnArg::Typed(pat_type) => <span class="prelude-val">Ok</span>(<span class="kw-2">*</span>pat_type.ty.clone()),
|
||||
<a href=#29 id=29 data-nosnippet>29</a> })
|
||||
<a href=#30 id=30 data-nosnippet>30</a> .collect()
|
||||
<a href=#31 id=31 data-nosnippet>31</a>}
|
||||
</code></pre></div></section></main></body></html>
|
||||
Reference in New Issue
Block a user