Files
SLikeNet/Samples/RankingServerDBTest/RankingServerDBTest.cpp
2025-11-24 14:19:51 +05:30

411 lines
17 KiB
C++

/*
* Copyright (c) 2014, Oculus VR, Inc.
* All rights reserved.
*
* This source code is licensed under the BSD-style license found in the
* RakNet License.txt file in the licenses directory of this source tree. An additional grant
* of patent rights can be found in the RakNet Patents.txt file in the same directory.
*
*/
// Common includes
#include <stdio.h>
#include <stdlib.h>
#include "Kbhit.h"
#include "GetTime.h"
#include "FunctionThread.h"
#include "RankingServer_PostgreSQL.h"
#ifdef _WIN32
#include <windows.h> // Sleep
#else
#include <unistd.h> // usleep
#endif
// localtime
#include <stdio.h>
#include <string.h>
#include <time.h>
// Database queries are asynchronous so the results are read in Functor::HandleResult. The default implementation passes it to a callback
class DBResultHandler : public RankingServerCBInterface
{
virtual void SubmitMatch_CB(SubmitMatch_PostgreSQLImpl *callResult, bool wasCancelled, void *context)
{
if (wasCancelled)
printf("SubmitMatch call canceled:\n");
else if (callResult->dbQuerySuccess==false)
{
printf("SubmitMatch call DB failure:\n");
printf("%s", callResult->rankingServer->GetLastError());
}
else
{
printf("SubmitMatch Completed (No data is returned)");
}
printf("\n");
}
virtual void GetRatingForParticipant_CB(GetRatingForParticipant_PostgreSQLImpl *callResult, bool wasCancelled, void *context)
{
if (wasCancelled)
printf("GetRatingForParticipant call cancelled:\n");
else if (callResult->dbQuerySuccess==false)
{
printf("GetRatingForParticipant call DB failure:\n");
printf("%s", callResult->rankingServer->GetLastError());
}
else
{
printf("GetRatingForParticipant result:\n");
printf("[in] participantDbId.primaryKey=%i\n", callResult->participantDbId.primaryKey);
printf("[in] participantDbId.secondaryKey=%i\n", callResult->participantDbId.secondaryKey);
printf("[in] gameDbId.primaryKey=%i\n", callResult->gameDbId.primaryKey);
printf("[in] gameDbId.secondaryKey=%i\n", callResult->gameDbId.secondaryKey);
printf("[out] participantFound=%i\n", callResult->participantFound); // False means no such participant
if (callResult->participantFound)
printf("[out] rating=%f\n", callResult->rating);
}
printf("\n");
}
virtual void GetRatingForParticipants_CB(GetRatingForParticipants_PostgreSQLImpl *callResult, bool wasCancelled, void *context)
{
if (wasCancelled)
printf("GetRatingForParticipants call canceled:\n");
else if (callResult->dbQuerySuccess==false)
{
printf("GetRatingForParticipants call DB failure:\n");
printf("%s", callResult->rankingServer->GetLastError());
}
else
{
printf("GetRatingForParticipants result:\n");
printf("[in] gameDbId.primaryKey=%i\n", callResult->gameDbId.primaryKey);
printf("[in] gameDbId.secondaryKey=%i\n", callResult->gameDbId.secondaryKey);
printf("[out] %i participants found.\n", callResult->outputList.Size());
unsigned i;
for (i=0; i < callResult->outputList.Size(); i++)
{
printf("[out] %i. participantDbId.primaryKey=%i\n", i+1, callResult->outputList[i].participantDbId.primaryKey);
printf("[out] %i. participantDbId.secondaryKey=%i\n", i+1, callResult->outputList[i].participantDbId.secondaryKey);
printf("[out] %i. newRating (most recent rating)=%f\n", i+1, callResult->outputList[i].newRating);
}
}
printf("\n");
}
virtual void GetHistoryForParticipant_CB(GetHistoryForParticipant_PostgreSQLImpl *callResult, bool wasCancelled, void *context)
{
if (wasCancelled)
printf("GetHistoryForParticipant call cancelled:\n");
else if (callResult->dbQuerySuccess==false)
{
printf("GetHistoryForParticipant call DB failure:\n");
printf("%s", callResult->rankingServer->GetLastError());
}
else
{
printf("GetHistoryForParticipant result:\n");
printf("[in] participantDbId.primaryKey=%i\n", callResult->participantDbId.primaryKey);
printf("[in] participantDbId.secondaryKey=%i\n", callResult->participantDbId.secondaryKey);
printf("[in] gameDbId.primaryKey=%i\n", callResult->gameDbId.primaryKey);
printf("[in] gameDbId.secondaryKey=%i\n", callResult->gameDbId.secondaryKey);
printf("[out] %i matches found for this participant.\n", callResult->matchHistoryList.Size());
unsigned i;
for (i=0; i < callResult->matchHistoryList.Size(); i++)
{
printf("[out] %i. participantBDbId.primaryKey (opponent) =%i\n", i+1, callResult->matchHistoryList[i]->participantBDbId.primaryKey);
printf("[out] %i. participantBDbId.secondaryKey (opponent) =%i\n", i+1, callResult->matchHistoryList[i]->participantBDbId.secondaryKey);
printf("[out] %i. participantAScore (us) =%f\n", i+1, callResult->matchHistoryList[i]->participantAScore);
printf("[out] %i. participantBScore (opponent) =%f\n", i+1, callResult->matchHistoryList[i]->participantBScore);
printf("[out] %i. participantAOldRating (us) =%f\n", i+1, callResult->matchHistoryList[i]->participantAOldRating);
printf("[out] %i. participantANewRating (us) =%f\n", i+1, callResult->matchHistoryList[i]->participantANewRating);
printf("[out] %i. participantBOldRating (opponent) =%f\n", i+1, callResult->matchHistoryList[i]->participantBOldRating);
printf("[out] %i. participantBNewRating (opponent) =%f\n", i+1, callResult->matchHistoryList[i]->participantBNewRating);
printf("[out] %i. Match Notes=%s\n", i+1, callResult->matchHistoryList[i]->matchNotes.C_String());
printf("[out] %i. matchBinaryDataLength=%i\n", i+1, callResult->matchHistoryList[i]->matchBinaryDataLength);
// Copied from the docs
struct tm *newtime;
newtime = _localtime64(& callResult->matchHistoryList[i]->matchTime);
char buff[30];
asctime_s( buff, sizeof(buff), newtime );
printf("[out] %i. matchTime (converted) =\n%s\n", i+1, buff );
}
}
printf("\n");
}
};
void main(int argc, char **argv)
{
printf("A sample on using the provided implementation\nof the Ranking Server specification based on PostgreSQL\n");
printf("Unlike the other samples, this is a server-only\nprocess so does not involve networking with RakNet.\n");
printf("The goal of this class is to allow you to save\nand retrieve a history of matches for a given participant and game ID\n");
printf("Difficulty: Intermediate\n\n");
// A function thread is class that spawns a thread that operates on functors.
// A functor is an instance of a class that has two pre-defined functions: One to perform processing, another to get the result.
// One per application is enough to not block
RakNet::FunctionThread ft;
// Start one thread. Starting more than one may be advantageous for multi-core processors. However, in this scenario we are
// blocking on database calls rather than processing.
ft.StartThreads(1);
// The real functionality of the RankingServer is contained in the functors defined in RankingServer_PostgreSQL.h/.cpp.
// However, this class contains some utility functions and members, such as the functionThread and a pointer to
// the postgreSQL interface.
RankingServer_PostgreSQL rankingServer;
// RankingServer_PostgreSQL internally uses a functionThread so that database queries can happen asynchronously,
// as opposed to blocking and slowing down the game.
// If you don't assign one it will create one automatically and start it with one thread.
rankingServer.AssignFunctionThread(&ft);
// The default implementation of the functors pass Functor::HandleResult through to a callback, instantiated here.
// Alternatively, you could derive from the *_PostgreSQLImpl functors found in RankingServer_PostgreSQL.h/.cpp
// and override the behavior of Functor::HandleResult
DBResultHandler resultHandler;
rankingServer.AssignCallback(&resultHandler);
printf("Enter database password:\n");
char connectionString[256],password[128];
char username[256];
strcpy(username, "postgres");
gets(password);
strcpy(connectionString, "user=");
strcat(connectionString, username);
strcat(connectionString, " password=");
strcat(connectionString, password);
// database=blah
if (rankingServer.Connect(connectionString)==false)
{
printf("Database connection failed.\n");
return;
}
printf("Database connection suceeded.\n");
printf("(D)rop tables\n"
"(C)reate tables.\n"
"(S)ubmit a match\n"
"Get (R)ating for one participant.\n"
"Get ratings for (A)ll participants\n"
"Get match (H)istory for one participant\n"
"(Q)uit\n");
char inputStr[512];
char ch;
while (1)
{
if (kbhit())
{
ch=getch();
if (ch=='q')
break;
else if (ch=='c')
{
if (rankingServer.CreateRankingServerTables()==false)
printf("%s", rankingServer.GetLastError());
else
printf("Tables created.\n");
}
else if (ch=='d')
{
if (rankingServer.DestroyRankingServerTables()==false)
printf("%s", rankingServer.GetLastError());
else
printf("Tables dropped.\n");
}
else if (ch=='s')
{
// We could do new and delete, but this ensures the class is allocated and deallocated in the same DLL, should we use one.
// By default deallocation takes place in HandleResult()
SubmitMatch_PostgreSQLImpl *functor = SubmitMatch_PostgreSQLImpl::Alloc();
printf("Submit a match\n");
/// Every participant is represented by a pair of numbers.
/// The primary ID can be used to reference a database primary key referring to another table holding all participant
/// The secondary ID can be used to indicate type
/// such as indicating the type of participant (player, squad, guild, clan, etc).
printf("Enter the primary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "0");
functor->participantADbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "1");
functor->participantADbId.secondaryKey = atoi(inputStr);
printf("Enter the primary ID of participant B (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "2");
functor->participantBDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of participant B (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "3");
functor->participantBDbId.secondaryKey = atoi(inputStr);
// Score is tracked in case you want to check the database for cheaters, or to modify ratings, etc.
// If you don't care about it, just use any number.
printf("Enter the score of participant A (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "10.0");
functor->participantAScore = (float) atof(inputStr);
printf("Enter the score of participant B (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "21.0");
functor->participantBScore = (float) atof(inputStr);
// The rating of the player before the match started. Elo.h/.cpp is provided as one way to calculate ratings
printf("Enter the pre-match rating of participant A (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "12.0");
functor->participantAOldRating = (float) atof(inputStr);
printf("Enter the post-match rating of participant A (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "5.0");
functor->participantANewRating = (float) atof(inputStr);
printf("Enter the pre-match rating of participant B (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "20.0");
functor->participantBOldRating = (float) atof(inputStr);
printf("Enter the post-match rating of participant B (float): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "30.0");
functor->participantBNewRating = (float) atof(inputStr);
/// Games are identified by a pair of integers the same way participants are
/// The primary key could be the game title (such as Asteroids) while the secondary key could
/// Indicate a submode (such as cooperative or deathmatch)
printf("Enter the primary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "12");
functor->gameDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "13");
functor->gameDbId.secondaryKey = atoi(inputStr);
/// A string is stored with each match.
printf("Enter the match notes: ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "Default Match Notes");
functor->matchNotes=inputStr;
printf("Store test binary data? (y/n): ");
gets(inputStr);
if (inputStr[0]!='n')
{
const char * data = "Default Match Binary Data";
functor->matchBinaryDataLength=(int) strlen(data)+1;
// As before, calling a function in the external code in case it was built to a DLL
functor->matchBinaryData = SubmitMatch_PostgreSQLImpl::AllocBytes(functor->matchBinaryDataLength);
memcpy(functor->matchBinaryData, data, functor->matchBinaryDataLength);
}
/// Puts this functor on the processing queue. It will process sometime later in a thread.
rankingServer.PushFunctor(functor);
}
else if (ch=='r')
{
// We could do new and delete, but this ensures the class is allocated and deallocated in the same DLL, should we use one.
// By default deallocation takes place in HandleResult()
GetRatingForParticipant_PostgreSQLImpl *functor = GetRatingForParticipant_PostgreSQLImpl::Alloc();
printf("Get the most recent rating for a participant\n");
printf("Enter the primary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "0");
functor->participantDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "1");
functor->participantDbId.secondaryKey = atoi(inputStr);
printf("Enter the primary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "12");
functor->gameDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "13");
functor->gameDbId.secondaryKey = atoi(inputStr);
/// Puts this functor on the processing queue. It will process sometime later in a thread.
rankingServer.PushFunctor(functor);
}
else if (ch=='a')
{
// We could do new and delete, but this ensures the class is allocated and deallocated in the same DLL, should we use one.
// By default deallocation takes place in HandleResult()
GetRatingForParticipants_PostgreSQLImpl *functor = GetRatingForParticipants_PostgreSQLImpl::Alloc();
printf("Get the most recent rating for all participants\n");
printf("Enter the primary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "12");
functor->gameDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "13");
functor->gameDbId.secondaryKey = atoi(inputStr);
/// Puts this functor on the processing queue. It will process sometime later in a thread.
rankingServer.PushFunctor(functor);
}
else if (ch=='h')
{
// We could do new and delete, but this ensures the class is allocated and deallocated in the same DLL, should we use one.
// By default deallocation takes place in HandleResult()
GetHistoryForParticipant_PostgreSQLImpl *functor = GetHistoryForParticipant_PostgreSQLImpl::Alloc();
printf("Get the match history for a participant\n");
printf("Enter the primary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "0");
functor->participantDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of participant A (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "1");
functor->participantDbId.secondaryKey = atoi(inputStr);
printf("Enter the primary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "12");
functor->gameDbId.primaryKey = atoi(inputStr);
printf("Enter the secondary ID of the game (int): ");
gets(inputStr);
if (inputStr[0]==0) strcpy(inputStr, "13");
functor->gameDbId.secondaryKey = atoi(inputStr);
/// Puts this functor on the processing queue. It will process sometime later in a thread.
rankingServer.PushFunctor(functor);
}
}
// Causes Functor::HandleResult calls. If this is forgotten you won't get processing result calls.
ft.CallResultHandlers();
Sleep(30);
}
}