check.h 1.03 KB
Newer Older
PanZezhong's avatar
PanZezhong committed
1
2
#ifndef INFINIUTILS_CHECK_H
#define INFINIUTILS_CHECK_H
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include <iostream>

#define CHECK_API_OR(API, EXPECT, ACTION)                                       \
    do {                                                                        \
        auto api_result_ = (API);                                               \
        if (api_result_ != (EXPECT)) {                                          \
            std::cerr << "Error Code " << api_result_ << " in `" << #API << "`" \
                      << " from " << __func__                                   \
                      << " at " << __FILE__ << ":" << __LINE__ << std::endl;    \
            { ACTION; }                                                         \
        }                                                                       \
    } while (0)

#define CHECK_INTERNAL(API, EXPECT) CHECK_API_OR(API, EXPECT, return INFINI_STATUS_INTERNAL_ERROR)

PanZezhong's avatar
PanZezhong committed
18
19
#define CHECK_STATUS(API) CHECK_API_OR(API, INFINI_STATUS_SUCCESS, return api_result_)

PanZezhong's avatar
PanZezhong committed
20
#endif // INFINIUTILS_CHECK_H