amxmodx/plugins/testsuite/sqlxtest.sma
2006-06-03 23:51:11 +00:00

287 lines
5.7 KiB
Plaintext

#include <amxmodx>
#include <amxmisc>
#include <dbi>
#include <sqlx>
new Handle:g_DbInfo
new g_QueryNum
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")
new configsDir[64]
get_configsdir(configsDir, 63)
server_cmd("exec %s/sql.cfg", configsDir)
server_exec()
}
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)
plugin_cfg()
server_print("Verification: %s", affinity)
}
}
public plugin_cfg()
{
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)
}
/**
* 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)
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)
{
server_print("Resolved query %d at: %f", data[0], get_gametime())
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)
} 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++
}
/**
* 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 plugin_end()
{
SQL_FreeHandle(g_DbInfo)
}