QuickTrack/CustomerDatabase.cpp at main · abdool86/QuickTrack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#include <stdlib.h>
#include <iostream>
#include <driver.h>
#include <exception.h>
#include <resultset.h>
#include <statement.h>
#include <vector>
#include <prepared_statement.h>
#include "Customer.h"
#include "CustomerDatabase.h"
#include "Database.h"

using namespace sql;
using namespace std;

	CustomerDatabase::CustomerDatabase(string serverName, string userName, string password):Database(serverName, userName, password){}

	Customer CustomerDatabase::selectCustomer(string userEmail){
		Customer user;
		this->stmt= con->createStatement();
    		this->res = stmt->executeQuery("SELECT * FROM tbl_customers WHERE customer_email ='"+ userEmail +"'");
		while(this->res->next()){
			user.setUniqueID(res->getInt("customer_id"));
			user.setUserName(res->getString("customer_userName"));
			user.setPhoneNum(res->getInt("customer_phoneNum"));
			user.setEmail(res->getString("customer_email"));
			user.setFirstName(res->getString("customer_firstName"));
			user.setLastName(res->getString("customer_lastName"));
			user.setPassword(res->getString("customer_password"));
		}

		return user;
	}

	Customer CustomerDatabase::selectCustomer(int uniqueID){
		Customer user;
		this->stmt= con->createStatement();
    		this->res = stmt->executeQuery("SELECT * FROM tbl_customers WHERE customer_id='"+ to_string(uniqueID) +"'");

		while(this->res->next()){
			user.setUniqueID(res->getInt("customer_id"));
			user.setUserName(res->getString("customer_userName"));
			user.setPhoneNum(res->getInt("customer_phoneNum"));
			user.setEmail(res->getString("customer_email"));
			user.setFirstName(res->getString("customer_firstName"));
			user.setLastName(res->getString("customer_lastName"));
			user.setPassword(res->getString("customer_password"));
		}

		return user;
	}

	int CustomerDatabase::searchCustomer(string username){
		this->stmt = con->createStatement();
		this->res = stmt->executeQuery("SELECT * FROM tbl_customers WHERE customer_username ='" + username+"'");
		return this->res->rowsCount();
		delete this->stmt;
		delete this->res;
	}

	int CustomerDatabase::searchCustomer(int id){
		this->stmt = con->createStatement();
		this->res = stmt->executeQuery("SELECT * FROM tbl_customers WHERE customer_id =" + to_string(id) +"");
		return this->res->rowsCount();
		delete this->stmt;
		delete this->res;
	}

	vector<Customer> CustomerDatabase::selectCustomers(string userEmail[], int arrSize){
		vector<Customer> userdb;
		for(int i = 0; i < arrSize; i++) {
			Customer user = selectCustomer(userEmail[i]);
			userdb.push_back(user);
		}
		return userdb;
	}

	vector<Customer> CustomerDatabase::selectCustomers(vector<int> listUniqueID){
		vector<Customer> userdb;
		int i =0;
		for(auto iterator = userdb.begin(); iterator != userdb.end(); iterator++) {
			Customer user = selectCustomer(listUniqueID[i]);
			userdb.push_back(user);
			i++;
		}
		return userdb;
	}

	void CustomerDatabase::addCustomer(Customer user){
		string values;
		values = "'"+user.getUserName()+"','"+
		to_string(user.getPhoneNum())+"','"+
		user.getEmail()+"','"+
		user.getFirstName()+"','"+
		user.getLastName()+"'";
		this->pstmt = con->prepareStatement("INSERT INTO tbl_customers(customer_username,customer_phoneNum, customer_email, customer_firstName, customer_lastName) VALUES ("+values+")");
		this->pstmt->executeUpdate();
		delete this->pstmt;
		return;
	}

	Customer CustomerDatabase::updateUniqueID(Customer user){
		string values;
		values = "customer_firstName = '" + user.getFirstName() + "' AND "+
		"customer_lastName = '" + user.getLastName() + "' AND " +
		"customer_email = '" + user.getEmail() + "'";
		this->stmt = con->createStatement();
		this->res = stmt->executeQuery("SELECT customer_id FROM tbl_customers WHERE " + values);
		while (this->res->next()){
                       	user.setUniqueID(res->getInt("customer_id"));
		}
		return user;
	}

	vector<Customer> CustomerDatabase::getAll(){
		vector<Customer> userdb;
		Customer user;
		this->stmt= con->createStatement();
    		this->res = stmt->executeQuery("SELECT * FROM tbl_customers");
		while(this->res->next()){
			user.setUniqueID(res->getInt("customer_id"));
			user.setUserName(res->getString("customer_username"));
			user.setPhoneNum(res->getInt("customer_phoneNum"));
			user.setEmail(res->getString("customer_email"));
			user.setFirstName(res->getString("customer_firstName"));
			user.setLastName(res->getString("customer_lastName"));
			user.setPassword(res->getString("customer_password"));
			userdb.push_back(user);
		}
		return userdb;
	}