Fixed tests.

[stg.git] / projects / stargazer / plugins / store / firebird / firebird_store.cpp

diff --git a/projects/stargazer/plugins/store/firebird/firebird_store.cpp b/projects/stargazer/plugins/store/firebird/firebird_store.cpp
index 9cc7c3048b3462d088cd91ee21c94c6f307f4603..3c4f32adf50b1560d2d8fa14427e6f479e4da087 100644 (file)
--- a/projects/stargazer/plugins/store/firebird/firebird_store.cpp
+++ b/projects/stargazer/plugins/store/firebird/firebird_store.cpp
@@ -26,13 +26,16 @@
  *
  */
 
  *
  */
 

+#include "firebird_store.h"
+
+#include "stg/ibpp.h"
+#include "stg/plugin_creator.h"
+

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

-#include "stg/ibpp.h"
-#include "stg/plugin_creator.h"
-#include "firebird_store.h"
+#include <cctype>

 
 namespace
 {
 
 namespace
 {


@@ -60,6 +63,7 @@ FIREBIRD_STORE::FIREBIRD_STORE()
       mutex(),
       til(IBPP::ilConcurrency),
       tlr(IBPP::lrWait),
       mutex(),
       til(IBPP::ilConcurrency),
       tlr(IBPP::lrWait),

+      schemaVersion(0),

       logger(GetPluginLogger(GetStgLogger(), "store_firebird"))
 {
 pthread_mutex_init(&mutex, NULL);
       logger(GetPluginLogger(GetStgLogger(), "store_firebird"))
 {
 pthread_mutex_init(&mutex, NULL);


@@ -78,55 +82,41 @@ std::string s;
 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
     {
     s = i->param;
 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
     {
     s = i->param;

-    std::transform(s.begin(), s.end(), s.begin(), ToLower());
+
+    std::transform(s.begin(), s.end(), s.begin(), ::tolower);
+

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

-        {

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

-        }
+

     if (s == "database")
     if (s == "database")

-        {

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

-        }
+

     if (s == "user")
     if (s == "user")

-        {

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

-        }
+

     if (s == "password")
     if (s == "password")

-        {

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

-        }

 
     // Advanced settings block
 
     if (s == "isolationLevel")
         {
         if (*(i->value.begin()) == "Concurrency")
 
     // Advanced settings block
 
     if (s == "isolationLevel")
         {
         if (*(i->value.begin()) == "Concurrency")

-            {

             til = IBPP::ilConcurrency;
             til = IBPP::ilConcurrency;

-            }

         else if (*(i->value.begin()) == "DirtyRead")
         else if (*(i->value.begin()) == "DirtyRead")

-            {

             til = IBPP::ilReadDirty;
             til = IBPP::ilReadDirty;

-            }

         else if (*(i->value.begin()) == "ReadCommitted")
         else if (*(i->value.begin()) == "ReadCommitted")

-            {

             til = IBPP::ilReadCommitted;
             til = IBPP::ilReadCommitted;

-            }

         else if (*(i->value.begin()) == "Consistency")
         else if (*(i->value.begin()) == "Consistency")

-            {

             til = IBPP::ilConsistency;
             til = IBPP::ilConsistency;

-            }

         }
         }

+

     if (s == "lockResolution")
         {
         if (*(i->value.begin()) == "Wait")
     if (s == "lockResolution")
         {
         if (*(i->value.begin()) == "Wait")

-            {

             tlr = IBPP::lrWait;
             tlr = IBPP::lrWait;

-            }

         else if (*(i->value.begin()) == "NoWait")
         else if (*(i->value.begin()) == "NoWait")

-            {

             tlr = IBPP::lrNoWait;
             tlr = IBPP::lrNoWait;

-            }

         }
     }
 
         }
     }
 


@@ -134,6 +124,7 @@ try
     {
     db = IBPP::DatabaseFactory(db_server, db_database, db_user, db_password, "", "KOI8U", "");
     db->Connect();
     {
     db = IBPP::DatabaseFactory(db_server, db_database, db_user, db_password, "", "KOI8U", "");
     db->Connect();

+    return CheckVersion();

     }
 catch (IBPP::Exception & ex)
     {
     }
 catch (IBPP::Exception & ex)
     {


@@ -145,3 +136,37 @@ catch (IBPP::Exception & ex)
 return 0;
 }
 //-----------------------------------------------------------------------------
 return 0;
 }
 //-----------------------------------------------------------------------------

+int FIREBIRD_STORE::CheckVersion()
+{
+IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amRead, til, tlr);
+IBPP::Statement st = IBPP::StatementFactory(db, tr);
+
+try
+    {
+    tr->Start();
+    st->Execute("SELECT RDB$RELATION_NAME FROM RDB$RELATIONS WHERE RDB$SYSTEM_FLAG=0 AND RDB$RELATION_NAME = 'TB_INFO'");
+    if (!st->Fetch())
+        {
+        schemaVersion = 0;
+        }
+    else
+        {
+        st->Execute("SELECT version FROM tb_info");
+        while (st->Fetch())
+            st->Get(1, schemaVersion);
+        }
+    tr->Commit();
+    logger("FIREBIRD_STORE: Current DB schema version: %d", schemaVersion);
+    }
+
+catch (IBPP::Exception & ex)
+    {
+    tr->Rollback();
+    strError = "IBPP exception";
+    printfd(__FILE__, ex.what());
+    return -1;
+    }
+
+return 0;
+}
+//-----------------------------------------------------------------------------