#include #include #include #include new Handle:g_DbInfo new g_QueryNum new bool:g_TestEnd = false public plugin_init() { register_plugin("SQLX Test", "1.0", "BAILOPAN") register_srvcmd("sqlx_test_normal", "SqlxTest_Normal") register_srvcmd("sqlx_test_thread", "SqlxTest_Thread") register_srvcmd("sqlx_test_old1", "SqlxTest_Old1") register_srvcmd("sqlx_test_old2", "SqlxTest_Old2") register_srvcmd("sqlx_test_thread_end", "SqlxTest_ThreadEnd") register_srvcmd("sqlx_test_bad", "SqlxTest_Bad") register_srvcmd("sqlx_test_quote", "SqlxTest_Quote") new configsDir[64] get_configsdir(configsDir, 63) server_cmd("exec %s/sql.cfg", configsDir) set_task(2.0, "start_map") } DoBasicInfo(affinities=0) { new type[12] new affinity[12] new wanted_type[12] dbi_type(type, 11) server_print("DBI type: %s", type) if (!affinities) return SQL_GetAffinity(affinity, 11); server_print("SQLX Affinity: %s", affinity) get_cvar_string("amx_sql_type", wanted_type, 11) if (!equal(wanted_type, affinity)) { if (g_DbInfo) { SQL_FreeHandle(g_DbInfo) g_DbInfo = Empty_Handle } new res = SQL_SetAffinity(wanted_type) server_print("Setting affinity from %s to %s: %s", affinity, wanted_type, res ? "Success" : "Failed") SQL_GetAffinity(affinity, 11) start_map() server_print("Verification: %s", affinity) } } public start_map() { new host[64] new user[64] new pass[64] new db[64] get_cvar_string("amx_sql_host", host, 63) get_cvar_string("amx_sql_user", user, 63) get_cvar_string("amx_sql_pass", pass, 63) get_cvar_string("amx_sql_db", db, 63) g_DbInfo = SQL_MakeDbTuple(host, user, pass, db) } public SqlxTest_Bad() { new errnum, error[255] new Handle:tempinfo = SQL_MakeDbTuple("1.2.3.4", "asdf", "gasdf", "gaben", 2) new Handle:db = SQL_Connect(tempinfo, errnum, error, 254) if (db == Empty_Handle) { server_print(" --> Errored out! %d, %s", errnum, error) } else { server_print(" --> Something is wrong here.") } return PLUGIN_HANDLED } /** * Note that this function works for both threaded and non-threaded queries. */ PrintQueryData(Handle:query) { new columns = SQL_NumColumns(query) new rows = SQL_NumResults(query) static querystring[2048] SQL_GetQueryString(query, querystring, 2047) server_print("Original query string: %s", querystring) server_print("Query columns: %d rows: %d", columns, rows) new num new row new str[32] new cols[2][32] SQL_FieldNumToName(query, 0, cols[0], 31) SQL_FieldNumToName(query, 1, cols[1], 31) while (SQL_MoreResults(query)) { num = SQL_ReadResult(query, 0) SQL_ReadResult(query, 1, str, 31) server_print("[%d]: %s=%d, %s=%s", row, cols[0], num, cols[1], str) SQL_NextRow(query) row++ } } /** * Handler for when a threaded query is resolved. */ public GetMyStuff(failstate, Handle:query, error[], errnum, data[], size, Float:queuetime) { server_print(" --> Resolved query %d, took %f seconds", data[0], queuetime) if (failstate) { if (failstate == TQUERY_CONNECT_FAILED) { server_print(" --> Connection failed!") } else if (failstate == TQUERY_QUERY_FAILED) { server_print(" --> Query failed!") } server_print(" --> Error code: %d (Message: ^"%s^")", errnum, error) new querystring[1024] SQL_GetQueryString(query, querystring, 1023) server_print(" --> Original query: %s", querystring) } else { PrintQueryData(query) } } /** * Starts a threaded query. */ public SqlxTest_Thread() { new query[512] new data[1] data[0] = g_QueryNum format(query, 511, "SELECT * FROM gaben") DoBasicInfo(1) server_print("Adding to %d queue at: %f", g_QueryNum, get_gametime()) SQL_ThreadQuery(g_DbInfo, "GetMyStuff", query, data, 1) g_QueryNum++ } /** * Tests string quoting */ public SqlxTest_Quote() { DoBasicInfo(1) new errno, error[255] new Handle:db = SQL_Connect(g_DbInfo, errno, error, sizeof(error)-1) if (!db) { server_print("Query failure: [%d] %s", errno, error) return } new buffer[500], num num = SQL_QuoteString(db, buffer, sizeof(buffer)-1, "Hi y'all! C\lam") server_print("num: %d str: %s", num, buffer) SQL_FreeHandle(db) } /** * Does a normal query. */ public SqlxTest_Normal() { new errnum, error[255] DoBasicInfo(1) new Handle:db = SQL_Connect(g_DbInfo, errnum, error, 254) if (!db) { server_print("Query failure: [%d] %s", errnum, error) return } new Handle:query = SQL_PrepareQuery(db, "SELECT * FROM gaben") if (!SQL_Execute(query)) { errnum = SQL_QueryError(query, error, 254) server_print("Query failure: [%d] %s", errnum, error) SQL_FreeHandle(query) SQL_FreeHandle(db) return } PrintQueryData(query) SQL_FreeHandle(query) SQL_FreeHandle(db) } /** * Wrapper for an old-style connection. */ Sql:OldInitDatabase() { new host[64] new user[64] new pass[64] new db[64] get_cvar_string("amx_sql_host", host, 63) get_cvar_string("amx_sql_user", user, 63) get_cvar_string("amx_sql_pass", pass, 63) get_cvar_string("amx_sql_db", db, 63) new error[255] new Sql:sql = dbi_connect(host, user, pass, db, error, 254) if (sql < SQL_OK) { server_print("Connection failure: %s", error) return SQL_FAILED } return sql } /** * Tests index-based lookup */ public SqlxTest_Old1() { DoBasicInfo() new Sql:sql = OldInitDatabase() if (sql < SQL_OK) return new Result:res = dbi_query(sql, "SELECT * FROM gaben") if (res == RESULT_FAILED) { new error[255] new code = dbi_error(sql, error, 254) server_print("Result failed! [%d]: %s", code, error) } else if (res == RESULT_NONE) { server_print("No result set returned.") } else { new cols[2][32] new str[32] new row, num new rows = dbi_num_rows(res) new columns = dbi_num_fields(res) dbi_field_name(res, 1, cols[0], 31) dbi_field_name(res, 2, cols[1], 31) server_print("Query columns: %d rows: %d", columns, rows) while (dbi_nextrow(res) > 0) { num = dbi_field(res, 1) dbi_field(res, 2, str, 31) server_print("[%d]: %s=%d, %s=%s", row, cols[0], num, cols[1], str) row++ } dbi_free_result(res) } dbi_close(sql) } /** * Tests name-based lookup */ public SqlxTest_Old2() { DoBasicInfo() new Sql:sql = OldInitDatabase() if (sql < SQL_OK) return new Result:res = dbi_query(sql, "SELECT * FROM gaben") if (res == RESULT_FAILED) { new error[255] new code = dbi_error(sql, error, 254) server_print("Result failed! [%d]: %s", code, error) } else if (res == RESULT_NONE) { server_print("No result set returned.") } else { new cols[2][32] new str[32] new row, num new rows = dbi_num_rows(res) new columns = dbi_num_fields(res) dbi_field_name(res, 1, cols[0], 31) dbi_field_name(res, 2, cols[1], 31) server_print("Query columns: %d rows: %d", columns, rows) while (dbi_nextrow(res) > 0) { num = dbi_result(res, cols[0]) dbi_result(res, cols[1], str, 31) server_print("[%d]: %s=%d, %s=%s", row, cols[0], num, cols[1], str) row++ } dbi_free_result(res) } dbi_close(sql) } public SqlxTest_ThreadEnd() { if (read_argc() < 2) { server_print("Requires mapname!") } else { new mapname[64] read_argv(1, mapname, 63) if (!is_map_valid(mapname)) { server_print("Invalid map: %s", mapname) } else { g_TestEnd = true server_cmd("changelevel %s", mapname) } } return PLUGIN_HANDLED } public plugin_end() { if (g_TestEnd) { SqlxTest_Thread() } else { SQL_FreeHandle(g_DbInfo) } }