Fixed process of creation update query

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>pl.timsixth</groupId>
     <artifactId>T-DataBasesAPI</artifactId>
-    <version>1.7</version>
+    <version>1.7.1</version>
 
     <repositories>
         <repository>

src/main/java/pl/timsixth/databasesapi/database/query/QueryBuilder.java

@@ -128,11 +128,23 @@ public QueryBuilder update(String table, Map<String, Object> data) {
         query.append("UPDATE ").append(table);
 
         data.forEach((key, value) -> {
-            if (value instanceof String)
-                query.append(" SET ").append(key)
-                        .append(" = ").append("'").append(value).append("'").append(",");
-            else query.append(" SET ").append(key)
-                    .append(" = ").append(value).append(",");
+            if (value instanceof String) {
+                if (!query.toString().contains("SET")) {
+                    query.append(" SET ").append(key)
+                            .append(" = ").append("'").append(value).append("'").append(",");
+                } else {
+                    query.append(key)
+                            .append(" = ").append("'").append(value).append("'").append(",");
+                }
+            } else {
+                if (!query.toString().contains("SET")) {
+                    query.append(" SET ").append(key)
+                            .append(" = ").append(value).append(",");
+                } else {
+                    query.append(key)
+                            .append(" = ").append(value).append(",");
+                }
+            }
         });
 
         query.deleteCharAt(query.lastIndexOf(","));

src/main/resources/plugin.yml

@@ -1,4 +1,4 @@
 name: T-DatabasesApi
 main: pl.timsixth.databasesapi.DatabasesApiPlugin
-version: '1.7'
+version: '1.7.1'
 author: timsixth

src/test/java/pl/timsixth/databasesapi/tests/QueryBuilderTest.java

@@ -73,7 +73,7 @@ public void shouldGenerateUpdateQuery() {
 
         String query = queryBuilder.update("users", data).build();
 
-        assertEquals("UPDATE users SET age = 12, SET username = 'test'", query);
+        assertEquals("UPDATE users SET age = 12,username = 'test'", query);
     }
 
     @Test