memgraph/include/query/util.hpp

70 lines
1.6 KiB
C++
Raw Normal View History

#pragma once
2016-07-18 01:32:35 +08:00
#include <iostream>
#include <sstream>
#include <string>
#include <map>
2016-08-08 16:32:34 +08:00
#include "fmt/format.h"
#include "logging/default.hpp"
#include "utils/exceptions/basic_exception.hpp"
2016-07-18 01:32:35 +08:00
using std::cout;
using std::endl;
// this is a nice way how to avoid multiple definition problem with
// headers because it will create a unique namespace for each compilation unit
// http://stackoverflow.com/questions/2727582/multiple-definition-in-header-file
// but sometimes that might be a problem
namespace {
2016-08-08 16:32:34 +08:00
class CodeLineFormatException : public BasicException {
public:
using BasicException::BasicException;
};
template<typename... Args>
std::string format(const std::string &format_str, const Args &... args) {
2016-08-08 16:32:34 +08:00
return fmt::format(format_str, args...);
}
2016-08-08 16:32:34 +08:00
template<typename... Args>
std::string code_line(const std::string &format_str, const Args &... args) {
try {
return "\t" + format(format_str, args...) + "\n";
} catch (std::runtime_error &e) {
throw CodeLineFormatException(std::string(e.what()) + " " + format_str);
}
}
class CoutSocket {
public:
CoutSocket() : logger(logging::log->logger("Cout Socket")) {}
int write(const std::string &str) {
logger.info(str);
return str.size();
}
int write(const char *data, size_t len) {
logger.info(std::string(data, len));
return len;
}
int write(const byte *data, size_t len) {
std::stringstream ss;
for (int i = 0; i < len; i++) {
ss << data[i];
}
std::string output(ss.str());
cout << output << endl;
logger.info(output);
return len;
}
private:
Logger logger;
};
}