bool IOLoginData::saveItems(const Player* player, const ItemBlockList& itemList, DBInsert& query_insert, PropWriteStream& propWriteStream)
{
std::ostringstream ss;
typedef std::pair<Container*, int32_t> containerBlock;
std::queue<containerBlock> queue;
int32_t parentId = 0;
int32_t runningId = 100;
Database* db = Database::getInstance();
for (const auto& it : itemList) {
int32_t pid = it.first;
Item* item = it.second;
++runningId;
propWriteStream.clear();
item->serializeAttr(propWriteStream);
size_t attributesSize;
const char* attributes = propWriteStream.getStream(attributesSize);
ss << player->getGUID() << ',' << pid << ',' << runningId << ',' << item->getID() << ',' << item->getSubType() << ',' << db->escapeBlob(attributes, attributesSize);
if (!query_insert.addRow(ss)) {
return false;
}
if (Container* container = item->getContainer()) {
queue.emplace(container, runningId);
}
}
while (!queue.empty()) {
const containerBlock& cb = queue.front();
Container* container = cb.first;
parentId = cb.second;
queue.pop();
for (Item* item : container->getItemList()) {
++runningId;
Container* subContainer = item->getContainer();
if (subContainer) {
queue.emplace(subContainer, runningId);
}
propWriteStream.clear();
item->serializeAttr(propWriteStream);
size_t attributesSize;
const char* attributes = propWriteStream.getStream(attributesSize);
ss << player->getGUID() << ',' << parentId << ',' << runningId << ',' << item->getID() << ',' << item->getSubType() << ',' << db->escapeBlob(attributes, attributesSize);
if (!query_insert.addRow(ss)) {
return false;
}
}
}
return query_insert.execute();
}
bool IOLoginData::saveItems(const Player* player, const ItemBlockList& itemList, DBInsert& query_insert)
{
std::ostringstream stream;
typedef std::pair<Container*, int32_t> containerBlock;
std::list<containerBlock> stack;
int32_t parentId = 0;
int32_t runningId = 100;
Database* db = Database::getInstance();
for (ItemBlockList::const_iterator it = itemList.begin(); it != itemList.end(); ++it) {
int32_t pid = it->first;
Item* item = it->second;
++runningId;
uint32_t attributesSize = 0;
PropWriteStream propWriteStream;
item->serializeAttr(propWriteStream);
const char* attributes = propWriteStream.getStream(attributesSize);
stream << player->getGUID() << ',' << pid << ',' << runningId << ',' << item->getID() << ',' << (int32_t)item->getSubType() << ',' << db->escapeBlob(attributes, attributesSize);
if (!query_insert.addRow(stream)) {
return false;
}
if (Container* container = item->getContainer()) {
stack.emplace_back(container, runningId);
}
}
while (!stack.empty()) {
const containerBlock& cb = stack.front();
Container* container = cb.first;
parentId = cb.second;
stack.pop_front();
for (Item* item : container->getItemList()) {
++runningId;
Container* subContainer = item->getContainer();
if (subContainer) {
stack.emplace_back(subContainer, runningId);
}
uint32_t attributesSize = 0;
PropWriteStream propWriteStream;
item->serializeAttr(propWriteStream);
const char* attributes = propWriteStream.getStream(attributesSize);
stream << player->getGUID() << ',' << parentId << ',' << runningId << ',' << item->getID() << ',' << (int32_t)item->getSubType() << ',' << db->escapeBlob(attributes, attributesSize);
if (!query_insert.addRow(stream)) {
return false;
}
}
}
return query_insert.execute();
}
bool IOMapSerialize::saveMap()
{
int64_t start = OTSYS_TIME();
Database* db = Database::getInstance();
std::ostringstream query;
//Start the transaction
DBTransaction transaction;
if (!transaction.begin()) {
return false;
}
//clear old tile data
if (!db->executeQuery("DELETE FROM `tile_store`")) {
return false;
}
DBInsert stmt;
stmt.setQuery("INSERT INTO `tile_store` (`house_id`, `data`) VALUES ");
PropWriteStream stream;
for (const auto& it : Houses::getInstance().getHouses()) {
//save house items
House* house = it.second;
for (HouseTile* tile : house->getTiles()) {
saveTile(stream, tile);
size_t attributesSize;
const char* attributes = stream.getStream(attributesSize);
if (attributesSize > 0) {
query << house->getId() << ',' << db->escapeBlob(attributes, attributesSize);
if (!stmt.addRow(query)) {
return false;
}
}
stream.clear();
}
}
if (!stmt.execute()) {
return false;
}
//End the transaction
bool success = transaction.commit();
std::cout << "> Saved house items in: " <<
(OTSYS_TIME() - start) / (1000.) << " s" << std::endl;
return success;
}
bool IOMapSerialize::saveHouseBinary(Database* db, DBInsert& stmt, House* house)
{
PropWriteStream stream;
for(HouseTileList::iterator tit = house->getHouseTileBegin(); tit != house->getHouseTileEnd(); ++tit)
{
if(!saveTile(stream, *tit))
continue;
}
uint32_t attributesSize = 0;
const char* attributes = stream.getStream(attributesSize);
if(!attributesSize)
return true;
DBQuery query;
query << house->getId() << ", " << g_config.getNumber(ConfigManager::WORLD_ID)
<< ", " << db->escapeBlob(attributes, attributesSize);
return stmt.addRow(query);
}
bool IOMapSerialize::saveHouseInfo()
{
Database* db = Database::getInstance();
DBTransaction transaction;
if (!transaction.begin()) {
return false;
}
if (!db->executeQuery("DELETE FROM `house_lists`")) {
return false;
}
std::ostringstream query;
for (const auto& it : Houses::getInstance().getHouses()) {
House* house = it.second;
query << "SELECT `id` FROM `houses` WHERE `id` = " << house->getId();
DBResult* result = db->storeQuery(query.str());
if (result) {
db->freeResult(result);
query.str("");
query << "UPDATE `houses` SET `owner` = " << house->getOwner() << ", `paid` = " << house->getPaidUntil() << ", `warnings` = " << house->getPayRentWarnings() << ", `name` = " << db->escapeString(house->getName()) << ", `town_id` = " << house->getTownId() << ", `rent` = " << house->getRent() << ", `size` = " << house->getTiles().size() << ", `beds` = " << house->getBedCount() << " WHERE `id` = " << house->getId();
} else {
query.str("");
query << "INSERT INTO `houses` (`id`, `owner`, `paid`, `warnings`, `name`, `town_id`, `rent`, `size`, `beds`) VALUES (" << house->getId() << ',' << house->getOwner() << ',' << house->getPaidUntil() << ',' << house->getPayRentWarnings() << ',' << db->escapeString(house->getName()) << ',' << house->getTownId() << ',' << house->getRent() << ',' << house->getTiles().size() << ',' << house->getBedCount() << ')';
}
db->executeQuery(query.str());
query.str("");
}
DBInsert stmt;
stmt.setQuery("INSERT INTO `house_lists` (`house_id` , `listid` , `list`) VALUES ");
for (const auto& it : Houses::getInstance().getHouses()) {
House* house = it.second;
std::string listText;
if (house->getAccessList(GUEST_LIST, listText) && !listText.empty()) {
query << house->getId() << ',' << GUEST_LIST << ',' << db->escapeString(listText);
if (!stmt.addRow(query)) {
return false;
}
listText.clear();
}
if (house->getAccessList(SUBOWNER_LIST, listText) && !listText.empty()) {
query << house->getId() << ',' << SUBOWNER_LIST << ',' << db->escapeString(listText);
if (!stmt.addRow(query)) {
return false;
}
listText.clear();
}
for (Door* door : house->getDoors()) {
if (door->getAccessList(listText) && !listText.empty()) {
query << house->getId() << ',' << door->getDoorId() << ',' << db->escapeString(listText);
if (!stmt.addRow(query)) {
return false;
}
listText.clear();
}
}
}
if (!stmt.execute()) {
return false;
}
return transaction.commit();
}
bool IOLoginData::savePlayer(Player* player)
{
if (player->getHealth() <= 0) {
player->changeHealth(1);
}
Database* db = Database::getInstance();
std::ostringstream query;
query << "SELECT `save` FROM `players` WHERE `id` = " << player->getGUID();
DBResult* result = db->storeQuery(query.str());
if (!result) {
return false;
}
if (result->getDataInt("save") == 0) {
db->freeResult(result);
query.str("");
query << "UPDATE `players` SET `lastlogin` = " << player->lastLoginSaved << ", `lastip` = " << player->lastIP << " WHERE `id` = " << player->getGUID();
return db->executeQuery(query.str());
}
db->freeResult(result);
//serialize conditions
PropWriteStream propWriteStream;
for (ConditionList::const_iterator it = player->conditions.begin(); it != player->conditions.end(); ++it) {
Condition* condition = *it;
if (condition->isPersistent()) {
if (!condition->serialize(propWriteStream)) {
return false;
}
propWriteStream.ADD_UCHAR(CONDITIONATTR_END);
}
}
uint32_t conditionsSize = 0;
const char* conditions = propWriteStream.getStream(conditionsSize);
//First, an UPDATE query to write the player itself
query.str("");
query << "UPDATE `players` SET ";
query << "`level` = " << player->level << ',';
query << "`group_id` = " << player->group->id << ',';
query << "`vocation` = " << (int32_t)player->getVocationId() << ',';
query << "`health` = " << player->health << ',';
query << "`healthmax` = " << player->healthMax << ',';
query << "`experience` = " << player->experience << ',';
query << "`lookbody` = " << (int32_t)player->defaultOutfit.lookBody << ',';
query << "`lookfeet` = " << (int32_t)player->defaultOutfit.lookFeet << ',';
query << "`lookhead` = " << (int32_t)player->defaultOutfit.lookHead << ',';
query << "`looklegs` = " << (int32_t)player->defaultOutfit.lookLegs << ',';
query << "`looktype` = " << (int32_t)player->defaultOutfit.lookType << ',';
query << "`lookaddons` = " << (int32_t)player->defaultOutfit.lookAddons << ',';
query << "`maglevel` = " << player->magLevel << ',';
query << "`mana` = " << player->mana << ',';
query << "`manamax` = " << player->manaMax << ',';
query << "`manaspent` = " << player->manaSpent << ',';
query << "`soul` = " << player->soul << ',';
query << "`town_id` = " << player->town->getID() << ',';
const Position& loginPosition = player->getLoginPosition();
query << "`posx` = " << loginPosition.getX() << ',';
query << "`posy` = " << loginPosition.getY() << ',';
query << "`posz` = " << loginPosition.getZ() << ',';
query << "`cap` = " << player->getCapacity() << ',';
query << "`sex` = " << player->sex << ',';
if (player->lastLoginSaved != 0) {
query << "`lastlogin` = " << player->lastLoginSaved << ',';
}
if (player->lastIP != 0) {
query << "`lastip` = " << player->lastIP << ',';
}
query << "`conditions` = " << db->escapeBlob(conditions, conditionsSize) << ',';
if (g_game.getWorldType() != WORLD_TYPE_PVP_ENFORCED) {
int32_t skullTime = 0;
if (player->skullTicks > 0) {
skullTime = time(nullptr) + player->skullTicks / 1000;
}
query << "`skulltime` = " << skullTime << ',';
int32_t skull = 0;
if (player->skull == SKULL_RED) {
skull = SKULL_RED;
} else if (player->skull == SKULL_BLACK) {
skull = SKULL_BLACK;
}
query << "`skull` = " << skull << ',';
}
query << "`lastlogout` = " << player->getLastLogout() << ',';
query << "`balance` = " << player->bankBalance << ',';
query << "`offlinetraining_time` = " << player->getOfflineTrainingTime() / 1000 << ',';
query << "`offlinetraining_skill` = " << player->getOfflineTrainingSkill() << ',';
query << "`stamina` = " << player->getStaminaMinutes() << ',';
query << "`skill_fist` = " << player->skills[SKILL_FIST][SKILL_LEVEL] << ',';
query << "`skill_fist_tries` = " << player->skills[SKILL_FIST][SKILL_TRIES] << ',';
query << "`skill_club` = " << player->skills[SKILL_CLUB][SKILL_LEVEL] << ',';
query << "`skill_club_tries` = " << player->skills[SKILL_CLUB][SKILL_TRIES] << ',';
query << "`skill_sword` = " << player->skills[SKILL_SWORD][SKILL_LEVEL] << ',';
query << "`skill_sword_tries` = " << player->skills[SKILL_SWORD][SKILL_TRIES] << ',';
query << "`skill_axe` = " << player->skills[SKILL_AXE][SKILL_LEVEL] << ',';
query << "`skill_axe_tries` = " << player->skills[SKILL_AXE][SKILL_TRIES] << ',';
query << "`skill_dist` = " << player->skills[SKILL_DIST][SKILL_LEVEL] << ',';
query << "`skill_dist_tries` = " << player->skills[SKILL_DIST][SKILL_TRIES] << ',';
query << "`skill_shielding` = " << player->skills[SKILL_SHIELD][SKILL_LEVEL] << ',';
query << "`skill_shielding_tries` = " << player->skills[SKILL_SHIELD][SKILL_TRIES] << ',';
query << "`skill_fishing` = " << player->skills[SKILL_FISH][SKILL_LEVEL] << ',';
query << "`skill_fishing_tries` = " << player->skills[SKILL_FISH][SKILL_TRIES] << ',';
if (!player->isOffline()) {
query << "`onlinetime` = `onlinetime` + " << (time(nullptr) - player->lastLoginSaved) << ',';
}
query << "`blessings` = " << player->blessings;
query << " WHERE `id` = " << player->getGUID();
DBTransaction transaction;
if (!transaction.begin()) {
return false;
}
if (!db->executeQuery(query.str())) {
return false;
}
// learned spells
query.str("");
query << "DELETE FROM `player_spells` WHERE `player_id` = " << player->getGUID();
if (!db->executeQuery(query.str())) {
return false;
}
query.str("");
DBInsert stmt;
stmt.setQuery("INSERT INTO `player_spells` (`player_id`, `name` ) VALUES ");
for (LearnedInstantSpellList::const_iterator it = player->learnedInstantSpellList.begin();
it != player->learnedInstantSpellList.end(); ++it) {
query << player->getGUID() << ',' << db->escapeString(*it);
if (!stmt.addRow(query)) {
return false;
}
}
if (!stmt.execute()) {
return false;
}
//item saving
query << "DELETE FROM `player_items` WHERE `player_id` = " << player->getGUID();
if (!db->executeQuery(query.str())) {
return false;
}
stmt.setQuery("INSERT INTO `player_items` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
ItemBlockList itemList;
for (int32_t slotId = 1; slotId <= 10; ++slotId) {
Item* item = player->inventory[slotId];
if (item) {
itemList.emplace_back(slotId, item);
}
}
if (!saveItems(player, itemList, stmt)) {
return false;
}
if (player->lastDepotId != -1) {
//save depot items
query.str("");
query << "DELETE FROM `player_depotitems` WHERE `player_id` = " << player->getGUID();
if (!db->executeQuery(query.str())) {
return false;
}
stmt.setQuery("INSERT INTO `player_depotitems` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
itemList.clear();
for (const auto& it : player->depotChests) {
DepotChest* depotChest = it.second;
for (Item* item : depotChest->getItemList()) {
itemList.emplace_back(it.first, item);
}
}
if (!saveItems(player, itemList, stmt)) {
return false;
}
}
//save inbox items
query.str("");
query << "DELETE FROM `player_inboxitems` WHERE `player_id` = " << player->getGUID();
if (!db->executeQuery(query.str())) {
return false;
}
stmt.setQuery("INSERT INTO `player_inboxitems` (`player_id`, `pid`, `sid`, `itemtype`, `count`, `attributes`) VALUES ");
itemList.clear();
for (Item* item : player->getInbox()->getItemList()) {
itemList.emplace_back(0, item);
}
if (!saveItems(player, itemList, stmt)) {
return false;
}
query.str("");
query << "DELETE FROM `player_storage` WHERE `player_id` = " << player->getGUID();
if (!db->executeQuery(query.str())) {
return false;
}
query.str("");
stmt.setQuery("INSERT INTO `player_storage` (`player_id`, `key`, `value`) VALUES ");
player->genReservedStorageRange();
for (StorageMap::const_iterator cit = player->getStorageIteratorBegin(), end = player->getStorageIteratorEnd(); cit != end; ++cit) {
query << player->getGUID() << ',' << cit->first << ',' << cit->second;
if (!stmt.addRow(query)) {
return false;
}
}
if (!stmt.execute()) {
return false;
}
//End the transaction
return transaction.commit();
}
bool IOPlayer::saveItems(Player* player, const ItemBlockList& itemList, DBInsert& query_insert)
{
std::list<Container*> listContainer;
std::stringstream stream;
typedef std::pair<Container*, int32_t> containerBlock;
std::list<containerBlock> stack;
int32_t parentId = 0;
int32_t runningId = 100;
Database* db = Database::instance();
Item* item;
int32_t pid;
for(ItemBlockList::const_iterator it = itemList.begin(); it != itemList.end(); ++it){
pid = it->first;
item = it->second;
++runningId;
uint32_t attributesSize;
PropWriteStream propWriteStream;
item->serializeAttr(propWriteStream);
const char* attributes = propWriteStream.getStream(attributesSize);
stream << player->getGUID() << ", " << pid << ", " << runningId << ", " << item->getID() << ", " << (int32_t)item->getSubType() << ", " << db->escapeBlob(attributes, attributesSize);
if(!query_insert.addRow(stream)){
return false;
}
if(Container* container = item->getContainer()){
stack.push_back(containerBlock(container, runningId));
}
}
while(stack.size() > 0){
const containerBlock& cb = stack.front();
Container* container = cb.first;
parentId = cb.second;
stack.pop_front();
for(uint32_t i = 0; i < container->size(); ++i){
++runningId;
item = container->getItem(i);
if(Container* sub = item->getContainer()){
stack.push_back(containerBlock(sub, runningId));
}
uint32_t attributesSize;
PropWriteStream propWriteStream;
item->serializeAttr(propWriteStream);
const char* attributes = propWriteStream.getStream(attributesSize);
stream << player->getGUID() << ", " << parentId << ", " << runningId << ", " << item->getID() << ", " << (int32_t)item->getSubType() << ", " << db->escapeBlob(attributes, attributesSize);
if(!query_insert.addRow(stream))
return false;
}
}
return true;
}
