proxysql/include/mysql_backend.h

/*

MySQL_Backend module

Classes:
MySQL_Server
MySQL_Hostgroup_Entry
MySQL_Hostgroup
MySQL_HostGroup_Handler


MySQL_HostGroup_Handler is a collections of MySQL_Hostgroup and MySQL_Server and function for their handling
MySQL_Hostgroup is a collections of MySQL_Hostgroup_Entry
MySQL_Hostgroup_Entry is a reference to MySQL_Server with further data


In other words:
- MySQL_Hostgroup_Entry links a MySQL_Server to a MySQL_Hostgroup
- MySQL_Hostgroups is a high-level class for the management of several MySQL_Hostgroup and several MySQL_Server

Classes MySQL_Server and MySQL_Hostgroup_Entry have very few functions, mainly constructor and deconstructor






MySQL_Hostgroup has the following private functions:
void add(MySQL_Hostgroup_Entry *)
		adds a MySQL_Hostgroup_Entry
void add(MySQL_Server *)
		creates a new MySQL_Hostgroup_Entry from a given MySQL_Server and adds it
void del(MySQL_Hostgroup_Entry *mshge)
		if exists, delete a given MySQL_Hostgroup_Entry
bool del(MySQL_Server *msptr)
		if exists, delete the MySQL_Hostgroup_Entry pointing to the given MySQL_Server

MySQL_Hostgroup has the following public functions:
MySQL_Hostgroup_Entry * MSHGE_find(MySQL_Server *)
		finds a MySQL_Hostgroup_Entry from a given MySQL_Server
MySQL_Hostgroup_Entry * server_add(MySQL_Server *)
		if the server is not already in the Hostgroup,
		creates a new MySQL_Hostgroup_Entry from a given MySQL_Server and adds it
size_t servers_in_hostgroup()
		returns the number of MySQL_Hostgroup_Entry
MySQL_Hostgroup_Entry * set_HG_entry_status(MySQL_Server *, enum proxysql_server_status)
		If a MySQL_Hostgroup_Entry exists using the given MySQL_Server, set the server_status and returns it




MySQL_HostGroups has the following private functions:
void reset()
		deletes all hostgroups and servers. This is called by the destructor
void insert_hostgroup(MySQL_Hostgroup *)
		adds a new hostgroup
void create_hostgroup(unsigned int)
		creates a new hostgroup and calls insert_hostgroup to add it
MySQL_Server * server_add(MySQL_Server *)
		addes and returns given MySQL_Server

MySQL_HostGroups has the following public functions:
size_t servers_in_hostgroup(int)
		returns the number of MySQL_Hostgroup_Entry from a given hostgroup id
MySQL_Server * server_find(char *, uint16_t)
		if exists, returns the MySQL_Server with the given address/port
MySQL_Server * server_add(char *add, uint16_t)
		if not exists, creates and add a new MySQL_Server with the given address/port
		returns the existing or new MySQL_Server
MySQL_Hostgroup_Entry * server_add_hg(unsigned int, char *add, uint16_t)
		adds a new server if doesn't exist, and then adds it the correct hostgroup
MySQL_Hostgroup_Entry * MSHGE_find(unsigned int hid, MySQL_Server *srv)
		finds and returns a MySQL_Hostgroup_Entry from a given MySQL_Server in the specified hostgroup id
MySQL_Hostgroup_Entry * MSHGE_find(unsigned int, char *, uint16_t)
		finds and returns a MySQL_Hostgroup_Entry from a given address/port in the specified hostgroup id




*/

