Added setters to scheduler interface

sql/scheduler.h

@@ -96,8 +96,10 @@ class thd_scheduler
   ~thd_scheduler();
 };
 
-void *thd_get_scheduler(THD *thd);
+void *thd_get_scheduler_data(THD *thd);
+void thd_set_scheduler_data(THD *thd, void *data);
 PSI_thread* thd_get_psi(THD *thd);
+void thd_set_psi(THD *thd, PSI_thread *psi);
 
 extern scheduler_functions *thread_scheduler;
 

sql/sql_class.cc

@@ -217,16 +217,26 @@ bool foreign_key_prefix(Key *a, Key *b)
 ** Thread specific functions
 ****************************************************************************/
 
-void* thd_get_scheduler(THD *thd)
+void *thd_get_scheduler_data(THD *thd)
 {
   return thd->scheduler.data;
 }
 
-PSI_thread* thd_get_psi(THD *thd)
+void thd_set_scheduler_data(THD *thd, void *data)
+{
+  thd->scheduler.data= data;
+}
+
+PSI_thread *thd_get_psi(THD *thd)
 {
   return thd->scheduler.m_psi;
 }
 
+void thd_set_psi(THD *thd, PSI_thread *psi)
+{
+  thd->scheduler.m_psi= psi;
+}
+
 /*
   The following functions form part of the C plugin API
 */