This is an automated email from the ASF dual-hosted git repository. github-bot pushed a commit to branch gh-pages in repository https://gitbox.apache.org/repos/asf/iceberg-rust.git
The following commit(s) were added to refs/heads/gh-pages by this push: new 48687d7e deploy: 40b055a779365b121ad070b900389d27f8122741 48687d7e is described below commit 48687d7ed5c54c87a43131a14619616510268f9b Author: liurenjie1024 <liurenjie1...@users.noreply.github.com> AuthorDate: Thu May 29 09:28:37 2025 +0000 deploy: 40b055a779365b121ad070b900389d27f8122741 --- api/src/iceberg/transaction/append.rs.html | 682 ++++++++++++++------------- api/src/iceberg/transaction/snapshot.rs.html | 501 ++++++++++---------- 2 files changed, 621 insertions(+), 562 deletions(-) diff --git a/api/src/iceberg/transaction/append.rs.html b/api/src/iceberg/transaction/append.rs.html index f04e35bc..73a49757 100644 --- a/api/src/iceberg/transaction/append.rs.html +++ b/api/src/iceberg/transaction/append.rs.html @@ -72,323 +72,373 @@ <a href=#72 id=72 data-nosnippet>72</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>) <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> <span class="doccomment">/// Adds existing parquet files -<a href=#76 id=76 data-nosnippet>76</a> /// -<a href=#77 id=77 data-nosnippet>77</a> /// Note: This API is not yet fully supported in version 0.5.x. -<a href=#78 id=78 data-nosnippet>78</a> /// It is currently incomplete and should not be used in production. -<a href=#79 id=79 data-nosnippet>79</a> /// Specifically, schema compatibility checks and support for adding to partitioned tables -<a href=#80 id=80 data-nosnippet>80</a> /// have not yet been implemented. -<a href=#81 id=81 data-nosnippet>81</a> </span><span class="attr">#[allow(dead_code)] -<a href=#82 id=82 data-nosnippet>82</a> </span><span class="kw">async fn </span>add_parquet_files(<span class="kw-2">mut </span><span class="self">self</span>, file_path: Vec<String>) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { -<a href=#83 id=83 data-nosnippet>83</a> <span class="kw">if </span>!<span class="self">self -<a href=#84 id=84 data-nosnippet>84</a> </span>.snapshot_produce_action -<a href=#85 id=85 data-nosnippet>85</a> .tx -<a href=#86 id=86 data-nosnippet>86</a> .current_table -<a href=#87 id=87 data-nosnippet>87</a> .metadata() -<a href=#88 id=88 data-nosnippet>88</a> .default_spec -<a href=#89 id=89 data-nosnippet>89</a> .is_unpartitioned() -<a href=#90 id=90 data-nosnippet>90</a> { -<a href=#91 id=91 data-nosnippet>91</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( -<a href=#92 id=92 data-nosnippet>92</a> ErrorKind::FeatureUnsupported, -<a href=#93 id=93 data-nosnippet>93</a> <span class="string">"Appending to partitioned tables is not supported"</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> -<a href=#97 id=97 data-nosnippet>97</a> <span class="kw">let </span>table_metadata = <span class="self">self</span>.snapshot_produce_action.tx.current_table.metadata(); -<a href=#98 id=98 data-nosnippet>98</a> -<a href=#99 id=99 data-nosnippet>99</a> <span class="kw">let </span>data_files = ParquetWriter::parquet_files_to_data_files( -<a href=#100 id=100 data-nosnippet>100</a> <span class="self">self</span>.snapshot_produce_action.tx.current_table.file_io(), -<a href=#101 id=101 data-nosnippet>101</a> file_path, -<a href=#102 id=102 data-nosnippet>102</a> table_metadata, -<a href=#103 id=103 data-nosnippet>103</a> ) -<a href=#104 id=104 data-nosnippet>104</a> .<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#105 id=105 data-nosnippet>105</a> -<a href=#106 id=106 data-nosnippet>106</a> <span class="self">self</span>.add_data_files(data_files)<span class="question-mark">?</span>; -<a href=#107 id=107 data-nosnippet>107</a> -<a href=#108 id=108 data-nosnippet>108</a> <span class="self">self</span>.apply().<span class="kw">await -<a href=#109 id=109 data-nosnippet>109</a> </span>} -<a href=#110 id=110 data-nosnippet>110</a> -<a href=#111 id=111 data-nosnippet>111</a> <span class="doccomment">/// Finished building the action and apply it to the transaction. -<a href=#112 id=112 data-nosnippet>112</a> </span><span class="kw">pub async fn </span>apply(<span class="self">self</span>) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { -<a href=#113 id=113 data-nosnippet>113</a> <span class="comment">// Checks duplicate files -<a href=#114 id=114 data-nosnippet>114</a> </span><span class="kw">if </span><span class="self">self</span>.check_duplicate { -<a href=#115 id=115 data-nosnippet>115</a> <span class="kw">let </span>new_files: HashSet<<span class="kw-2">&</span>str> = <span class="self">self -<a href=#116 id=116 data-nosnippet>116</a> </span>.snapshot_produce_action -<a href=#117 id=117 data-nosnippet>117</a> .added_data_files -<a href=#118 id=118 data-nosnippet>118</a> .iter() -<a href=#119 id=119 data-nosnippet>119</a> .map(|df| df.file_path.as_str()) -<a href=#120 id=120 data-nosnippet>120</a> .collect(); -<a href=#121 id=121 data-nosnippet>121</a> -<a href=#122 id=122 data-nosnippet>122</a> <span class="kw">let </span><span class="kw-2">mut </span>manifest_stream = <span class="self">self -<a href=#123 id=123 data-nosnippet>123</a> </span>.snapshot_produce_action -<a href=#124 id=124 data-nosnippet>124</a> .tx -<a href=#125 id=125 data-nosnippet>125</a> .current_table -<a href=#126 id=126 data-nosnippet>126</a> .inspect() -<a href=#127 id=127 data-nosnippet>127</a> .manifests() -<a href=#128 id=128 data-nosnippet>128</a> .scan() -<a href=#129 id=129 data-nosnippet>129</a> .<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#130 id=130 data-nosnippet>130</a> <span class="kw">let </span><span class="kw-2">mut </span>referenced_files = Vec::new(); +<a href=#75 id=75 data-nosnippet>75</a> <span class="doccomment">/// Set snapshot summary properties. +<a href=#76 id=76 data-nosnippet>76</a> </span><span class="kw">pub fn </span>set_snapshot_properties( +<a href=#77 id=77 data-nosnippet>77</a> <span class="kw-2">&mut </span><span class="self">self</span>, +<a href=#78 id=78 data-nosnippet>78</a> snapshot_properties: HashMap<String, String>, +<a href=#79 id=79 data-nosnippet>79</a> ) -> <span class="prelude-ty">Result</span><<span class="kw-2">&mut </span><span class="self">Self</span>> { +<a href=#80 id=80 data-nosnippet>80</a> <span class="self">self</span>.snapshot_produce_action +<a href=#81 id=81 data-nosnippet>81</a> .set_snapshot_properties(snapshot_properties)<span class="question-mark">?</span>; +<a href=#82 id=82 data-nosnippet>82</a> <span class="prelude-val">Ok</span>(<span class="self">self</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="doccomment">/// Adds existing parquet files +<a href=#86 id=86 data-nosnippet>86</a> /// +<a href=#87 id=87 data-nosnippet>87</a> /// Note: This API is not yet fully supported in version 0.5.x. +<a href=#88 id=88 data-nosnippet>88</a> /// It is currently incomplete and should not be used in production. +<a href=#89 id=89 data-nosnippet>89</a> /// Specifically, schema compatibility checks and support for adding to partitioned tables +<a href=#90 id=90 data-nosnippet>90</a> /// have not yet been implemented. +<a href=#91 id=91 data-nosnippet>91</a> </span><span class="attr">#[allow(dead_code)] +<a href=#92 id=92 data-nosnippet>92</a> </span><span class="kw">async fn </span>add_parquet_files(<span class="kw-2">mut </span><span class="self">self</span>, file_path: Vec<String>) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { +<a href=#93 id=93 data-nosnippet>93</a> <span class="kw">if </span>!<span class="self">self +<a href=#94 id=94 data-nosnippet>94</a> </span>.snapshot_produce_action +<a href=#95 id=95 data-nosnippet>95</a> .tx +<a href=#96 id=96 data-nosnippet>96</a> .current_table +<a href=#97 id=97 data-nosnippet>97</a> .metadata() +<a href=#98 id=98 data-nosnippet>98</a> .default_spec +<a href=#99 id=99 data-nosnippet>99</a> .is_unpartitioned() +<a href=#100 id=100 data-nosnippet>100</a> { +<a href=#101 id=101 data-nosnippet>101</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( +<a href=#102 id=102 data-nosnippet>102</a> ErrorKind::FeatureUnsupported, +<a href=#103 id=103 data-nosnippet>103</a> <span class="string">"Appending to partitioned tables is not supported"</span>, +<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> +<a href=#107 id=107 data-nosnippet>107</a> <span class="kw">let </span>table_metadata = <span class="self">self</span>.snapshot_produce_action.tx.current_table.metadata(); +<a href=#108 id=108 data-nosnippet>108</a> +<a href=#109 id=109 data-nosnippet>109</a> <span class="kw">let </span>data_files = ParquetWriter::parquet_files_to_data_files( +<a href=#110 id=110 data-nosnippet>110</a> <span class="self">self</span>.snapshot_produce_action.tx.current_table.file_io(), +<a href=#111 id=111 data-nosnippet>111</a> file_path, +<a href=#112 id=112 data-nosnippet>112</a> table_metadata, +<a href=#113 id=113 data-nosnippet>113</a> ) +<a href=#114 id=114 data-nosnippet>114</a> .<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#115 id=115 data-nosnippet>115</a> +<a href=#116 id=116 data-nosnippet>116</a> <span class="self">self</span>.add_data_files(data_files)<span class="question-mark">?</span>; +<a href=#117 id=117 data-nosnippet>117</a> +<a href=#118 id=118 data-nosnippet>118</a> <span class="self">self</span>.apply().<span class="kw">await +<a href=#119 id=119 data-nosnippet>119</a> </span>} +<a href=#120 id=120 data-nosnippet>120</a> +<a href=#121 id=121 data-nosnippet>121</a> <span class="doccomment">/// Finished building the action and apply it to the transaction. +<a href=#122 id=122 data-nosnippet>122</a> </span><span class="kw">pub async fn </span>apply(<span class="self">self</span>) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { +<a href=#123 id=123 data-nosnippet>123</a> <span class="comment">// Checks duplicate files +<a href=#124 id=124 data-nosnippet>124</a> </span><span class="kw">if </span><span class="self">self</span>.check_duplicate { +<a href=#125 id=125 data-nosnippet>125</a> <span class="kw">let </span>new_files: HashSet<<span class="kw-2">&</span>str> = <span class="self">self +<a href=#126 id=126 data-nosnippet>126</a> </span>.snapshot_produce_action +<a href=#127 id=127 data-nosnippet>127</a> .added_data_files +<a href=#128 id=128 data-nosnippet>128</a> .iter() +<a href=#129 id=129 data-nosnippet>129</a> .map(|df| df.file_path.as_str()) +<a href=#130 id=130 data-nosnippet>130</a> .collect(); <a href=#131 id=131 data-nosnippet>131</a> -<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">while let </span><span class="prelude-val">Some</span>(batch) = manifest_stream.try_next().<span class="kw">await</span><span class="question-mark">? </span>{ -<a href=#133 id=133 data-nosnippet>133</a> <span class="kw">let </span>file_path_array = batch -<a href=#134 id=134 data-nosnippet>134</a> .column(<span class="number">1</span>) -<a href=#135 id=135 data-nosnippet>135</a> .as_any() -<a href=#136 id=136 data-nosnippet>136</a> .downcast_ref::<StringArray>() -<a href=#137 id=137 data-nosnippet>137</a> .ok_or_else(|| { -<a href=#138 id=138 data-nosnippet>138</a> Error::new( -<a href=#139 id=139 data-nosnippet>139</a> ErrorKind::DataInvalid, -<a href=#140 id=140 data-nosnippet>140</a> <span class="string">"Failed to downcast file_path column to StringArray"</span>, -<a href=#141 id=141 data-nosnippet>141</a> ) -<a href=#142 id=142 data-nosnippet>142</a> })<span class="question-mark">?</span>; -<a href=#143 id=143 data-nosnippet>143</a> -<a href=#144 id=144 data-nosnippet>144</a> <span class="kw">for </span>i <span class="kw">in </span><span class="number">0</span>..batch.num_rows() { -<a href=#145 id=145 data-nosnippet>145</a> <span class="kw">let </span>file_path = file_path_array.value(i); -<a href=#146 id=146 data-nosnippet>146</a> <span class="kw">if </span>new_files.contains(file_path) { -<a href=#147 id=147 data-nosnippet>147</a> referenced_files.push(file_path.to_string()); -<a href=#148 id=148 data-nosnippet>148</a> } -<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> -<a href=#152 id=152 data-nosnippet>152</a> <span class="kw">if </span>!referenced_files.is_empty() { -<a href=#153 id=153 data-nosnippet>153</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( -<a href=#154 id=154 data-nosnippet>154</a> ErrorKind::DataInvalid, -<a href=#155 id=155 data-nosnippet>155</a> <span class="macro">format!</span>( -<a href=#156 id=156 data-nosnippet>156</a> <span class="string">"Cannot add files that are already referenced by table, files: {}"</span>, -<a href=#157 id=157 data-nosnippet>157</a> referenced_files.join(<span class="string">", "</span>) -<a href=#158 id=158 data-nosnippet>158</a> ), -<a href=#159 id=159 data-nosnippet>159</a> )); +<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">let </span><span class="kw-2">mut </span>manifest_stream = <span class="self">self +<a href=#133 id=133 data-nosnippet>133</a> </span>.snapshot_produce_action +<a href=#134 id=134 data-nosnippet>134</a> .tx +<a href=#135 id=135 data-nosnippet>135</a> .current_table +<a href=#136 id=136 data-nosnippet>136</a> .inspect() +<a href=#137 id=137 data-nosnippet>137</a> .manifests() +<a href=#138 id=138 data-nosnippet>138</a> .scan() +<a href=#139 id=139 data-nosnippet>139</a> .<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#140 id=140 data-nosnippet>140</a> <span class="kw">let </span><span class="kw-2">mut </span>referenced_files = Vec::new(); +<a href=#141 id=141 data-nosnippet>141</a> +<a href=#142 id=142 data-nosnippet>142</a> <span class="kw">while let </span><span class="prelude-val">Some</span>(batch) = manifest_stream.try_next().<span class="kw">await</span><span class="question-mark">? </span>{ +<a href=#143 id=143 data-nosnippet>143</a> <span class="kw">let </span>file_path_array = batch +<a href=#144 id=144 data-nosnippet>144</a> .column(<span class="number">1</span>) +<a href=#145 id=145 data-nosnippet>145</a> .as_any() +<a href=#146 id=146 data-nosnippet>146</a> .downcast_ref::<StringArray>() +<a href=#147 id=147 data-nosnippet>147</a> .ok_or_else(|| { +<a href=#148 id=148 data-nosnippet>148</a> Error::new( +<a href=#149 id=149 data-nosnippet>149</a> ErrorKind::DataInvalid, +<a href=#150 id=150 data-nosnippet>150</a> <span class="string">"Failed to downcast file_path column to StringArray"</span>, +<a href=#151 id=151 data-nosnippet>151</a> ) +<a href=#152 id=152 data-nosnippet>152</a> })<span class="question-mark">?</span>; +<a href=#153 id=153 data-nosnippet>153</a> +<a href=#154 id=154 data-nosnippet>154</a> <span class="kw">for </span>i <span class="kw">in </span><span class="number">0</span>..batch.num_rows() { +<a href=#155 id=155 data-nosnippet>155</a> <span class="kw">let </span>file_path = file_path_array.value(i); +<a href=#156 id=156 data-nosnippet>156</a> <span class="kw">if </span>new_files.contains(file_path) { +<a href=#157 id=157 data-nosnippet>157</a> referenced_files.push(file_path.to_string()); +<a href=#158 id=158 data-nosnippet>158</a> } +<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> -<a href=#163 id=163 data-nosnippet>163</a> <span class="self">self</span>.snapshot_produce_action -<a href=#164 id=164 data-nosnippet>164</a> .apply(FastAppendOperation, DefaultManifestProcess) -<a href=#165 id=165 data-nosnippet>165</a> .<span class="kw">await -<a href=#166 id=166 data-nosnippet>166</a> </span>} -<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><span class="kw">struct </span>FastAppendOperation; -<a href=#170 id=170 data-nosnippet>170</a> -<a href=#171 id=171 data-nosnippet>171</a><span class="kw">impl </span>SnapshotProduceOperation <span class="kw">for </span>FastAppendOperation { -<a href=#172 id=172 data-nosnippet>172</a> <span class="kw">fn </span>operation(<span class="kw-2">&</span><span class="self">self</span>) -> Operation { -<a href=#173 id=173 data-nosnippet>173</a> Operation::Append -<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="kw">async fn </span>delete_entries( -<a href=#177 id=177 data-nosnippet>177</a> <span class="kw-2">&</span><span class="self">self</span>, -<a href=#178 id=178 data-nosnippet>178</a> _snapshot_produce: <span class="kw-2">&</span>SnapshotProduceAction<<span class="lifetime">'_</span>>, -<a href=#179 id=179 data-nosnippet>179</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestEntry>> { -<a href=#180 id=180 data-nosnippet>180</a> <span class="prelude-val">Ok</span>(<span class="macro">vec!</span>[]) -<a href=#181 id=181 data-nosnippet>181</a> } -<a href=#182 id=182 data-nosnippet>182</a> -<a href=#183 id=183 data-nosnippet>183</a> <span class="kw">async fn </span>existing_manifest( -<a href=#184 id=184 data-nosnippet>184</a> <span class="kw-2">&</span><span class="self">self</span>, -<a href=#185 id=185 data-nosnippet>185</a> snapshot_produce: <span class="kw-2">&</span>SnapshotProduceAction<<span class="lifetime">'_</span>>, -<a href=#186 id=186 data-nosnippet>186</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestFile>> { -<a href=#187 id=187 data-nosnippet>187</a> <span class="kw">let </span><span class="prelude-val">Some</span>(snapshot) = snapshot_produce -<a href=#188 id=188 data-nosnippet>188</a> .tx -<a href=#189 id=189 data-nosnippet>189</a> .current_table -<a href=#190 id=190 data-nosnippet>190</a> .metadata() -<a href=#191 id=191 data-nosnippet>191</a> .current_snapshot() -<a href=#192 id=192 data-nosnippet>192</a> <span class="kw">else </span>{ -<a href=#193 id=193 data-nosnippet>193</a> <span class="kw">return </span><span class="prelude-val">Ok</span>(<span class="macro">vec!</span>[]); -<a href=#194 id=194 data-nosnippet>194</a> }; -<a href=#195 id=195 data-nosnippet>195</a> -<a href=#196 id=196 data-nosnippet>196</a> <span class="kw">let </span>manifest_list = snapshot -<a href=#197 id=197 data-nosnippet>197</a> .load_manifest_list( -<a href=#198 id=198 data-nosnippet>198</a> snapshot_produce.tx.current_table.file_io(), -<a href=#199 id=199 data-nosnippet>199</a> <span class="kw-2">&</span>snapshot_produce.tx.current_table.metadata_ref(), -<a href=#200 id=200 data-nosnippet>200</a> ) -<a href=#201 id=201 data-nosnippet>201</a> .<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#202 id=202 data-nosnippet>202</a> -<a href=#203 id=203 data-nosnippet>203</a> <span class="prelude-val">Ok</span>(manifest_list -<a href=#204 id=204 data-nosnippet>204</a> .entries() -<a href=#205 id=205 data-nosnippet>205</a> .iter() -<a href=#206 id=206 data-nosnippet>206</a> .filter(|entry| entry.has_added_files() || entry.has_existing_files()) -<a href=#207 id=207 data-nosnippet>207</a> .cloned() -<a href=#208 id=208 data-nosnippet>208</a> .collect()) -<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><span class="attr">#[cfg(test)] -<a href=#213 id=213 data-nosnippet>213</a></span><span class="kw">mod </span>tests { -<a href=#214 id=214 data-nosnippet>214</a> <span class="kw">use </span><span class="kw">crate</span>::scan::tests::TableTestFixture; -<a href=#215 id=215 data-nosnippet>215</a> <span class="kw">use </span><span class="kw">crate</span>::spec::{ -<a href=#216 id=216 data-nosnippet>216</a> DataContentType, DataFileBuilder, DataFileFormat, Literal, MAIN_BRANCH, Struct, -<a href=#217 id=217 data-nosnippet>217</a> }; -<a href=#218 id=218 data-nosnippet>218</a> <span class="kw">use </span><span class="kw">crate</span>::transaction::Transaction; -<a href=#219 id=219 data-nosnippet>219</a> <span class="kw">use </span><span class="kw">crate</span>::transaction::tests::make_v2_minimal_table; -<a href=#220 id=220 data-nosnippet>220</a> <span class="kw">use crate</span>::{TableRequirement, TableUpdate}; +<a href=#161 id=161 data-nosnippet>161</a> +<a href=#162 id=162 data-nosnippet>162</a> <span class="kw">if </span>!referenced_files.is_empty() { +<a href=#163 id=163 data-nosnippet>163</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( +<a href=#164 id=164 data-nosnippet>164</a> ErrorKind::DataInvalid, +<a href=#165 id=165 data-nosnippet>165</a> <span class="macro">format!</span>( +<a href=#166 id=166 data-nosnippet>166</a> <span class="string">"Cannot add files that are already referenced by table, files: {}"</span>, +<a href=#167 id=167 data-nosnippet>167</a> referenced_files.join(<span class="string">", "</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> } +<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="self">self</span>.snapshot_produce_action +<a href=#174 id=174 data-nosnippet>174</a> .apply(FastAppendOperation, DefaultManifestProcess) +<a href=#175 id=175 data-nosnippet>175</a> .<span class="kw">await +<a href=#176 id=176 data-nosnippet>176</a> </span>} +<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><span class="kw">struct </span>FastAppendOperation; +<a href=#180 id=180 data-nosnippet>180</a> +<a href=#181 id=181 data-nosnippet>181</a><span class="kw">impl </span>SnapshotProduceOperation <span class="kw">for </span>FastAppendOperation { +<a href=#182 id=182 data-nosnippet>182</a> <span class="kw">fn </span>operation(<span class="kw-2">&</span><span class="self">self</span>) -> Operation { +<a href=#183 id=183 data-nosnippet>183</a> Operation::Append +<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> <span class="kw">async fn </span>delete_entries( +<a href=#187 id=187 data-nosnippet>187</a> <span class="kw-2">&</span><span class="self">self</span>, +<a href=#188 id=188 data-nosnippet>188</a> _snapshot_produce: <span class="kw-2">&</span>SnapshotProduceAction<<span class="lifetime">'_</span>>, +<a href=#189 id=189 data-nosnippet>189</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestEntry>> { +<a href=#190 id=190 data-nosnippet>190</a> <span class="prelude-val">Ok</span>(<span class="macro">vec!</span>[]) +<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">async fn </span>existing_manifest( +<a href=#194 id=194 data-nosnippet>194</a> <span class="kw-2">&</span><span class="self">self</span>, +<a href=#195 id=195 data-nosnippet>195</a> snapshot_produce: <span class="kw-2">&</span>SnapshotProduceAction<<span class="lifetime">'_</span>>, +<a href=#196 id=196 data-nosnippet>196</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestFile>> { +<a href=#197 id=197 data-nosnippet>197</a> <span class="kw">let </span><span class="prelude-val">Some</span>(snapshot) = snapshot_produce +<a href=#198 id=198 data-nosnippet>198</a> .tx +<a href=#199 id=199 data-nosnippet>199</a> .current_table +<a href=#200 id=200 data-nosnippet>200</a> .metadata() +<a href=#201 id=201 data-nosnippet>201</a> .current_snapshot() +<a href=#202 id=202 data-nosnippet>202</a> <span class="kw">else </span>{ +<a href=#203 id=203 data-nosnippet>203</a> <span class="kw">return </span><span class="prelude-val">Ok</span>(<span class="macro">vec!</span>[]); +<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>manifest_list = snapshot +<a href=#207 id=207 data-nosnippet>207</a> .load_manifest_list( +<a href=#208 id=208 data-nosnippet>208</a> snapshot_produce.tx.current_table.file_io(), +<a href=#209 id=209 data-nosnippet>209</a> <span class="kw-2">&</span>snapshot_produce.tx.current_table.metadata_ref(), +<a href=#210 id=210 data-nosnippet>210</a> ) +<a href=#211 id=211 data-nosnippet>211</a> .<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#212 id=212 data-nosnippet>212</a> +<a href=#213 id=213 data-nosnippet>213</a> <span class="prelude-val">Ok</span>(manifest_list +<a href=#214 id=214 data-nosnippet>214</a> .entries() +<a href=#215 id=215 data-nosnippet>215</a> .iter() +<a href=#216 id=216 data-nosnippet>216</a> .filter(|entry| entry.has_added_files() || entry.has_existing_files()) +<a href=#217 id=217 data-nosnippet>217</a> .cloned() +<a href=#218 id=218 data-nosnippet>218</a> .collect()) +<a href=#219 id=219 data-nosnippet>219</a> } +<a href=#220 id=220 data-nosnippet>220</a>} <a href=#221 id=221 data-nosnippet>221</a> -<a href=#222 id=222 data-nosnippet>222</a> <span class="attr">#[tokio::test] -<a href=#223 id=223 data-nosnippet>223</a> </span><span class="kw">async fn </span>test_empty_data_append_action() { -<a href=#224 id=224 data-nosnippet>224</a> <span class="kw">let </span>table = make_v2_minimal_table(); -<a href=#225 id=225 data-nosnippet>225</a> <span class="kw">let </span>tx = Transaction::new(<span class="kw-2">&</span>table); -<a href=#226 id=226 data-nosnippet>226</a> <span class="kw">let </span><span class="kw-2">mut </span>action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); -<a href=#227 id=227 data-nosnippet>227</a> action.add_data_files(<span class="macro">vec!</span>[]).unwrap(); -<a href=#228 id=228 data-nosnippet>228</a> <span class="macro">assert!</span>(action.apply().<span class="kw">await</span>.is_err()); -<a href=#229 id=229 data-nosnippet>229</a> } -<a href=#230 id=230 data-nosnippet>230</a> -<a href=#231 id=231 data-nosnippet>231</a> <span class="attr">#[tokio::test] -<a href=#232 id=232 data-nosnippet>232</a> </span><span class="kw">async fn </span>test_fast_append_action() { -<a href=#233 id=233 data-nosnippet>233</a> <span class="kw">let </span>table = make_v2_minimal_table(); -<a href=#234 id=234 data-nosnippet>234</a> <span class="kw">let </span>tx = Transaction::new(<span class="kw-2">&</span>table); -<a href=#235 id=235 data-nosnippet>235</a> <span class="kw">let </span><span class="kw-2">mut </span>action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); -<a href=#236 id=236 data-nosnippet>236</a> -<a href=#237 id=237 data-nosnippet>237</a> <span class="comment">// check add data file with incompatible partition value -<a href=#238 id=238 data-nosnippet>238</a> </span><span class="kw">let </span>data_file = DataFileBuilder::default() -<a href=#239 id=239 data-nosnippet>239</a> .content(DataContentType::Data) -<a href=#240 id=240 data-nosnippet>240</a> .file_path(<span class="string">"test/3.parquet"</span>.to_string()) -<a href=#241 id=241 data-nosnippet>241</a> .file_format(DataFileFormat::Parquet) -<a href=#242 id=242 data-nosnippet>242</a> .file_size_in_bytes(<span class="number">100</span>) -<a href=#243 id=243 data-nosnippet>243</a> .record_count(<span class="number">1</span>) -<a href=#244 id=244 data-nosnippet>244</a> .partition_spec_id(table.metadata().default_partition_spec_id()) -<a href=#245 id=245 data-nosnippet>245</a> .partition(Struct::from_iter([<span class="prelude-val">Some</span>(Literal::string(<span class="string">"test"</span>))])) -<a href=#246 id=246 data-nosnippet>246</a> .build() -<a href=#247 id=247 data-nosnippet>247</a> .unwrap(); -<a href=#248 id=248 data-nosnippet>248</a> <span class="macro">assert!</span>(action.add_data_files(<span class="macro">vec!</span>[data_file.clone()]).is_err()); -<a href=#249 id=249 data-nosnippet>249</a> -<a href=#250 id=250 data-nosnippet>250</a> <span class="kw">let </span>data_file = DataFileBuilder::default() -<a href=#251 id=251 data-nosnippet>251</a> .content(DataContentType::Data) -<a href=#252 id=252 data-nosnippet>252</a> .file_path(<span class="string">"test/3.parquet"</span>.to_string()) -<a href=#253 id=253 data-nosnippet>253</a> .file_format(DataFileFormat::Parquet) -<a href=#254 id=254 data-nosnippet>254</a> .file_size_in_bytes(<span class="number">100</span>) -<a href=#255 id=255 data-nosnippet>255</a> .record_count(<span class="number">1</span>) -<a href=#256 id=256 data-nosnippet>256</a> .partition_spec_id(table.metadata().default_partition_spec_id()) -<a href=#257 id=257 data-nosnippet>257</a> .partition(Struct::from_iter([<span class="prelude-val">Some</span>(Literal::long(<span class="number">300</span>))])) -<a href=#258 id=258 data-nosnippet>258</a> .build() -<a href=#259 id=259 data-nosnippet>259</a> .unwrap(); -<a href=#260 id=260 data-nosnippet>260</a> action.add_data_files(<span class="macro">vec!</span>[data_file.clone()]).unwrap(); -<a href=#261 id=261 data-nosnippet>261</a> <span class="kw">let </span>tx = action.apply().<span class="kw">await</span>.unwrap(); -<a href=#262 id=262 data-nosnippet>262</a> -<a href=#263 id=263 data-nosnippet>263</a> <span class="comment">// check updates and requirements -<a href=#264 id=264 data-nosnippet>264</a> </span><span class="macro">assert!</span>( -<a href=#265 id=265 data-nosnippet>265</a> <span class="macro">matches!</span>((<span class="kw-2">&</span>tx.updates[<span class="number">0</span>],<span class="kw-2">&</span>tx.updates[<span class="number">1</span>]), (TableUpdate::AddSnapshot { snapshot },TableUpdate::SetSnapshotRef { reference,ref_name }) <span class="kw">if </span>snapshot.snapshot_id() == reference.snapshot_id && ref_name == MAIN_BRANCH) -<a href=#266 id=266 data-nosnippet>266</a> ); -<a href=#267 id=267 data-nosnippet>267</a> <span class="macro">assert_eq!</span>( -<a href=#268 id=268 data-nosnippet>268</a> <span class="macro">vec!</span>[ -<a href=#269 id=269 data-nosnippet>269</a> TableRequirement::UuidMatch { -<a href=#270 id=270 data-nosnippet>270</a> uuid: table.metadata().uuid() -<a href=#271 id=271 data-nosnippet>271</a> }, -<a href=#272 id=272 data-nosnippet>272</a> TableRequirement::RefSnapshotIdMatch { -<a href=#273 id=273 data-nosnippet>273</a> r#ref: MAIN_BRANCH.to_string(), -<a href=#274 id=274 data-nosnippet>274</a> snapshot_id: table.metadata().current_snapshot_id -<a href=#275 id=275 data-nosnippet>275</a> } -<a href=#276 id=276 data-nosnippet>276</a> ], -<a href=#277 id=277 data-nosnippet>277</a> tx.requirements -<a href=#278 id=278 data-nosnippet>278</a> ); -<a href=#279 id=279 data-nosnippet>279</a> -<a href=#280 id=280 data-nosnippet>280</a> <span class="comment">// check manifest list -<a href=#281 id=281 data-nosnippet>281</a> </span><span class="kw">let </span>new_snapshot = <span class="kw">if let </span>TableUpdate::AddSnapshot { snapshot } = <span class="kw-2">&</span>tx.updates[<span class="number">0</span>] { -<a href=#282 id=282 data-nosnippet>282</a> snapshot -<a href=#283 id=283 data-nosnippet>283</a> } <span class="kw">else </span>{ -<a href=#284 id=284 data-nosnippet>284</a> <span class="macro">unreachable!</span>() -<a href=#285 id=285 data-nosnippet>285</a> }; -<a href=#286 id=286 data-nosnippet>286</a> <span class="kw">let </span>manifest_list = new_snapshot -<a href=#287 id=287 data-nosnippet>287</a> .load_manifest_list(table.file_io(), table.metadata()) -<a href=#288 id=288 data-nosnippet>288</a> .<span class="kw">await -<a href=#289 id=289 data-nosnippet>289</a> </span>.unwrap(); -<a href=#290 id=290 data-nosnippet>290</a> <span class="macro">assert_eq!</span>(<span class="number">1</span>, manifest_list.entries().len()); -<a href=#291 id=291 data-nosnippet>291</a> <span class="macro">assert_eq!</span>( -<a href=#292 id=292 data-nosnippet>292</a> manifest_list.entries()[<span class="number">0</span>].sequence_number, -<a href=#293 id=293 data-nosnippet>293</a> new_snapshot.sequence_number() -<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="comment">// check manifest -<a href=#297 id=297 data-nosnippet>297</a> </span><span class="kw">let </span>manifest = manifest_list.entries()[<span class="number">0</span>] -<a href=#298 id=298 data-nosnippet>298</a> .load_manifest(table.file_io()) -<a href=#299 id=299 data-nosnippet>299</a> .<span class="kw">await -<a href=#300 id=300 data-nosnippet>300</a> </span>.unwrap(); -<a href=#301 id=301 data-nosnippet>301</a> <span class="macro">assert_eq!</span>(<span class="number">1</span>, manifest.entries().len()); -<a href=#302 id=302 data-nosnippet>302</a> <span class="macro">assert_eq!</span>( -<a href=#303 id=303 data-nosnippet>303</a> new_snapshot.sequence_number(), -<a href=#304 id=304 data-nosnippet>304</a> manifest.entries()[<span class="number">0</span>] -<a href=#305 id=305 data-nosnippet>305</a> .sequence_number() -<a href=#306 id=306 data-nosnippet>306</a> .expect(<span class="string">"Inherit sequence number by load manifest"</span>) -<a href=#307 id=307 data-nosnippet>307</a> ); -<a href=#308 id=308 data-nosnippet>308</a> -<a href=#309 id=309 data-nosnippet>309</a> <span class="macro">assert_eq!</span>( -<a href=#310 id=310 data-nosnippet>310</a> new_snapshot.snapshot_id(), -<a href=#311 id=311 data-nosnippet>311</a> manifest.entries()[<span class="number">0</span>].snapshot_id().unwrap() -<a href=#312 id=312 data-nosnippet>312</a> ); -<a href=#313 id=313 data-nosnippet>313</a> <span class="macro">assert_eq!</span>(data_file, <span class="kw-2">*</span>manifest.entries()[<span class="number">0</span>].data_file()); -<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> <span class="attr">#[tokio::test] -<a href=#317 id=317 data-nosnippet>317</a> </span><span class="kw">async fn </span>test_add_existing_parquet_files_to_unpartitioned_table() { -<a href=#318 id=318 data-nosnippet>318</a> <span class="kw">let </span><span class="kw-2">mut </span>fixture = TableTestFixture::new_unpartitioned(); -<a href=#319 id=319 data-nosnippet>319</a> fixture.setup_unpartitioned_manifest_files().<span class="kw">await</span>; -<a href=#320 id=320 data-nosnippet>320</a> <span class="kw">let </span>tx = <span class="kw">crate</span>::transaction::Transaction::new(<span class="kw-2">&</span>fixture.table); -<a href=#321 id=321 data-nosnippet>321</a> -<a href=#322 id=322 data-nosnippet>322</a> <span class="kw">let </span>file_paths = <span class="macro">vec!</span>[ -<a href=#323 id=323 data-nosnippet>323</a> <span class="macro">format!</span>(<span class="string">"{}/1.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), -<a href=#324 id=324 data-nosnippet>324</a> <span class="macro">format!</span>(<span class="string">"{}/2.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), -<a href=#325 id=325 data-nosnippet>325</a> <span class="macro">format!</span>(<span class="string">"{}/3.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), -<a href=#326 id=326 data-nosnippet>326</a> ]; -<a href=#327 id=327 data-nosnippet>327</a> -<a href=#328 id=328 data-nosnippet>328</a> <span class="kw">let </span>fast_append_action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); +<a href=#222 id=222 data-nosnippet>222</a><span class="attr">#[cfg(test)] +<a href=#223 id=223 data-nosnippet>223</a></span><span class="kw">mod </span>tests { +<a href=#224 id=224 data-nosnippet>224</a> <span class="kw">use </span>std::collections::HashMap; +<a href=#225 id=225 data-nosnippet>225</a> +<a href=#226 id=226 data-nosnippet>226</a> <span class="kw">use </span><span class="kw">crate</span>::scan::tests::TableTestFixture; +<a href=#227 id=227 data-nosnippet>227</a> <span class="kw">use </span><span class="kw">crate</span>::spec::{ +<a href=#228 id=228 data-nosnippet>228</a> DataContentType, DataFileBuilder, DataFileFormat, Literal, MAIN_BRANCH, Struct, +<a href=#229 id=229 data-nosnippet>229</a> }; +<a href=#230 id=230 data-nosnippet>230</a> <span class="kw">use </span><span class="kw">crate</span>::transaction::Transaction; +<a href=#231 id=231 data-nosnippet>231</a> <span class="kw">use </span><span class="kw">crate</span>::transaction::tests::make_v2_minimal_table; +<a href=#232 id=232 data-nosnippet>232</a> <span class="kw">use crate</span>::{TableRequirement, TableUpdate}; +<a href=#233 id=233 data-nosnippet>233</a> +<a href=#234 id=234 data-nosnippet>234</a> <span class="attr">#[tokio::test] +<a href=#235 id=235 data-nosnippet>235</a> </span><span class="kw">async fn </span>test_empty_data_append_action() { +<a href=#236 id=236 data-nosnippet>236</a> <span class="kw">let </span>table = make_v2_minimal_table(); +<a href=#237 id=237 data-nosnippet>237</a> <span class="kw">let </span>tx = Transaction::new(<span class="kw-2">&</span>table); +<a href=#238 id=238 data-nosnippet>238</a> <span class="kw">let </span><span class="kw-2">mut </span>action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); +<a href=#239 id=239 data-nosnippet>239</a> action.add_data_files(<span class="macro">vec!</span>[]).unwrap(); +<a href=#240 id=240 data-nosnippet>240</a> <span class="macro">assert!</span>(action.apply().<span class="kw">await</span>.is_err()); +<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> <span class="attr">#[tokio::test] +<a href=#244 id=244 data-nosnippet>244</a> </span><span class="kw">async fn </span>test_set_snapshot_properties() { +<a href=#245 id=245 data-nosnippet>245</a> <span class="kw">let </span>table = make_v2_minimal_table(); +<a href=#246 id=246 data-nosnippet>246</a> <span class="kw">let </span>tx = Transaction::new(<span class="kw-2">&</span>table); +<a href=#247 id=247 data-nosnippet>247</a> <span class="kw">let </span><span class="kw-2">mut </span>action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); +<a href=#248 id=248 data-nosnippet>248</a> +<a href=#249 id=249 data-nosnippet>249</a> <span class="kw">let </span><span class="kw-2">mut </span>snapshot_properties = HashMap::new(); +<a href=#250 id=250 data-nosnippet>250</a> snapshot_properties.insert(<span class="string">"key"</span>.to_string(), <span class="string">"val"</span>.to_string()); +<a href=#251 id=251 data-nosnippet>251</a> action.set_snapshot_properties(snapshot_properties).unwrap(); +<a href=#252 id=252 data-nosnippet>252</a> <span class="kw">let </span>data_file = DataFileBuilder::default() +<a href=#253 id=253 data-nosnippet>253</a> .content(DataContentType::Data) +<a href=#254 id=254 data-nosnippet>254</a> .file_path(<span class="string">"test/1.parquet"</span>.to_string()) +<a href=#255 id=255 data-nosnippet>255</a> .file_format(DataFileFormat::Parquet) +<a href=#256 id=256 data-nosnippet>256</a> .file_size_in_bytes(<span class="number">100</span>) +<a href=#257 id=257 data-nosnippet>257</a> .record_count(<span class="number">1</span>) +<a href=#258 id=258 data-nosnippet>258</a> .partition_spec_id(table.metadata().default_partition_spec_id()) +<a href=#259 id=259 data-nosnippet>259</a> .partition(Struct::from_iter([<span class="prelude-val">Some</span>(Literal::long(<span class="number">300</span>))])) +<a href=#260 id=260 data-nosnippet>260</a> .build() +<a href=#261 id=261 data-nosnippet>261</a> .unwrap(); +<a href=#262 id=262 data-nosnippet>262</a> action.add_data_files(<span class="macro">vec!</span>[data_file]).unwrap(); +<a href=#263 id=263 data-nosnippet>263</a> <span class="kw">let </span>tx = action.apply().<span class="kw">await</span>.unwrap(); +<a href=#264 id=264 data-nosnippet>264</a> +<a href=#265 id=265 data-nosnippet>265</a> <span class="comment">// Check customized properties is contained in snapshot summary properties. +<a href=#266 id=266 data-nosnippet>266</a> </span><span class="kw">let </span>new_snapshot = <span class="kw">if let </span>TableUpdate::AddSnapshot { snapshot } = <span class="kw-2">&</span>tx.updates[<span class="number">0</span>] { +<a href=#267 id=267 data-nosnippet>267</a> snapshot +<a href=#268 id=268 data-nosnippet>268</a> } <span class="kw">else </span>{ +<a href=#269 id=269 data-nosnippet>269</a> <span class="macro">unreachable!</span>() +<a href=#270 id=270 data-nosnippet>270</a> }; +<a href=#271 id=271 data-nosnippet>271</a> <span class="macro">assert_eq!</span>( +<a href=#272 id=272 data-nosnippet>272</a> new_snapshot +<a href=#273 id=273 data-nosnippet>273</a> .summary() +<a href=#274 id=274 data-nosnippet>274</a> .additional_properties +<a href=#275 id=275 data-nosnippet>275</a> .get(<span class="string">"key"</span>) +<a href=#276 id=276 data-nosnippet>276</a> .unwrap(), +<a href=#277 id=277 data-nosnippet>277</a> <span class="string">"val" +<a href=#278 id=278 data-nosnippet>278</a> </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="attr">#[tokio::test] +<a href=#282 id=282 data-nosnippet>282</a> </span><span class="kw">async fn </span>test_fast_append_action() { +<a href=#283 id=283 data-nosnippet>283</a> <span class="kw">let </span>table = make_v2_minimal_table(); +<a href=#284 id=284 data-nosnippet>284</a> <span class="kw">let </span>tx = Transaction::new(<span class="kw-2">&</span>table); +<a href=#285 id=285 data-nosnippet>285</a> <span class="kw">let </span><span class="kw-2">mut </span>action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); +<a href=#286 id=286 data-nosnippet>286</a> +<a href=#287 id=287 data-nosnippet>287</a> <span class="comment">// check add data file with incompatible partition value +<a href=#288 id=288 data-nosnippet>288</a> </span><span class="kw">let </span>data_file = DataFileBuilder::default() +<a href=#289 id=289 data-nosnippet>289</a> .content(DataContentType::Data) +<a href=#290 id=290 data-nosnippet>290</a> .file_path(<span class="string">"test/3.parquet"</span>.to_string()) +<a href=#291 id=291 data-nosnippet>291</a> .file_format(DataFileFormat::Parquet) +<a href=#292 id=292 data-nosnippet>292</a> .file_size_in_bytes(<span class="number">100</span>) +<a href=#293 id=293 data-nosnippet>293</a> .record_count(<span class="number">1</span>) +<a href=#294 id=294 data-nosnippet>294</a> .partition_spec_id(table.metadata().default_partition_spec_id()) +<a href=#295 id=295 data-nosnippet>295</a> .partition(Struct::from_iter([<span class="prelude-val">Some</span>(Literal::string(<span class="string">"test"</span>))])) +<a href=#296 id=296 data-nosnippet>296</a> .build() +<a href=#297 id=297 data-nosnippet>297</a> .unwrap(); +<a href=#298 id=298 data-nosnippet>298</a> <span class="macro">assert!</span>(action.add_data_files(<span class="macro">vec!</span>[data_file.clone()]).is_err()); +<a href=#299 id=299 data-nosnippet>299</a> +<a href=#300 id=300 data-nosnippet>300</a> <span class="kw">let </span>data_file = DataFileBuilder::default() +<a href=#301 id=301 data-nosnippet>301</a> .content(DataContentType::Data) +<a href=#302 id=302 data-nosnippet>302</a> .file_path(<span class="string">"test/3.parquet"</span>.to_string()) +<a href=#303 id=303 data-nosnippet>303</a> .file_format(DataFileFormat::Parquet) +<a href=#304 id=304 data-nosnippet>304</a> .file_size_in_bytes(<span class="number">100</span>) +<a href=#305 id=305 data-nosnippet>305</a> .record_count(<span class="number">1</span>) +<a href=#306 id=306 data-nosnippet>306</a> .partition_spec_id(table.metadata().default_partition_spec_id()) +<a href=#307 id=307 data-nosnippet>307</a> .partition(Struct::from_iter([<span class="prelude-val">Some</span>(Literal::long(<span class="number">300</span>))])) +<a href=#308 id=308 data-nosnippet>308</a> .build() +<a href=#309 id=309 data-nosnippet>309</a> .unwrap(); +<a href=#310 id=310 data-nosnippet>310</a> action.add_data_files(<span class="macro">vec!</span>[data_file.clone()]).unwrap(); +<a href=#311 id=311 data-nosnippet>311</a> <span class="kw">let </span>tx = action.apply().<span class="kw">await</span>.unwrap(); +<a href=#312 id=312 data-nosnippet>312</a> +<a href=#313 id=313 data-nosnippet>313</a> <span class="comment">// check updates and requirements +<a href=#314 id=314 data-nosnippet>314</a> </span><span class="macro">assert!</span>( +<a href=#315 id=315 data-nosnippet>315</a> <span class="macro">matches!</span>((<span class="kw-2">&</span>tx.updates[<span class="number">0</span>],<span class="kw-2">&</span>tx.updates[<span class="number">1</span>]), (TableUpdate::AddSnapshot { snapshot },TableUpdate::SetSnapshotRef { reference,ref_name }) <span class="kw">if </span>snapshot.snapshot_id() == reference.snapshot_id && ref_name == MAIN_BRANCH) +<a href=#316 id=316 data-nosnippet>316</a> ); +<a href=#317 id=317 data-nosnippet>317</a> <span class="macro">assert_eq!</span>( +<a href=#318 id=318 data-nosnippet>318</a> <span class="macro">vec!</span>[ +<a href=#319 id=319 data-nosnippet>319</a> TableRequirement::UuidMatch { +<a href=#320 id=320 data-nosnippet>320</a> uuid: table.metadata().uuid() +<a href=#321 id=321 data-nosnippet>321</a> }, +<a href=#322 id=322 data-nosnippet>322</a> TableRequirement::RefSnapshotIdMatch { +<a href=#323 id=323 data-nosnippet>323</a> r#ref: MAIN_BRANCH.to_string(), +<a href=#324 id=324 data-nosnippet>324</a> snapshot_id: table.metadata().current_snapshot_id +<a href=#325 id=325 data-nosnippet>325</a> } +<a href=#326 id=326 data-nosnippet>326</a> ], +<a href=#327 id=327 data-nosnippet>327</a> tx.requirements +<a href=#328 id=328 data-nosnippet>328</a> ); <a href=#329 id=329 data-nosnippet>329</a> -<a href=#330 id=330 data-nosnippet>330</a> <span class="comment">// Attempt to add the existing Parquet files with fast append. -<a href=#331 id=331 data-nosnippet>331</a> </span><span class="kw">let </span>new_tx = fast_append_action -<a href=#332 id=332 data-nosnippet>332</a> .add_parquet_files(file_paths.clone()) -<a href=#333 id=333 data-nosnippet>333</a> .<span class="kw">await -<a href=#334 id=334 data-nosnippet>334</a> </span>.expect(<span class="string">"Adding existing Parquet files should succeed"</span>); -<a href=#335 id=335 data-nosnippet>335</a> -<a href=#336 id=336 data-nosnippet>336</a> <span class="kw">let </span><span class="kw-2">mut </span>found_add_snapshot = <span class="bool-val">false</span>; -<a href=#337 id=337 data-nosnippet>337</a> <span class="kw">let </span><span class="kw-2">mut </span>found_set_snapshot_ref = <span class="bool-val">false</span>; -<a href=#338 id=338 data-nosnippet>338</a> <span class="kw">for </span>update <span class="kw">in </span>new_tx.updates.iter() { -<a href=#339 id=339 data-nosnippet>339</a> <span class="kw">match </span>update { -<a href=#340 id=340 data-nosnippet>340</a> TableUpdate::AddSnapshot { .. } => { -<a href=#341 id=341 data-nosnippet>341</a> found_add_snapshot = <span class="bool-val">true</span>; -<a href=#342 id=342 data-nosnippet>342</a> } -<a href=#343 id=343 data-nosnippet>343</a> TableUpdate::SetSnapshotRef { -<a href=#344 id=344 data-nosnippet>344</a> ref_name, -<a href=#345 id=345 data-nosnippet>345</a> reference, -<a href=#346 id=346 data-nosnippet>346</a> } => { -<a href=#347 id=347 data-nosnippet>347</a> found_set_snapshot_ref = <span class="bool-val">true</span>; -<a href=#348 id=348 data-nosnippet>348</a> <span class="macro">assert_eq!</span>(ref_name, MAIN_BRANCH); -<a href=#349 id=349 data-nosnippet>349</a> <span class="macro">assert!</span>(reference.snapshot_id > <span class="number">0</span>); -<a href=#350 id=350 data-nosnippet>350</a> } -<a href=#351 id=351 data-nosnippet>351</a> <span class="kw">_ </span>=> {} -<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> <span class="macro">assert!</span>(found_add_snapshot); -<a href=#355 id=355 data-nosnippet>355</a> <span class="macro">assert!</span>(found_set_snapshot_ref); -<a href=#356 id=356 data-nosnippet>356</a> -<a href=#357 id=357 data-nosnippet>357</a> <span class="kw">let </span>new_snapshot = <span class="kw">if let </span>TableUpdate::AddSnapshot { snapshot } = <span class="kw-2">&</span>new_tx.updates[<span class="number">0</span>] { -<a href=#358 id=358 data-nosnippet>358</a> snapshot -<a href=#359 id=359 data-nosnippet>359</a> } <span class="kw">else </span>{ -<a href=#360 id=360 data-nosnippet>360</a> <span class="macro">panic!</span>(<span class="string">"Expected the first update to be an AddSnapshot update"</span>); -<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> <span class="kw">let </span>manifest_list = new_snapshot -<a href=#364 id=364 data-nosnippet>364</a> .load_manifest_list(fixture.table.file_io(), fixture.table.metadata()) -<a href=#365 id=365 data-nosnippet>365</a> .<span class="kw">await -<a href=#366 id=366 data-nosnippet>366</a> </span>.expect(<span class="string">"Failed to load manifest list"</span>); -<a href=#367 id=367 data-nosnippet>367</a> -<a href=#368 id=368 data-nosnippet>368</a> <span class="macro">assert_eq!</span>( -<a href=#369 id=369 data-nosnippet>369</a> manifest_list.entries().len(), -<a href=#370 id=370 data-nosnippet>370</a> <span class="number">2</span>, -<a href=#371 id=371 data-nosnippet>371</a> <span class="string">"Expected 2 manifest list entries, got {}"</span>, -<a href=#372 id=372 data-nosnippet>372</a> manifest_list.entries().len() -<a href=#373 id=373 data-nosnippet>373</a> ); -<a href=#374 id=374 data-nosnippet>374</a> -<a href=#375 id=375 data-nosnippet>375</a> <span class="comment">// Load the manifest from the manifest list -<a href=#376 id=376 data-nosnippet>376</a> </span><span class="kw">let </span>manifest = manifest_list.entries()[<span class="number">0</span>] -<a href=#377 id=377 data-nosnippet>377</a> .load_manifest(fixture.table.file_io()) -<a href=#378 id=378 data-nosnippet>378</a> .<span class="kw">await -<a href=#379 id=379 data-nosnippet>379</a> </span>.expect(<span class="string">"Failed to load manifest"</span>); -<a href=#380 id=380 data-nosnippet>380</a> -<a href=#381 id=381 data-nosnippet>381</a> <span class="comment">// Check that the manifest contains three entries. -<a href=#382 id=382 data-nosnippet>382</a> </span><span class="macro">assert_eq!</span>(manifest.entries().len(), <span class="number">3</span>); -<a href=#383 id=383 data-nosnippet>383</a> -<a href=#384 id=384 data-nosnippet>384</a> <span class="comment">// Verify each file path appears in manifest. -<a href=#385 id=385 data-nosnippet>385</a> </span><span class="kw">let </span>manifest_paths: Vec<String> = manifest -<a href=#386 id=386 data-nosnippet>386</a> .entries() -<a href=#387 id=387 data-nosnippet>387</a> .iter() -<a href=#388 id=388 data-nosnippet>388</a> .map(|entry| entry.data_file().file_path.clone()) -<a href=#389 id=389 data-nosnippet>389</a> .collect(); -<a href=#390 id=390 data-nosnippet>390</a> <span class="kw">for </span>path <span class="kw">in </span>file_paths { -<a href=#391 id=391 data-nosnippet>391</a> <span class="macro">assert!</span>(manifest_paths.contains(<span class="kw-2">&</span>path)); -<a href=#392 id=392 data-nosnippet>392</a> } -<a href=#393 id=393 data-nosnippet>393</a> } -<a href=#394 id=394 data-nosnippet>394</a>}</code></pre></div></section></main></body></html> \ No newline at end of file +<a href=#330 id=330 data-nosnippet>330</a> <span class="comment">// check manifest list +<a href=#331 id=331 data-nosnippet>331</a> </span><span class="kw">let </span>new_snapshot = <span class="kw">if let </span>TableUpdate::AddSnapshot { snapshot } = <span class="kw-2">&</span>tx.updates[<span class="number">0</span>] { +<a href=#332 id=332 data-nosnippet>332</a> snapshot +<a href=#333 id=333 data-nosnippet>333</a> } <span class="kw">else </span>{ +<a href=#334 id=334 data-nosnippet>334</a> <span class="macro">unreachable!</span>() +<a href=#335 id=335 data-nosnippet>335</a> }; +<a href=#336 id=336 data-nosnippet>336</a> <span class="kw">let </span>manifest_list = new_snapshot +<a href=#337 id=337 data-nosnippet>337</a> .load_manifest_list(table.file_io(), table.metadata()) +<a href=#338 id=338 data-nosnippet>338</a> .<span class="kw">await +<a href=#339 id=339 data-nosnippet>339</a> </span>.unwrap(); +<a href=#340 id=340 data-nosnippet>340</a> <span class="macro">assert_eq!</span>(<span class="number">1</span>, manifest_list.entries().len()); +<a href=#341 id=341 data-nosnippet>341</a> <span class="macro">assert_eq!</span>( +<a href=#342 id=342 data-nosnippet>342</a> manifest_list.entries()[<span class="number">0</span>].sequence_number, +<a href=#343 id=343 data-nosnippet>343</a> new_snapshot.sequence_number() +<a href=#344 id=344 data-nosnippet>344</a> ); +<a href=#345 id=345 data-nosnippet>345</a> +<a href=#346 id=346 data-nosnippet>346</a> <span class="comment">// check manifest +<a href=#347 id=347 data-nosnippet>347</a> </span><span class="kw">let </span>manifest = manifest_list.entries()[<span class="number">0</span>] +<a href=#348 id=348 data-nosnippet>348</a> .load_manifest(table.file_io()) +<a href=#349 id=349 data-nosnippet>349</a> .<span class="kw">await +<a href=#350 id=350 data-nosnippet>350</a> </span>.unwrap(); +<a href=#351 id=351 data-nosnippet>351</a> <span class="macro">assert_eq!</span>(<span class="number">1</span>, manifest.entries().len()); +<a href=#352 id=352 data-nosnippet>352</a> <span class="macro">assert_eq!</span>( +<a href=#353 id=353 data-nosnippet>353</a> new_snapshot.sequence_number(), +<a href=#354 id=354 data-nosnippet>354</a> manifest.entries()[<span class="number">0</span>] +<a href=#355 id=355 data-nosnippet>355</a> .sequence_number() +<a href=#356 id=356 data-nosnippet>356</a> .expect(<span class="string">"Inherit sequence number by load manifest"</span>) +<a href=#357 id=357 data-nosnippet>357</a> ); +<a href=#358 id=358 data-nosnippet>358</a> +<a href=#359 id=359 data-nosnippet>359</a> <span class="macro">assert_eq!</span>( +<a href=#360 id=360 data-nosnippet>360</a> new_snapshot.snapshot_id(), +<a href=#361 id=361 data-nosnippet>361</a> manifest.entries()[<span class="number">0</span>].snapshot_id().unwrap() +<a href=#362 id=362 data-nosnippet>362</a> ); +<a href=#363 id=363 data-nosnippet>363</a> <span class="macro">assert_eq!</span>(data_file, <span class="kw-2">*</span>manifest.entries()[<span class="number">0</span>].data_file()); +<a href=#364 id=364 data-nosnippet>364</a> } +<a href=#365 id=365 data-nosnippet>365</a> +<a href=#366 id=366 data-nosnippet>366</a> <span class="attr">#[tokio::test] +<a href=#367 id=367 data-nosnippet>367</a> </span><span class="kw">async fn </span>test_add_existing_parquet_files_to_unpartitioned_table() { +<a href=#368 id=368 data-nosnippet>368</a> <span class="kw">let </span><span class="kw-2">mut </span>fixture = TableTestFixture::new_unpartitioned(); +<a href=#369 id=369 data-nosnippet>369</a> fixture.setup_unpartitioned_manifest_files().<span class="kw">await</span>; +<a href=#370 id=370 data-nosnippet>370</a> <span class="kw">let </span>tx = <span class="kw">crate</span>::transaction::Transaction::new(<span class="kw-2">&</span>fixture.table); +<a href=#371 id=371 data-nosnippet>371</a> +<a href=#372 id=372 data-nosnippet>372</a> <span class="kw">let </span>file_paths = <span class="macro">vec!</span>[ +<a href=#373 id=373 data-nosnippet>373</a> <span class="macro">format!</span>(<span class="string">"{}/1.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), +<a href=#374 id=374 data-nosnippet>374</a> <span class="macro">format!</span>(<span class="string">"{}/2.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), +<a href=#375 id=375 data-nosnippet>375</a> <span class="macro">format!</span>(<span class="string">"{}/3.parquet"</span>, <span class="kw-2">&</span>fixture.table_location), +<a href=#376 id=376 data-nosnippet>376</a> ]; +<a href=#377 id=377 data-nosnippet>377</a> +<a href=#378 id=378 data-nosnippet>378</a> <span class="kw">let </span>fast_append_action = tx.fast_append(<span class="prelude-val">None</span>, <span class="macro">vec!</span>[]).unwrap(); +<a href=#379 id=379 data-nosnippet>379</a> +<a href=#380 id=380 data-nosnippet>380</a> <span class="comment">// Attempt to add the existing Parquet files with fast append. +<a href=#381 id=381 data-nosnippet>381</a> </span><span class="kw">let </span>new_tx = fast_append_action +<a href=#382 id=382 data-nosnippet>382</a> .add_parquet_files(file_paths.clone()) +<a href=#383 id=383 data-nosnippet>383</a> .<span class="kw">await +<a href=#384 id=384 data-nosnippet>384</a> </span>.expect(<span class="string">"Adding existing Parquet files should succeed"</span>); +<a href=#385 id=385 data-nosnippet>385</a> +<a href=#386 id=386 data-nosnippet>386</a> <span class="kw">let </span><span class="kw-2">mut </span>found_add_snapshot = <span class="bool-val">false</span>; +<a href=#387 id=387 data-nosnippet>387</a> <span class="kw">let </span><span class="kw-2">mut </span>found_set_snapshot_ref = <span class="bool-val">false</span>; +<a href=#388 id=388 data-nosnippet>388</a> <span class="kw">for </span>update <span class="kw">in </span>new_tx.updates.iter() { +<a href=#389 id=389 data-nosnippet>389</a> <span class="kw">match </span>update { +<a href=#390 id=390 data-nosnippet>390</a> TableUpdate::AddSnapshot { .. } => { +<a href=#391 id=391 data-nosnippet>391</a> found_add_snapshot = <span class="bool-val">true</span>; +<a href=#392 id=392 data-nosnippet>392</a> } +<a href=#393 id=393 data-nosnippet>393</a> TableUpdate::SetSnapshotRef { +<a href=#394 id=394 data-nosnippet>394</a> ref_name, +<a href=#395 id=395 data-nosnippet>395</a> reference, +<a href=#396 id=396 data-nosnippet>396</a> } => { +<a href=#397 id=397 data-nosnippet>397</a> found_set_snapshot_ref = <span class="bool-val">true</span>; +<a href=#398 id=398 data-nosnippet>398</a> <span class="macro">assert_eq!</span>(ref_name, MAIN_BRANCH); +<a href=#399 id=399 data-nosnippet>399</a> <span class="macro">assert!</span>(reference.snapshot_id > <span class="number">0</span>); +<a href=#400 id=400 data-nosnippet>400</a> } +<a href=#401 id=401 data-nosnippet>401</a> <span class="kw">_ </span>=> {} +<a href=#402 id=402 data-nosnippet>402</a> } +<a href=#403 id=403 data-nosnippet>403</a> } +<a href=#404 id=404 data-nosnippet>404</a> <span class="macro">assert!</span>(found_add_snapshot); +<a href=#405 id=405 data-nosnippet>405</a> <span class="macro">assert!</span>(found_set_snapshot_ref); +<a href=#406 id=406 data-nosnippet>406</a> +<a href=#407 id=407 data-nosnippet>407</a> <span class="kw">let </span>new_snapshot = <span class="kw">if let </span>TableUpdate::AddSnapshot { snapshot } = <span class="kw-2">&</span>new_tx.updates[<span class="number">0</span>] { +<a href=#408 id=408 data-nosnippet>408</a> snapshot +<a href=#409 id=409 data-nosnippet>409</a> } <span class="kw">else </span>{ +<a href=#410 id=410 data-nosnippet>410</a> <span class="macro">panic!</span>(<span class="string">"Expected the first update to be an AddSnapshot update"</span>); +<a href=#411 id=411 data-nosnippet>411</a> }; +<a href=#412 id=412 data-nosnippet>412</a> +<a href=#413 id=413 data-nosnippet>413</a> <span class="kw">let </span>manifest_list = new_snapshot +<a href=#414 id=414 data-nosnippet>414</a> .load_manifest_list(fixture.table.file_io(), fixture.table.metadata()) +<a href=#415 id=415 data-nosnippet>415</a> .<span class="kw">await +<a href=#416 id=416 data-nosnippet>416</a> </span>.expect(<span class="string">"Failed to load manifest list"</span>); +<a href=#417 id=417 data-nosnippet>417</a> +<a href=#418 id=418 data-nosnippet>418</a> <span class="macro">assert_eq!</span>( +<a href=#419 id=419 data-nosnippet>419</a> manifest_list.entries().len(), +<a href=#420 id=420 data-nosnippet>420</a> <span class="number">2</span>, +<a href=#421 id=421 data-nosnippet>421</a> <span class="string">"Expected 2 manifest list entries, got {}"</span>, +<a href=#422 id=422 data-nosnippet>422</a> manifest_list.entries().len() +<a href=#423 id=423 data-nosnippet>423</a> ); +<a href=#424 id=424 data-nosnippet>424</a> +<a href=#425 id=425 data-nosnippet>425</a> <span class="comment">// Load the manifest from the manifest list +<a href=#426 id=426 data-nosnippet>426</a> </span><span class="kw">let </span>manifest = manifest_list.entries()[<span class="number">0</span>] +<a href=#427 id=427 data-nosnippet>427</a> .load_manifest(fixture.table.file_io()) +<a href=#428 id=428 data-nosnippet>428</a> .<span class="kw">await +<a href=#429 id=429 data-nosnippet>429</a> </span>.expect(<span class="string">"Failed to load manifest"</span>); +<a href=#430 id=430 data-nosnippet>430</a> +<a href=#431 id=431 data-nosnippet>431</a> <span class="comment">// Check that the manifest contains three entries. +<a href=#432 id=432 data-nosnippet>432</a> </span><span class="macro">assert_eq!</span>(manifest.entries().len(), <span class="number">3</span>); +<a href=#433 id=433 data-nosnippet>433</a> +<a href=#434 id=434 data-nosnippet>434</a> <span class="comment">// Verify each file path appears in manifest. +<a href=#435 id=435 data-nosnippet>435</a> </span><span class="kw">let </span>manifest_paths: Vec<String> = manifest +<a href=#436 id=436 data-nosnippet>436</a> .entries() +<a href=#437 id=437 data-nosnippet>437</a> .iter() +<a href=#438 id=438 data-nosnippet>438</a> .map(|entry| entry.data_file().file_path.clone()) +<a href=#439 id=439 data-nosnippet>439</a> .collect(); +<a href=#440 id=440 data-nosnippet>440</a> <span class="kw">for </span>path <span class="kw">in </span>file_paths { +<a href=#441 id=441 data-nosnippet>441</a> <span class="macro">assert!</span>(manifest_paths.contains(<span class="kw-2">&</span>path)); +<a href=#442 id=442 data-nosnippet>442</a> } +<a href=#443 id=443 data-nosnippet>443</a> } +<a href=#444 id=444 data-nosnippet>444</a>}</code></pre></div></section></main></body></html> \ No newline at end of file diff --git a/api/src/iceberg/transaction/snapshot.rs.html b/api/src/iceberg/transaction/snapshot.rs.html index 672b1b3f..dd789cb2 100644 --- a/api/src/iceberg/transaction/snapshot.rs.html +++ b/api/src/iceberg/transaction/snapshot.rs.html @@ -122,258 +122,267 @@ <a href=#122 id=122 data-nosnippet>122</a> <span class="prelude-val">Ok</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> <span class="doccomment">/// Add data files to the snapshot. -<a href=#126 id=126 data-nosnippet>126</a> </span><span class="kw">pub fn </span>add_data_files( +<a href=#125 id=125 data-nosnippet>125</a> <span class="doccomment">/// Set snapshot summary properties. +<a href=#126 id=126 data-nosnippet>126</a> </span><span class="kw">pub fn </span>set_snapshot_properties( <a href=#127 id=127 data-nosnippet>127</a> <span class="kw-2">&mut </span><span class="self">self</span>, -<a href=#128 id=128 data-nosnippet>128</a> data_files: <span class="kw">impl </span>IntoIterator<Item = DataFile>, +<a href=#128 id=128 data-nosnippet>128</a> snapshot_properties: HashMap<String, String>, <a href=#129 id=129 data-nosnippet>129</a> ) -> <span class="prelude-ty">Result</span><<span class="kw-2">&mut </span><span class="self">Self</span>> { -<a href=#130 id=130 data-nosnippet>130</a> <span class="kw">let </span>data_files: Vec<DataFile> = data_files.into_iter().collect(); -<a href=#131 id=131 data-nosnippet>131</a> <span class="kw">for </span>data_file <span class="kw">in </span><span class="kw-2">&</span>data_files { -<a href=#132 id=132 data-nosnippet>132</a> <span class="kw">if </span>data_file.content_type() != <span class="kw">crate</span>::spec::DataContentType::Data { -<a href=#133 id=133 data-nosnippet>133</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( -<a href=#134 id=134 data-nosnippet>134</a> ErrorKind::DataInvalid, -<a href=#135 id=135 data-nosnippet>135</a> <span class="string">"Only data content type is allowed for fast append"</span>, -<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> <span class="comment">// Check if the data file partition spec id matches the table default partition spec id. -<a href=#139 id=139 data-nosnippet>139</a> </span><span class="kw">if </span><span class="self">self</span>.tx.current_table.metadata().default_partition_spec_id() -<a href=#140 id=140 data-nosnippet>140</a> != data_file.partition_spec_id -<a href=#141 id=141 data-nosnippet>141</a> { +<a href=#130 id=130 data-nosnippet>130</a> <span class="self">self</span>.snapshot_properties = snapshot_properties; +<a href=#131 id=131 data-nosnippet>131</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>) +<a href=#132 id=132 data-nosnippet>132</a> } +<a href=#133 id=133 data-nosnippet>133</a> +<a href=#134 id=134 data-nosnippet>134</a> <span class="doccomment">/// Add data files to the snapshot. +<a href=#135 id=135 data-nosnippet>135</a> </span><span class="kw">pub fn </span>add_data_files( +<a href=#136 id=136 data-nosnippet>136</a> <span class="kw-2">&mut </span><span class="self">self</span>, +<a href=#137 id=137 data-nosnippet>137</a> data_files: <span class="kw">impl </span>IntoIterator<Item = DataFile>, +<a href=#138 id=138 data-nosnippet>138</a> ) -> <span class="prelude-ty">Result</span><<span class="kw-2">&mut </span><span class="self">Self</span>> { +<a href=#139 id=139 data-nosnippet>139</a> <span class="kw">let </span>data_files: Vec<DataFile> = data_files.into_iter().collect(); +<a href=#140 id=140 data-nosnippet>140</a> <span class="kw">for </span>data_file <span class="kw">in </span><span class="kw-2">&</span>data_files { +<a href=#141 id=141 data-nosnippet>141</a> <span class="kw">if </span>data_file.content_type() != <span class="kw">crate</span>::spec::DataContentType::Data { <a href=#142 id=142 data-nosnippet>142</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( <a href=#143 id=143 data-nosnippet>143</a> ErrorKind::DataInvalid, -<a href=#144 id=144 data-nosnippet>144</a> <span class="string">"Data file partition spec id does not match table default partition spec id"</span>, +<a href=#144 id=144 data-nosnippet>144</a> <span class="string">"Only data content type is allowed for fast append"</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> <span class="self">Self</span>::validate_partition_value( -<a href=#148 id=148 data-nosnippet>148</a> data_file.partition(), -<a href=#149 id=149 data-nosnippet>149</a> <span class="self">self</span>.tx.current_table.metadata().default_partition_type(), -<a href=#150 id=150 data-nosnippet>150</a> )<span class="question-mark">?</span>; -<a href=#151 id=151 data-nosnippet>151</a> } -<a href=#152 id=152 data-nosnippet>152</a> <span class="self">self</span>.added_data_files.extend(data_files); -<a href=#153 id=153 data-nosnippet>153</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>) -<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> <span class="kw">fn </span>new_manifest_output(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><OutputFile> { -<a href=#157 id=157 data-nosnippet>157</a> <span class="kw">let </span>new_manifest_path = <span class="macro">format!</span>( -<a href=#158 id=158 data-nosnippet>158</a> <span class="string">"{}/{}/{}-m{}.{}"</span>, -<a href=#159 id=159 data-nosnippet>159</a> <span class="self">self</span>.tx.current_table.metadata().location(), -<a href=#160 id=160 data-nosnippet>160</a> META_ROOT_PATH, -<a href=#161 id=161 data-nosnippet>161</a> <span class="self">self</span>.commit_uuid, -<a href=#162 id=162 data-nosnippet>162</a> <span class="self">self</span>.manifest_counter.next().unwrap(), -<a href=#163 id=163 data-nosnippet>163</a> DataFileFormat::Avro -<a href=#164 id=164 data-nosnippet>164</a> ); -<a href=#165 id=165 data-nosnippet>165</a> <span class="self">self</span>.tx -<a href=#166 id=166 data-nosnippet>166</a> .current_table -<a href=#167 id=167 data-nosnippet>167</a> .file_io() -<a href=#168 id=168 data-nosnippet>168</a> .new_output(new_manifest_path) -<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="comment">// Write manifest file for added data files and return the ManifestFile for ManifestList. -<a href=#172 id=172 data-nosnippet>172</a> </span><span class="kw">async fn </span>write_added_manifest(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><ManifestFile> { -<a href=#173 id=173 data-nosnippet>173</a> <span class="kw">let </span>added_data_files = std::mem::take(<span class="kw-2">&mut </span><span class="self">self</span>.added_data_files); -<a href=#174 id=174 data-nosnippet>174</a> <span class="kw">if </span>added_data_files.is_empty() { -<a href=#175 id=175 data-nosnippet>175</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( -<a href=#176 id=176 data-nosnippet>176</a> ErrorKind::PreconditionFailed, -<a href=#177 id=177 data-nosnippet>177</a> <span class="string">"No added data files found when write a manifest file"</span>, -<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>snapshot_id = <span class="self">self</span>.snapshot_id; -<a href=#182 id=182 data-nosnippet>182</a> <span class="kw">let </span>format_version = <span class="self">self</span>.tx.current_table.metadata().format_version(); -<a href=#183 id=183 data-nosnippet>183</a> <span class="kw">let </span>manifest_entries = added_data_files.into_iter().map(|data_file| { -<a href=#184 id=184 data-nosnippet>184</a> <span class="kw">let </span>builder = ManifestEntry::builder() -<a href=#185 id=185 data-nosnippet>185</a> .status(<span class="kw">crate</span>::spec::ManifestStatus::Added) -<a href=#186 id=186 data-nosnippet>186</a> .data_file(data_file); -<a href=#187 id=187 data-nosnippet>187</a> <span class="kw">if </span>format_version == FormatVersion::V1 { -<a href=#188 id=188 data-nosnippet>188</a> builder.snapshot_id(snapshot_id).build() -<a href=#189 id=189 data-nosnippet>189</a> } <span class="kw">else </span>{ -<a href=#190 id=190 data-nosnippet>190</a> <span class="comment">// For format version > 1, we set the snapshot id at the inherited time to avoid rewrite the manifest file when -<a href=#191 id=191 data-nosnippet>191</a> // commit failed. -<a href=#192 id=192 data-nosnippet>192</a> </span>builder.build() -<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">let </span><span class="kw-2">mut </span>writer = { -<a href=#196 id=196 data-nosnippet>196</a> <span class="kw">let </span>builder = ManifestWriterBuilder::new( -<a href=#197 id=197 data-nosnippet>197</a> <span class="self">self</span>.new_manifest_output()<span class="question-mark">?</span>, -<a href=#198 id=198 data-nosnippet>198</a> <span class="prelude-val">Some</span>(<span class="self">self</span>.snapshot_id), -<a href=#199 id=199 data-nosnippet>199</a> <span class="self">self</span>.key_metadata.clone(), -<a href=#200 id=200 data-nosnippet>200</a> <span class="self">self</span>.tx.current_table.metadata().current_schema().clone(), -<a href=#201 id=201 data-nosnippet>201</a> <span class="self">self</span>.tx -<a href=#202 id=202 data-nosnippet>202</a> .current_table -<a href=#203 id=203 data-nosnippet>203</a> .metadata() -<a href=#204 id=204 data-nosnippet>204</a> .default_partition_spec() -<a href=#205 id=205 data-nosnippet>205</a> .as_ref() -<a href=#206 id=206 data-nosnippet>206</a> .clone(), -<a href=#207 id=207 data-nosnippet>207</a> ); -<a href=#208 id=208 data-nosnippet>208</a> <span class="kw">if </span><span class="self">self</span>.tx.current_table.metadata().format_version() == FormatVersion::V1 { -<a href=#209 id=209 data-nosnippet>209</a> builder.build_v1() -<a href=#210 id=210 data-nosnippet>210</a> } <span class="kw">else </span>{ -<a href=#211 id=211 data-nosnippet>211</a> builder.build_v2_data() -<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="kw">for </span>entry <span class="kw">in </span>manifest_entries { -<a href=#215 id=215 data-nosnippet>215</a> writer.add_entry(entry)<span class="question-mark">?</span>; -<a href=#216 id=216 data-nosnippet>216</a> } -<a href=#217 id=217 data-nosnippet>217</a> writer.write_manifest_file().<span class="kw">await -<a href=#218 id=218 data-nosnippet>218</a> </span>} -<a href=#219 id=219 data-nosnippet>219</a> -<a href=#220 id=220 data-nosnippet>220</a> <span class="kw">async fn </span>manifest_file<OP: SnapshotProduceOperation, MP: ManifestProcess>( -<a href=#221 id=221 data-nosnippet>221</a> <span class="kw-2">&mut </span><span class="self">self</span>, -<a href=#222 id=222 data-nosnippet>222</a> snapshot_produce_operation: <span class="kw-2">&</span>OP, -<a href=#223 id=223 data-nosnippet>223</a> manifest_process: <span class="kw-2">&</span>MP, -<a href=#224 id=224 data-nosnippet>224</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestFile>> { -<a href=#225 id=225 data-nosnippet>225</a> <span class="kw">let </span>added_manifest = <span class="self">self</span>.write_added_manifest().<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#226 id=226 data-nosnippet>226</a> <span class="kw">let </span>existing_manifests = snapshot_produce_operation.existing_manifest(<span class="self">self</span>).<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#227 id=227 data-nosnippet>227</a> <span class="comment">// # TODO -<a href=#228 id=228 data-nosnippet>228</a> // Support process delete entries. -<a href=#229 id=229 data-nosnippet>229</a> -<a href=#230 id=230 data-nosnippet>230</a> </span><span class="kw">let </span><span class="kw-2">mut </span>manifest_files = <span class="macro">vec!</span>[added_manifest]; -<a href=#231 id=231 data-nosnippet>231</a> manifest_files.extend(existing_manifests); -<a href=#232 id=232 data-nosnippet>232</a> <span class="kw">let </span>manifest_files = manifest_process.process_manifests(manifest_files); -<a href=#233 id=233 data-nosnippet>233</a> <span class="prelude-val">Ok</span>(manifest_files) -<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="comment">// Returns a `Summary` of the current snapshot -<a href=#237 id=237 data-nosnippet>237</a> </span><span class="kw">fn </span>summary<OP: SnapshotProduceOperation>( -<a href=#238 id=238 data-nosnippet>238</a> <span class="kw-2">&</span><span class="self">self</span>, -<a href=#239 id=239 data-nosnippet>239</a> snapshot_produce_operation: <span class="kw-2">&</span>OP, -<a href=#240 id=240 data-nosnippet>240</a> ) -> <span class="prelude-ty">Result</span><Summary> { -<a href=#241 id=241 data-nosnippet>241</a> <span class="kw">let </span><span class="kw-2">mut </span>summary_collector = SnapshotSummaryCollector::default(); -<a href=#242 id=242 data-nosnippet>242</a> <span class="kw">let </span>table_metadata = <span class="self">self</span>.tx.current_table.metadata_ref(); -<a href=#243 id=243 data-nosnippet>243</a> -<a href=#244 id=244 data-nosnippet>244</a> <span class="kw">let </span>partition_summary_limit = <span class="kw">if let </span><span class="prelude-val">Some</span>(limit) = table_metadata -<a href=#245 id=245 data-nosnippet>245</a> .properties() -<a href=#246 id=246 data-nosnippet>246</a> .get(PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT) -<a href=#247 id=247 data-nosnippet>247</a> { -<a href=#248 id=248 data-nosnippet>248</a> <span class="kw">if let </span><span class="prelude-val">Ok</span>(limit) = limit.parse::<u64>() { -<a href=#249 id=249 data-nosnippet>249</a> limit -<a href=#250 id=250 data-nosnippet>250</a> } <span class="kw">else </span>{ -<a href=#251 id=251 data-nosnippet>251</a> PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT_DEFAULT -<a href=#252 id=252 data-nosnippet>252</a> } -<a href=#253 id=253 data-nosnippet>253</a> } <span class="kw">else </span>{ -<a href=#254 id=254 data-nosnippet>254</a> PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT_DEFAULT -<a href=#255 id=255 data-nosnippet>255</a> }; -<a href=#256 id=256 data-nosnippet>256</a> -<a href=#257 id=257 data-nosnippet>257</a> summary_collector.set_partition_summary_limit(partition_summary_limit); -<a href=#258 id=258 data-nosnippet>258</a> -<a href=#259 id=259 data-nosnippet>259</a> <span class="kw">for </span>data_file <span class="kw">in </span><span class="kw-2">&</span><span class="self">self</span>.added_data_files { -<a href=#260 id=260 data-nosnippet>260</a> summary_collector.add_file( -<a href=#261 id=261 data-nosnippet>261</a> data_file, -<a href=#262 id=262 data-nosnippet>262</a> table_metadata.current_schema().clone(), -<a href=#263 id=263 data-nosnippet>263</a> table_metadata.default_partition_spec().clone(), -<a href=#264 id=264 data-nosnippet>264</a> ); -<a href=#265 id=265 data-nosnippet>265</a> } -<a href=#266 id=266 data-nosnippet>266</a> -<a href=#267 id=267 data-nosnippet>267</a> <span class="kw">let </span>previous_snapshot = table_metadata -<a href=#268 id=268 data-nosnippet>268</a> .snapshot_by_id(<span class="self">self</span>.snapshot_id) -<a href=#269 id=269 data-nosnippet>269</a> .and_then(|snapshot| snapshot.parent_snapshot_id()) -<a href=#270 id=270 data-nosnippet>270</a> .and_then(|parent_id| table_metadata.snapshot_by_id(parent_id)); -<a href=#271 id=271 data-nosnippet>271</a> -<a href=#272 id=272 data-nosnippet>272</a> <span class="kw">let </span><span class="kw-2">mut </span>additional_properties = summary_collector.build(); -<a href=#273 id=273 data-nosnippet>273</a> additional_properties.extend(<span class="self">self</span>.snapshot_properties.clone()); -<a href=#274 id=274 data-nosnippet>274</a> -<a href=#275 id=275 data-nosnippet>275</a> <span class="kw">let </span>summary = Summary { -<a href=#276 id=276 data-nosnippet>276</a> operation: snapshot_produce_operation.operation(), -<a href=#277 id=277 data-nosnippet>277</a> additional_properties, -<a href=#278 id=278 data-nosnippet>278</a> }; -<a href=#279 id=279 data-nosnippet>279</a> -<a href=#280 id=280 data-nosnippet>280</a> update_snapshot_summaries( -<a href=#281 id=281 data-nosnippet>281</a> summary, -<a href=#282 id=282 data-nosnippet>282</a> previous_snapshot.map(|s| s.summary()), -<a href=#283 id=283 data-nosnippet>283</a> snapshot_produce_operation.operation() == Operation::Overwrite, -<a href=#284 id=284 data-nosnippet>284</a> ) -<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">fn </span>generate_manifest_list_file_path(<span class="kw-2">&</span><span class="self">self</span>, attempt: i64) -> String { -<a href=#288 id=288 data-nosnippet>288</a> <span class="macro">format!</span>( -<a href=#289 id=289 data-nosnippet>289</a> <span class="string">"{}/{}/snap-{}-{}-{}.{}"</span>, -<a href=#290 id=290 data-nosnippet>290</a> <span class="self">self</span>.tx.current_table.metadata().location(), -<a href=#291 id=291 data-nosnippet>291</a> META_ROOT_PATH, -<a href=#292 id=292 data-nosnippet>292</a> <span class="self">self</span>.snapshot_id, -<a href=#293 id=293 data-nosnippet>293</a> attempt, -<a href=#294 id=294 data-nosnippet>294</a> <span class="self">self</span>.commit_uuid, -<a href=#295 id=295 data-nosnippet>295</a> DataFileFormat::Avro -<a href=#296 id=296 data-nosnippet>296</a> ) -<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> <span class="doccomment">/// Finished building the action and apply it to the transaction. -<a href=#300 id=300 data-nosnippet>300</a> </span><span class="kw">pub async fn </span>apply<OP: SnapshotProduceOperation, MP: ManifestProcess>( -<a href=#301 id=301 data-nosnippet>301</a> <span class="kw-2">mut </span><span class="self">self</span>, -<a href=#302 id=302 data-nosnippet>302</a> snapshot_produce_operation: OP, -<a href=#303 id=303 data-nosnippet>303</a> process: MP, -<a href=#304 id=304 data-nosnippet>304</a> ) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { -<a href=#305 id=305 data-nosnippet>305</a> <span class="kw">let </span>new_manifests = <span class="self">self -<a href=#306 id=306 data-nosnippet>306</a> </span>.manifest_file(<span class="kw-2">&</span>snapshot_produce_operation, <span class="kw-2">&</span>process) -<a href=#307 id=307 data-nosnippet>307</a> .<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#308 id=308 data-nosnippet>308</a> <span class="kw">let </span>next_seq_num = <span class="self">self</span>.tx.current_table.metadata().next_sequence_number(); -<a href=#309 id=309 data-nosnippet>309</a> -<a href=#310 id=310 data-nosnippet>310</a> <span class="kw">let </span>summary = <span class="self">self -<a href=#311 id=311 data-nosnippet>311</a> </span>.summary(<span class="kw-2">&</span>snapshot_produce_operation) -<a href=#312 id=312 data-nosnippet>312</a> .map_err(|err| { -<a href=#313 id=313 data-nosnippet>313</a> Error::new(ErrorKind::Unexpected, <span class="string">"Failed to create snapshot summary."</span>) -<a href=#314 id=314 data-nosnippet>314</a> .with_source(err) -<a href=#315 id=315 data-nosnippet>315</a> }) -<a href=#316 id=316 data-nosnippet>316</a> .unwrap(); -<a href=#317 id=317 data-nosnippet>317</a> -<a href=#318 id=318 data-nosnippet>318</a> <span class="kw">let </span>manifest_list_path = <span class="self">self</span>.generate_manifest_list_file_path(<span class="number">0</span>); -<a href=#319 id=319 data-nosnippet>319</a> -<a href=#320 id=320 data-nosnippet>320</a> <span class="kw">let </span><span class="kw-2">mut </span>manifest_list_writer = <span class="kw">match </span><span class="self">self</span>.tx.current_table.metadata().format_version() { -<a href=#321 id=321 data-nosnippet>321</a> FormatVersion::V1 => ManifestListWriter::v1( -<a href=#322 id=322 data-nosnippet>322</a> <span class="self">self</span>.tx -<a href=#323 id=323 data-nosnippet>323</a> .current_table -<a href=#324 id=324 data-nosnippet>324</a> .file_io() -<a href=#325 id=325 data-nosnippet>325</a> .new_output(manifest_list_path.clone())<span class="question-mark">?</span>, -<a href=#326 id=326 data-nosnippet>326</a> <span class="self">self</span>.snapshot_id, -<a href=#327 id=327 data-nosnippet>327</a> <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), -<a href=#328 id=328 data-nosnippet>328</a> ), -<a href=#329 id=329 data-nosnippet>329</a> FormatVersion::V2 => ManifestListWriter::v2( -<a href=#330 id=330 data-nosnippet>330</a> <span class="self">self</span>.tx -<a href=#331 id=331 data-nosnippet>331</a> .current_table -<a href=#332 id=332 data-nosnippet>332</a> .file_io() -<a href=#333 id=333 data-nosnippet>333</a> .new_output(manifest_list_path.clone())<span class="question-mark">?</span>, -<a href=#334 id=334 data-nosnippet>334</a> <span class="self">self</span>.snapshot_id, -<a href=#335 id=335 data-nosnippet>335</a> <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), -<a href=#336 id=336 data-nosnippet>336</a> next_seq_num, +<a href=#147 id=147 data-nosnippet>147</a> <span class="comment">// Check if the data file partition spec id matches the table default partition spec id. +<a href=#148 id=148 data-nosnippet>148</a> </span><span class="kw">if </span><span class="self">self</span>.tx.current_table.metadata().default_partition_spec_id() +<a href=#149 id=149 data-nosnippet>149</a> != data_file.partition_spec_id +<a href=#150 id=150 data-nosnippet>150</a> { +<a href=#151 id=151 data-nosnippet>151</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( +<a href=#152 id=152 data-nosnippet>152</a> ErrorKind::DataInvalid, +<a href=#153 id=153 data-nosnippet>153</a> <span class="string">"Data file partition spec id does not match table default partition spec id"</span>, +<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> <span class="self">Self</span>::validate_partition_value( +<a href=#157 id=157 data-nosnippet>157</a> data_file.partition(), +<a href=#158 id=158 data-nosnippet>158</a> <span class="self">self</span>.tx.current_table.metadata().default_partition_type(), +<a href=#159 id=159 data-nosnippet>159</a> )<span class="question-mark">?</span>; +<a href=#160 id=160 data-nosnippet>160</a> } +<a href=#161 id=161 data-nosnippet>161</a> <span class="self">self</span>.added_data_files.extend(data_files); +<a href=#162 id=162 data-nosnippet>162</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>) +<a href=#163 id=163 data-nosnippet>163</a> } +<a href=#164 id=164 data-nosnippet>164</a> +<a href=#165 id=165 data-nosnippet>165</a> <span class="kw">fn </span>new_manifest_output(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><OutputFile> { +<a href=#166 id=166 data-nosnippet>166</a> <span class="kw">let </span>new_manifest_path = <span class="macro">format!</span>( +<a href=#167 id=167 data-nosnippet>167</a> <span class="string">"{}/{}/{}-m{}.{}"</span>, +<a href=#168 id=168 data-nosnippet>168</a> <span class="self">self</span>.tx.current_table.metadata().location(), +<a href=#169 id=169 data-nosnippet>169</a> META_ROOT_PATH, +<a href=#170 id=170 data-nosnippet>170</a> <span class="self">self</span>.commit_uuid, +<a href=#171 id=171 data-nosnippet>171</a> <span class="self">self</span>.manifest_counter.next().unwrap(), +<a href=#172 id=172 data-nosnippet>172</a> DataFileFormat::Avro +<a href=#173 id=173 data-nosnippet>173</a> ); +<a href=#174 id=174 data-nosnippet>174</a> <span class="self">self</span>.tx +<a href=#175 id=175 data-nosnippet>175</a> .current_table +<a href=#176 id=176 data-nosnippet>176</a> .file_io() +<a href=#177 id=177 data-nosnippet>177</a> .new_output(new_manifest_path) +<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> <span class="comment">// Write manifest file for added data files and return the ManifestFile for ManifestList. +<a href=#181 id=181 data-nosnippet>181</a> </span><span class="kw">async fn </span>write_added_manifest(<span class="kw-2">&mut </span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><ManifestFile> { +<a href=#182 id=182 data-nosnippet>182</a> <span class="kw">let </span>added_data_files = std::mem::take(<span class="kw-2">&mut </span><span class="self">self</span>.added_data_files); +<a href=#183 id=183 data-nosnippet>183</a> <span class="kw">if </span>added_data_files.is_empty() { +<a href=#184 id=184 data-nosnippet>184</a> <span class="kw">return </span><span class="prelude-val">Err</span>(Error::new( +<a href=#185 id=185 data-nosnippet>185</a> ErrorKind::PreconditionFailed, +<a href=#186 id=186 data-nosnippet>186</a> <span class="string">"No added data files found when write a manifest file"</span>, +<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> <span class="kw">let </span>snapshot_id = <span class="self">self</span>.snapshot_id; +<a href=#191 id=191 data-nosnippet>191</a> <span class="kw">let </span>format_version = <span class="self">self</span>.tx.current_table.metadata().format_version(); +<a href=#192 id=192 data-nosnippet>192</a> <span class="kw">let </span>manifest_entries = added_data_files.into_iter().map(|data_file| { +<a href=#193 id=193 data-nosnippet>193</a> <span class="kw">let </span>builder = ManifestEntry::builder() +<a href=#194 id=194 data-nosnippet>194</a> .status(<span class="kw">crate</span>::spec::ManifestStatus::Added) +<a href=#195 id=195 data-nosnippet>195</a> .data_file(data_file); +<a href=#196 id=196 data-nosnippet>196</a> <span class="kw">if </span>format_version == FormatVersion::V1 { +<a href=#197 id=197 data-nosnippet>197</a> builder.snapshot_id(snapshot_id).build() +<a href=#198 id=198 data-nosnippet>198</a> } <span class="kw">else </span>{ +<a href=#199 id=199 data-nosnippet>199</a> <span class="comment">// For format version > 1, we set the snapshot id at the inherited time to avoid rewrite the manifest file when +<a href=#200 id=200 data-nosnippet>200</a> // commit failed. +<a href=#201 id=201 data-nosnippet>201</a> </span>builder.build() +<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">let </span><span class="kw-2">mut </span>writer = { +<a href=#205 id=205 data-nosnippet>205</a> <span class="kw">let </span>builder = ManifestWriterBuilder::new( +<a href=#206 id=206 data-nosnippet>206</a> <span class="self">self</span>.new_manifest_output()<span class="question-mark">?</span>, +<a href=#207 id=207 data-nosnippet>207</a> <span class="prelude-val">Some</span>(<span class="self">self</span>.snapshot_id), +<a href=#208 id=208 data-nosnippet>208</a> <span class="self">self</span>.key_metadata.clone(), +<a href=#209 id=209 data-nosnippet>209</a> <span class="self">self</span>.tx.current_table.metadata().current_schema().clone(), +<a href=#210 id=210 data-nosnippet>210</a> <span class="self">self</span>.tx +<a href=#211 id=211 data-nosnippet>211</a> .current_table +<a href=#212 id=212 data-nosnippet>212</a> .metadata() +<a href=#213 id=213 data-nosnippet>213</a> .default_partition_spec() +<a href=#214 id=214 data-nosnippet>214</a> .as_ref() +<a href=#215 id=215 data-nosnippet>215</a> .clone(), +<a href=#216 id=216 data-nosnippet>216</a> ); +<a href=#217 id=217 data-nosnippet>217</a> <span class="kw">if </span><span class="self">self</span>.tx.current_table.metadata().format_version() == FormatVersion::V1 { +<a href=#218 id=218 data-nosnippet>218</a> builder.build_v1() +<a href=#219 id=219 data-nosnippet>219</a> } <span class="kw">else </span>{ +<a href=#220 id=220 data-nosnippet>220</a> builder.build_v2_data() +<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">for </span>entry <span class="kw">in </span>manifest_entries { +<a href=#224 id=224 data-nosnippet>224</a> writer.add_entry(entry)<span class="question-mark">?</span>; +<a href=#225 id=225 data-nosnippet>225</a> } +<a href=#226 id=226 data-nosnippet>226</a> writer.write_manifest_file().<span class="kw">await +<a href=#227 id=227 data-nosnippet>227</a> </span>} +<a href=#228 id=228 data-nosnippet>228</a> +<a href=#229 id=229 data-nosnippet>229</a> <span class="kw">async fn </span>manifest_file<OP: SnapshotProduceOperation, MP: ManifestProcess>( +<a href=#230 id=230 data-nosnippet>230</a> <span class="kw-2">&mut </span><span class="self">self</span>, +<a href=#231 id=231 data-nosnippet>231</a> snapshot_produce_operation: <span class="kw-2">&</span>OP, +<a href=#232 id=232 data-nosnippet>232</a> manifest_process: <span class="kw-2">&</span>MP, +<a href=#233 id=233 data-nosnippet>233</a> ) -> <span class="prelude-ty">Result</span><Vec<ManifestFile>> { +<a href=#234 id=234 data-nosnippet>234</a> <span class="kw">let </span>added_manifest = <span class="self">self</span>.write_added_manifest().<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#235 id=235 data-nosnippet>235</a> <span class="kw">let </span>existing_manifests = snapshot_produce_operation.existing_manifest(<span class="self">self</span>).<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#236 id=236 data-nosnippet>236</a> <span class="comment">// # TODO +<a href=#237 id=237 data-nosnippet>237</a> // Support process delete entries. +<a href=#238 id=238 data-nosnippet>238</a> +<a href=#239 id=239 data-nosnippet>239</a> </span><span class="kw">let </span><span class="kw-2">mut </span>manifest_files = <span class="macro">vec!</span>[added_manifest]; +<a href=#240 id=240 data-nosnippet>240</a> manifest_files.extend(existing_manifests); +<a href=#241 id=241 data-nosnippet>241</a> <span class="kw">let </span>manifest_files = manifest_process.process_manifests(manifest_files); +<a href=#242 id=242 data-nosnippet>242</a> <span class="prelude-val">Ok</span>(manifest_files) +<a href=#243 id=243 data-nosnippet>243</a> } +<a href=#244 id=244 data-nosnippet>244</a> +<a href=#245 id=245 data-nosnippet>245</a> <span class="comment">// Returns a `Summary` of the current snapshot +<a href=#246 id=246 data-nosnippet>246</a> </span><span class="kw">fn </span>summary<OP: SnapshotProduceOperation>( +<a href=#247 id=247 data-nosnippet>247</a> <span class="kw-2">&</span><span class="self">self</span>, +<a href=#248 id=248 data-nosnippet>248</a> snapshot_produce_operation: <span class="kw-2">&</span>OP, +<a href=#249 id=249 data-nosnippet>249</a> ) -> <span class="prelude-ty">Result</span><Summary> { +<a href=#250 id=250 data-nosnippet>250</a> <span class="kw">let </span><span class="kw-2">mut </span>summary_collector = SnapshotSummaryCollector::default(); +<a href=#251 id=251 data-nosnippet>251</a> <span class="kw">let </span>table_metadata = <span class="self">self</span>.tx.current_table.metadata_ref(); +<a href=#252 id=252 data-nosnippet>252</a> +<a href=#253 id=253 data-nosnippet>253</a> <span class="kw">let </span>partition_summary_limit = <span class="kw">if let </span><span class="prelude-val">Some</span>(limit) = table_metadata +<a href=#254 id=254 data-nosnippet>254</a> .properties() +<a href=#255 id=255 data-nosnippet>255</a> .get(PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT) +<a href=#256 id=256 data-nosnippet>256</a> { +<a href=#257 id=257 data-nosnippet>257</a> <span class="kw">if let </span><span class="prelude-val">Ok</span>(limit) = limit.parse::<u64>() { +<a href=#258 id=258 data-nosnippet>258</a> limit +<a href=#259 id=259 data-nosnippet>259</a> } <span class="kw">else </span>{ +<a href=#260 id=260 data-nosnippet>260</a> PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT_DEFAULT +<a href=#261 id=261 data-nosnippet>261</a> } +<a href=#262 id=262 data-nosnippet>262</a> } <span class="kw">else </span>{ +<a href=#263 id=263 data-nosnippet>263</a> PROPERTY_WRITE_PARTITION_SUMMARY_LIMIT_DEFAULT +<a href=#264 id=264 data-nosnippet>264</a> }; +<a href=#265 id=265 data-nosnippet>265</a> +<a href=#266 id=266 data-nosnippet>266</a> summary_collector.set_partition_summary_limit(partition_summary_limit); +<a href=#267 id=267 data-nosnippet>267</a> +<a href=#268 id=268 data-nosnippet>268</a> <span class="kw">for </span>data_file <span class="kw">in </span><span class="kw-2">&</span><span class="self">self</span>.added_data_files { +<a href=#269 id=269 data-nosnippet>269</a> summary_collector.add_file( +<a href=#270 id=270 data-nosnippet>270</a> data_file, +<a href=#271 id=271 data-nosnippet>271</a> table_metadata.current_schema().clone(), +<a href=#272 id=272 data-nosnippet>272</a> table_metadata.default_partition_spec().clone(), +<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="kw">let </span>previous_snapshot = table_metadata +<a href=#277 id=277 data-nosnippet>277</a> .snapshot_by_id(<span class="self">self</span>.snapshot_id) +<a href=#278 id=278 data-nosnippet>278</a> .and_then(|snapshot| snapshot.parent_snapshot_id()) +<a href=#279 id=279 data-nosnippet>279</a> .and_then(|parent_id| table_metadata.snapshot_by_id(parent_id)); +<a href=#280 id=280 data-nosnippet>280</a> +<a href=#281 id=281 data-nosnippet>281</a> <span class="kw">let </span><span class="kw-2">mut </span>additional_properties = summary_collector.build(); +<a href=#282 id=282 data-nosnippet>282</a> additional_properties.extend(<span class="self">self</span>.snapshot_properties.clone()); +<a href=#283 id=283 data-nosnippet>283</a> +<a href=#284 id=284 data-nosnippet>284</a> <span class="kw">let </span>summary = Summary { +<a href=#285 id=285 data-nosnippet>285</a> operation: snapshot_produce_operation.operation(), +<a href=#286 id=286 data-nosnippet>286</a> additional_properties, +<a href=#287 id=287 data-nosnippet>287</a> }; +<a href=#288 id=288 data-nosnippet>288</a> +<a href=#289 id=289 data-nosnippet>289</a> update_snapshot_summaries( +<a href=#290 id=290 data-nosnippet>290</a> summary, +<a href=#291 id=291 data-nosnippet>291</a> previous_snapshot.map(|s| s.summary()), +<a href=#292 id=292 data-nosnippet>292</a> snapshot_produce_operation.operation() == Operation::Overwrite, +<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>generate_manifest_list_file_path(<span class="kw-2">&</span><span class="self">self</span>, attempt: i64) -> String { +<a href=#297 id=297 data-nosnippet>297</a> <span class="macro">format!</span>( +<a href=#298 id=298 data-nosnippet>298</a> <span class="string">"{}/{}/snap-{}-{}-{}.{}"</span>, +<a href=#299 id=299 data-nosnippet>299</a> <span class="self">self</span>.tx.current_table.metadata().location(), +<a href=#300 id=300 data-nosnippet>300</a> META_ROOT_PATH, +<a href=#301 id=301 data-nosnippet>301</a> <span class="self">self</span>.snapshot_id, +<a href=#302 id=302 data-nosnippet>302</a> attempt, +<a href=#303 id=303 data-nosnippet>303</a> <span class="self">self</span>.commit_uuid, +<a href=#304 id=304 data-nosnippet>304</a> DataFileFormat::Avro +<a href=#305 id=305 data-nosnippet>305</a> ) +<a href=#306 id=306 data-nosnippet>306</a> } +<a href=#307 id=307 data-nosnippet>307</a> +<a href=#308 id=308 data-nosnippet>308</a> <span class="doccomment">/// Finished building the action and apply it to the transaction. +<a href=#309 id=309 data-nosnippet>309</a> </span><span class="kw">pub async fn </span>apply<OP: SnapshotProduceOperation, MP: ManifestProcess>( +<a href=#310 id=310 data-nosnippet>310</a> <span class="kw-2">mut </span><span class="self">self</span>, +<a href=#311 id=311 data-nosnippet>311</a> snapshot_produce_operation: OP, +<a href=#312 id=312 data-nosnippet>312</a> process: MP, +<a href=#313 id=313 data-nosnippet>313</a> ) -> <span class="prelude-ty">Result</span><Transaction<<span class="lifetime">'a</span>>> { +<a href=#314 id=314 data-nosnippet>314</a> <span class="kw">let </span>new_manifests = <span class="self">self +<a href=#315 id=315 data-nosnippet>315</a> </span>.manifest_file(<span class="kw-2">&</span>snapshot_produce_operation, <span class="kw-2">&</span>process) +<a href=#316 id=316 data-nosnippet>316</a> .<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#317 id=317 data-nosnippet>317</a> <span class="kw">let </span>next_seq_num = <span class="self">self</span>.tx.current_table.metadata().next_sequence_number(); +<a href=#318 id=318 data-nosnippet>318</a> +<a href=#319 id=319 data-nosnippet>319</a> <span class="kw">let </span>summary = <span class="self">self +<a href=#320 id=320 data-nosnippet>320</a> </span>.summary(<span class="kw-2">&</span>snapshot_produce_operation) +<a href=#321 id=321 data-nosnippet>321</a> .map_err(|err| { +<a href=#322 id=322 data-nosnippet>322</a> Error::new(ErrorKind::Unexpected, <span class="string">"Failed to create snapshot summary."</span>) +<a href=#323 id=323 data-nosnippet>323</a> .with_source(err) +<a href=#324 id=324 data-nosnippet>324</a> }) +<a href=#325 id=325 data-nosnippet>325</a> .unwrap(); +<a href=#326 id=326 data-nosnippet>326</a> +<a href=#327 id=327 data-nosnippet>327</a> <span class="kw">let </span>manifest_list_path = <span class="self">self</span>.generate_manifest_list_file_path(<span class="number">0</span>); +<a href=#328 id=328 data-nosnippet>328</a> +<a href=#329 id=329 data-nosnippet>329</a> <span class="kw">let </span><span class="kw-2">mut </span>manifest_list_writer = <span class="kw">match </span><span class="self">self</span>.tx.current_table.metadata().format_version() { +<a href=#330 id=330 data-nosnippet>330</a> FormatVersion::V1 => ManifestListWriter::v1( +<a href=#331 id=331 data-nosnippet>331</a> <span class="self">self</span>.tx +<a href=#332 id=332 data-nosnippet>332</a> .current_table +<a href=#333 id=333 data-nosnippet>333</a> .file_io() +<a href=#334 id=334 data-nosnippet>334</a> .new_output(manifest_list_path.clone())<span class="question-mark">?</span>, +<a href=#335 id=335 data-nosnippet>335</a> <span class="self">self</span>.snapshot_id, +<a href=#336 id=336 data-nosnippet>336</a> <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), <a href=#337 id=337 data-nosnippet>337</a> ), -<a href=#338 id=338 data-nosnippet>338</a> }; -<a href=#339 id=339 data-nosnippet>339</a> manifest_list_writer.add_manifests(new_manifests.into_iter())<span class="question-mark">?</span>; -<a href=#340 id=340 data-nosnippet>340</a> manifest_list_writer.close().<span class="kw">await</span><span class="question-mark">?</span>; -<a href=#341 id=341 data-nosnippet>341</a> -<a href=#342 id=342 data-nosnippet>342</a> <span class="kw">let </span>commit_ts = chrono::Utc::now().timestamp_millis(); -<a href=#343 id=343 data-nosnippet>343</a> <span class="kw">let </span>new_snapshot = Snapshot::builder() -<a href=#344 id=344 data-nosnippet>344</a> .with_manifest_list(manifest_list_path) -<a href=#345 id=345 data-nosnippet>345</a> .with_snapshot_id(<span class="self">self</span>.snapshot_id) -<a href=#346 id=346 data-nosnippet>346</a> .with_parent_snapshot_id(<span class="self">self</span>.tx.current_table.metadata().current_snapshot_id()) -<a href=#347 id=347 data-nosnippet>347</a> .with_sequence_number(next_seq_num) -<a href=#348 id=348 data-nosnippet>348</a> .with_summary(summary) -<a href=#349 id=349 data-nosnippet>349</a> .with_schema_id(<span class="self">self</span>.tx.current_table.metadata().current_schema_id()) -<a href=#350 id=350 data-nosnippet>350</a> .with_timestamp_ms(commit_ts) -<a href=#351 id=351 data-nosnippet>351</a> .build(); -<a href=#352 id=352 data-nosnippet>352</a> -<a href=#353 id=353 data-nosnippet>353</a> <span class="self">self</span>.tx.apply( -<a href=#354 id=354 data-nosnippet>354</a> <span class="macro">vec!</span>[ -<a href=#355 id=355 data-nosnippet>355</a> TableUpdate::AddSnapshot { -<a href=#356 id=356 data-nosnippet>356</a> snapshot: new_snapshot, -<a href=#357 id=357 data-nosnippet>357</a> }, -<a href=#358 id=358 data-nosnippet>358</a> TableUpdate::SetSnapshotRef { -<a href=#359 id=359 data-nosnippet>359</a> ref_name: MAIN_BRANCH.to_string(), -<a href=#360 id=360 data-nosnippet>360</a> reference: SnapshotReference::new( -<a href=#361 id=361 data-nosnippet>361</a> <span class="self">self</span>.snapshot_id, -<a href=#362 id=362 data-nosnippet>362</a> SnapshotRetention::branch(<span class="prelude-val">None</span>, <span class="prelude-val">None</span>, <span class="prelude-val">None</span>), -<a href=#363 id=363 data-nosnippet>363</a> ), -<a href=#364 id=364 data-nosnippet>364</a> }, -<a href=#365 id=365 data-nosnippet>365</a> ], -<a href=#366 id=366 data-nosnippet>366</a> <span class="macro">vec!</span>[ -<a href=#367 id=367 data-nosnippet>367</a> TableRequirement::UuidMatch { -<a href=#368 id=368 data-nosnippet>368</a> uuid: <span class="self">self</span>.tx.current_table.metadata().uuid(), -<a href=#369 id=369 data-nosnippet>369</a> }, -<a href=#370 id=370 data-nosnippet>370</a> TableRequirement::RefSnapshotIdMatch { -<a href=#371 id=371 data-nosnippet>371</a> r#ref: MAIN_BRANCH.to_string(), -<a href=#372 id=372 data-nosnippet>372</a> snapshot_id: <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), +<a href=#338 id=338 data-nosnippet>338</a> FormatVersion::V2 => ManifestListWriter::v2( +<a href=#339 id=339 data-nosnippet>339</a> <span class="self">self</span>.tx +<a href=#340 id=340 data-nosnippet>340</a> .current_table +<a href=#341 id=341 data-nosnippet>341</a> .file_io() +<a href=#342 id=342 data-nosnippet>342</a> .new_output(manifest_list_path.clone())<span class="question-mark">?</span>, +<a href=#343 id=343 data-nosnippet>343</a> <span class="self">self</span>.snapshot_id, +<a href=#344 id=344 data-nosnippet>344</a> <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), +<a href=#345 id=345 data-nosnippet>345</a> next_seq_num, +<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> manifest_list_writer.add_manifests(new_manifests.into_iter())<span class="question-mark">?</span>; +<a href=#349 id=349 data-nosnippet>349</a> manifest_list_writer.close().<span class="kw">await</span><span class="question-mark">?</span>; +<a href=#350 id=350 data-nosnippet>350</a> +<a href=#351 id=351 data-nosnippet>351</a> <span class="kw">let </span>commit_ts = chrono::Utc::now().timestamp_millis(); +<a href=#352 id=352 data-nosnippet>352</a> <span class="kw">let </span>new_snapshot = Snapshot::builder() +<a href=#353 id=353 data-nosnippet>353</a> .with_manifest_list(manifest_list_path) +<a href=#354 id=354 data-nosnippet>354</a> .with_snapshot_id(<span class="self">self</span>.snapshot_id) +<a href=#355 id=355 data-nosnippet>355</a> .with_parent_snapshot_id(<span class="self">self</span>.tx.current_table.metadata().current_snapshot_id()) +<a href=#356 id=356 data-nosnippet>356</a> .with_sequence_number(next_seq_num) +<a href=#357 id=357 data-nosnippet>357</a> .with_summary(summary) +<a href=#358 id=358 data-nosnippet>358</a> .with_schema_id(<span class="self">self</span>.tx.current_table.metadata().current_schema_id()) +<a href=#359 id=359 data-nosnippet>359</a> .with_timestamp_ms(commit_ts) +<a href=#360 id=360 data-nosnippet>360</a> .build(); +<a href=#361 id=361 data-nosnippet>361</a> +<a href=#362 id=362 data-nosnippet>362</a> <span class="self">self</span>.tx.apply( +<a href=#363 id=363 data-nosnippet>363</a> <span class="macro">vec!</span>[ +<a href=#364 id=364 data-nosnippet>364</a> TableUpdate::AddSnapshot { +<a href=#365 id=365 data-nosnippet>365</a> snapshot: new_snapshot, +<a href=#366 id=366 data-nosnippet>366</a> }, +<a href=#367 id=367 data-nosnippet>367</a> TableUpdate::SetSnapshotRef { +<a href=#368 id=368 data-nosnippet>368</a> ref_name: MAIN_BRANCH.to_string(), +<a href=#369 id=369 data-nosnippet>369</a> reference: SnapshotReference::new( +<a href=#370 id=370 data-nosnippet>370</a> <span class="self">self</span>.snapshot_id, +<a href=#371 id=371 data-nosnippet>371</a> SnapshotRetention::branch(<span class="prelude-val">None</span>, <span class="prelude-val">None</span>, <span class="prelude-val">None</span>), +<a href=#372 id=372 data-nosnippet>372</a> ), <a href=#373 id=373 data-nosnippet>373</a> }, <a href=#374 id=374 data-nosnippet>374</a> ], -<a href=#375 id=375 data-nosnippet>375</a> )<span class="question-mark">?</span>; -<a href=#376 id=376 data-nosnippet>376</a> -<a href=#377 id=377 data-nosnippet>377</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>.tx) -<a href=#378 id=378 data-nosnippet>378</a> } -<a href=#379 id=379 data-nosnippet>379</a>}</code></pre></div></section></main></body></html> \ No newline at end of file +<a href=#375 id=375 data-nosnippet>375</a> <span class="macro">vec!</span>[ +<a href=#376 id=376 data-nosnippet>376</a> TableRequirement::UuidMatch { +<a href=#377 id=377 data-nosnippet>377</a> uuid: <span class="self">self</span>.tx.current_table.metadata().uuid(), +<a href=#378 id=378 data-nosnippet>378</a> }, +<a href=#379 id=379 data-nosnippet>379</a> TableRequirement::RefSnapshotIdMatch { +<a href=#380 id=380 data-nosnippet>380</a> r#ref: MAIN_BRANCH.to_string(), +<a href=#381 id=381 data-nosnippet>381</a> snapshot_id: <span class="self">self</span>.tx.current_table.metadata().current_snapshot_id(), +<a href=#382 id=382 data-nosnippet>382</a> }, +<a href=#383 id=383 data-nosnippet>383</a> ], +<a href=#384 id=384 data-nosnippet>384</a> )<span class="question-mark">?</span>; +<a href=#385 id=385 data-nosnippet>385</a> +<a href=#386 id=386 data-nosnippet>386</a> <span class="prelude-val">Ok</span>(<span class="self">self</span>.tx) +<a href=#387 id=387 data-nosnippet>387</a> } +<a href=#388 id=388 data-nosnippet>388</a>}</code></pre></div></section></main></body></html> \ No newline at end of file