be0064f730
Array type lookup failed previously because the typemapping class was not yet receiving the list of types from the database. Fix was to pass this data
90 lines
2.3 KiB
C++
90 lines
2.3 KiB
C++
#include "DefaultConfigs.h"
|
|
#include "IndentationConfig.h"
|
|
#include "LanguageConfig.h"
|
|
#include "NameManglingRules.h"
|
|
#include "ResultLoopTemplate.h"
|
|
#include "StructureTemplate.h"
|
|
#include "TypeMappings.h"
|
|
#include "Pgsql_oids.h"
|
|
#include "PgDatabaseCatalog.h"
|
|
|
|
using namespace Pgsql;
|
|
|
|
class PgTypeContainer;
|
|
|
|
std::shared_ptr<const TypeMappings> GetPglabCppTypeMappings(std::shared_ptr<const PgTypeContainer> types)
|
|
{
|
|
auto tm = std::make_shared<TypeMappings>(types, std::initializer_list<TypeMappings::Mapping>
|
|
{
|
|
{ bool_oid, "bool" },
|
|
{ char_oid, "char" },
|
|
{ name_oid, "std::string" },
|
|
{ int8_oid, "int64_t" },
|
|
{ int2_oid, "int16_t" },
|
|
{ int4_oid, "int32_t" },
|
|
{ text_oid, "std::string" },
|
|
{ oid_oid, "Oid" },
|
|
{ float4_oid, "float" },
|
|
{ float8_oid, "double" }
|
|
});
|
|
tm->setDefaultStringType("std::string");
|
|
tm->setDefaultContainerType("std::vector<%1>");
|
|
return tm;
|
|
}
|
|
|
|
std::shared_ptr<StructureTemplate> buildPglabStructureTemplate()
|
|
{
|
|
auto t = std::make_shared<StructureTemplate>();
|
|
t->m_structTemplate =
|
|
R"__(class /%structname%/ {
|
|
public:
|
|
/%structfields%/
|
|
};
|
|
using /%structname%/Lst = std::vector</%structname%/>;
|
|
)__";
|
|
|
|
t->m_fieldTemplate = "/%typename%/ /%varname%/;";
|
|
//st_templ->m_fieldSeparator;
|
|
|
|
return t;
|
|
}
|
|
|
|
std::shared_ptr<ResultLoopTemplate> buildPglabResultLoopTemplate()
|
|
{
|
|
auto t = std::make_shared<ResultLoopTemplate>();
|
|
t->m_loopTemplate =
|
|
R"__(Pgsql::Result result = conn.query(/%queryliteral%/);
|
|
/%structname%/Lst lst;
|
|
for (auto row: result) {
|
|
Pgsql::Col col(row);
|
|
/%structname%/ v;
|
|
col
|
|
/%assignfields%/;
|
|
lst.push_back(v);
|
|
}
|
|
)__";
|
|
t->m_retrieveValueTemplate = " >> v./%varname%/";
|
|
|
|
return t;
|
|
}
|
|
|
|
|
|
std::shared_ptr<LanguageConfig> buildPglabCppLanguageConfig(std::shared_ptr<PgDatabaseCatalog> catalog)
|
|
{
|
|
auto config = std::make_shared<LanguageConfig>();
|
|
config->setTypeMappings(GetPglabCppTypeMappings(catalog->types()));
|
|
config->setNameManglingRules(std::make_shared<NameManglingRules>());
|
|
config->setStructureTemplate(buildPglabStructureTemplate());
|
|
config->setIndentationConfig(std::make_shared<IndentationConfig>());
|
|
config->setResultLoopTemplate(buildPglabResultLoopTemplate());
|
|
return config;
|
|
}
|
|
|
|
|
|
std::shared_ptr<const LanguageConfig> getPglabCppLanguageConfig(std::shared_ptr<PgDatabaseCatalog> catalog)
|
|
{
|
|
static auto config = buildPglabCppLanguageConfig(catalog);
|
|
return config;
|
|
}
|
|
|
|
|