Skip to content

Commit 0a7738c

Browse files
committed
feat: Makes error macros hygienic
Use an absolute path for DatafusionError Allows removing a bunch of imports
1 parent 70edcc1 commit 0a7738c

File tree

78 files changed

+89
-103
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

78 files changed

+89
-103
lines changed

datafusion/catalog-listing/src/helpers.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ use std::sync::Arc;
2222

2323
use datafusion_catalog::Session;
2424
use datafusion_common::{
25-
assert_or_internal_err, DataFusionError, HashMap, Result, ScalarValue,
25+
assert_or_internal_err, HashMap, Result, ScalarValue,
2626
};
2727
use datafusion_datasource::ListingTableUrl;
2828
use datafusion_datasource::PartitionedFile;

datafusion/common/src/error.rs

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,7 @@ impl DataFusionErrorBuilder {
768768
macro_rules! unwrap_or_internal_err {
769769
($Value: ident) => {
770770
$Value.ok_or_else(|| {
771-
DataFusionError::Internal(format!(
771+
$crate::DataFusionError::Internal(format!(
772772
"{} should not be None",
773773
stringify!($Value)
774774
))
@@ -789,15 +789,15 @@ macro_rules! unwrap_or_internal_err {
789789
macro_rules! assert_or_internal_err {
790790
($cond:expr) => {
791791
if !$cond {
792-
return Err(DataFusionError::Internal(format!(
792+
return Err($crate::DataFusionError::Internal(format!(
793793
"Assertion failed: {}",
794794
stringify!($cond)
795795
)));
796796
}
797797
};
798798
($cond:expr, $($arg:tt)+) => {
799799
if !$cond {
800-
return Err(DataFusionError::Internal(format!(
800+
return Err($crate::DataFusionError::Internal(format!(
801801
"Assertion failed: {}: {}",
802802
stringify!($cond),
803803
format!($($arg)+)
@@ -821,7 +821,7 @@ macro_rules! assert_eq_or_internal_err {
821821
let left_val = &$left;
822822
let right_val = &$right;
823823
if left_val != right_val {
824-
return Err(DataFusionError::Internal(format!(
824+
return Err($crate::DataFusionError::Internal(format!(
825825
"Assertion failed: {} == {} (left: {:?}, right: {:?})",
826826
stringify!($left),
827827
stringify!($right),
@@ -834,7 +834,7 @@ macro_rules! assert_eq_or_internal_err {
834834
let left_val = &$left;
835835
let right_val = &$right;
836836
if left_val != right_val {
837-
return Err(DataFusionError::Internal(format!(
837+
return Err($crate::DataFusionError::Internal(format!(
838838
"Assertion failed: {} == {} (left: {:?}, right: {:?}): {}",
839839
stringify!($left),
840840
stringify!($right),
@@ -861,7 +861,7 @@ macro_rules! assert_ne_or_internal_err {
861861
let left_val = &$left;
862862
let right_val = &$right;
863863
if left_val == right_val {
864-
return Err(DataFusionError::Internal(format!(
864+
return Err($crate::DataFusionError::Internal(format!(
865865
"Assertion failed: {} != {} (left: {:?}, right: {:?})",
866866
stringify!($left),
867867
stringify!($right),
@@ -874,7 +874,7 @@ macro_rules! assert_ne_or_internal_err {
874874
let left_val = &$left;
875875
let right_val = &$right;
876876
if left_val == right_val {
877-
return Err(DataFusionError::Internal(format!(
877+
return Err($crate::DataFusionError::Internal(format!(
878878
"Assertion failed: {} != {} (left: {:?}, right: {:?}): {}",
879879
stringify!($left),
880880
stringify!($right),
@@ -973,7 +973,7 @@ make_error!(resources_err, resources_datafusion_err, ResourcesExhausted);
973973
#[macro_export]
974974
macro_rules! sql_datafusion_err {
975975
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {{
976-
let err = DataFusionError::SQL(Box::new($ERR), Some(DataFusionError::get_back_trace()));
976+
let err = $crate::DataFusionError::SQL(Box::new($ERR), Some($crate::DataFusionError::get_back_trace()));
977977
$(
978978
let err = err.with_diagnostic($DIAG);
979979
)?
@@ -985,7 +985,7 @@ macro_rules! sql_datafusion_err {
985985
#[macro_export]
986986
macro_rules! sql_err {
987987
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {{
988-
let err = datafusion_common::sql_datafusion_err!($ERR);
988+
let err = $crate::sql_datafusion_err!($ERR);
989989
$(
990990
let err = err.with_diagnostic($DIAG);
991991
)?
@@ -997,7 +997,7 @@ macro_rules! sql_err {
997997
#[macro_export]
998998
macro_rules! arrow_datafusion_err {
999999
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {{
1000-
let err = DataFusionError::ArrowError(Box::new($ERR), Some(DataFusionError::get_back_trace()));
1000+
let err = $crate::DataFusionError::ArrowError(Box::new($ERR), Some($crate::DataFusionError::get_back_trace()));
10011001
$(
10021002
let err = err.with_diagnostic($DIAG);
10031003
)?
@@ -1010,7 +1010,7 @@ macro_rules! arrow_datafusion_err {
10101010
macro_rules! arrow_err {
10111011
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {
10121012
{
1013-
let err = datafusion_common::arrow_datafusion_err!($ERR);
1013+
let err = $crate::arrow_datafusion_err!($ERR);
10141014
$(
10151015
let err = err.with_diagnostic($DIAG);
10161016
)?
@@ -1022,9 +1022,9 @@ macro_rules! arrow_err {
10221022
#[macro_export]
10231023
macro_rules! schema_datafusion_err {
10241024
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {{
1025-
let err = $crate::error::DataFusionError::SchemaError(
1025+
let err = $crate::DataFusionError::SchemaError(
10261026
Box::new($ERR),
1027-
Box::new(Some($crate::error::DataFusionError::get_back_trace())),
1027+
Box::new(Some($crate::DataFusionError::get_back_trace())),
10281028
);
10291029
$(
10301030
let err = err.with_diagnostic($DIAG);
@@ -1037,9 +1037,9 @@ macro_rules! schema_datafusion_err {
10371037
#[macro_export]
10381038
macro_rules! schema_err {
10391039
($ERR:expr $(; diagnostic = $DIAG:expr)?) => {{
1040-
let err = $crate::error::DataFusionError::SchemaError(
1040+
let err = $crate::DataFusionError::SchemaError(
10411041
Box::new($ERR),
1042-
Box::new(Some($crate::error::DataFusionError::get_back_trace())),
1042+
Box::new(Some($crate::DataFusionError::get_back_trace())),
10431043
);
10441044
$(
10451045
let err = err.with_diagnostic($DIAG);

datafusion/common/src/utils/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ pub mod string_utils;
2424

2525
use crate::assert_or_internal_err;
2626
use crate::error::{_exec_datafusion_err, _internal_datafusion_err};
27-
use crate::{DataFusionError, Result, ScalarValue};
27+
use crate::{Result, ScalarValue};
2828
use arrow::array::{
2929
cast::AsArray, Array, ArrayRef, FixedSizeListArray, LargeListArray, ListArray,
3030
OffsetSizeTrait,

datafusion/core/src/datasource/listing_table_factory.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ use crate::datasource::listing::{
2828
use crate::execution::context::SessionState;
2929

3030
use arrow::datatypes::DataType;
31-
use datafusion_common::{arrow_datafusion_err, plan_err, DataFusionError, ToDFSchema};
31+
use datafusion_common::{arrow_datafusion_err, plan_err, ToDFSchema};
3232
use datafusion_common::{config_datafusion_err, Result};
3333
use datafusion_expr::CreateExternalTable;
3434

datafusion/datasource-avro/src/avro_to_arrow/arrow_array_reader.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ use arrow::error::Result as ArrowResult;
4646
use arrow::record_batch::RecordBatch;
4747
use arrow::util::bit_util;
4848
use datafusion_common::arrow_err;
49-
use datafusion_common::error::{DataFusionError, Result};
49+
use datafusion_common::error::Result;
5050
use num_traits::NumCast;
5151
use std::collections::BTreeMap;
5252
use std::io::Read;

datafusion/datasource-parquet/src/access_plan.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
// specific language governing permissions and limitations
1616
// under the License.
1717

18-
use datafusion_common::{assert_eq_or_internal_err, DataFusionError, Result};
18+
use datafusion_common::{assert_eq_or_internal_err, Result};
1919
use parquet::arrow::arrow_reader::{RowSelection, RowSelector};
2020
use parquet::file::metadata::RowGroupMetaData;
2121

datafusion/datasource/src/memory.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ use crate::source::{DataSource, DataSourceExec};
3030
use arrow::array::{RecordBatch, RecordBatchOptions};
3131
use arrow::datatypes::{Schema, SchemaRef};
3232
use datafusion_common::{
33-
assert_or_internal_err, plan_err, project_schema, DataFusionError, Result,
33+
assert_or_internal_err, plan_err, project_schema, Result,
3434
ScalarValue,
3535
};
3636
use datafusion_execution::TaskContext;

datafusion/datasource/src/sink.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ use std::sync::Arc;
2424

2525
use arrow::array::{ArrayRef, RecordBatch, UInt64Array};
2626
use arrow::datatypes::{DataType, Field, Schema, SchemaRef};
27-
use datafusion_common::{assert_eq_or_internal_err, DataFusionError, Result};
27+
use datafusion_common::{assert_eq_or_internal_err, Result};
2828
use datafusion_execution::TaskContext;
2929
use datafusion_physical_expr::{Distribution, EquivalenceProperties};
3030
use datafusion_physical_expr_common::sort_expr::{LexRequirement, OrderingRequirements};

datafusion/expr-common/src/statistics.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ use arrow::datatypes::DataType;
2626
use datafusion_common::rounding::alter_fp_rounding_mode;
2727
use datafusion_common::{
2828
assert_eq_or_internal_err, assert_ne_or_internal_err, assert_or_internal_err,
29-
internal_err, not_impl_err, DataFusionError, Result, ScalarValue,
29+
internal_err, not_impl_err, Result, ScalarValue,
3030
};
3131

3232
/// This object defines probabilistic distributions that encode uncertain

datafusion/expr/src/logical_plan/invariants.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
use datafusion_common::{
1919
assert_or_internal_err, plan_err,
2020
tree_node::{TreeNode, TreeNodeRecursion},
21-
DFSchemaRef, DataFusionError, Result,
21+
DFSchemaRef, Result,
2222
};
2323

2424
use crate::{

0 commit comments

Comments
 (0)