From f69f3fba104a2209a4c5fce8adfb7d7739463efe Mon Sep 17 00:00:00 2001
From: iprunache <ionut@vitaminsoftware.com>
Date: Fri, 4 Mar 2016 13:18:15 +0200
Subject: [PATCH] #498 Add SET command parsing method stub

---
 lib/ProxySQL_Admin.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/lib/ProxySQL_Admin.cpp b/lib/ProxySQL_Admin.cpp
index 21603a533..1d29e194f 100644
--- a/lib/ProxySQL_Admin.cpp
+++ b/lib/ProxySQL_Admin.cpp
@@ -434,6 +434,19 @@ bool admin_handler_command_proxysql(char *query_no_space, unsigned int query_no_
 	return true;
 }
 
+// This method translates a 'SET variable=value' command into an equivalent UPDATE.
+//
+// It modifies the original query.
+bool admin_handler_command_set(char *query_no_space, unsigned int query_no_space_length, MySQL_Session *sess, ProxySQL_Admin *pa, char **q, unsigned int *ql) {
+	proxy_debug(PROXY_DEBUG_ADMIN, 4, "Received command %s\n", query_no_space);
+
+	// TODO(iprunache) - parse command to extract variables and values (,=)
+	// TODO(iprunache) - build UPDATE command
+	// TODO(iprunache) - run UPDATE command
+
+	return true;
+}
+
 /* Note:
  * This function can modify the original query
  */
@@ -1420,6 +1433,13 @@ void admin_session_handler(MySQL_Session *sess, ProxySQL_Admin *pa, PtrSize_t *p
 		sprintf(query,q,PROXYSQL_VERSION);
 		goto __run_query;
 	}
+
+	if (!strncasecmp("SET ", query_no_space, 4)) {
+		proxy_debug(PROXY_DEBUG_ADMIN, 4, "Received SET\n");
+		run_query = admin_handler_command_set(query_no_space, query_no_space_length, sess, pa, &query, &query_length);
+		goto __run_query;
+	}
+
 	if(!strncasecmp("CHECKSUM ", query_no_space, 9)){
 		proxy_debug(PROXYSQL_DEBUG_ADMIN, 4, "Received CHECKSUM command\n");
 		ProxySQL_Admin *SPA=(ProxySQL_Admin *)pa;