Merge remote-tracking branch 'github/master'

[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 fc15d035fd787883a1da5bf5a4e24c63e2d415cf..2146d66483f32dc9179122828cbe0b146584694d 100644 (file)
--- a/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
+++ b/projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
@@ -38,36 +38,22 @@
  *
  */
 
  *
  */
 

+#include "postgresql_store.h"
+
+#include "postgresql_store_utils.h"
+#include "postgresql_store.h"
+
+#include "stg/common.h" // str2x, printfd
+

 #include <string>
 #include <vector>
 #include <string>
 #include <vector>

-#include <algorithm>

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

-#include "postgresql_store.h"
-#include "postgresql_store_utils.h"
-#include "module_settings.h"
-
-class POSTGRESQL_STORE_CREATOR
-{
-public:
-    POSTGRESQL_STORE_CREATOR()
-        : pqStore(new POSTGRESQL_STORE())
-        {
-        };
-    ~POSTGRESQL_STORE_CREATOR()
-        {
-        delete pqStore;
-        };
-    POSTGRESQL_STORE * GetStore() { return pqStore; };
-private:
-    POSTGRESQL_STORE * pqStore;
-} pqStoreeCreator;
-
-//-----------------------------------------------------------------------------
-STORE * GetStore()
+extern "C" STG::Store* GetStore()

 {
 {

-return pqStoreeCreator.GetStore();
+    static POSTGRESQL_STORE plugin;
+    return &plugin;

 }
 
 //-----------------------------------------------------------------------------
 }
 
 //-----------------------------------------------------------------------------


@@ -77,8 +63,11 @@ POSTGRESQL_STORE::POSTGRESQL_STORE()
       database("stargazer"),
       user("stg"),
       password("123456"),
       database("stargazer"),
       user("stg"),
       password("123456"),

+      clientEncoding("KOI8"),

       version(0),
       version(0),

-      connection(NULL)
+      retries(3),
+      connection(NULL),
+      logger(STG::PluginLogger::get("store_postgresql"))

 {
 pthread_mutex_init(&mutex, NULL);
 }
 {
 pthread_mutex_init(&mutex, NULL);
 }


@@ -94,28 +83,37 @@ pthread_mutex_destroy(&mutex);
 //-----------------------------------------------------------------------------
 int POSTGRESQL_STORE::ParseSettings()
 {
 //-----------------------------------------------------------------------------
 int POSTGRESQL_STORE::ParseSettings()
 {

-std::vector<PARAM_VALUE>::iterator i;
-string s;
+std::vector<STG::ParamValue>::iterator i;

 
 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 (i->value.empty())
+        continue;
+    std::string s = ToLower(i->param);

     if (s == "server")
         {
     if (s == "server")
         {

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

         }
     if (s == "database")
         {
         }
     if (s == "database")
         {

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

         }
     if (s == "user")
         {
         }
     if (s == "user")
         {

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

         }
     if (s == "password")
         {
         }
     if (s == "password")
         {

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

         }
     }
 
         }
     }
 


@@ -138,7 +136,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()))


@@ -153,7 +152,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)
     {


@@ -166,7 +170,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();


@@ -224,6 +228,8 @@ if (CommitTransaction())
     return -1;
     }
 
     return -1;
     }
 

+logger("POSTGRESQL_STORE: Current DB schema version: %d", version);
+

 return 0;
 }
 //-----------------------------------------------------------------------------
 return 0;
 }
 //-----------------------------------------------------------------------------