include only one variant in compiled code (#2438)

litentry · Jan 29, 2024 · 7e9b430 · 7e9b430
1 parent 7503916
commit 7e9b430
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/primitives/core/macros/src/lib.rs b/primitives/core/macros/src/lib.rs
@@ -17,19 +17,23 @@
 
 #[macro_export]
 macro_rules! if_production_or {
-	($prod_variant:expr, $non_prod_variant:expr) => {
-		if cfg!(feature = "production") {
-			$prod_variant
-		} else {
+	($prod_variant:expr, $non_prod_variant:expr) => {{
+		#[cfg(not(feature = "production"))]
+		{
 			$non_prod_variant
 		}
-	};
+		#[cfg(feature = "production")]
+		{
+			$prod_variant
+		}
+	}};
 }
 
 #[macro_export]
 macro_rules! if_not_production {
 	($expression:expr) => {
-		if cfg!(not(feature = "production")) {
+		#[cfg(not(feature = "production"))]
+		{
 			$expression
 		}
 	};