mirror of
https://github.com/EQEmu/Server.git
synced 2026-06-28 00:57:15 +00:00
Split repositories out between a base immutable repository (that does not change) and an extended repository of which developers can add more custom methods to
This commit is contained in:
Akkadius
@@ -23,326 +23,45 @@
|
||||
|
||||
#include "../database.h"
|
||||
#include "../string_util.h"
|
||||
#include "base/base_character_pet_info_repository.h"
|
||||
|
||||
class CharacterPetInfoRepository {
|
||||
class CharacterPetInfoRepository: public BaseCharacterPetInfoRepository {
|
||||
public:
|
||||
struct CharacterPetInfo {
|
||||
int char_id;
|
||||
int pet;
|
||||
std::string petname;
|
||||
int petpower;
|
||||
int spell_id;
|
||||
int hp;
|
||||
int mana;
|
||||
float size;
|
||||
};
|
||||
|
||||
static std::string PrimaryKey()
|
||||
{
|
||||
return std::string("pet");
|
||||
}
|
||||
/**
|
||||
* This file was auto generated on Apr 5, 2020 and can be modified and extended upon
|
||||
*
|
||||
* Base repository methods are automatically
|
||||
* generated in the "base" version of this repository. The base repository
|
||||
* is immutable and to be left untouched, while methods in this class
|
||||
* are used as extension methods for more specific persistence-layer
|
||||
* accessors or mutators
|
||||
*
|
||||
* Base Methods (Subject to be expanded upon in time)
|
||||
*
|
||||
* InsertOne
|
||||
* UpdateOne
|
||||
* DeleteOne
|
||||
* FindOne
|
||||
* GetWhere(std::string where_filter)
|
||||
* DeleteWhere(std::string where_filter)
|
||||
* InsertMany
|
||||
* All
|
||||
*
|
||||
* Example custom methods in a repository
|
||||
*
|
||||
* CharacterPetInfoRepository::GetByZoneAndVersion(int zone_id, int zone_version)
|
||||
* CharacterPetInfoRepository::GetWhereNeverExpires()
|
||||
* CharacterPetInfoRepository::GetWhereXAndY()
|
||||
* CharacterPetInfoRepository::DeleteWhereXAndY()
|
||||
*
|
||||
* Most of the above could be covered by base methods, but if you as a developer
|
||||
* find yourself re-using logic for other parts of the code, its best to just make a
|
||||
* method that can be re-used easily elsewhere especially if it can use a base repository
|
||||
* method and encapsulate filters there
|
||||
*/
|
||||
|
||||
static std::vector<std::string> Columns()
|
||||
{
|
||||
return {
|
||||
"char_id",
|
||||
"pet",
|
||||
"petname",
|
||||
"petpower",
|
||||
"spell_id",
|
||||
"hp",
|
||||
"mana",
|
||||
"size",
|
||||
};
|
||||
}
|
||||
|
||||
static std::string ColumnsRaw()
|
||||
{
|
||||
return std::string(implode(", ", Columns()));
|
||||
}
|
||||
|
||||
static std::string InsertColumnsRaw()
|
||||
{
|
||||
std::vector<std::string> insert_columns;
|
||||
|
||||
for (auto &column : Columns()) {
|
||||
if (column == PrimaryKey()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
insert_columns.push_back(column);
|
||||
}
|
||||
|
||||
return std::string(implode(", ", insert_columns));
|
||||
}
|
||||
|
||||
static std::string TableName()
|
||||
{
|
||||
return std::string("character_pet_info");
|
||||
}
|
||||
|
||||
static std::string BaseSelect()
|
||||
{
|
||||
return fmt::format(
|
||||
"SELECT {} FROM {}",
|
||||
ColumnsRaw(),
|
||||
TableName()
|
||||
);
|
||||
}
|
||||
|
||||
static std::string BaseInsert()
|
||||
{
|
||||
return fmt::format(
|
||||
"INSERT INTO {} ({}) ",
|
||||
TableName(),
|
||||
InsertColumnsRaw()
|
||||
);
|
||||
}
|
||||
|
||||
static CharacterPetInfo NewEntity()
|
||||
{
|
||||
CharacterPetInfo entry{};
|
||||
|
||||
entry.char_id = 0;
|
||||
entry.pet = 0;
|
||||
entry.petname = "";
|
||||
entry.petpower = 0;
|
||||
entry.spell_id = 0;
|
||||
entry.hp = 0;
|
||||
entry.mana = 0;
|
||||
entry.size = 0;
|
||||
|
||||
return entry;
|
||||
}
|
||||
|
||||
static CharacterPetInfo GetCharacterPetInfoEntry(
|
||||
const std::vector<CharacterPetInfo> &character_pet_infos,
|
||||
int character_pet_info_id
|
||||
)
|
||||
{
|
||||
for (auto &character_pet_info : character_pet_infos) {
|
||||
if (character_pet_info.pet == character_pet_info_id) {
|
||||
return character_pet_info;
|
||||
}
|
||||
}
|
||||
|
||||
return NewEntity();
|
||||
}
|
||||
|
||||
static CharacterPetInfo FindOne(
|
||||
int character_pet_info_id
|
||||
)
|
||||
{
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"{} WHERE id = {} LIMIT 1",
|
||||
BaseSelect(),
|
||||
character_pet_info_id
|
||||
)
|
||||
);
|
||||
|
||||
auto row = results.begin();
|
||||
if (results.RowCount() == 1) {
|
||||
CharacterPetInfo entry{};
|
||||
|
||||
entry.char_id = atoi(row[0]);
|
||||
entry.pet = atoi(row[1]);
|
||||
entry.petname = row[2] ? row[2] : "";
|
||||
entry.petpower = atoi(row[3]);
|
||||
entry.spell_id = atoi(row[4]);
|
||||
entry.hp = atoi(row[5]);
|
||||
entry.mana = atoi(row[6]);
|
||||
entry.size = atof(row[7]);
|
||||
|
||||
return entry;
|
||||
}
|
||||
|
||||
return NewEntity();
|
||||
}
|
||||
|
||||
static int DeleteOne(
|
||||
int character_pet_info_id
|
||||
)
|
||||
{
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"DELETE FROM {} WHERE {} = {}",
|
||||
TableName(),
|
||||
PrimaryKey(),
|
||||
character_pet_info_id
|
||||
)
|
||||
);
|
||||
|
||||
return (results.Success() ? results.RowsAffected() : 0);
|
||||
}
|
||||
|
||||
static int UpdateOne(
|
||||
CharacterPetInfo character_pet_info_entry
|
||||
)
|
||||
{
|
||||
std::vector<std::string> update_values;
|
||||
|
||||
auto columns = Columns();
|
||||
|
||||
update_values.push_back(columns[2] + " = '" + EscapeString(character_pet_info_entry.petname) + "'");
|
||||
update_values.push_back(columns[3] + " = " + std::to_string(character_pet_info_entry.petpower));
|
||||
update_values.push_back(columns[4] + " = " + std::to_string(character_pet_info_entry.spell_id));
|
||||
update_values.push_back(columns[5] + " = " + std::to_string(character_pet_info_entry.hp));
|
||||
update_values.push_back(columns[6] + " = " + std::to_string(character_pet_info_entry.mana));
|
||||
update_values.push_back(columns[7] + " = " + std::to_string(character_pet_info_entry.size));
|
||||
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"UPDATE {} SET {} WHERE {} = {}",
|
||||
TableName(),
|
||||
implode(", ", update_values),
|
||||
PrimaryKey(),
|
||||
character_pet_info_entry.pet
|
||||
)
|
||||
);
|
||||
|
||||
return (results.Success() ? results.RowsAffected() : 0);
|
||||
}
|
||||
|
||||
static CharacterPetInfo InsertOne(
|
||||
CharacterPetInfo character_pet_info_entry
|
||||
)
|
||||
{
|
||||
std::vector<std::string> insert_values;
|
||||
|
||||
insert_values.push_back("'" + EscapeString(character_pet_info_entry.petname) + "'");
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.petpower));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.spell_id));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.hp));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.mana));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.size));
|
||||
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"{} VALUES ({})",
|
||||
BaseInsert(),
|
||||
implode(",", insert_values)
|
||||
)
|
||||
);
|
||||
|
||||
if (results.Success()) {
|
||||
character_pet_info_entry.id = results.LastInsertedID();
|
||||
return character_pet_info_entry;
|
||||
}
|
||||
|
||||
character_pet_info_entry = CharacterPetInfoRepository::NewEntity();
|
||||
|
||||
return character_pet_info_entry;
|
||||
}
|
||||
|
||||
static int InsertMany(
|
||||
std::vector<CharacterPetInfo> character_pet_info_entries
|
||||
)
|
||||
{
|
||||
std::vector<std::string> insert_chunks;
|
||||
|
||||
for (auto &character_pet_info_entry: character_pet_info_entries) {
|
||||
std::vector<std::string> insert_values;
|
||||
|
||||
insert_values.push_back("'" + EscapeString(character_pet_info_entry.petname) + "'");
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.petpower));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.spell_id));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.hp));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.mana));
|
||||
insert_values.push_back(std::to_string(character_pet_info_entry.size));
|
||||
|
||||
insert_chunks.push_back("(" + implode(",", insert_values) + ")");
|
||||
}
|
||||
|
||||
std::vector<std::string> insert_values;
|
||||
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"{} VALUES {}",
|
||||
BaseInsert(),
|
||||
implode(",", insert_chunks)
|
||||
)
|
||||
);
|
||||
|
||||
return (results.Success() ? results.RowsAffected() : 0);
|
||||
}
|
||||
|
||||
static std::vector<CharacterPetInfo> All()
|
||||
{
|
||||
std::vector<CharacterPetInfo> all_entries;
|
||||
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"{}",
|
||||
BaseSelect()
|
||||
)
|
||||
);
|
||||
|
||||
all_entries.reserve(results.RowCount());
|
||||
|
||||
for (auto row = results.begin(); row != results.end(); ++row) {
|
||||
CharacterPetInfo entry{};
|
||||
|
||||
entry.char_id = atoi(row[0]);
|
||||
entry.pet = atoi(row[1]);
|
||||
entry.petname = row[2] ? row[2] : "";
|
||||
entry.petpower = atoi(row[3]);
|
||||
entry.spell_id = atoi(row[4]);
|
||||
entry.hp = atoi(row[5]);
|
||||
entry.mana = atoi(row[6]);
|
||||
entry.size = atof(row[7]);
|
||||
|
||||
all_entries.push_back(entry);
|
||||
}
|
||||
|
||||
return all_entries;
|
||||
}
|
||||
|
||||
static std::vector<CharacterPetInfo> GetWhere(std::string where_filter)
|
||||
{
|
||||
std::vector<CharacterPetInfo> all_entries;
|
||||
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"{} WHERE {}",
|
||||
BaseSelect(),
|
||||
where_filter
|
||||
)
|
||||
);
|
||||
|
||||
all_entries.reserve(results.RowCount());
|
||||
|
||||
for (auto row = results.begin(); row != results.end(); ++row) {
|
||||
CharacterPetInfo entry{};
|
||||
|
||||
entry.char_id = atoi(row[0]);
|
||||
entry.pet = atoi(row[1]);
|
||||
entry.petname = row[2] ? row[2] : "";
|
||||
entry.petpower = atoi(row[3]);
|
||||
entry.spell_id = atoi(row[4]);
|
||||
entry.hp = atoi(row[5]);
|
||||
entry.mana = atoi(row[6]);
|
||||
entry.size = atof(row[7]);
|
||||
|
||||
all_entries.push_back(entry);
|
||||
}
|
||||
|
||||
return all_entries;
|
||||
}
|
||||
|
||||
static int DeleteWhere(std::string where_filter)
|
||||
{
|
||||
auto results = database.QueryDatabase(
|
||||
fmt::format(
|
||||
"DELETE FROM {} WHERE {}",
|
||||
TableName(),
|
||||
PrimaryKey(),
|
||||
where_filter
|
||||
)
|
||||
);
|
||||
|
||||
return (results.Success() ? results.RowsAffected() : 0);
|
||||
}
|
||||
// Custom extended repository methods here
|
||||
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user