#ifndef __CLASS_MYSQL_BACKEND_H
#define __CLASS_MYSQL_BACKEND_H
#include "proxysql.h"
#include "cpp.h"
/*
class MySQL_Server {
	public:
	char *address;
	uint16_t port;
	uint16_t flags;
	unsigned int hostgroup_references;
	unsigned int connections;
	unsigned char alive;
	MySQL_Server(char *add=NULL, uint16_t p=3306) {
		address=( add ? strdup(add) : NULL);
		port=p;
		flags=0;
		hostgroup_references=0;
		connections=0;
		alive=0;
	};
	void set_addr(char *add) {
		if (address) free(address);
		address=( add ? strdup(add) : NULL);
	};
	~MySQL_Server() {
		if (address) free(address);
	};
};
*/
/*
class MySQL_Hostgroup_Entry {
	public:
	unsigned int hostgroup_id;
	MySQL_Server *MSptr;
	unsigned int weight;
	unsigned int connections_created;
	unsigned int connections_active;
	unsigned int references;
	enum proxysql_server_status status;
	unsigned long long status_change_time;
	MySQL_Hostgroup_Entry(unsigned int hid, MySQL_Server *msptr, unsigned int _weight) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Creating MySQL_Hostgroup_Entry for HID %d and MySQL_Server %p at %p\n", hid, msptr, this);
		hostgroup_id=hid;
		MSptr=msptr;
		weight=_weight;
		connections_created=0;
		connections_active=0;
		references=0;
		MSptr->hostgroup_references++;
		status=PROXYSQL_SERVER_STATUS_OFFLINE_HARD;
	};
	~MySQL_Hostgroup_Entry() {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Destroying MySQL_Hostgroup_Entry at %p for HID %d and MySQL_Server %p\n", this, hostgroup_id, MSptr);
		MSptr->hostgroup_references--;
	};
	enum proxysql_server_status set_status(enum proxysql_server_status _s) {
		if (status != _s) {
			status=_s;
			status_change_time=monotonic_time();
		};
		return status;
	};
};

class MySQL_Hostgroup {
	private:
	void add(MySQL_Hostgroup_Entry *mshge);
	void add(MySQL_Server *msptr, unsigned int _weight=1);
	bool del(MySQL_Hostgroup_Entry *mshge);
	bool del(MySQL_Server *msptr);
*//*
	void add(MySQL_Hostgroup_Entry *mshge) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Adding MySQL_Hostgroup_Entry %p to Hostgroup %p with HID %d\n", mshge, this, hostgroup_id);	
		MSHGEs.push_back(mshge);
	};
	void add(MySQL_Server *msptr, unsigned int _weight=1) {
		MySQL_Hostgroup_Entry *mshge=new MySQL_Hostgroup_Entry(hostgroup_id, msptr, _weight);
		this->add(mshge);
	};
	bool del(MySQL_Hostgroup_Entry *mshge) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Deleting MySQL_Hostgroup_Entry %p from Hostgroup %p with HID %d\n", mshge, this, hostgroup_id);	
		std::vector<MySQL_Hostgroup_Entry *>::iterator it = find(MSHGEs.begin(), MSHGEs.end(), mshge);
		if (it != MSHGEs.end()) {
			MSHGEs.erase(it);
			delete mshge;
			return true;
		}
		return false;
	};
	bool del(MySQL_Server *msptr) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Deleting MySQL_Hostgroup_Entry with MySQL_Server %p from Hostgroup %p with HID %d\n", msptr, this, hostgroup_id);	
		for (std::vector<MySQL_Hostgroup_Entry *>::iterator it = MSHGEs.begin() ; it != MSHGEs.end(); ++it) {
			MySQL_Hostgroup_Entry *mshge=*it;
			if (mshge->MSptr==msptr) {
				MSHGEs.erase(it);
				delete mshge;
				return true;
			}
		}
		return false;
	};
*//*
	public:
	unsigned int hostgroup_id;
	//std::vector<MySQL_Hostgroup_Entry *> MSHGEs;
	PtrArray *MSHGEs;
	MySQL_Hostgroup(unsigned int hid) {
		hostgroup_id=hid;
		MSHGEs = new PtrArray();
	};
	~MySQL_Hostgroup() {
//		for (std::vector<MySQL_Hostgroup_Entry *>::iterator it = MSHGEs.begin() ; it != MSHGEs.end(); ++it) {
//			MySQL_Hostgroup_Entry *mshge=*it;
//			delete mshge;
//		};
		while (MSHGEs->len) {
			MySQL_Hostgroup_Entry *mshge=(MySQL_Hostgroup_Entry *)MSHGEs->remove_index_fast(0);
			delete mshge;
		}
	};
	MySQL_Hostgroup_Entry * MSHGE_find(MySQL_Server *msptr);
*//*
	MySQL_Hostgroup_Entry * MSHGE_find(MySQL_Server *msptr) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Searching MySQL_Hostgroup_Entry for MySQL_Server %p from Hostgroup %p with HID %d\n", msptr, this, hostgroup_id);	
		for (std::vector<MySQL_Hostgroup_Entry *>::iterator it = MSHGEs.begin(); it != MSHGEs.end(); ++it) {
			MySQL_Hostgroup_Entry *mshge=*it;
			if (mshge->MSptr==msptr) {
				proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Found MySQL_Hostgroup_Entry %p\n",mshge);
				return mshge;
			};
		}
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "MySQL_Hostgroup_Entry not found\n");
		return NULL;
	};
*//*
	MySQL_Hostgroup_Entry * server_add(MySQL_Server *msptr, unsigned int _weight) {
		MySQL_Hostgroup_Entry *mshge=NULL;
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Trying to add MySQL_Server %p to MSHGE %p with HID %d\n",msptr, this, hostgroup_id);	
		mshge=MSHGE_find(msptr);
		if (mshge==NULL) {
			proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "MySQL_Server not found, adding to MSHGE %p\n", this);
			MySQL_Hostgroup_Entry *mshge=new MySQL_Hostgroup_Entry(hostgroup_id, msptr, _weight);
			this->add(mshge);
		} else {
			proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "MySQL_Server found at %p, updating weight from %d to %d\n", mshge, mshge->weight, _weight);
			mshge->weight=_weight;
		}
		return mshge;
	};
	size_t servers_in_hostgroup();
//	size_t servers_in_hostgroup() {
//		return MSHGEs.size();
//	};
	void set_HG_entry_status(MySQL_Hostgroup_Entry *mshge, enum proxysql_server_status _status) {
		mshge->set_status(_status);
	};
	MySQL_Hostgroup_Entry * set_HG_entry_status(MySQL_Server *msptr, enum proxysql_server_status _status) {
		MySQL_Hostgroup_Entry *mshge=NULL;
		mshge=MSHGE_find(msptr);
		if (mshge) {
			set_HG_entry_status(mshge,_status);
			//mshge->status=_status;
		}
		return mshge;
	};
	MySQL_Hostgroup_Entry * get_random_hostgroup_entry();
};

class MySQL_HostGroups_Handler {
	private:
	pthread_rwlock_t rwlock;
	//std::vector<MySQL_Hostgroup *> MyHostGroups;
	//std::vector<MySQL_Server *> Servers;
	PtrArray * MyHostGroups;
	PtrArray * Servers;
	void reset();
*//*
	void reset() {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Resetting all MSHGE from Global Handler\n");
		if (MyHostGroups.empty()==false) MyHostGroups.clear();
//		for (std::vector<MySQL_Hostgroup *>::iterator it = MyHostGroups.begin(); it != MyHostGroups.end(); ++it) {
//			MySQL_Hostgroup *myhg=*it;
//			delete myhg;
//		}
//		if (Servers.empty()==false) Servers.clear();
			while (Servers->len) {
				MySQL_Server *srv=(MySQL_Server *)Servers->remove_index_fast(0);
				delete srv;
			}
//		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Resetting all MySQL Server from Global Handler\n");	
//		for (std::vector<MySQL_Server *>::iterator it = Servers.begin(); it != Servers.end(); ++it) {
//			MySQL_Server *srv=*it;
//			delete srv;
//		}
	};
*//*
	MySQL_Hostgroup * MyHostGroups_idx(unsigned int hid);
	void insert_hostgroup(MySQL_Hostgroup *myhg);
*//*
	void insert_hostgroup(MySQL_Hostgroup *myhg) {
		unsigned int p=myhg->hostgroup_id;
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Inserting hostgroup %p with HID %d in Global Handler\n", myhg, p);
		if (MyHostGroups.capacity() < (p+1) ) {
			MyHostGroups.reserve(p);
		}
		//MyHostGroups.insert(MyHostGroups.begin()+p,myhg);
		MyHostGroups[p]=myhg;
	};
*//*
	void create_hostgroup(unsigned int hid) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Creating hostgroup %d in Global Handler\n", hid);	
		MySQL_Hostgroup *myhg=new MySQL_Hostgroup(hid);	
		insert_hostgroup(myhg);
	};
	MySQL_Server * server_add(MySQL_Server *srv);
//	MySQL_Server * server_add(MySQL_Server *srv) {
//		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Adding MySQL_Server %p in Global Handler\n", srv);	
//		Servers.push_back(srv);
//		return srv;
//	};
	public:
	MySQL_HostGroups_Handler() {
		Servers = new PtrArray();
		MyHostGroups = new PtrArray();
		int rc=pthread_rwlock_init(&rwlock, NULL);
		assert(rc==0);
	};
	~MySQL_HostGroups_Handler() {
		reset();
		delete Servers;
	};
	void wrlock() {
		int rc=pthread_rwlock_wrlock(&rwlock);
		assert(rc==0);
	};
	void rdlock() {
		int rc=pthread_rwlock_rdlock(&rwlock);
		assert(rc==0);
	};
	void wrunlock() {
		int rc=pthread_rwlock_unlock(&rwlock);
		assert(rc==0);
	};
	void rdunlock() {
		int rc=pthread_rwlock_unlock(&rwlock);
		assert(rc==0);
	};
	size_t servers_in_hostgroup(int hid);
//	size_t servers_in_hostgroup(int hid) {
//		return MyHostGroups[hid]->servers_in_hostgroup();
//	};
	MySQL_Server * server_find(char *add, uint16_t p);
*//*
	MySQL_Server * server_find(char *add, uint16_t p) {
		for (std::vector<MySQL_Server *>::iterator it = Servers.begin(); it != Servers.end(); ++it) {
			MySQL_Server *srv=*it;
			if (strcmp(srv->address,add)==0 && srv->port==p) {
				return srv;
			};
		}
		return NULL;
	};
*//*
	MySQL_Server * server_add(char *add=NULL, uint16_t p=3306) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Adding MySQL server %s:%d in Global Handler\n", add, p);
		MySQL_Server *srv=NULL;
		srv=server_find(add,p);
		if (srv==NULL) {
			srv=new MySQL_Server(add,p);
			server_add(srv);
		}
		return srv;
	};
*//*
	MySQL_Hostgroup_Entry * server_add_hg(unsigned int hid, char *add=NULL, uint16_t p=3306, unsigned int _weight=1) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Adding MySQL server %s:%d in Global Handler in hostgroup\n", add, p, hid);
		if (hid>=MyHostGroups.size()) {
			create_hostgroup(hid);
		};
		MySQL_Server *srv=server_add(add,p);
		return MyHostGroups[hid]->server_add(srv, _weight);
	};
	MySQL_Hostgroup_Entry * MSHGE_find(unsigned int hid, MySQL_Server *srv) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Searching MSHGE for MySQL_Server %p into HID %d\n", srv, hid);
		return MyHostGroups[hid]->MSHGE_find(srv);
	};
	MySQL_Hostgroup_Entry * MSHGE_find(unsigned int hid, char *add, uint16_t p) {
		proxy_debug(PROXY_DEBUG_MYSQL_CONNPOOL, 4, "Searching MSHGE for MySQL server %s:%d into HID %d\n", add, p, hid);
		MySQL_Server *srv=server_find(add,p);
		if (srv==NULL) return NULL;	// server not found
		return MSHGE_find(hid,srv);
	};
*//*
	MySQL_Hostgroup_Entry * server_add_hg(unsigned int hid, char *add=NULL, uint16_t p=3306, unsigned int _weight=1);
	MySQL_Hostgroup_Entry * MSHGE_find(unsigned int hid, MySQL_Server *srv);
	MySQL_Hostgroup_Entry * MSHGE_find(unsigned int hid, char *add, uint16_t p);
	MySQL_Hostgroup_Entry * set_HG_entry_status(unsigned int hid, MySQL_Server *msptr, enum proxysql_server_status _status);
//	MySQL_Hostgroup_Entry * set_HG_entry_status(unsigned int hid, MySQL_Server *msptr, enum proxysql_server_status _status) {
//		return MyHostGroups[hid]->set_HG_entry_status(msptr,_status);
//	};
	MySQL_Hostgroup_Entry * set_HG_entry_status(unsigned int hid, char *add, uint16_t p, enum proxysql_server_status _status);
*//*
	MySQL_Hostgroup_Entry * set_HG_entry_status(unsigned int hid, char *add, uint16_t p, enum proxysql_server_status _status) {
		MySQL_Server *msptr=server_find(add,p);
		if (msptr==NULL) return NULL;	// server not found
		return MyHostGroups[hid]->set_HG_entry_status(msptr,_status);
	};
*//*
	MySQL_Hostgroup_Entry * get_random_hostgroup_entry(unsigned int hid);
};
*/


class MySQL_Backend
{
	public:
	void * operator new(size_t);
	void operator delete(void *);
	int hostgroup_id;
	MySQL_Data_Stream *server_myds;
//  mysql_cp_entry_t *server_mycpe;
  bytes_stats_t server_bytes_at_cmd;
	//MySQL_Hostgroup_Entry *mshge;
	//MySQL_Connection *myconn;
	MySQL_Backend();
	~MySQL_Backend();
	void reset();
};

#endif /* __CLASS_MYSQL_BACKEND_H */