85 lines
2.3 KiB
C++
85 lines
2.3 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <memory>
|
|
|
|
// #define NOT_LOG_INFO
|
|
|
|
#include "memgraph_dynamic_lib.hpp"
|
|
#include "query_stripper.hpp"
|
|
#include "code_compiler.hpp"
|
|
#include "code_generator.hpp"
|
|
#include "utils/hashing/fnv.hpp"
|
|
#include "config/config.hpp"
|
|
#include "utils/log/logger.hpp"
|
|
|
|
using std::string;
|
|
using std::cout;
|
|
using std::endl;
|
|
|
|
|
|
class CodeLoader
|
|
{
|
|
public:
|
|
|
|
using sptr_code_lib = std::shared_ptr<CodeLib>;
|
|
|
|
CodeLoader()
|
|
: stripper(make_query_stripper(TK_INT, TK_FLOAT, TK_STR))
|
|
{
|
|
}
|
|
|
|
ICodeCPU* load_code_cpu(const string& query)
|
|
{
|
|
auto stripped = stripper.strip(query);
|
|
LOG_INFO("stripped_query=" + stripped);
|
|
auto stripped_hash = fnv(stripped);
|
|
auto hash_string = std::to_string(stripped_hash);
|
|
LOG_INFO("query_hash=" + hash_string);
|
|
|
|
auto code_lib_iter = code_libs.find(stripped_hash);
|
|
|
|
// code is already compiled and loaded, just return runnable
|
|
// instance
|
|
if (code_lib_iter != code_libs.end())
|
|
// TODO also return extracted arguments
|
|
return code_lib_iter->second->instance();
|
|
|
|
// code has to be generated, compiled and loaded
|
|
// TODO load output path from config
|
|
auto base_path = config::Config::instance()[config::COMPILE_CPU_PATH];
|
|
auto path_cpp = base_path + hash_string + ".cpp";
|
|
code_generator.generate_cpp(query, stripped_hash, path_cpp);
|
|
|
|
// TODO compile generated code
|
|
auto path_so = base_path + hash_string + ".so";
|
|
code_compiler.compile(path_cpp, path_so);
|
|
|
|
// loads dynamic lib and store it
|
|
auto code_lib = load_code_lib(path_so);
|
|
code_libs.insert({{stripped_hash, code_lib}});
|
|
|
|
// return instance of runnable code (ICodeCPU)
|
|
return code_lib->instance();
|
|
}
|
|
|
|
private:
|
|
// TODO somehow remove int.. from here
|
|
QueryStripper<int, int, int> stripper;
|
|
// TODO ifdef MEMGRAPH64 problem, how to use this kind
|
|
// of ifdef functions?
|
|
// uint64_t depends on fnv function
|
|
std::unordered_map<uint64_t, sptr_code_lib> code_libs;
|
|
|
|
CodeGenerator code_generator;
|
|
CodeCompiler code_compiler;
|
|
|
|
sptr_code_lib load_code_lib(const string& path)
|
|
{
|
|
sptr_code_lib code_lib = std::make_shared<CodeLib>(path);
|
|
code_lib->load();
|
|
return code_lib;
|
|
}
|
|
};
|