1 files changed, 93 insertions, 0 deletions

diff --git a/src/server/shared/Database/PreparedStatements.cpp b/src/server/shared/Database/PreparedStatements.cpp
new file mode 100644
index 00000000000..40a910acf9e
--- /dev/null
+++ b/src/server/shared/Database/PreparedStatements.cpp
@@ -0,0 +1,93 @@
+#include "PreparedStatements.h"
+
+void PreparedStatementHolder::_prepareStatement(const char* name, const char* sql, Database *db, uint32 &count)
+{
+    const char prefix[] = "PREPARE ";
+    size_t querySize = 8 + strlen(name) + 6 + strlen(sql) + 2 + 1;
+    char* query = new char[querySize];
+    strcpy(query, prefix);
+    strcat(query, name);
+    strcat(query, " FROM ");
+    strcat(query, "'");
+    strcat(query, sql);
+    strcat(query, "'");
+    
+    DEBUG_LOG("Preparing statement: %s", query);
+    db->Execute(query);
+    
+    delete[] query;
+    ++count;    
+}
+
+void PreparedStatementHolder::LoadAuthserver(Database *db, uint32 &count)
+{
+    _prepareStatement("auth_ping", "SELECT 1 FROM realmlist LIMIT 1", db, count);
+};
+
+void PreparedStatementHolder::Execute(Database *db, const char *name)
+{
+    const char prefix[] = "EXECUTE ";
+    size_t querySize = 8 + strlen(name) + 1;
+    char* query = new char[querySize];
+    strcpy(query, prefix);
+    strcat(query, name);
+
+    DEBUG_LOG("Prepared statement: %s", query);
+    db->Execute(query);
+    delete[] query;
+}
+
+void PreparedStatementHolder::PExecute(Database *db, const char *name, const char* args)
+{
+    // NOTE: if args == NULL, we're crashing here. No need to waste performance on checking;
+    // devs must make sure they use PExecute for args and Execute for no args.
+    
+    const char prefix[] = "EXECUTE ";
+    size_t querySize = 8 + strlen(name) + 7 + strlen(args) + 1;
+    char* query = new char[querySize];
+    strcpy(query, prefix);
+    strcat(query, name);
+    strcat(query, " USING ");
+    strcat(query, args);
+    
+    DEBUG_LOG("Prepared statement (parsed args): %s", query);
+    db->Execute(query);
+    delete[] query;
+}
+
+QueryResult_AutoPtr PreparedStatementHolder::Query(Database *db, const char *name)
+{
+    QueryResult_AutoPtr _return = QueryResult_AutoPtr(NULL);
+
+    const char prefix[] = "EXECUTE ";
+    size_t querySize = 8 + strlen(name) + 1;
+    char* query = new char[querySize];
+    strcpy(query, prefix);
+    strcat(query, name);
+
+    DEBUG_LOG("Prepared statement with resultset: %s", query);
+    _return = db->Query(query);
+    delete[] query;
+    return _return;
+}
+
+QueryResult_AutoPtr PreparedStatementHolder::PQuery(Database *db, const char *name, const char *args)
+{
+    // NOTE: if args == NULL, we're crashing here. No need to waste performance on checking;
+    // devs must make sure they use PQuery for args and Query for no args.
+
+    QueryResult_AutoPtr _return = QueryResult_AutoPtr(NULL);
+
+    const char prefix[] = "EXECUTE ";
+    size_t querySize = 8 + strlen(name) + 7 + strlen(args) + 1;
+    char* query = new char[querySize];
+    strcpy(query, prefix);
+    strcat(query, name);
+    strcat(query, " USING ");
+    strcat(query, args);
+
+    DEBUG_LOG("Prepared statement with resultset (parsed args): %s", query);
+    _return = db->Query(query);
+    delete[] query;
+    return _return;
+}
\ No newline at end of file