mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2025-10-12 18:38:10 +00:00
76c2f380bf
* feat: re-write persistent object ID tracker Features: - Remove random objectIDs entirely - Replace random objectIDs with persistentIDs - Remove the need to contact the MASTER server for a persistent ID - Add persistent ID logic to WorldServers that use transactions to guarantee unique IDs no matter when they are generated - Default character xml version to be the most recent one Fixes: - Return optional from GetModel (and check for nullopt where it may exist) - Regenerate inventory item ids on first login to be unique item IDs (fixes all those random IDs Pet IDs and subkeys are left alone and are assumed to be reserved (checks are there to prevent this) There is also duplicate check logic in place for properties and UGC/Models * Update comment and log * fix: sqlite transaction bug * fix colliding temp item ids temp items should not be saved. would cause issues between worlds as experienced before this commit
287 lines
13 KiB
C++
287 lines
13 KiB
C++
#ifndef SQLITEDATABASE_H
|
|
#define SQLITEDATABASE_H
|
|
|
|
#include "CppSQLite3.h"
|
|
|
|
#include "GameDatabase.h"
|
|
|
|
using PreppedStmtRef = CppSQLite3Statement&;
|
|
|
|
// Purposefully no definition for this to provide linker errors in the case someone tries to
|
|
// bind a parameter to a type that isn't defined.
|
|
template<typename ParamType>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const ParamType param);
|
|
|
|
// This is a function to set each parameter in a prepared statement.
|
|
// This is accomplished with a combination of parameter packing and Fold Expressions.
|
|
// The constexpr if statement is used to prevent the compiler from trying to call SetParam with 0 arguments.
|
|
template<typename... Args>
|
|
void SetParams(PreppedStmtRef stmt, Args&&... args) {
|
|
if constexpr (sizeof...(args) != 0) {
|
|
int i = 1;
|
|
(SetParam(stmt, i++, args), ...);
|
|
}
|
|
}
|
|
|
|
class SQLiteDatabase : public GameDatabase {
|
|
public:
|
|
void Connect() override;
|
|
void Destroy(std::string source = "") override;
|
|
|
|
void Commit() override;
|
|
bool GetAutoCommit() override;
|
|
void SetAutoCommit(bool value) override;
|
|
void ExecuteCustomQuery(const std::string_view query) override;
|
|
|
|
// Overloaded queries
|
|
std::optional<IServers::MasterInfo> GetMasterInfo() override;
|
|
|
|
std::vector<std::string> GetApprovedCharacterNames() override;
|
|
|
|
std::vector<FriendData> GetFriendsList(LWOOBJID charID) override;
|
|
|
|
std::optional<IFriends::BestFriendStatus> GetBestFriendStatus(const LWOOBJID playerCharacterId, const LWOOBJID friendCharacterId) override;
|
|
void SetBestFriendStatus(const LWOOBJID playerAccountId, const LWOOBJID friendAccountId, const uint32_t bestFriendStatus) override;
|
|
void AddFriend(const LWOOBJID playerAccountId, const LWOOBJID friendAccountId) override;
|
|
void RemoveFriend(const LWOOBJID playerAccountId, const LWOOBJID friendAccountId) override;
|
|
void UpdateActivityLog(const LWOOBJID characterId, const eActivityType activityType, const LWOMAPID mapId) override;
|
|
void DeleteUgcModelData(const LWOOBJID& modelId) override;
|
|
void UpdateUgcModelData(const LWOOBJID& modelId, std::stringstream& lxfml) override;
|
|
std::vector<IUgc::Model> GetAllUgcModels() override;
|
|
void CreateMigrationHistoryTable() override;
|
|
bool IsMigrationRun(const std::string_view str) override;
|
|
void InsertMigration(const std::string_view str) override;
|
|
std::optional<ICharInfo::Info> GetCharacterInfo(const LWOOBJID charId) override;
|
|
std::optional<ICharInfo::Info> GetCharacterInfo(const std::string_view charId) override;
|
|
std::string GetCharacterXml(const LWOOBJID accountId) override;
|
|
void UpdateCharacterXml(const LWOOBJID characterId, const std::string_view lxfml) override;
|
|
std::optional<IAccounts::Info> GetAccountInfo(const std::string_view username) override;
|
|
void InsertNewCharacter(const ICharInfo::Info info) override;
|
|
void InsertCharacterXml(const LWOOBJID accountId, const std::string_view lxfml) override;
|
|
std::vector<LWOOBJID> GetAccountCharacterIds(LWOOBJID accountId) override;
|
|
void DeleteCharacter(const LWOOBJID characterId) override;
|
|
void SetCharacterName(const LWOOBJID characterId, const std::string_view name) override;
|
|
void SetPendingCharacterName(const LWOOBJID characterId, const std::string_view name) override;
|
|
void UpdateLastLoggedInCharacter(const LWOOBJID characterId) override;
|
|
void SetPetNameModerationStatus(const LWOOBJID& petId, const IPetNames::Info& info) override;
|
|
std::optional<IPetNames::Info> GetPetNameInfo(const LWOOBJID& petId) override;
|
|
std::optional<IProperty::Info> GetPropertyInfo(const LWOMAPID mapId, const LWOCLONEID cloneId) override;
|
|
void UpdatePropertyModerationInfo(const IProperty::Info& info) override;
|
|
void UpdatePropertyDetails(const IProperty::Info& info) override;
|
|
void UpdateLastSave(const IProperty::Info& info) override;
|
|
void InsertNewProperty(const IProperty::Info& info, const uint32_t templateId, const LWOZONEID& zoneId) override;
|
|
std::vector<IPropertyContents::Model> GetPropertyModels(const LWOOBJID& propertyId) override;
|
|
void RemoveUnreferencedUgcModels() override;
|
|
void InsertNewPropertyModel(const LWOOBJID& propertyId, const IPropertyContents::Model& model, const std::string_view name) override;
|
|
void UpdateModel(const LWOOBJID& modelID, const NiPoint3& position, const NiQuaternion& rotation, const std::array<std::pair<LWOOBJID, std::string>, 5>& behaviors) override;
|
|
void RemoveModel(const LWOOBJID& modelId) override;
|
|
void UpdatePerformanceCost(const LWOZONEID& zoneId, const float performanceCost) override;
|
|
void InsertNewBugReport(const IBugReports::Info& info) override;
|
|
void InsertCheatDetection(const IPlayerCheatDetections::Info& info) override;
|
|
void InsertNewMail(const MailInfo& mail) override;
|
|
void InsertNewUgcModel(
|
|
std::stringstream& sd0Data,
|
|
const uint64_t blueprintId,
|
|
const uint32_t accountId,
|
|
const LWOOBJID characterId) override;
|
|
std::vector<MailInfo> GetMailForPlayer(const LWOOBJID characterId, const uint32_t numberOfMail) override;
|
|
std::optional<MailInfo> GetMail(const uint64_t mailId) override;
|
|
uint32_t GetUnreadMailCount(const LWOOBJID characterId) override;
|
|
void MarkMailRead(const uint64_t mailId) override;
|
|
void DeleteMail(const uint64_t mailId) override;
|
|
void ClaimMailItem(const uint64_t mailId) override;
|
|
void InsertSlashCommandUsage(const LWOOBJID characterId, const std::string_view command) override;
|
|
void UpdateAccountUnmuteTime(const uint32_t accountId, const uint64_t timeToUnmute) override;
|
|
void UpdateAccountBan(const uint32_t accountId, const bool banned) override;
|
|
void UpdateAccountPassword(const uint32_t accountId, const std::string_view bcryptpassword) override;
|
|
void InsertNewAccount(const std::string_view username, const std::string_view bcryptpassword) override;
|
|
void SetMasterInfo(const IServers::MasterInfo& info) override;
|
|
std::optional<uint64_t> GetCurrentPersistentId() override;
|
|
IObjectIdTracker::Range GetPersistentIdRange() override;
|
|
void InsertDefaultPersistentId() override;
|
|
std::optional<uint32_t> GetDonationTotal(const uint32_t activityId) override;
|
|
std::optional<bool> IsPlaykeyActive(const int32_t playkeyId) override;
|
|
std::vector<IUgc::Model> GetUgcModels(const LWOOBJID& propertyId) override;
|
|
void AddIgnore(const LWOOBJID playerId, const LWOOBJID ignoredPlayerId) override;
|
|
void RemoveIgnore(const LWOOBJID playerId, const LWOOBJID ignoredPlayerId) override;
|
|
std::vector<IIgnoreList::Info> GetIgnoreList(const LWOOBJID playerId) override;
|
|
void InsertRewardCode(const uint32_t account_id, const uint32_t reward_code) override;
|
|
std::vector<uint32_t> GetRewardCodesByAccountID(const uint32_t account_id) override;
|
|
void AddBehavior(const IBehaviors::Info& info) override;
|
|
std::string GetBehavior(const LWOOBJID behaviorId) override;
|
|
void RemoveBehavior(const LWOOBJID characterId) override;
|
|
void UpdateAccountGmLevel(const uint32_t accountId, const eGameMasterLevel gmLevel) override;
|
|
std::optional<IProperty::PropertyEntranceResult> GetProperties(const IProperty::PropertyLookup& params) override;
|
|
std::vector<ILeaderboard::Entry> GetDescendingLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetAscendingLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetNsLeaderboard(const uint32_t activityId) override;
|
|
std::vector<ILeaderboard::Entry> GetAgsLeaderboard(const uint32_t activityId) override;
|
|
void SaveScore(const LWOOBJID playerId, const uint32_t gameId, const Score& score) override;
|
|
void UpdateScore(const LWOOBJID playerId, const uint32_t gameId, const Score& score) override;
|
|
std::optional<ILeaderboard::Score> GetPlayerScore(const LWOOBJID playerId, const uint32_t gameId) override;
|
|
void IncrementNumWins(const LWOOBJID playerId, const uint32_t gameId) override;
|
|
void IncrementTimesPlayed(const LWOOBJID playerId, const uint32_t gameId) override;
|
|
void InsertUgcBuild(const std::string& modules, const LWOOBJID bigId, const std::optional<LWOOBJID> characterId) override;
|
|
void DeleteUgcBuild(const LWOOBJID bigId) override;
|
|
uint32_t GetAccountCount() override;
|
|
bool IsNameInUse(const std::string_view name) override;
|
|
std::optional<IPropertyContents::Model> GetModel(const LWOOBJID modelID) override;
|
|
std::optional<IUgc::Model> GetUgcModel(const LWOOBJID ugcId) override;
|
|
std::optional<IProperty::Info> GetPropertyInfo(const LWOOBJID id) override;
|
|
private:
|
|
CppSQLite3Statement CreatePreppedStmt(const std::string& query);
|
|
|
|
// Generic query functions that can be used for any query.
|
|
// Return type may be different depending on the query, so it is up to the caller to check the return type.
|
|
// The first argument is the query string, and the rest are the parameters to bind to the query.
|
|
// The return type is a unique_ptr to the result set, which is deleted automatically when it goes out of scope
|
|
template<typename... Args>
|
|
inline std::pair<CppSQLite3Statement, CppSQLite3Query> ExecuteSelect(const std::string& query, Args&&... args) {
|
|
std::pair<CppSQLite3Statement, CppSQLite3Query> toReturn;
|
|
toReturn.first = CreatePreppedStmt(query);
|
|
SetParams(toReturn.first, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(toReturn.second = toReturn.first.execQuery());
|
|
return toReturn;
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline void ExecuteDelete(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(preppedStmt.execDML());
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline int32_t ExecuteUpdate(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(return preppedStmt.execDML());
|
|
}
|
|
|
|
template<typename... Args>
|
|
inline int ExecuteInsert(const std::string& query, Args&&... args) {
|
|
auto preppedStmt = CreatePreppedStmt(query);
|
|
SetParams(preppedStmt, std::forward<Args>(args)...);
|
|
DLU_SQL_TRY_CATCH_RETHROW(return preppedStmt.execDML());
|
|
}
|
|
};
|
|
|
|
// Below are each of the definitions of SetParam for each supported type.
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::string_view param) {
|
|
LOG("%s", param.data());
|
|
stmt.bind(index, param.data());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const char* param) {
|
|
LOG("%s", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::string param) {
|
|
LOG("%s", param.c_str());
|
|
stmt.bind(index, param.c_str());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int8_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint8_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int16_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint16_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint32_t param) {
|
|
LOG("%u", param);
|
|
stmt.bind(index, static_cast<int32_t>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int32_t param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const int64_t param) {
|
|
LOG("%llu", param);
|
|
stmt.bind(index, static_cast<sqlite_int64>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const uint64_t param) {
|
|
LOG("%llu", param);
|
|
stmt.bind(index, static_cast<sqlite_int64>(param));
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const float param) {
|
|
LOG("%f", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const double param) {
|
|
LOG("%f", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const bool param) {
|
|
LOG("%d", param);
|
|
stmt.bind(index, param);
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::istream* param) {
|
|
LOG("Blob");
|
|
// This is the one time you will ever see me use const_cast.
|
|
std::stringstream stream;
|
|
stream << param->rdbuf();
|
|
stmt.bind(index, reinterpret_cast<const unsigned char*>(stream.str().c_str()), stream.str().size());
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::optional<uint32_t> param) {
|
|
if (param) {
|
|
LOG("%d", param.value());
|
|
stmt.bind(index, static_cast<int>(param.value()));
|
|
} else {
|
|
LOG("Null");
|
|
stmt.bindNull(index);
|
|
}
|
|
}
|
|
|
|
template<>
|
|
inline void SetParam(PreppedStmtRef stmt, const int index, const std::optional<LWOOBJID> param) {
|
|
if (param) {
|
|
LOG("%d", param.value());
|
|
stmt.bind(index, static_cast<sqlite_int64>(param.value()));
|
|
} else {
|
|
LOG("Null");
|
|
stmt.bindNull(index);
|
|
}
|
|
}
|
|
|
|
#endif //!SQLITEDATABASE_H
|