frame_support_procedural/
transactional.rs

1// This file is part of Substrate.
2
3// Copyright (C) Parity Technologies (UK) Ltd.
4// SPDX-License-Identifier: Apache-2.0
5
6// Licensed under the Apache License, Version 2.0 (the "License");
7// you may not use this file except in compliance with the License.
8// You may obtain a copy of the License at
9//
10// 	http://www.apache.org/licenses/LICENSE-2.0
11//
12// Unless required by applicable law or agreed to in writing, software
13// distributed under the License is distributed on an "AS IS" BASIS,
14// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15// See the License for the specific language governing permissions and
16// limitations under the License.
17
18use frame_support_procedural_tools::generate_access_from_frame_or_crate;
19use proc_macro::TokenStream;
20use quote::quote;
21use syn::{ItemFn, Result};
22
23pub fn transactional(_attr: TokenStream, input: TokenStream) -> Result<TokenStream> {
24	let ItemFn { attrs, vis, sig, block } = syn::parse(input)?;
25
26	let crate_ = generate_access_from_frame_or_crate("frame-support")?;
27	let output = quote! {
28		#(#attrs)*
29		#vis #sig {
30			use #crate_::storage::{with_transaction, TransactionOutcome};
31			with_transaction(|| {
32				let r = (|| { #block })();
33				if r.is_ok() {
34					TransactionOutcome::Commit(r)
35				} else {
36					TransactionOutcome::Rollback(r)
37				}
38			})
39		}
40	};
41
42	Ok(output.into())
43}
44
45pub fn require_transactional(_attr: TokenStream, input: TokenStream) -> Result<TokenStream> {
46	let ItemFn { attrs, vis, sig, block } = syn::parse(input)?;
47
48	let crate_ = generate_access_from_frame_or_crate("frame-support")?;
49	let output = quote! {
50		#(#attrs)*
51		#vis #sig {
52			if !#crate_::storage::transactional::is_transactional() {
53				return Err(#crate_::sp_runtime::TransactionalError::NoLayer.into());
54			}
55			#block
56		}
57	};
58
59	Ok(output.into())
60}
frame_support_procedural/transactional.rs

frame_support_procedural/
transactional.rs
