bugfix getting id after insert

Author: tkvitko

DbManager.cpp

@@ -91,10 +91,15 @@ unsigned int DbManager::insertWord(std::string word)
 {
     try {
         pqxx::transaction<> tx{ *conn };
-        pqxx::result r = tx.exec("insert into words (word)"
-                "values ('" + tx.esc(word) + "') returning id;");
+        pqxx::result r = tx.exec(
+            "insert into words (word)"
+            "values ('" + tx.esc(word) + "') returning id;"
+            );
         tx.commit();
-        return r.inserted_oid();
+        if (!r.empty()) {
+            return r[0][0].as<unsigned int>();
+        }
+        return 0;
 
     } catch(std::exception const& e) {
         return 0;
@@ -105,10 +110,15 @@ unsigned int DbManager::insertUrl(std::string url)
 {
     try {
         pqxx::transaction<> tx{ *conn };
-        pqxx::result r = tx.exec("insert into urls (url)"
-                "values ('" + tx.esc(url) + "') RETURNING id;");
+        pqxx::result r = tx.exec(
+            "insert into urls (url)"
+            "values ('" + tx.esc(url) + "') RETURNING id;"
+            );
         tx.commit();
-        return r.inserted_oid();
+        if (!r.empty()) {
+            return r[0][0].as<unsigned int>();
+        }
+        return 0;
 
     } catch(std::exception const& e) {
         return 0;
@@ -118,29 +128,27 @@ unsigned int DbManager::insertUrl(std::string url)
 bool DbManager::insertPresence(WordPresence presence)
 {
     try {
-        pqxx::transaction<> tx{ *conn };
         std::string word = presence.word;
         std::string url = presence.url;
         unsigned short frequency = presence.frequency;
 
         // если слова нет в таблице слов, добавляем
         unsigned int wordId = getWordId(word);
         if (wordId == 0) {
-            wordId = insertWord(word); // здесь почему-то всегда 0, потому получаем отдельным запросом ниже
-            wordId = getWordId(word);
+            wordId = insertWord(word);
         }
 
         // если ресурса нет в таблице ресурсов, добавляем
         unsigned int urlId = getUrlId(url);
         if (urlId == 0) {
-            urlId = insertWord(url); // здесь почему-то всегда 0, потому получаем отдельным запросом ниже
-            urlId = getUrlId(url);
+            urlId = insertUrl(url);
         }
 
         std::string wordIdStr = std::to_string(wordId);
         std::string urlIdStr = std::to_string(urlId);
 
         // добавляем новую частоту слова на ресурсе
+        pqxx::transaction<> tx{ *conn };
         tx.exec("insert into frequencies (word_id, url_id, frequency)"
                 "values ('" + tx.esc(wordIdStr) + "', '" + tx.esc(urlIdStr) + "', '" + tx.esc(std::to_string(frequency)) + "') RETURNING id;");
         tx.commit();

main.cpp

@@ -31,16 +31,17 @@ int main()
 
     // dbManager.insertWord("привет");
     // dbManager.insertWord("мир");
-    // dbManager.insertWord("пока");
+    // unsigned int id = dbManager.insertWord("пока3");
+    // std::cout << id << std::endl;
     // dbManager.insertUrl("www.yandex.ru");
     // dbManager.insertUrl("www.google.com");
     // dbManager.insertUrl("www.yahoo.com");
 
-    unsigned int id = dbManager.getWordId("привет");
-    std::cout << id << std::endl;
+    // unsigned int id = dbManager.getWordId("привет");
+    // std::cout << id << std::endl;
 
-    std::string url = "www.yandex.ru";
-    std::string word = "привет";
+    std::string url = "www.yandex3.ru";
+    std::string word = "новое3";
     unsigned short frequency = 2;
     WordPresence presence = {word, url, frequency};
     dbManager.insertPresence(presence);