38 lines
891 B
C
38 lines
891 B
C
|
#ifndef STREAM_HANDLER_H
|
|||
|
|
#define STREAM_HANDLER_H
|
|||
|
|
|
|||
|
|
#include <iosfwd>
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <sstream>
|
|||
|
|
|
|||
|
|
#ifdef __OHOS__
|
|||
|
|
#define CC_OHOS_ 1
|
|||
|
|
#include "hilog/log.h"
|
|||
|
|
#undef LOG_DOMAIN
|
|||
|
|
#undef LOG_TAG
|
|||
|
|
#define LOG_DOMAIN 0x0728 // 全局domain宏，标识业务领域
|
|||
|
|
#define LOG_TAG "CC_SDK" // 全局tag宏，标识模块日志tag
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
class StreamHandler {
|
|||
|
|
public:
|
|||
|
|
// 构造函数
|
|||
|
|
explicit StreamHandler(std::ostream& stream) : os(stream) {}
|
|||
|
|
// 重载 << 操作符，处理输入值
|
|||
|
|
template <typename T>
|
|||
|
|
StreamHandler& operator<<(const T& value) {
|
|||
|
|
os << value;
|
|||
|
|
return *this;
|
|||
|
|
}
|
|||
|
|
// 专门处理函数触发逻辑
|
|||
|
|
StreamHandler& operator<<(std::ostream& (*func)(std::ostream&)) {
|
|||
|
|
func(os); // 调用传入的函数
|
|||
|
|
return *this;
|
|||
|
|
}
|
|||
|
|
private:
|
|||
|
|
std::ostream& os;
|
|||
|
|
};
|
|||
|
|
inline StreamHandler customCout(std::cout);
|
|||
|
|
|
|||
|
|
#endif
|