updateMatches.cpp

#include<iostream>
#include <pqxx/pqxx>

using namespace std;
using namespace pqxx;


bool isInList(const string& target, result list){
	for (int i=0; i < list.size(); ++i){
		if (list[i]["questionid"].as<string>() == target){
			return true;
		}
	}
	return false;
}

int matchOneQuestion(work& conn, const string& questionID, const string& myUserName, const string& yourUserName){
	// Pull answers for each user
	string myExpQuery = "select answerother, importance from users_answer_questions where questionID=" + questionID + "and username='" + string(myUserName) + "';";
	string yourAnsQuery = "select answerself from users_answer_questions where questionID=" + questionID + "and username='" + string(yourUserName) + "';";
	
	result myExpectation = conn.exec(myExpQuery);
	result yourAnswer = conn.exec(yourAnsQuery);
	
	string answerOther = myExpectation[0]["answerother"].as<string>();
	int index = yourAnswer[0]["answerself"].as<int>();
	char theAns = answerOther[ index-1 ];
	
	cout << "QID = " << questionID << ", myAnswer = " << answerOther << " yourAnswer = " << theAns << endl;	
	
	if ( theAns == '1' ){
		// cout << ", it's a match!";
		return myExpectation[0]["importance"].as<int>();
	}
	// cout << ", it's not a match.";
	return 0;
	
}

void matchOneUserWithOthers(work& conn, const string& myUserName){
	// First, grab a list of all users...
	result userList = conn.exec("select username from users;");
	for (int i=0; i < userList.size(); ++i){
		// For each user
		int totalScore = 0;
		int totalPossibleScore = 0;
		cout << "myUserName = " << myUserName << ", yourUserName = " << userList[i]["username"].as<string>() << endl;
		if (myUserName != userList[i]["username"].as<string>()){
			// We don't want to self-match
			// Grab a list of questions I have answered
			string myQIDQuery = "select questionid from users_answer_questions where username='" + myUserName + "';";
			string yourQIDQuery = "select questionid from users_answer_questions where username='" + userList[i]["username"].as<string>() + "';";
			result myQuestionNums = conn.exec(myQIDQuery);
			result yourQuestionNums = conn.exec(yourQIDQuery);
			
			for (int j=0; j < myQuestionNums.size(); ++j){
				if ( isInList(myQuestionNums[j]["questionid"].as<string>(), yourQuestionNums) ){
					// If this question is answered by you
					int subScore = matchOneQuestion(conn, myQuestionNums[j]["questionid"].as<string>(), myUserName, userList[i]["username"].as<string>());
					totalPossibleScore += 5;
					totalScore += subScore;
					// cout << " Score value added is " << subScore << endl;
				}
			}
			
			// cout << "Final score is " << totalScore << ", highest possible score is " << totalPossibleScore;
			int intMatchRate = 0;
			if (totalPossibleScore != 0){
				float matchRate = float(totalScore) / float(totalPossibleScore);
				intMatchRate = 100-int(matchRate*100);
			}

			
			string matchQuery = "select matchuser('" + myUserName + "', '" + userList[i]["username"].as<string>() + "', " + to_string(intMatchRate) + ");";
			// cout << ", your match rate is " << intMatchRate << endl;\
			// cout << endl << matchQuery << endl;
			conn.exec(matchQuery);
		}
	}
	

}

void matchAllUsers(work& conn){
	// First, grab a list of all users...
	result userList = conn.exec("select username from users;");
	for (int i=0; i < userList.size(); ++i){
		// For each user, match with everyone else
		matchOneUserWithOthers(conn, userList[i]["username"].as<string>());
	}		
}

int main(int argc, char* argv[]){
	// Connect to database
	cout << "Starting!" << endl;
	connection db("dbname=bt773 user=bt773 password=bt773");
	cout << "Connection successful!" << endl;
	work conn(db);
	if (argc == 2){
		matchOneUserWithOthers(conn, argv[1]);	
	}
	else{
		matchAllUsers(conn);
	}
	
	conn.commit();
	db.disconnect();
	cout << "Executed\n";
	return 0;
}