From f34fa290b56c53a646e38fb5bdbdd3212115e8d8 Mon Sep 17 00:00:00 2001
From: David Hall <david.hall@mariadb.com>
Date: Thu, 1 Nov 2018 11:04:41 -0500
Subject: [PATCH] MCOL-1642 Add SQL command that shows Primary Front-End
 MariaDB ColumnStore Module

---
 dbcon/mysql/ha_calpont_impl.cpp      | 52 ++++++++++++++++++++++++++++
 dbcon/mysql/install_calpont_mysql.sh |  1 +
 2 files changed, 53 insertions(+)

diff --git a/dbcon/mysql/ha_calpont_impl.cpp b/dbcon/mysql/ha_calpont_impl.cpp
index ad83f48db..3f6d6912c 100644
--- a/dbcon/mysql/ha_calpont_impl.cpp
+++ b/dbcon/mysql/ha_calpont_impl.cpp
@@ -2096,6 +2096,58 @@ extern "C"
     {
     }
 
+#ifdef _MSC_VER
+    __declspec(dllexport)
+#endif
+// Return non-zero if this is the primary UM; 0 if not primary
+    long long mcssystemprimary(UDF_INIT* initid, UDF_ARGS* args,
+                                char* is_null, char* error)
+    {
+        long long rtn = 0;
+        Oam oam;
+        string PrimaryUMModuleName;
+        string localModule;
+        oamModuleInfo_t st;
+
+        try
+        {
+            st = oam.getModuleInfo();
+            localModule = boost::get<0>(st);
+            PrimaryUMModuleName = config::Config::makeConfig()->getConfig("SystemConfig", "PrimaryUMModuleName");
+
+            if (boost::iequals(localModule, PrimaryUMModuleName))
+                rtn = 1;
+            if (PrimaryUMModuleName == "unassigned")
+                rtn = 1;
+        }
+        catch (runtime_error& e)
+        {
+            // It's difficult to return an error message from a numerical UDF
+            //string msg = string("ERROR: Problem getting Primary UM Module Name. ") + e.what();
+            *error = 1;
+        }
+        catch (...)
+        {
+            *error = 1;
+        }
+        return rtn;
+    }
+
+#ifdef _MSC_VER
+    __declspec(dllexport)
+#endif
+    my_bool mcssystemprimary_init(UDF_INIT* initid, UDF_ARGS* args, char* message)
+    {
+        return 0;
+    }
+
+#ifdef _MSC_VER
+    __declspec(dllexport)
+#endif
+    void mcssystemprimary_deinit(UDF_INIT* initid)
+    {
+    }
+
 #define MAXSTRINGLENGTH 50
 
     const char* PmSmallSideMaxMemory = "pmmaxmemorysmallside";
diff --git a/dbcon/mysql/install_calpont_mysql.sh b/dbcon/mysql/install_calpont_mysql.sh
index 589894c01..366d1ae85 100755
--- a/dbcon/mysql/install_calpont_mysql.sh
+++ b/dbcon/mysql/install_calpont_mysql.sh
@@ -84,6 +84,7 @@ CREATE FUNCTION idbpartition RETURNS STRING soname 'libcalmysql.so';
 CREATE FUNCTION idblocalpm RETURNS INTEGER soname 'libcalmysql.so';
 CREATE FUNCTION mcssystemready RETURNS INTEGER soname 'libcalmysql.so';
 CREATE FUNCTION mcssystemreadonly RETURNS INTEGER soname 'libcalmysql.so';
+CREATE FUNCTION mcssystemprimary RETURNS INTEGER soname 'libcalmysql.so';
 CREATE AGGREGATE FUNCTION regr_avgx RETURNS REAL soname 'libregr_mysql.so';
 CREATE AGGREGATE FUNCTION regr_avgy RETURNS REAL soname 'libregr_mysql.so';
 CREATE AGGREGATE FUNCTION regr_count RETURNS INTEGER soname 'libregr_mysql.so';