amxmodx/plugins/testsuite/sqlxtest.sma
2007-10-26 00:28:17 +00:00

413 lines
8.4 KiB
Plaintext

#include <amxmodx>
#include <amxmisc>
#include <dbi>
#include <sqlx>
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_proc", "SqlxTest_Proc")
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")
register_srvcmd("sqlx_test_affinity", "SqlxTest_Affinity")
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)
}
}
public SqlxTest_Affinity()
{
server_print("[Access Manager] try SetAffinity to sqlite");
SQL_SetAffinity("sqlite");
server_print("[Access Manager] try SetAffinity to mysql");
SQL_SetAffinity("mysql");
server_print("[Access Manager] try SetAffinity to sqlite again");
SQL_SetAffinity("sqlite");
}
/**
* 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)
}
public SqlxTest_Proc()
{
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, "CALL ExampleProc()")
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)
server_print("Next result: %d", SQL_NextResultSet(query));
PrintQueryData(query)
SQL_FreeHandle(query)
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)
server_print("Next result: %d", SQL_NextResultSet(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)
}
}