From 3f2987b576ddfc93c639a81b3da5335c50aa2f42 Mon Sep 17 00:00:00 2001
From: mathias <germannewsmaker@gmail.com>
Date: Thu, 19 Aug 2021 19:59:58 +0200
Subject: [PATCH] Cleaned up code of the insert manager

---
 .../SQLToolkit/manager/InsertManager.java     | 27 +++++++++----------
 1 file changed, 13 insertions(+), 14 deletions(-)

diff --git a/src/main/java/de/gnmyt/SQLToolkit/manager/InsertManager.java b/src/main/java/de/gnmyt/SQLToolkit/manager/InsertManager.java
index 59353a0..8570a7a 100644
--- a/src/main/java/de/gnmyt/SQLToolkit/manager/InsertManager.java
+++ b/src/main/java/de/gnmyt/SQLToolkit/manager/InsertManager.java
@@ -3,7 +3,6 @@ package de.gnmyt.SQLToolkit.manager;
 import de.gnmyt.SQLToolkit.drivers.MySQLConnection;
 
 import java.util.HashMap;
-import java.util.concurrent.atomic.AtomicBoolean;
 
 public class InsertManager {
 
@@ -60,26 +59,26 @@ public class InsertManager {
      * @return the SQL Query
      */
     public String prepareStatement() {
-        StringBuilder query = new StringBuilder();
-        query.append("INSERT INTO ").append(tableName).append(" ").append("(");
-        AtomicBoolean used = new AtomicBoolean(false);
-        values.forEach((field, object) -> {
-            if (used.get()) query.append(", ");
-            used.set(true);
-            query.append("`").append(field).append("`");
-        });
-        query.append(")");
-        if (values.size() > 0) query.append(" VALUES (");
-        AtomicBoolean used_values = new AtomicBoolean(false);
+        StringBuilder query = new StringBuilder().append("INSERT INTO ").append(tableName).append(" (");
+
         for (int i = 0; i < values.size(); i++) {
-            if (used_values.get()) query.append(", ");
-            used_values.set(true);
+            if (i > 0) query.append(", ");
+            query.append("`").append(values.keySet().toArray()[i]).append("`");
+        }
+
+        query.append(")").append(values.size() > 0 ? "VALUES (" : "");
+
+        for (int i = 0; i < values.size(); i++) {
+            if (i > 0) query.append(", ");
             query.append("?");
         }
+
         if (values.size() > 0) query.append(")");
+
         return query.toString();
     }
 
+
     /**
      * Execute the current SQL query
      *