[NY Flight] Improvements in parser dispatching.

[stg.git] / projects / stargazer / plugins / store / postgresql / postgresql_store.cpp

diff --git a/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp b/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
index e316beafb87d68d4dceb9090b2830cced8793086..5e2d8ae6de9e29aa608686d0cc3351f406355c97 100644 (file)
--- a/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
+++ b/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
@@ -22,7 +22,7 @@
  *  This file contains a realization of a base postgresql-storage plugin class
  *
  *  v. 1.3
  *  This file contains a realization of a base postgresql-storage plugin class
  *
  *  v. 1.3

- *  FreeMb logging on dosconnects added
+ *  FreeMb logging on disconnects added

  *
  *  v. 1.2
  *  Reconnection on faults added
  *
  *  v. 1.2
  *  Reconnection on faults added


@@ -38,47 +38,45 @@
  *
  */
 
  *
  */
 

-#include <string>
-#include <vector>
-#include <algorithm>
+#include "postgresql_store.h"
+
+#include "stg/module_settings.h"
+#include "stg/plugin_creator.h"

 
 #include <libpq-fe.h>
 
 
 #include <libpq-fe.h>
 

-#include "postgresql_store.h"
-#include "postgresql_store_utils.h"
-#include "base_settings.h"
+#include <string>
+#include <vector>

 
 

-class POSTGRESQL_STORE_CREATOR
+namespace

 {
 {

-public:
-    POSTGRESQL_STORE_CREATOR()
-        : pqStore(new POSTGRESQL_STORE())
-        {
-        };
-    ~POSTGRESQL_STORE_CREATOR()
-        {
-        delete pqStore;
-        };
-    POSTGRESQL_STORE * GetStore() { return pqStore; };
-private:
-    POSTGRESQL_STORE * pqStore;
-} pqStoreeCreator;
+PLUGIN_CREATOR<POSTGRESQL_STORE> pgsc;
+}
+
+extern "C" STORE * GetStore();

 
 //-----------------------------------------------------------------------------
 
 //-----------------------------------------------------------------------------

-BASE_STORE * GetStore()
+STORE * GetStore()

 {
 {

-return pqStoreeCreator.GetStore();
+return pgsc.GetPlugin();

 }
 
 //-----------------------------------------------------------------------------
 POSTGRESQL_STORE::POSTGRESQL_STORE()
 }
 
 //-----------------------------------------------------------------------------
 POSTGRESQL_STORE::POSTGRESQL_STORE()

-    : connection(NULL)
+    : versionString("postgresql_store v.1.3"),
+      strError(),
+      server("localhost"),
+      database("stargazer"),
+      user("stg"),
+      password("123456"),
+      clientEncoding("KOI8"),
+      settings(),
+      mutex(),
+      version(0),
+      retries(3),
+      connection(NULL),
+      logger(GetPluginLogger(GetStgLogger(), "store_postgresql"))

 {
 {

-server = "localhost";
-database = "stargazer";
-user = "stg";
-password = "123456";
-versionString = "postgresql_store v.1.3";

 pthread_mutex_init(&mutex, NULL);
 }
 //-----------------------------------------------------------------------------
 pthread_mutex_init(&mutex, NULL);
 }
 //-----------------------------------------------------------------------------


@@ -94,28 +92,25 @@ pthread_mutex_destroy(&mutex);
 int POSTGRESQL_STORE::ParseSettings()
 {
 std::vector<PARAM_VALUE>::iterator i;
 int POSTGRESQL_STORE::ParseSettings()
 {
 std::vector<PARAM_VALUE>::iterator i;

-string s;

 
 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
     {
 
 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
     {

-    s = i->param;
-    std::transform(s.begin(), s.end(), s.begin(), ToLower());
-    if (s == "server")
-        {
+    std::string param(ToLower(i->param));
+    if (param == "server")

         server = *(i->value.begin());
         server = *(i->value.begin());

-        }
-    if (s == "database")
-        {
+    else if (param == "database")

         database = *(i->value.begin());
         database = *(i->value.begin());

-        }
-    if (s == "user")
-        {
+    else if (param == "user")

         user = *(i->value.begin());
         user = *(i->value.begin());

-        }
-    if (s == "password")
-        {
+    else if (param == "password")

         password = *(i->value.begin());
         password = *(i->value.begin());

-        }
+    else if (param == "retries")
+        if (str2x(*(i->value.begin()), retries))
+            {
+            strError = "Invalid 'retries' value";
+            printfd(__FILE__, "POSTGRESQL_STORE::ParseSettings(): '%s'\n", strError.c_str());
+            return -1;
+            }

     }
 
 clientEncoding = "KOI8";
     }
 
 clientEncoding = "KOI8";


@@ -137,7 +132,8 @@ if (PQstatus(connection) != CONNECTION_OK)
     {
     strError = PQerrorMessage(connection);
     printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());
     {
     strError = PQerrorMessage(connection);
     printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());

-    return 1;
+    // Will try to connect later
+    return 0;

     }
 
 if (PQsetClientEncoding(connection, clientEncoding.c_str()))
     }
 
 if (PQsetClientEncoding(connection, clientEncoding.c_str()))


@@ -152,7 +148,12 @@ return CheckVersion();
 //-----------------------------------------------------------------------------
 int POSTGRESQL_STORE::Reset() const
 {
 //-----------------------------------------------------------------------------
 int POSTGRESQL_STORE::Reset() const
 {

-PQreset(connection);
+for (int i = 0; i < retries && PQstatus(connection) != CONNECTION_OK; ++i)
+    {
+    struct timespec ts = {1, 0};
+    nanosleep(&ts, NULL);
+    PQreset(connection);
+    }

 
 if (PQstatus(connection) != CONNECTION_OK)
     {
 
 if (PQstatus(connection) != CONNECTION_OK)
     {


@@ -165,7 +166,7 @@ if (PQsetClientEncoding(connection, clientEncoding.c_str()))
     {
     strError = PQerrorMessage(connection);
     printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());
     {
     strError = PQerrorMessage(connection);
     printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());

-    return 1;
+    return -1;

     }
 
 return CheckVersion();
     }
 
 return CheckVersion();