Mathias/sqltoolkit
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Removed the delete feature from the update manager & cleaned up the update manager

Browse Source
This commit is contained in:
mathias 2021-08-19 20:19:03 +02:00
parent cebb2c8c98
commit 42f2b37b8b
No known key found for this signature in database
GPG Key ID: 8950DF62139C852A
1 changed files with 25 additions and 125 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

150
src/main/java/de/gnmyt/SQLToolkit/manager/UpdateManager.java
View File

@ -1,12 +1,10 @@
package de.gnmyt.SQLToolkit.manager;
import de.gnmyt.SQLToolkit.drivers.MySQLConnection;
import de.gnmyt.SQLToolkit.generator.TableGenerator;
import org.slf4j.Logger;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicBoolean;
public class UpdateManager {
@ -41,6 +39,17 @@ public class UpdateManager {
this.setList = new HashMap<>();
}
/**
* Change the tableName
*
* @param tableName New tableName
* @return this class
*/
public UpdateManager toTable(String tableName) {
this.tableName = tableName;
return this;
}
/**
* Get the parameters for the SQL statement
*
@ -53,64 +62,6 @@ public class UpdateManager {
return tempParams;
}
/**
* Get the current connection
*
* @return MySQL connection
*/
public MySQLConnection getConnection() {
return connection;
}
/**
* Print the deletion statement
*
* @return this class
*/
public UpdateManager printDeleteStatement() {
StringBuilder messageBuilder = new StringBuilder();
messageBuilder.append(processDeleteStatement());
messageBuilder.append(" | params» ");
AtomicBoolean added = new AtomicBoolean(false);
whereList.values().forEach(v -> {
messageBuilder.append((added.get()) ? ", " : "").append(v);
added.set(true);
});
LOG.debug("Statement: " + messageBuilder);
return this;
}
/**
* Print the update statement
*
* @return this class
*/
public UpdateManager printUpdateStatement() {
StringBuilder messageBuilder = new StringBuilder();
messageBuilder.append(processUpdateStatement());
messageBuilder.append(" | params» ");
AtomicBoolean added = new AtomicBoolean(false);
getTempParams().forEach(v -> {
messageBuilder.append((added.get()) ? ", " : "").append(v);
added.set(true);
});
StackTraceElement[] st = Thread.currentThread().getStackTrace();
StackTraceElement stack = st[st.length - 1];
LOG.debug("DEBUG <" + stack.getFileName() + ":" + stack.getLineNumber() + "> Statement: " + messageBuilder);
return this;
}
/**
* Change the tableName
*
* @param tableName New tableName
* @return this class
*/
public UpdateManager toTable(String tableName) {
this.tableName = tableName;
return this;
}
/**
* Add a 'where'-clause
*
@ -135,23 +86,13 @@ public class UpdateManager {
return this;
}
/**
* Delete the entries with your current conditions
*
* @return this class
*/
public UpdateManager delete() {
connection.update(processDeleteStatement(), whereList.values().toArray());
return this;
}
/**
* Update the entries with your current conditions
*
* @return this class
*/
public UpdateManager update() {
connection.update(processUpdateStatement(), getTempParams().toArray());
connection.update(prepareUpdateStatement(), getTempParams().toArray());
return this;
}
@ -160,64 +101,23 @@ public class UpdateManager {
*
* @return the Statement
*/
private String processUpdateStatement() {
StringBuilder sb = new StringBuilder();
sb.append("UPDATE ").append(tableName);
if (!setList.isEmpty()) sb.append(" SET ");
AtomicBoolean used = new AtomicBoolean(false);
private String prepareUpdateStatement() {
StringBuilder query = new StringBuilder().append("UPDATE ").append(tableName);
if (!setList.isEmpty()) query.append(" SET ");
setList.forEach((str, obj) -> {
if (used.get()) sb.append(", ");
sb.append(str).append(" = ?");
used.set(true);
});
for (int i = 0; i < setList.size(); i++) {
if (i > 0) query.append(", ");
query.append(setList.keySet().toArray()[i]);
}
if (!whereList.isEmpty()) sb.append(" WHERE ");
AtomicBoolean used2 = new AtomicBoolean(false);
whereList.forEach((str, obj) -> {
if (used2.get()) sb.append(" AND ");
sb.append(str).append(" = ?");
used2.set(true);
});
return sb.toString();
}
if (!whereList.isEmpty()) query.append(" WHERE ");
/**
* Get the Query of the 'delete'-Statement
*
* @return
*/
private String processDeleteStatement() {
StringBuilder sb = new StringBuilder();
sb.append("DELETE FROM ").append(tableName);
if (!whereList.isEmpty()) sb.append(" WHERE ");
AtomicBoolean used = new AtomicBoolean(false);
whereList.forEach((str, obj) -> {
if (used.get()) sb.append(" AND ");
sb.append(str).append(" = ?");
used.set(true);
});
return sb.toString();
}
for (int i = 0; i < whereList.size(); i++) {
if (i > 0) query.append(" AND ");
query.append(whereList.keySet().toArray()[i]).append(" = ?");
}
/**
* Generate a new Table (with prefilled tableName)
*
* @param tableName Name of the new table
* @return the de.gnmyt.SQLToolkit.generator
*/
public TableGenerator generateTable(String tableName) {
return new TableGenerator(this.getConnection(), tableName);
}
/**
* Generate a new Table
*
* @return the de.gnmyt.SQLToolkit.generator
*/
public TableGenerator generateTable() {
return (tableName.isEmpty()) ? null : new TableGenerator(this.getConnection(), tableName);
return query.toString();
}
}