Added reuse Insert with clearValues()

Author: sea5kg

src/tests/test_insert.cpp

@@ -53,5 +53,25 @@ int main() {
         ;
         return -1;
     }
+
+    builder.findInsertOrCreate("table2")
+        .clearValues()
+        .val("val2")
+        .val(2)
+        .val(10.0)
+    ;
+
+    sqlQuery = builder.sql();
+    sqlQueryExpected = "INSERT INTO table2(col1, col2, col3) VALUES('val2', 2, 10.000000)";
+    if (sqlQuery != sqlQueryExpected) {
+        std::cerr
+            << "Expected:" << std::endl
+            << "   {" << sqlQueryExpected << "}" << std::endl
+            << ", but got:" << std::endl
+            << "   {" << sqlQuery << "}" << std::endl
+        ;
+        return -1;
+    }
+
     return 0;
 }

src/wsjcpp_sql_builder.cpp

@@ -260,6 +260,11 @@ WsjcppSqlInsert &WsjcppSqlInsert::addColums(const std::vector<std::string> &cols
   return *this;
 }
 
+WsjcppSqlInsert &WsjcppSqlInsert::clearValues() {
+  m_values.clear();
+  return *this;
+}
+
 WsjcppSqlInsert &WsjcppSqlInsert::val(const std::string &val) {
   m_values.push_back(WsjcppSqlBuilderHelpers::escapingStringValue(val));
   return *this;
@@ -320,7 +325,16 @@ WsjcppSqlSelect &WsjcppSqlBuilder::selectFrom(const std::string &tableName) {
 
 WsjcppSqlInsert &WsjcppSqlBuilder::insertInto(const std::string &tableName) {
   m_queries.push_back(std::make_shared<WsjcppSqlInsert>(tableName, this));
-  return *(WsjcppSqlInsert *)(m_queries[m_queries.size() -1].get());;
+  return *(WsjcppSqlInsert *)(m_queries[m_queries.size() -1].get());
+}
+
+WsjcppSqlInsert &WsjcppSqlBuilder::findInsertOrCreate(const std::string &tableName) {
+  for (auto query : m_queries) {
+    if (query->sqlType() == WsjcppSqlQueryType::INSERT && query->tableName() == tableName) {
+      return *(WsjcppSqlInsert *)(query.get());
+    }
+  }
+  return insertInto(tableName);
 }
 
 // WsjcppSqlBuilder &WsjcppSqlBuilder::makeUpdate(const std::string &tableName) {

src/wsjcpp_sql_builder.h

@@ -241,6 +241,7 @@ class WsjcppSqlInsert : public WsjcppSqlQuery {
   WsjcppSqlInsert(const std::string &tableName, WsjcppSqlBuilder *builder);
   WsjcppSqlInsert &colum(const std::string &col);
   WsjcppSqlInsert &addColums(const std::vector<std::string> &cols);
+  WsjcppSqlInsert &clearValues();
 
   WsjcppSqlInsert &val(const std::string &col);
   WsjcppSqlInsert &val(int col);
@@ -261,6 +262,7 @@ class WsjcppSqlBuilder {
 
   WsjcppSqlSelect &selectFrom(const std::string &tableName);
   WsjcppSqlInsert &insertInto(const std::string &tableName);
+  WsjcppSqlInsert &findInsertOrCreate(const std::string &tableName);
   // WsjcppSqlBuilder &update(const std::string &sSqlTable);
   // WsjcppSqlBuilder &deleteFrom(const std::string &sSqlTable);