64 lines
1.6 KiB
C++
64 lines
1.6 KiB
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include "memgraph_dynamic_lib.hpp"
|
||
|
|
#include "query_stripper.hpp"
|
||
|
|
#include "code_compiler.hpp"
|
||
|
|
#include "code_generator.hpp"
|
||
|
|
#include "utils/hashing/fnv.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)
|
||
|
|
{
|
||
|
|
// TODO implement me
|
||
|
|
// for now returns already compiled code
|
||
|
|
|
||
|
|
auto stripped = stripper.strip(query);
|
||
|
|
// TODO move to logger
|
||
|
|
cout << "Stripped query is: " << stripped << endl;
|
||
|
|
auto stripped_hash = fnv(stripped);
|
||
|
|
// TODO move to logger
|
||
|
|
cout << "Query hash is: " << stripped_hash << endl;
|
||
|
|
|
||
|
|
auto code_lib = load_code_lib("./compiled/cpu/create_return.so");
|
||
|
|
code_libs.insert({{stripped_hash, code_lib}});
|
||
|
|
|
||
|
|
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;
|
||
|
|
}
|
||
|
|
};
|