diff options
Diffstat (limited to 'src/map/mapreg_sql.c')
-rw-r--r-- | src/map/mapreg_sql.c | 228 |
1 files changed, 121 insertions, 107 deletions
diff --git a/src/map/mapreg_sql.c b/src/map/mapreg_sql.c index 3cf92454d..926df2994 100644 --- a/src/map/mapreg_sql.c +++ b/src/map/mapreg_sql.c @@ -20,19 +20,35 @@ struct mapreg_interface mapreg_s; #define MAPREG_AUTOSAVE_INTERVAL (300*1000) -/// Looks up the value of an integer variable using its uid. +/** + * Looks up the value of an integer variable using its uid. + * + * @param uid variable's unique identifier. + * @return variable's integer value + */ int mapreg_readreg(int64 uid) { struct mapreg_save *m = i64db_get(mapreg->db, uid); return m?m->u.i:0; } -/// Looks up the value of a string variable using its uid. +/** + * Looks up the value of a string variable using its uid. + * + * @param uid variable's unique identifier + * @return variable's string value + */ char* mapreg_readregstr(int64 uid) { - struct mapreg_save *m = i64db_get(mapreg->str_db, uid); + struct mapreg_save *m = i64db_get(mapreg->db, uid); return m?m->u.str:NULL; } -/// Modifies the value of an integer variable. +/** + * Modifies the value of an integer variable. + * + * @param uid variable's unique identifier + * @param val new value + * @retval true value was successfully set + */ bool mapreg_setreg(int64 uid, int val) { struct mapreg_save *m; int num = script_getvarid(uid); @@ -44,7 +60,7 @@ bool mapreg_setreg(int64 uid, int val) { m->u.i = val; if(name[1] != '@') { m->save = true; - mapreg->i_dirty = true; + mapreg->dirty = true; } } else { if( i ) @@ -55,6 +71,7 @@ bool mapreg_setreg(int64 uid, int val) { m->u.i = val; m->uid = uid; m->save = false; + m->is_string = false; if(name[1] != '@' && !mapreg->skip_insert) {// write new variable to database char tmp_str[32*2+1]; @@ -81,13 +98,19 @@ bool mapreg_setreg(int64 uid, int val) { return true; } -/// Modifies the value of a string variable. +/** + * Modifies the value of a string variable. + * + * @param uid variable's unique identifier + * @param str new value + * @retval true value was successfully set + */ bool mapreg_setregstr(int64 uid, const char* str) { struct mapreg_save *m; int num = script_getvarid(uid); unsigned int i = script_getvaridx(uid); const char* name = script->get_str(num); - + if( str == NULL || *str == 0 ) { if( i ) script->array_update(&mapreg->array_db,uid,true); @@ -95,31 +118,32 @@ bool mapreg_setregstr(int64 uid, const char* str) { if( SQL_ERROR == SQL->Query(map->mysql_handle, "DELETE FROM `%s` WHERE `varname`='%s' AND `index`='%d'", mapreg->table, name, i) ) Sql_ShowDebug(map->mysql_handle); } - if( (m = i64db_get(mapreg->str_db,uid)) ) { + if( (m = i64db_get(mapreg->db,uid)) ) { if( m->u.str != NULL ) aFree(m->u.str); ers_free(mapreg->ers, m); } - i64db_remove(mapreg->str_db,uid); + i64db_remove(mapreg->db,uid); } else { - if( (m = i64db_get(mapreg->str_db,uid)) ) { + if( (m = i64db_get(mapreg->db,uid)) ) { if( m->u.str != NULL ) aFree(m->u.str); m->u.str = aStrdup(str); if(name[1] != '@') { - mapreg->str_dirty = true; + mapreg->dirty = true; m->save = true; } } else { if( i ) script->array_update(&mapreg->array_db,uid,false); - + m = ers_alloc(mapreg->ers, struct mapreg_save); m->uid = uid; m->u.str = aStrdup(str); m->save = false; - + m->is_string = true; + if(name[1] != '@' && !mapreg->skip_insert) { //put returned null, so we must insert. char tmp_str[32*2+1]; char tmp_str2[255*2+1]; @@ -128,14 +152,16 @@ bool mapreg_setregstr(int64 uid, const char* str) { if( SQL_ERROR == SQL->Query(map->mysql_handle, "INSERT INTO `%s`(`varname`,`index`,`value`) VALUES ('%s','%d','%s')", mapreg->table, tmp_str, i, tmp_str2) ) Sql_ShowDebug(map->mysql_handle); } - i64db_put(mapreg->str_db, uid, m); + i64db_put(mapreg->db, uid, m); } } return true; } -/// Loads permanent variables from database +/** + * Loads permanent variables from database. + */ void script_load_mapreg(void) { /* 0 1 2 @@ -158,7 +184,7 @@ void script_load_mapreg(void) { } mapreg->skip_insert = true; - + SQL->StmtBindColumn(stmt, 0, SQLDT_STRING, &varname[0], sizeof(varname), &length, NULL); SQL->StmtBindColumn(stmt, 1, SQLDT_INT, &index, 0, NULL, NULL); SQL->StmtBindColumn(stmt, 2, SQLDT_STRING, &value[0], sizeof(value), NULL, NULL); @@ -166,145 +192,127 @@ void script_load_mapreg(void) { while ( SQL_SUCCESS == SQL->StmtNextRow(stmt) ) { int s = script->add_str(varname); int i = index; - + + if( i64db_exists(mapreg->db, reference_uid(s, i)) ) { + ShowWarning("load_mapreg: duplicate! '%s' => '%s' skipping...\n",varname,value); + continue; + } if( varname[length-1] == '$' ) { - if( i64db_exists(mapreg->str_db, reference_uid(s, i)) ) { - ShowWarning("load_mapreg: duplicate! '%s' => '%s' skipping...\n",varname,value); - continue; - } mapreg->setregstr(reference_uid(s, i),value); } else { - if( i64db_exists(mapreg->db, reference_uid(s, i)) ) { - ShowWarning("load_mapreg: duplicate! '%s' => '%s' skipping...\n",varname,value); - continue; - } mapreg->setreg(reference_uid(s, i),atoi(value)); } } - + SQL->StmtFree(stmt); mapreg->skip_insert = false; - - mapreg->i_dirty = false; - mapreg->str_dirty = false; + + mapreg->dirty = false; } -/// Saves permanent variables to database +/** + * Saves permanent variables to database. + */ void script_save_mapreg(void) { DBIterator* iter; struct mapreg_save *m = NULL; - if( mapreg->i_dirty ) { + if( mapreg->dirty ) { iter = db_iterator(mapreg->db); for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { if( m->save ) { int num = script_getvarid(m->uid); int i = script_getvaridx(m->uid); const char* name = script->get_str(num); - - if( SQL_ERROR == SQL->Query(map->mysql_handle, "UPDATE `%s` SET `value`='%d' WHERE `varname`='%s' AND `index`='%d' LIMIT 1", mapreg->table, m->u.i, name, i) ) - Sql_ShowDebug(map->mysql_handle); - m->save = false; - } - } - dbi_destroy(iter); - mapreg->i_dirty = false; - } - - if( mapreg->str_dirty ) { - iter = db_iterator(mapreg->str_db); - for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { - if( m->save ) { - int num = script_getvarid(m->uid); - int i = script_getvaridx(m->uid); - const char* name = script->get_str(num); - char tmp_str2[2*255+1]; - - SQL->EscapeStringLen(map->mysql_handle, tmp_str2, m->u.str, safestrnlen(m->u.str, 255)); - if( SQL_ERROR == SQL->Query(map->mysql_handle, "UPDATE `%s` SET `value`='%s' WHERE `varname`='%s' AND `index`='%d' LIMIT 1", mapreg->table, tmp_str2, name, i) ) - Sql_ShowDebug(map->mysql_handle); + if (!m->is_string) { + if( SQL_ERROR == SQL->Query(map->mysql_handle, "UPDATE `%s` SET `value`='%d' WHERE `varname`='%s' AND `index`='%d' LIMIT 1", mapreg->table, m->u.i, name, i) ) + Sql_ShowDebug(map->mysql_handle); + } else { + char tmp_str2[2*255+1]; + SQL->EscapeStringLen(map->mysql_handle, tmp_str2, m->u.str, safestrnlen(m->u.str, 255)); + if( SQL_ERROR == SQL->Query(map->mysql_handle, "UPDATE `%s` SET `value`='%s' WHERE `varname`='%s' AND `index`='%d' LIMIT 1", mapreg->table, tmp_str2, name, i) ) + Sql_ShowDebug(map->mysql_handle); + } m->save = false; } } dbi_destroy(iter); - mapreg->str_dirty = false; + mapreg->dirty = false; } } +/** + * Timer event to auto-save permanent variables. + * + * @see timer->do_timer + */ int script_autosave_mapreg(int tid, int64 tick, int id, intptr_t data) { mapreg->save(); return 0; } - -void mapreg_reload(void) { - DBIterator* iter; +/** + * Destroys a mapreg_save structure, freeing the contained string, if any. + * + * @see DBApply + */ +int mapreg_destroyreg(DBKey key, DBData *data, va_list ap) { struct mapreg_save *m = NULL; - mapreg->save(); + if (data->type != DB_DATA_PTR) // Sanity check + return 0; - iter = db_iterator(mapreg->db); - for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { - ers_free(mapreg->ers, m); - } - dbi_destroy(iter); - - iter = db_iterator(mapreg->str_db); - for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { - if( m->u.str != NULL ) { + m = DB->data2ptr(data); + + if (m->is_string) { + if (m->u.str) aFree(m->u.str); - } - ers_free(mapreg->ers, m); } - dbi_destroy(iter); - - db_clear(mapreg->db); - db_clear(mapreg->str_db); - + ers_free(mapreg->ers, m); + + return 0; +} + +/** + * Reloads mapregs, saving to database beforehand. + * + * This has the effect of clearing the temporary variables, and + * reloading the permanent ones. + */ +void mapreg_reload(void) { + mapreg->save(); + + mapreg->db->clear(mapreg->db, mapreg->destroyreg); + if( mapreg->array_db ) { mapreg->array_db->destroy(mapreg->array_db,script->array_free_db); mapreg->array_db = NULL; } - - + mapreg->load(); } +/** + * Finalizer. + */ void mapreg_final(void) { - DBIterator* iter; - struct mapreg_save *m = NULL; - mapreg->save(); - iter = db_iterator(mapreg->db); - for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { - ers_free(mapreg->ers, m); - } - dbi_destroy(iter); - - iter = db_iterator(mapreg->str_db); - for( m = dbi_first(iter); dbi_exists(iter); m = dbi_next(iter) ) { - if( m->u.str != NULL ) { - aFree(m->u.str); - } - ers_free(mapreg->ers, m); - } - dbi_destroy(iter); - - db_destroy(mapreg->db); - db_destroy(mapreg->str_db); - + mapreg->db->destroy(mapreg->db, mapreg->destroyreg); + ers_destroy(mapreg->ers); - + if( mapreg->array_db ) mapreg->array_db->destroy(mapreg->array_db,script->array_free_db); } +/** + * Initializer. + */ void mapreg_init(void) { mapreg->db = i64db_alloc(DB_OPT_BASE); - mapreg->str_db = i64db_alloc(DB_OPT_BASE); mapreg->ers = ers_new(sizeof(struct mapreg_save), "mapreg_sql.c::mapreg_ers", ERS_OPT_CLEAN); mapreg->load(); @@ -313,6 +321,9 @@ void mapreg_init(void) { timer->add_interval(timer->gettick() + MAPREG_AUTOSAVE_INTERVAL, mapreg->save_timer, 0, 0, MAPREG_AUTOSAVE_INTERVAL); } +/** + * Loads the mapreg configuration file. + */ bool mapreg_config_read(const char* w1, const char* w2) { if(!strcmpi(w1, "mapreg_db")) safestrncpy(mapreg->table, w2, sizeof(mapreg->table)); @@ -321,26 +332,28 @@ bool mapreg_config_read(const char* w1, const char* w2) { return true; } + +/** + * Interface defaults initializer. + */ void mapreg_defaults(void) { mapreg = &mapreg_s; - + /* */ mapreg->db = NULL; - mapreg->str_db = NULL; mapreg->ers = NULL; mapreg->skip_insert = false; - + safestrncpy(mapreg->table, "mapreg", sizeof(mapreg->table)); - mapreg->i_dirty = false; - mapreg->str_dirty = false; - + mapreg->dirty = false; + /* */ mapreg->array_db = NULL; - + /* */ mapreg->init = mapreg_init; mapreg->final = mapreg_final; - + /* */ mapreg->readreg = mapreg_readreg; mapreg->readregstr = mapreg_readregstr; @@ -349,6 +362,7 @@ void mapreg_defaults(void) { mapreg->load = script_load_mapreg; mapreg->save = script_save_mapreg; mapreg->save_timer = script_autosave_mapreg; + mapreg->destroyreg = mapreg_destroyreg; mapreg->reload = mapreg_reload; mapreg->config_read = mapreg_config_read; |