From 80928395ccd5843cbb33c8ffa6e2d0f72928f87a Mon Sep 17 00:00:00 2001
From: Rene Cannao <rene@proxysql.com>
Date: Sun, 7 Jul 2024 15:12:11 +0000
Subject: [PATCH] Template for reset_all_backends()

---
 include/Base_Session.h  |  1 +
 include/MySQL_Session.h |  2 +-
 include/PgSQL_Session.h |  2 +-
 lib/Base_Session.cpp    | 23 +++++++++++++++++++++++
 lib/MySQL_Session.cpp   |  2 +-
 lib/PgSQL_Session.cpp   |  2 +-
 6 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/include/Base_Session.h b/include/Base_Session.h
index fd6176bf7..92ef0f15f 100644
--- a/include/Base_Session.h
+++ b/include/Base_Session.h
@@ -112,6 +112,7 @@ class Base_Session {
 	virtual void RequestEnd(DS *) = 0;
 	virtual void SQLite3_to_MySQL(SQLite3_result*, char*, int, MySQL_Protocol*, bool in_transaction = false, bool deprecate_eof_active = false) = 0;
 	bool has_any_backend();
+	void reset_all_backends();
 };
 
 #endif // CLASS_BASE_SESSION_H
diff --git a/include/MySQL_Session.h b/include/MySQL_Session.h
index 6c878d662..6272e1375 100644
--- a/include/MySQL_Session.h
+++ b/include/MySQL_Session.h
@@ -416,7 +416,7 @@ class MySQL_Session: public Base_Session<MySQL_Session, MySQL_Data_Stream, MySQL
 	int FindOneActiveTransaction(bool check_savepoint=false);
 	unsigned long long IdleTime();
 
-	void reset_all_backends();
+	//void reset_all_backends();
 	//void writeout();
 	void Memory_Stats();
 	void create_new_session_and_reset_connection(MySQL_Data_Stream *_myds);
diff --git a/include/PgSQL_Session.h b/include/PgSQL_Session.h
index 62b6d7373..01b3b5d4e 100644
--- a/include/PgSQL_Session.h
+++ b/include/PgSQL_Session.h
@@ -394,7 +394,7 @@ public:
 	int FindOneActiveTransaction(bool check_savepoint = false);
 	unsigned long long IdleTime();
 
-	void reset_all_backends();
+	//void reset_all_backends();
 	//void writeout();
 	void Memory_Stats();
 	void create_new_session_and_reset_connection(PgSQL_Data_Stream* _myds);
diff --git a/lib/Base_Session.cpp b/lib/Base_Session.cpp
index 36cf8a38c..8118e0301 100644
--- a/lib/Base_Session.cpp
+++ b/lib/Base_Session.cpp
@@ -48,6 +48,9 @@ template void Base_Session<PgSQL_Session, PgSQL_Data_Stream, PgSQL_Backend, PgSQ
 template bool Base_Session<MySQL_Session, MySQL_Data_Stream, MySQL_Backend, MySQL_Thread>::has_any_backend();
 template bool Base_Session<PgSQL_Session, PgSQL_Data_Stream, PgSQL_Backend, PgSQL_Thread>::has_any_backend();
 
+template void Base_Session<MySQL_Session, MySQL_Data_Stream, MySQL_Backend, MySQL_Thread>::reset_all_backends();
+template void Base_Session<PgSQL_Session, PgSQL_Data_Stream, PgSQL_Backend, PgSQL_Thread>::reset_all_backends();
+
 template<typename S, typename DS, typename B, typename T>
 Base_Session<S,DS,B,T>::Base_Session() {
 };
@@ -347,3 +350,23 @@ bool Base_Session<S,DS,B,T>::has_any_backend() {
 	}
 	return false;
 }
+
+
+
+
+/**
+ * @brief Reset all MySQL backends associated with this session.
+ * 
+ * This function resets all MySQL backends associated with the current session.
+ * It iterates over all backends stored in the session, resets each backend, and then deletes it.
+ * 
+ */
+template<typename S, typename DS, typename B, typename T>
+void Base_Session<S,DS,B,T>::reset_all_backends() {
+	B *mybe;
+	while(mybes->len) {
+		mybe=(B *)mybes->remove_index_fast(0);
+		mybe->reset();
+		delete mybe;
+	}
+};
diff --git a/lib/MySQL_Session.cpp b/lib/MySQL_Session.cpp
index 6a486fcf4..435e15820 100644
--- a/lib/MySQL_Session.cpp
+++ b/lib/MySQL_Session.cpp
@@ -822,6 +822,7 @@ void MySQL_Session::update_expired_conns(const vector<function<bool(MySQL_Connec
 	}
 }
 
+#if 0
 /**
  * @brief Reset all MySQL backends associated with this session.
  * 
@@ -838,7 +839,6 @@ void MySQL_Session::reset_all_backends() {
 	}
 };
 
-#if 0
 /**
  * @brief Writes data from the session to the network with optional throttling and flow control.
  *
diff --git a/lib/PgSQL_Session.cpp b/lib/PgSQL_Session.cpp
index f88342a15..d228b6a16 100644
--- a/lib/PgSQL_Session.cpp
+++ b/lib/PgSQL_Session.cpp
@@ -732,6 +732,7 @@ void PgSQL_Session::update_expired_conns(const vector<function<bool(PgSQL_Connec
 	}
 }
 
+#if 0
 void PgSQL_Session::reset_all_backends() {
 	PgSQL_Backend* mybe;
 	while (mybes->len) {
@@ -741,7 +742,6 @@ void PgSQL_Session::reset_all_backends() {
 	}
 };
 
-#if 0
 void PgSQL_Session::writeout() {
 	int tps = 10; // throttling per second , by default every 100ms
 	int total_written = 0;