Fix local variables in sql queries within the same tick & first startup errors

This commit is contained in:
Govindas 2020-07-31 11:50:44 +03:00
parent 11063c166d
commit a63a4c6d6f
4 changed files with 59 additions and 51 deletions

View File

@ -1,5 +1,5 @@
group 'com.btk5h.skript-db' group 'com.btk5h.skript-db'
version '1.1.0' version '1.2.0'
buildscript { buildscript {
repositories { repositories {

View File

@ -91,6 +91,11 @@ public final class SkriptDB extends JavaPlugin {
@Override @Override
public void onEnable() { public void onEnable() {
try {
setupConfig();
} catch (IOException e) {
e.printStackTrace();
} finally {
try { try {
rowSetFactory = RowSetProvider.newFactory(); rowSetFactory = RowSetProvider.newFactory();
@ -100,10 +105,6 @@ public final class SkriptDB extends JavaPlugin {
} catch (IOException e) { } catch (IOException e) {
e.printStackTrace(); e.printStackTrace();
} }
try {
setupConfig();
} catch (IOException e) {
e.printStackTrace();
} }
} }

View File

@ -40,7 +40,7 @@ import java.util.concurrent.Executors;
* @example execute "select * from %{table variable}%" in {sql} and store the result in {output::*} * @example execute "select * from %{table variable}%" in {sql} and store the result in {output::*}
* @since 0.1.0 * @since 0.1.0
*/ */
public class EffExecuteStatement extends Delay { public class EffExecuteStatement extends Effect {
static { static {
Skript.registerEffect(EffExecuteStatement.class, Skript.registerEffect(EffExecuteStatement.class,
"execute %string% (in|on) %datasource% " + "execute %string% (in|on) %datasource% " +
@ -58,18 +58,20 @@ public class EffExecuteStatement extends Delay {
private VariableString var; private VariableString var;
private boolean isLocal; private boolean isLocal;
private boolean isList; private boolean isList;
private Map<String, Object> doLater = new HashMap<>();
private boolean isSync; private boolean isSync;
private void continueScriptExecution(Event e, String res) { private void continueScriptExecution(Event e, Object populatedVariables) {
lastError = res; lastError = null;
if (populatedVariables instanceof String) {
lastError = (String) populatedVariables;
} else {
if (getNext() != null) { if (getNext() != null) {
doLater.forEach((name, value) -> setVariable(e, name, value)); ((Map<String, Object>) populatedVariables).forEach((name, value) -> setVariable(e, name, value));
doLater.clear();
TriggerItem.walk(getNext(), e);
} }
} }
TriggerItem.walk(getNext(), e);
}
@Override @Override
protected void execute(Event e) { protected void execute(Event e) {
DataSource ds = dataSource.getSingle(e); DataSource ds = dataSource.getSingle(e);
@ -79,11 +81,11 @@ public class EffExecuteStatement extends Delay {
if (ds == null) if (ds == null)
return; return;
if (isSync) { if (isSync) {
String result = executeStatement(ds, baseVariable, query); Object populatedVariables = executeStatement(ds, baseVariable, query);
continueScriptExecution(e, result); continueScriptExecution(e, populatedVariables);
} else { } else {
Object locals = Variables.removeLocals(e); Object locals = Variables.removeLocals(e);
CompletableFuture<String> sql = CompletableFuture<Object> sql =
CompletableFuture.supplyAsync(() -> executeStatement(ds, baseVariable, query), threadPool); CompletableFuture.supplyAsync(() -> executeStatement(ds, baseVariable, query), threadPool);
sql.whenComplete((res, err) -> { sql.whenComplete((res, err) -> {
@ -92,13 +94,19 @@ public class EffExecuteStatement extends Delay {
} }
Bukkit.getScheduler().runTask(SkriptDB.getInstance(), () -> { Bukkit.getScheduler().runTask(SkriptDB.getInstance(), () -> {
lastError = res; lastError = null;
if (res instanceof String) {
lastError = (String) res;
}
if (getNext() != null) { if (getNext() != null) {
if (locals != null) if (locals != null)
Variables.setLocalVariables(e, locals); Variables.setLocalVariables(e, locals);
doLater.forEach((name, value) -> setVariable(e, name, value));
doLater.clear(); if (!(res instanceof String)) {
((Map<String, Object>) res).forEach((name, value) -> setVariable(e, name, value));
}
//doLater.clear();
TriggerItem.walk(getNext(), e); TriggerItem.walk(getNext(), e);
Variables.removeLocals(e); Variables.removeLocals(e);
} }
@ -171,11 +179,11 @@ public class EffExecuteStatement extends Delay {
return new Pair<>(sb.toString(), parameters); return new Pair<>(sb.toString(), parameters);
} }
private String executeStatement(DataSource ds, String baseVariable, Pair<String, List<Object>> query) { private Object executeStatement(DataSource ds, String baseVariable, Pair<String, List<Object>> query) {
if (ds == null) { if (ds == null) {
return "Data source is not set"; return "Data source is not set";
} }
Map<String, Object> variableList = new HashMap<>();
try (Connection conn = ds.getConnection(); try (Connection conn = ds.getConnection();
PreparedStatement stmt = createStatement(conn, query)) { PreparedStatement stmt = createStatement(conn, query)) {
@ -191,19 +199,38 @@ public class EffExecuteStatement extends Delay {
crs.populate(stmt.getResultSet()); crs.populate(stmt.getResultSet());
if (isList) { if (isList) {
populateVariable(crs, baseVariable); ResultSetMetaData meta = crs.getMetaData();
int columnCount = meta.getColumnCount();
for (int i = 1; i <= columnCount; i++) {
String label = meta.getColumnLabel(i);
variableList.put(baseVariable + label, label);
}
int rowNumber = 1;
try {
while (crs.next()) {
for (int i = 1; i <= columnCount; i++) {
variableList.put(baseVariable + meta.getColumnLabel(i).toLowerCase(Locale.ENGLISH)
+ Variable.SEPARATOR + rowNumber, crs.getObject(i));
}
rowNumber++;
}
} catch (SQLException ex) {
return ex.getMessage();
}
} else { } else {
crs.last(); crs.last();
doLater.put(baseVariable, crs.getRow()); variableList.put(baseVariable, crs.getRow());
} }
} else if (!isList) { } else if (!isList) {
doLater.put(baseVariable, stmt.getUpdateCount()); variableList.put(baseVariable, stmt.getUpdateCount());
} }
} }
} catch (SQLException ex) { } catch (SQLException ex) {
return ex.getMessage(); return ex.getMessage();
} }
return null; return variableList;
} }
private PreparedStatement createStatement(Connection conn, Pair<String, List<Object>> query) throws SQLException { private PreparedStatement createStatement(Connection conn, Pair<String, List<Object>> query) throws SQLException {
@ -237,26 +264,6 @@ public class EffExecuteStatement extends Delay {
Variables.setVariable(name.toLowerCase(Locale.ENGLISH), obj, e, isLocal); Variables.setVariable(name.toLowerCase(Locale.ENGLISH), obj, e, isLocal);
} }
private void populateVariable(CachedRowSet crs, String baseVariable)
throws SQLException {
ResultSetMetaData meta = crs.getMetaData();
int columnCount = meta.getColumnCount();
for (int i = 1; i <= columnCount; i++) {
String label = meta.getColumnLabel(i);
doLater.put(baseVariable + label, label);
}
int rowNumber = 1;
while (crs.next()) {
for (int i = 1; i <= columnCount; i++) {
doLater.put(baseVariable + meta.getColumnLabel(i).toLowerCase(Locale.ENGLISH)
+ Variable.SEPARATOR + rowNumber, crs.getObject(i));
}
rowNumber++;
}
}
@Override @Override
public String toString(Event e, boolean debug) { public String toString(Event e, boolean debug) {
return "execute " + query.toString(e, debug) + " in " + dataSource.toString(e, debug); return "execute " + query.toString(e, debug) + " in " + dataSource.toString(e, debug);

View File

@ -1,5 +1,5 @@
name: skript-db name: skript-db
version: 1.1.0 version: 1.2.0
main: com.btk5h.skriptdb.SkriptDB main: com.btk5h.skriptdb.SkriptDB
depend: [Skript] depend: [Skript]
authors: [btk5h, FranKusmiruk, Govindas] authors: [btk5h, FranKusmiruk, Govindas]