Compare commits
61 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 3466a04ec8 | |||
| d0191007a5 | |||
| b1df041ccb | |||
| b9e14652ea | |||
| 0098450441 | |||
| 9c41039217 | |||
| 46a72639af | |||
| cbd1565896 | |||
| 0332ad9334 | |||
| 32f40af484 | |||
| 2a24a974ef | |||
| d80df616c3 | |||
| 0b3b183a14 | |||
| 70fa7b7105 | |||
| c85d579645 | |||
| 1e039c1bfe | |||
| 2a4d7f6a6d | |||
| 2dcd4edf58 | |||
| 9b694206d3 | |||
| d8cb6fedb5 | |||
| 2eda1418cf | |||
| 83d2795519 | |||
| 66459c6190 | |||
| b1ef437b96 | |||
|
|
35cfb0b4bb | ||
| 4da64624d8 | |||
| 0c16b98553 | |||
| 9e7c5fcbee | |||
| 8b3d26cf78 | |||
| 5a22864650 | |||
| 05554a0043 | |||
| f6a43f0b53 | |||
| 5803be55b7 | |||
| d57b921b4f | |||
| 4fd654c743 | |||
| 233d21a30b | |||
| 8e760c4d8b | |||
| fe2231ae23 | |||
| a132912f6b | |||
| d4f9a394fd | |||
| a63a4c6d6f | |||
| 11063c166d | |||
| b5f2d56263 | |||
| 8b5121d5fb | |||
| b435696385 | |||
| 4d795edf01 | |||
| 2570408906 | |||
| 8d94a85779 | |||
| 6d0ec81e2e | |||
| c4379d45b7 | |||
| 434ea07410 | |||
| 8ca308802f | |||
| 2ce1dd924e | |||
| b20740b020 | |||
|
|
7906dd8eb1 | ||
|
|
33f4d5ff54 | ||
|
|
9a20da02a0 | ||
|
|
f13bf78007 | ||
|
|
55081e5d28 | ||
|
|
d88e431f07 | ||
|
|
e6606a012a |
25
README.md
25
README.md
@@ -5,10 +5,17 @@
|
|||||||
|
|
||||||
### Difference from original skript-db
|
### Difference from original skript-db
|
||||||
- Fixed local variables disappearance in newer Skript versions (very hacky fix, but it works, so that's good!)
|
- Fixed local variables disappearance in newer Skript versions (very hacky fix, but it works, so that's good!)
|
||||||
- Thread-pool size is now automatically increasing on demand to use of CachedThreadPool, instead of a fixed hard-coded number
|
- Thread-pool size is now automatically increasing on demand with use of CachedThreadPool, instead of a fixed hard-coded number
|
||||||
- Uses a newer version of HikariCP
|
- Uses newer versions of dependencies (Increased performance and security)
|
||||||
- Only meant to be used by newer Minecraft versions (1.8 is not supported)
|
- Replaced `synchronously execute` with `quickly execute`, which allows to speed up queries by 50ms with some risk
|
||||||
|
- SQL Driver is configurable
|
||||||
|
- A few variable type related bugs fixed
|
||||||
|
- Uses Java 11 instead of Java 8
|
||||||
|
|
||||||
|
### Installation
|
||||||
|
1. Use Skript 2.5+
|
||||||
|
2. Use Java 11+
|
||||||
|
3. Put skript-db in plugins folder and restart the server
|
||||||
### Expression `Data Source` => `datasource`
|
### Expression `Data Source` => `datasource`
|
||||||
Stores the connection information for a data source. This should be saved to a variable in a
|
Stores the connection information for a data source. This should be saved to a variable in a
|
||||||
`script load` event or manually through an effect command.
|
`script load` event or manually through an effect command.
|
||||||
@@ -36,10 +43,10 @@ Executes a statement on a database and optionally stores the result in a variabl
|
|||||||
If a list variable, such as `{test::*}`, is passed, the query result will be mapped to the list
|
If a list variable, such as `{test::*}`, is passed, the query result will be mapped to the list
|
||||||
variable in the form `{test::<column name>::<row number>}`
|
variable in the form `{test::<column name>::<row number>}`
|
||||||
|
|
||||||
If `synchronously` is specified, the SQL query will be done on the current thread.
|
If `quickly` is specified, the SQL query will be done without jumping back to main thread, which speeds it up by 50ms, however that makes code after it to also be on separate thread, you can jump back to main thread by adding `wait a tick`
|
||||||
#### Syntax
|
#### Syntax
|
||||||
```
|
```
|
||||||
[synchronously] execute %string% (in|on) %datasource% [and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]
|
[quickly] execute %string% (in|on) %datasource% [and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]
|
||||||
```
|
```
|
||||||
|
|
||||||
#### Examples
|
#### Examples
|
||||||
@@ -77,4 +84,10 @@ execute unsafe {fully dynamic query} in {sql}
|
|||||||
```
|
```
|
||||||
|
|
||||||
---
|
---
|
||||||
|
### Configuration
|
||||||
|
plugins/skript-db/config.yml
|
||||||
|
```
|
||||||
|
# Only change this if you wish to use a different driver than Java's default, like MariaDB driver.
|
||||||
|
# If you use MariaDB, its driver is shaded together with skript-db, so you can just specify: "org.mariadb.jdbc.Driver"
|
||||||
|
sql-driver-class-name: "default"
|
||||||
|
```
|
||||||
12
pom.xml
12
pom.xml
@@ -6,7 +6,7 @@
|
|||||||
|
|
||||||
<groupId>com.btk5h</groupId>
|
<groupId>com.btk5h</groupId>
|
||||||
<artifactId>skript-db</artifactId>
|
<artifactId>skript-db</artifactId>
|
||||||
<version>1.3.1</version>
|
<version>1.3.5</version>
|
||||||
<packaging>jar</packaging>
|
<packaging>jar</packaging>
|
||||||
|
|
||||||
<repositories>
|
<repositories>
|
||||||
@@ -45,8 +45,8 @@
|
|||||||
<artifactId>maven-compiler-plugin</artifactId>
|
<artifactId>maven-compiler-plugin</artifactId>
|
||||||
<version>3.6.1</version>
|
<version>3.6.1</version>
|
||||||
<configuration>
|
<configuration>
|
||||||
<source>8</source>
|
<source>11</source>
|
||||||
<target>8</target>
|
<target>11</target>
|
||||||
<encoding>UTF-8</encoding>
|
<encoding>UTF-8</encoding>
|
||||||
</configuration>
|
</configuration>
|
||||||
</plugin>
|
</plugin>
|
||||||
@@ -81,7 +81,7 @@
|
|||||||
<dependency>
|
<dependency>
|
||||||
<groupId>com.zaxxer</groupId>
|
<groupId>com.zaxxer</groupId>
|
||||||
<artifactId>HikariCP</artifactId>
|
<artifactId>HikariCP</artifactId>
|
||||||
<version>4.0.3</version>
|
<version>5.0.1</version>
|
||||||
<scope>compile</scope>
|
<scope>compile</scope>
|
||||||
</dependency>
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
@@ -93,14 +93,14 @@
|
|||||||
<dependency>
|
<dependency>
|
||||||
<groupId>com.github.SkriptLang</groupId>
|
<groupId>com.github.SkriptLang</groupId>
|
||||||
<artifactId>Skript</artifactId>
|
<artifactId>Skript</artifactId>
|
||||||
<version>2.5.3</version>
|
<version>2.6-beta3</version>
|
||||||
<scope>provided</scope>
|
<scope>provided</scope>
|
||||||
</dependency>
|
</dependency>
|
||||||
<!-- https://mvnrepository.com/artifact/org.mariadb.jdbc/mariadb-java-client -->
|
<!-- https://mvnrepository.com/artifact/org.mariadb.jdbc/mariadb-java-client -->
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>org.mariadb.jdbc</groupId>
|
<groupId>org.mariadb.jdbc</groupId>
|
||||||
<artifactId>mariadb-java-client</artifactId>
|
<artifactId>mariadb-java-client</artifactId>
|
||||||
<version>2.7.2</version>
|
<version>3.0.3</version>
|
||||||
<scope>compile</scope>
|
<scope>compile</scope>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
|
||||||
|
|||||||
@@ -5,14 +5,18 @@ import org.bukkit.event.HandlerList;
|
|||||||
|
|
||||||
public class SQLQueryCompleteEvent extends Event {
|
public class SQLQueryCompleteEvent extends Event {
|
||||||
private final static HandlerList HANDLERS = new HandlerList();
|
private final static HandlerList HANDLERS = new HandlerList();
|
||||||
private String argument;
|
private final String argument;
|
||||||
private Object variables;
|
|
||||||
|
|
||||||
public SQLQueryCompleteEvent(String argument) {
|
public SQLQueryCompleteEvent(String argument) {
|
||||||
super(true);
|
super(true);
|
||||||
this.argument = argument;
|
this.argument = argument;
|
||||||
// this.variables = variables;
|
// this.variables = variables;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static HandlerList getHandlerList() {
|
||||||
|
return HANDLERS;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getEventName() {
|
public String getEventName() {
|
||||||
return super.getEventName();
|
return super.getEventName();
|
||||||
@@ -22,13 +26,10 @@ public class SQLQueryCompleteEvent extends Event {
|
|||||||
public HandlerList getHandlers() {
|
public HandlerList getHandlers() {
|
||||||
return HANDLERS;
|
return HANDLERS;
|
||||||
}
|
}
|
||||||
public static HandlerList getHandlerList() {
|
|
||||||
return HANDLERS;
|
|
||||||
}
|
|
||||||
|
|
||||||
public String getArgument() {
|
public String getQuery() {
|
||||||
return argument;
|
return argument;
|
||||||
}
|
}
|
||||||
|
|
||||||
// public String getVariables() {return;}
|
// public String getVariables() {return;}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,13 +8,16 @@ import ch.njol.util.Kleenean;
|
|||||||
import ch.njol.util.Pair;
|
import ch.njol.util.Pair;
|
||||||
import com.btk5h.skriptdb.SkriptDB;
|
import com.btk5h.skriptdb.SkriptDB;
|
||||||
import com.btk5h.skriptdb.SkriptUtil;
|
import com.btk5h.skriptdb.SkriptUtil;
|
||||||
import com.btk5h.skriptdb.events.SQLQueryCompleteEvent;
|
|
||||||
import com.zaxxer.hikari.HikariDataSource;
|
import com.zaxxer.hikari.HikariDataSource;
|
||||||
import org.bukkit.Bukkit;
|
import org.bukkit.Bukkit;
|
||||||
|
import org.bukkit.entity.Player;
|
||||||
import org.bukkit.event.Event;
|
import org.bukkit.event.Event;
|
||||||
|
|
||||||
import javax.sql.DataSource;
|
import javax.sql.DataSource;
|
||||||
import javax.sql.rowset.CachedRowSet;
|
import javax.sql.rowset.CachedRowSet;
|
||||||
|
import javax.sql.rowset.serial.SerialBlob;
|
||||||
|
import javax.sql.rowset.serial.SerialException;
|
||||||
|
import java.io.IOException;
|
||||||
import java.sql.Connection;
|
import java.sql.Connection;
|
||||||
import java.sql.PreparedStatement;
|
import java.sql.PreparedStatement;
|
||||||
import java.sql.ResultSetMetaData;
|
import java.sql.ResultSetMetaData;
|
||||||
@@ -42,283 +45,312 @@ import java.util.concurrent.Executors;
|
|||||||
* @since 0.1.0
|
* @since 0.1.0
|
||||||
*/
|
*/
|
||||||
public class EffExecuteStatement extends Effect {
|
public class EffExecuteStatement extends Effect {
|
||||||
static {
|
private static final ExecutorService threadPool =
|
||||||
Skript.registerEffect(EffExecuteStatement.class,
|
Executors.newCachedThreadPool();
|
||||||
"execute %string% (in|on) %datasource% " +
|
static String lastError;
|
||||||
"[and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]", "synchronously execute %string% (in|on) %datasource% " +
|
|
||||||
"[and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]");
|
|
||||||
}
|
|
||||||
|
|
||||||
static String lastError;
|
static {
|
||||||
|
Skript.registerEffect(EffExecuteStatement.class,
|
||||||
private static final ExecutorService threadPool =
|
"execute %string% (in|on) %datasource% " +
|
||||||
Executors.newCachedThreadPool();
|
"[and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]", "quickly execute %string% (in|on) %datasource% " +
|
||||||
|
"[and store [[the] (output|result)[s]] (to|in) [the] [var[iable]] %-objects%]");
|
||||||
private Expression<String> query;
|
|
||||||
private Expression<HikariDataSource> dataSource;
|
|
||||||
private VariableString var;
|
|
||||||
private boolean isLocal;
|
|
||||||
private boolean isList;
|
|
||||||
private boolean isSync;
|
|
||||||
|
|
||||||
private void continueScriptExecution(Event e, Object populatedVariables) {
|
|
||||||
lastError = null;
|
|
||||||
if (populatedVariables instanceof String) {
|
|
||||||
lastError = (String) populatedVariables;
|
|
||||||
} else {
|
|
||||||
|
|
||||||
if (getNext() != null) {
|
|
||||||
((Map<String, Object>) populatedVariables).forEach((name, value) -> setVariable(e, name, value));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
TriggerItem.walk(getNext(), e);
|
|
||||||
}
|
|
||||||
@Override
|
|
||||||
protected void execute(Event e) {
|
|
||||||
DataSource ds = dataSource.getSingle(e);
|
|
||||||
Pair<String, List<Object>> query = parseQuery(e);
|
|
||||||
String baseVariable = var != null ? var.toString(e).toLowerCase(Locale.ENGLISH) : null;
|
|
||||||
|
|
||||||
//if data source isn't set
|
|
||||||
if (ds == null) return;
|
|
||||||
|
|
||||||
boolean sync = false;
|
|
||||||
if (!Bukkit.isPrimaryThread()) {
|
|
||||||
sync = true;
|
|
||||||
}
|
|
||||||
Object locals = Variables.removeLocals(e);
|
|
||||||
|
|
||||||
//execute SQL statement
|
|
||||||
CompletableFuture<Object> sql =
|
|
||||||
CompletableFuture.supplyAsync(() -> executeStatement(ds, baseVariable, query), threadPool);
|
|
||||||
|
|
||||||
//when SQL statement is completed
|
|
||||||
boolean finalSync = sync;
|
|
||||||
sql.whenComplete((res, err) -> {
|
|
||||||
if (err != null) { err.printStackTrace(); }
|
|
||||||
|
|
||||||
//handle last error syntax data
|
|
||||||
lastError = null;
|
|
||||||
if (res instanceof String) {
|
|
||||||
lastError = (String) res;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (getNext() != null) {
|
|
||||||
//if local variables are present
|
|
||||||
if (locals != null)
|
|
||||||
//bring back local variables
|
|
||||||
|
|
||||||
//populate SQL data into variables
|
|
||||||
if (!(res instanceof String)) {
|
|
||||||
|
|
||||||
//also set variables in the sql query complete event
|
|
||||||
|
|
||||||
SQLQueryCompleteEvent event = new SQLQueryCompleteEvent("something");
|
|
||||||
((Map<String, Object>) res).forEach((name, value) -> setVariable(event, name, value));
|
|
||||||
SkriptDB.getPlugin(SkriptDB.class).getServer().getPluginManager().callEvent(event);
|
|
||||||
}
|
|
||||||
if (isSync || finalSync) {
|
|
||||||
|
|
||||||
Variables.setLocalVariables(e, locals);
|
|
||||||
if (!(res instanceof String)) { ((Map<String, Object>) res).forEach((name, value) -> setVariable(e, name, value)); }
|
|
||||||
TriggerItem.walk(getNext(), e);
|
|
||||||
Variables.removeLocals(e);
|
|
||||||
} else {
|
|
||||||
Bukkit.getScheduler().runTask(SkriptDB.getInstance(), () -> {
|
|
||||||
Variables.setLocalVariables(e, locals);
|
|
||||||
if (!(res instanceof String)) { ((Map<String, Object>) res).forEach((name, value) -> setVariable(e, name, value)); }
|
|
||||||
TriggerItem.walk(getNext(), e);
|
|
||||||
//the line below is required to prevent memory leaks
|
|
||||||
//no functionality difference notice with it being removed from my test, but the memory gets filled with leaks
|
|
||||||
//so it must be kept
|
|
||||||
Variables.removeLocals(e);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected TriggerItem walk(Event e) {
|
|
||||||
debug(e, true);
|
|
||||||
//I think no longer needed as of 1.3.0, uncomment if something breaks
|
|
||||||
if (!isSync) {
|
|
||||||
Delay.addDelayedEvent(e);
|
|
||||||
}
|
|
||||||
execute(e);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private Pair<String, List<Object>> parseQuery(Event e) {
|
|
||||||
if (!(query instanceof VariableString)) {
|
|
||||||
return new Pair<>(query.getSingle(e), null);
|
|
||||||
}
|
|
||||||
VariableString q = (VariableString) query;
|
|
||||||
if (q.isSimple()) {
|
|
||||||
return new Pair<>(q.toString(e), null);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
StringBuilder sb = new StringBuilder();
|
private Expression<String> query;
|
||||||
List<Object> parameters = new ArrayList<>();
|
private Expression<HikariDataSource> dataSource;
|
||||||
Object[] objects = SkriptUtil.getTemplateString(q);
|
private VariableString var;
|
||||||
|
private boolean isLocal;
|
||||||
|
private boolean isList;
|
||||||
|
private boolean isSync;
|
||||||
|
|
||||||
for (int i = 0; i < objects.length; i++) {
|
private void continueScriptExecution(Event e, Object populatedVariables) {
|
||||||
Object o = objects[i];
|
lastError = null;
|
||||||
if (o instanceof String) {
|
if (populatedVariables instanceof String) {
|
||||||
sb.append(o);
|
lastError = (String) populatedVariables;
|
||||||
} else {
|
|
||||||
Expression<?> expr = SkriptUtil.getExpressionFromInfo(o);
|
|
||||||
|
|
||||||
String before = getString(objects, i - 1);
|
|
||||||
String after = getString(objects, i + 1);
|
|
||||||
boolean standaloneString = false;
|
|
||||||
|
|
||||||
if (before != null && after != null) {
|
|
||||||
if (before.endsWith("'") && after.endsWith("'")) {
|
|
||||||
standaloneString = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Object expressionValue = expr.getSingle(e);
|
|
||||||
|
|
||||||
if (expr instanceof ExprUnsafe) {
|
|
||||||
sb.append(expressionValue);
|
|
||||||
|
|
||||||
if (standaloneString && expressionValue instanceof String) {
|
|
||||||
String rawExpression = ((ExprUnsafe) expr).getRawExpression();
|
|
||||||
Skript.warning(
|
|
||||||
String.format("Unsafe may have been used unnecessarily. Try replacing 'unsafe %1$s' with %1$s",
|
|
||||||
rawExpression));
|
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
parameters.add(expressionValue);
|
|
||||||
sb.append('?');
|
|
||||||
|
|
||||||
if (standaloneString) {
|
if (getNext() != null) {
|
||||||
Skript.warning("Do not surround expressions with quotes!");
|
((Map<String, Object>) populatedVariables).forEach((name, value) -> setVariable(e, name, value));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
TriggerItem.walk(getNext(), e);
|
||||||
}
|
}
|
||||||
return new Pair<>(sb.toString(), parameters);
|
|
||||||
}
|
|
||||||
|
|
||||||
private Object executeStatement(DataSource ds, String baseVariable, Pair<String, List<Object>> query) {
|
@Override
|
||||||
if (ds == null) {
|
protected void execute(Event e) {
|
||||||
return "Data source is not set";
|
DataSource ds = dataSource.getSingle(e);
|
||||||
}
|
Pair<String, List<Object>> query = parseQuery(e);
|
||||||
Map<String, Object> variableList = new HashMap<>();
|
String baseVariable = var != null ? var.toString(e).toLowerCase(Locale.ENGLISH) : null;
|
||||||
try (Connection conn = ds.getConnection();
|
//if data source isn't set
|
||||||
PreparedStatement stmt = createStatement(conn, query)) {
|
if (ds == null) return;
|
||||||
|
|
||||||
boolean hasResultSet = stmt.execute();
|
boolean sync = !Bukkit.isPrimaryThread();
|
||||||
|
|
||||||
if (baseVariable != null) {
|
//if current thread is not main thread, then make this query to not have delays
|
||||||
if (isList) {
|
|
||||||
baseVariable = baseVariable.substring(0, baseVariable.length() - 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (hasResultSet) {
|
Object locals = Variables.removeLocals(e);
|
||||||
CachedRowSet crs = SkriptDB.getRowSetFactory().createCachedRowSet();
|
|
||||||
crs.populate(stmt.getResultSet());
|
|
||||||
|
|
||||||
if (isList) {
|
//execute SQL statement
|
||||||
ResultSetMetaData meta = crs.getMetaData();
|
CompletableFuture<Object> sql =
|
||||||
int columnCount = meta.getColumnCount();
|
CompletableFuture.supplyAsync(() -> executeStatement(ds, baseVariable, query), threadPool);
|
||||||
|
|
||||||
for (int i = 1; i <= columnCount; i++) {
|
//when SQL statement is completed
|
||||||
String label = meta.getColumnLabel(i);
|
boolean finalSync = sync;
|
||||||
variableList.put(baseVariable + label, label);
|
sql.whenComplete((res, err) -> {
|
||||||
}
|
if (err != null) {
|
||||||
|
err.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
int rowNumber = 1;
|
//handle last error syntax data
|
||||||
try {
|
lastError = null;
|
||||||
while (crs.next()) {
|
if (res instanceof String) {
|
||||||
for (int i = 1; i <= columnCount; i++) {
|
lastError = (String) res;
|
||||||
variableList.put(baseVariable + meta.getColumnLabel(i).toLowerCase(Locale.ENGLISH)
|
}
|
||||||
+ Variable.SEPARATOR + rowNumber, crs.getObject(i));
|
|
||||||
}
|
if (getNext() != null) {
|
||||||
rowNumber++;
|
//if local variables are present
|
||||||
|
//bring back local variables
|
||||||
|
|
||||||
|
//populate SQL data into variables
|
||||||
|
if (!(res instanceof String)) {
|
||||||
|
|
||||||
|
//also set variables in the sql query complete event
|
||||||
|
|
||||||
|
//TEMPORARILY DISABLED, AS THIS WOULD WORSEN PERFORMANCE OF THE QUERIES AND NOT BE USED BY MOST PEOPLE.
|
||||||
|
//I may add config option to enable this later?
|
||||||
|
|
||||||
|
//SQLQueryCompleteEvent event = new SQLQueryCompleteEvent(this.query.getSingle(e));
|
||||||
|
//((Map<String, Object>) res).forEach((name, value) -> setVariable(event, name, value));
|
||||||
|
//SkriptDB.getPlugin(SkriptDB.class).getServer().getPluginManager().callEvent(event);
|
||||||
}
|
}
|
||||||
} catch (SQLException ex) {
|
if (isSync || finalSync) {
|
||||||
return ex.getMessage();
|
if (locals != null) {
|
||||||
}
|
Variables.setLocalVariables(e, locals);
|
||||||
} else {
|
}
|
||||||
crs.last();
|
if (!(res instanceof String)) {
|
||||||
variableList.put(baseVariable, crs.getRow());
|
((Map<String, Object>) res).forEach((name, value) -> setVariable(e, name, value));
|
||||||
}
|
}
|
||||||
} else if (!isList) {
|
TriggerItem.walk(getNext(), e);
|
||||||
//if no results are returned and the specified variable isn't a list variable, put the affected rows count in the variable
|
Variables.removeLocals(e);
|
||||||
variableList.put(baseVariable, stmt.getUpdateCount());
|
} else {
|
||||||
|
Bukkit.getScheduler().runTask(SkriptDB.getInstance(), () -> {
|
||||||
|
if (locals != null) {
|
||||||
|
Variables.setLocalVariables(e, locals);
|
||||||
|
}
|
||||||
|
if (!(res instanceof String)) {
|
||||||
|
((Map<String, Object>) res).forEach((name, value) -> setVariable(e, name, value));
|
||||||
|
}
|
||||||
|
TriggerItem.walk(getNext(), e);
|
||||||
|
//the line below is required to prevent memory leaks
|
||||||
|
//no functionality difference notice with it being removed from my test, but the memory gets filled with leaks
|
||||||
|
//so it must be kept
|
||||||
|
Variables.removeLocals(e);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected TriggerItem walk(Event e) {
|
||||||
|
debug(e, true);
|
||||||
|
//I think no longer needed as of 1.3.0, uncomment if something breaks
|
||||||
|
if (!isSync) {
|
||||||
|
Delay.addDelayedEvent(e);
|
||||||
}
|
}
|
||||||
}
|
execute(e);
|
||||||
} catch (SQLException ex) {
|
return null;
|
||||||
return ex.getMessage();
|
|
||||||
}
|
|
||||||
return variableList;
|
|
||||||
}
|
|
||||||
|
|
||||||
private PreparedStatement createStatement(Connection conn, Pair<String, List<Object>> query) throws SQLException {
|
|
||||||
PreparedStatement stmt = conn.prepareStatement(query.getFirst());
|
|
||||||
List<Object> parameters = query.getSecond();
|
|
||||||
|
|
||||||
if (parameters != null) {
|
|
||||||
for (int i = 0; i < parameters.size(); i++) {
|
|
||||||
stmt.setObject(i + 1, parameters.get(i));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return stmt;
|
private Pair<String, List<Object>> parseQuery(Event e) {
|
||||||
}
|
if (!(query instanceof VariableString)) {
|
||||||
|
return new Pair<>(query.getSingle(e), null);
|
||||||
|
}
|
||||||
|
VariableString q = (VariableString) query;
|
||||||
|
if (q.isSimple()) {
|
||||||
|
return new Pair<>(q.toString(e), null);
|
||||||
|
}
|
||||||
|
|
||||||
private String getString(Object[] objects, int index) {
|
StringBuilder sb = new StringBuilder();
|
||||||
if (index < 0 || index >= objects.length) {
|
List<Object> parameters = new ArrayList<>();
|
||||||
return null;
|
Object[] objects = SkriptUtil.getTemplateString(q);
|
||||||
|
|
||||||
|
for (int i = 0; i < objects.length; i++) {
|
||||||
|
Object o = objects[i];
|
||||||
|
if (o instanceof String) {
|
||||||
|
sb.append(o);
|
||||||
|
} else {
|
||||||
|
Expression<?> expr;
|
||||||
|
if (o instanceof Expression)
|
||||||
|
expr = (Expression<?>) o;
|
||||||
|
else
|
||||||
|
expr = SkriptUtil.getExpressionFromInfo(o);
|
||||||
|
|
||||||
|
String before = getString(objects, i - 1);
|
||||||
|
String after = getString(objects, i + 1);
|
||||||
|
boolean standaloneString = false;
|
||||||
|
|
||||||
|
if (before != null && after != null) {
|
||||||
|
if (before.endsWith("'") && after.endsWith("'")) {
|
||||||
|
standaloneString = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Object expressionValue = expr.getSingle(e);
|
||||||
|
|
||||||
|
if (expr instanceof ExprUnsafe) {
|
||||||
|
sb.append(expressionValue);
|
||||||
|
|
||||||
|
if (standaloneString && expressionValue instanceof String) {
|
||||||
|
String rawExpression = ((ExprUnsafe) expr).getRawExpression();
|
||||||
|
Skript.warning(
|
||||||
|
String.format("Unsafe may have been used unnecessarily. Try replacing 'unsafe %1$s' with %1$s",
|
||||||
|
rawExpression));
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
parameters.add(expressionValue);
|
||||||
|
sb.append('?');
|
||||||
|
|
||||||
|
if (standaloneString) {
|
||||||
|
Skript.warning("Do not surround expressions with quotes!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return new Pair<>(sb.toString(), parameters);
|
||||||
}
|
}
|
||||||
|
|
||||||
Object object = objects[index];
|
private Object executeStatement(DataSource ds, String baseVariable, Pair<String, List<Object>> query) {
|
||||||
|
if (ds == null) {
|
||||||
|
return "Data source is not set";
|
||||||
|
}
|
||||||
|
Map<String, Object> variableList = new HashMap<>();
|
||||||
|
try (Connection conn = ds.getConnection();
|
||||||
|
PreparedStatement stmt = createStatement(conn, query)) {
|
||||||
|
|
||||||
if (object instanceof String) {
|
boolean hasResultSet = stmt.execute();
|
||||||
return (String) object;
|
|
||||||
|
if (baseVariable != null) {
|
||||||
|
if (isList) {
|
||||||
|
baseVariable = baseVariable.substring(0, baseVariable.length() - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (hasResultSet) {
|
||||||
|
CachedRowSet crs = SkriptDB.getRowSetFactory().createCachedRowSet();
|
||||||
|
crs.populate(stmt.getResultSet());
|
||||||
|
|
||||||
|
if (isList) {
|
||||||
|
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 {
|
||||||
|
crs.last();
|
||||||
|
variableList.put(baseVariable, crs.getRow());
|
||||||
|
}
|
||||||
|
} else if (!isList) {
|
||||||
|
//if no results are returned and the specified variable isn't a list variable, put the affected rows count in the variable
|
||||||
|
variableList.put(baseVariable, stmt.getUpdateCount());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (SQLException ex) {
|
||||||
|
return ex.getMessage();
|
||||||
|
}
|
||||||
|
return variableList;
|
||||||
}
|
}
|
||||||
|
|
||||||
return null;
|
private PreparedStatement createStatement(Connection conn, Pair<String, List<Object>> query) throws SQLException {
|
||||||
}
|
PreparedStatement stmt = conn.prepareStatement(query.getFirst());
|
||||||
|
List<Object> parameters = query.getSecond();
|
||||||
|
|
||||||
private void setVariable(Event e, String name, Object obj) {
|
if (parameters != null) {
|
||||||
Variables.setVariable(name.toLowerCase(Locale.ENGLISH), obj, e, isLocal);
|
for (int i = 0; i < parameters.size(); i++) {
|
||||||
}
|
stmt.setObject(i + 1, parameters.get(i));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
return stmt;
|
||||||
public String toString(Event e, boolean debug) {
|
|
||||||
return "execute " + query.toString(e, debug) + " in " + dataSource.toString(e, debug);
|
|
||||||
}
|
|
||||||
|
|
||||||
@SuppressWarnings("unchecked")
|
|
||||||
@Override
|
|
||||||
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
|
||||||
SkriptParser.ParseResult parseResult) {
|
|
||||||
Expression<String> statementExpr = (Expression<String>) exprs[0];
|
|
||||||
if (statementExpr instanceof VariableString || statementExpr instanceof ExprUnsafe) {
|
|
||||||
query = statementExpr;
|
|
||||||
} else {
|
|
||||||
Skript.error("Database statements must be string literals. If you must use an expression, " +
|
|
||||||
"you may use \"%unsafe (your expression)%\", but keep in mind, you may be vulnerable " +
|
|
||||||
"to SQL injection attacks!");
|
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
dataSource = (Expression<HikariDataSource>) exprs[1];
|
|
||||||
Expression<?> expr = exprs[2];
|
private String getString(Object[] objects, int index) {
|
||||||
isSync = matchedPattern == 1;
|
if (index < 0 || index >= objects.length) {
|
||||||
if (expr instanceof Variable) {
|
return null;
|
||||||
Variable<?> varExpr = (Variable<?>) expr;
|
}
|
||||||
var = varExpr.getName();
|
|
||||||
isLocal = varExpr.isLocal();
|
Object object = objects[index];
|
||||||
isList = varExpr.isList();
|
|
||||||
} else if (expr != null) {
|
if (object instanceof String) {
|
||||||
Skript.error(expr + " is not a variable");
|
return (String) object;
|
||||||
return false;
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void setVariable(Event e, String name, Object obj) {
|
||||||
|
|
||||||
|
//fix mediumblob and similar column types, so they return a String correctly
|
||||||
|
if (obj != null) {
|
||||||
|
if (obj.getClass().getName().equals("[B")) {
|
||||||
|
obj = new String((byte[]) obj);
|
||||||
|
|
||||||
|
//in some servers instead of being byte array, it appears as SerialBlob (depends on mc version, 1.12.2 is bvte array, 1.16.5 SerialBlob)
|
||||||
|
} else if (obj instanceof SerialBlob) {
|
||||||
|
try {
|
||||||
|
obj = new String(((SerialBlob) obj).getBinaryStream().readAllBytes());
|
||||||
|
} catch (IOException | SerialException ex) {
|
||||||
|
ex.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Variables.setVariable(name.toLowerCase(Locale.ENGLISH), obj, e, isLocal);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString(Event e, boolean debug) {
|
||||||
|
return "execute " + query.toString(e, debug) + " in " + dataSource.toString(e, debug);
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
@Override
|
||||||
|
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
||||||
|
SkriptParser.ParseResult parseResult) {
|
||||||
|
Expression<String> statementExpr = (Expression<String>) exprs[0];
|
||||||
|
if (statementExpr instanceof VariableString || statementExpr instanceof ExprUnsafe) {
|
||||||
|
query = statementExpr;
|
||||||
|
} else {
|
||||||
|
Skript.error("Database statements must be string literals. If you must use an expression, " +
|
||||||
|
"you may use \"%unsafe (your expression)%\", but keep in mind, you may be vulnerable " +
|
||||||
|
"to SQL injection attacks!");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
dataSource = (Expression<HikariDataSource>) exprs[1];
|
||||||
|
Expression<?> expr = exprs[2];
|
||||||
|
isSync = matchedPattern == 1;
|
||||||
|
if (expr instanceof Variable) {
|
||||||
|
Variable<?> varExpr = (Variable<?>) expr;
|
||||||
|
var = varExpr.getName();
|
||||||
|
isLocal = varExpr.isLocal();
|
||||||
|
isList = varExpr.isList();
|
||||||
|
} else if (expr != null) {
|
||||||
|
Skript.error(expr + " is not a variable");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
@@ -6,24 +6,24 @@ import ch.njol.skript.lang.SkriptEvent;
|
|||||||
import ch.njol.skript.lang.SkriptParser;
|
import ch.njol.skript.lang.SkriptParser;
|
||||||
import com.btk5h.skriptdb.events.SQLQueryCompleteEvent;
|
import com.btk5h.skriptdb.events.SQLQueryCompleteEvent;
|
||||||
import org.bukkit.event.Event;
|
import org.bukkit.event.Event;
|
||||||
import org.eclipse.jdt.annotation.Nullable;
|
|
||||||
|
|
||||||
public class EvtSQLQueryComplete extends SkriptEvent {
|
public class EvtSQLQueryComplete extends SkriptEvent {
|
||||||
static {
|
static {
|
||||||
Skript.registerEvent("complete of sql query", EvtSQLQueryComplete.class, SQLQueryCompleteEvent.class, "complete of [(sql|database)] query");
|
Skript.registerEvent("complete of sql query", EvtSQLQueryComplete.class, SQLQueryCompleteEvent.class, "complete of [(sql|database)] query");
|
||||||
}
|
}
|
||||||
@Override
|
|
||||||
public boolean init(final Literal<?>[] literals, final int i, final SkriptParser.ParseResult parseResult) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean check(Event event) {
|
|
||||||
return (event instanceof SQLQueryCompleteEvent);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String toString(@Nullable Event event, boolean debug) {
|
public boolean init(final Literal<?>[] literals, final int i, final SkriptParser.ParseResult parseResult) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean check(Event event) {
|
||||||
|
return (event instanceof SQLQueryCompleteEvent);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString(Event event, boolean debug) {
|
||||||
return "complete of sql query";
|
return "complete of sql query";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,12 @@
|
|||||||
package com.btk5h.skriptdb.skript;
|
package com.btk5h.skriptdb.skript;
|
||||||
|
|
||||||
import org.bukkit.event.Event;
|
|
||||||
|
|
||||||
import ch.njol.skript.Skript;
|
import ch.njol.skript.Skript;
|
||||||
import ch.njol.skript.lang.Expression;
|
import ch.njol.skript.lang.Expression;
|
||||||
import ch.njol.skript.lang.ExpressionType;
|
import ch.njol.skript.lang.ExpressionType;
|
||||||
import ch.njol.skript.lang.SkriptParser;
|
import ch.njol.skript.lang.SkriptParser;
|
||||||
import ch.njol.skript.lang.util.SimpleExpression;
|
import ch.njol.skript.lang.util.SimpleExpression;
|
||||||
import ch.njol.util.Kleenean;
|
import ch.njol.util.Kleenean;
|
||||||
|
import org.bukkit.event.Event;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Stores the error from the last executed statement, if there was one.
|
* Stores the error from the last executed statement, if there was one.
|
||||||
@@ -18,33 +17,33 @@ import ch.njol.util.Kleenean;
|
|||||||
* @since 0.1.0
|
* @since 0.1.0
|
||||||
*/
|
*/
|
||||||
public class ExprDBError extends SimpleExpression<String> {
|
public class ExprDBError extends SimpleExpression<String> {
|
||||||
static {
|
static {
|
||||||
Skript.registerExpression(ExprDBError.class, String.class,
|
Skript.registerExpression(ExprDBError.class, String.class,
|
||||||
ExpressionType.SIMPLE, "[the] [last] (sql|db|data(base|[ ]source)) error");
|
ExpressionType.SIMPLE, "[the] [last] (sql|db|data(base|[ ]source)) error");
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected String[] get(Event e) {
|
protected String[] get(Event e) {
|
||||||
return new String[]{EffExecuteStatement.lastError};
|
return new String[]{EffExecuteStatement.lastError};
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean isSingle() {
|
public boolean isSingle() {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Class<? extends String> getReturnType() {
|
public Class<? extends String> getReturnType() {
|
||||||
return String.class;
|
return String.class;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String toString(Event e, boolean debug) {
|
public String toString(Event e, boolean debug) {
|
||||||
return "last database error";
|
return "last database error";
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed, SkriptParser.ParseResult parseResult) {
|
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed, SkriptParser.ParseResult parseResult) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,5 @@
|
|||||||
package com.btk5h.skriptdb.skript;
|
package com.btk5h.skriptdb.skript;
|
||||||
|
|
||||||
import com.btk5h.skriptdb.SkriptDB;
|
|
||||||
import com.zaxxer.hikari.HikariDataSource;
|
|
||||||
|
|
||||||
import org.bukkit.event.Event;
|
|
||||||
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.Map;
|
|
||||||
|
|
||||||
import ch.njol.skript.Skript;
|
import ch.njol.skript.Skript;
|
||||||
import ch.njol.skript.lang.Expression;
|
import ch.njol.skript.lang.Expression;
|
||||||
import ch.njol.skript.lang.ExpressionType;
|
import ch.njol.skript.lang.ExpressionType;
|
||||||
@@ -15,11 +7,17 @@ import ch.njol.skript.lang.SkriptParser;
|
|||||||
import ch.njol.skript.lang.util.SimpleExpression;
|
import ch.njol.skript.lang.util.SimpleExpression;
|
||||||
import ch.njol.skript.util.Timespan;
|
import ch.njol.skript.util.Timespan;
|
||||||
import ch.njol.util.Kleenean;
|
import ch.njol.util.Kleenean;
|
||||||
|
import com.btk5h.skriptdb.SkriptDB;
|
||||||
|
import com.zaxxer.hikari.HikariDataSource;
|
||||||
|
import org.bukkit.event.Event;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Stores the connection information for a data source. This should be saved to a variable in a
|
* Stores the connection information for a data source. This should be saved to a variable in a
|
||||||
* `script load` event or manually through an effect command.
|
* `script load` event or manually through an effect command.
|
||||||
*
|
* <p>
|
||||||
* The url format for your database may vary! The example provided uses a MySQL database.
|
* The url format for your database may vary! The example provided uses a MySQL database.
|
||||||
*
|
*
|
||||||
* @name Data Source
|
* @name Data Source
|
||||||
@@ -30,74 +28,74 @@ import ch.njol.util.Kleenean;
|
|||||||
* @since 0.1.0
|
* @since 0.1.0
|
||||||
*/
|
*/
|
||||||
public class ExprDataSource extends SimpleExpression<HikariDataSource> {
|
public class ExprDataSource extends SimpleExpression<HikariDataSource> {
|
||||||
static {
|
private static final Map<String, HikariDataSource> connectionCache = new HashMap<>();
|
||||||
Skript.registerExpression(ExprDataSource.class, HikariDataSource.class,
|
|
||||||
ExpressionType.COMBINED, "[the] data(base|[ ]source) [(of|at)] %string% " +
|
|
||||||
"[with [a] [max[imum]] [connection] life[ ]time of %-timespan%]");
|
|
||||||
}
|
|
||||||
|
|
||||||
private static Map<String, HikariDataSource> connectionCache = new HashMap<>();
|
static {
|
||||||
|
Skript.registerExpression(ExprDataSource.class, HikariDataSource.class,
|
||||||
private Expression<String> url;
|
ExpressionType.COMBINED, "[the] data(base|[ ]source) [(of|at)] %string% " +
|
||||||
private Expression<Timespan> maxLifetime;
|
"[with [a] [max[imum]] [connection] life[ ]time of %-timespan%]");
|
||||||
|
|
||||||
@Override
|
|
||||||
protected HikariDataSource[] get(Event e) {
|
|
||||||
String jdbcUrl = url.getSingle(e);
|
|
||||||
if (jdbcUrl == null) {
|
|
||||||
return null;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!jdbcUrl.startsWith("jdbc:")) {
|
private Expression<String> url;
|
||||||
jdbcUrl = "jdbc:" + jdbcUrl;
|
private Expression<Timespan> maxLifetime;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected HikariDataSource[] get(Event e) {
|
||||||
|
String jdbcUrl = url.getSingle(e);
|
||||||
|
if (jdbcUrl == null) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!jdbcUrl.startsWith("jdbc:")) {
|
||||||
|
jdbcUrl = "jdbc:" + jdbcUrl;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (connectionCache.containsKey(jdbcUrl)) {
|
||||||
|
return new HikariDataSource[]{connectionCache.get(jdbcUrl)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HikariDataSource ds = new HikariDataSource();
|
||||||
|
|
||||||
|
//allow specifying of own sql driver class name
|
||||||
|
if (!SkriptDB.getInstance().getConfig().getString("sql-driver-class-name", "default").equals("default")) {
|
||||||
|
ds.setDriverClassName(SkriptDB.getInstance().getConfig().getString("sql-driver-class-name"));
|
||||||
|
}
|
||||||
|
ds.setJdbcUrl(jdbcUrl);
|
||||||
|
|
||||||
|
if (maxLifetime != null) {
|
||||||
|
Timespan l = maxLifetime.getSingle(e);
|
||||||
|
|
||||||
|
if (l != null) {
|
||||||
|
ds.setMaxLifetime(l.getMilliSeconds());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
connectionCache.put(jdbcUrl, ds);
|
||||||
|
|
||||||
|
return new HikariDataSource[]{ds};
|
||||||
}
|
}
|
||||||
|
|
||||||
if (connectionCache.containsKey(jdbcUrl)) {
|
@Override
|
||||||
return new HikariDataSource[]{connectionCache.get(jdbcUrl)};
|
public boolean isSingle() {
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
HikariDataSource ds = new HikariDataSource();
|
@Override
|
||||||
|
public Class<? extends HikariDataSource> getReturnType() {
|
||||||
//allow specifying of own sql driver class name
|
return HikariDataSource.class;
|
||||||
if (!SkriptDB.getInstance().getConfig().getString("sql-driver-class-name", "default").equals("default")) {
|
|
||||||
ds.setDriverClassName(SkriptDB.getInstance().getConfig().getString("sql-driver-class-name"));
|
|
||||||
}
|
|
||||||
ds.setJdbcUrl(jdbcUrl);
|
|
||||||
|
|
||||||
if (maxLifetime != null) {
|
|
||||||
Timespan l = maxLifetime.getSingle(e);
|
|
||||||
|
|
||||||
if (l != null) {
|
|
||||||
ds.setMaxLifetime(l.getMilliSeconds());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
connectionCache.put(jdbcUrl, ds);
|
@Override
|
||||||
|
public String toString(Event e, boolean debug) {
|
||||||
|
return "datasource " + url.toString(e, debug);
|
||||||
|
}
|
||||||
|
|
||||||
return new HikariDataSource[]{ds};
|
@SuppressWarnings("unchecked")
|
||||||
}
|
@Override
|
||||||
|
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
||||||
@Override
|
SkriptParser.ParseResult parseResult) {
|
||||||
public boolean isSingle() {
|
url = (Expression<String>) exprs[0];
|
||||||
return true;
|
maxLifetime = (Expression<Timespan>) exprs[1];
|
||||||
}
|
return true;
|
||||||
|
}
|
||||||
@Override
|
|
||||||
public Class<? extends HikariDataSource> getReturnType() {
|
|
||||||
return HikariDataSource.class;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String toString(Event e, boolean debug) {
|
|
||||||
return "datasource " + url.toString(e, debug);
|
|
||||||
}
|
|
||||||
|
|
||||||
@SuppressWarnings("unchecked")
|
|
||||||
@Override
|
|
||||||
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
|
||||||
SkriptParser.ParseResult parseResult) {
|
|
||||||
url = (Expression<String>) exprs[0];
|
|
||||||
maxLifetime = (Expression<Timespan>) exprs[1];
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
60
src/main/java/com/btk5h/skriptdb/skript/ExprSQLQuery.java
Normal file
60
src/main/java/com/btk5h/skriptdb/skript/ExprSQLQuery.java
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
package com.btk5h.skriptdb.skript;
|
||||||
|
|
||||||
|
import ch.njol.skript.ScriptLoader;
|
||||||
|
import ch.njol.skript.Skript;
|
||||||
|
import ch.njol.skript.lang.Expression;
|
||||||
|
import ch.njol.skript.lang.ExpressionType;
|
||||||
|
import ch.njol.skript.lang.SkriptParser;
|
||||||
|
import ch.njol.skript.lang.util.SimpleExpression;
|
||||||
|
import ch.njol.skript.log.ErrorQuality;
|
||||||
|
import ch.njol.util.Kleenean;
|
||||||
|
import com.btk5h.skriptdb.events.SQLQueryCompleteEvent;
|
||||||
|
import org.bukkit.event.Event;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Stores the error from the last executed statement, if there was one.
|
||||||
|
*
|
||||||
|
* @name Last Data Source Error
|
||||||
|
* @pattern [the] [last] (sql|db|data(base|[ ]source)) error
|
||||||
|
* @return text
|
||||||
|
* @since 0.1.0
|
||||||
|
*/
|
||||||
|
public class ExprSQLQuery extends SimpleExpression<String> {
|
||||||
|
static {
|
||||||
|
Skript.registerExpression(ExprSQLQuery.class, String.class,
|
||||||
|
ExpressionType.SIMPLE, "sql query");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String[] get(Event e) {
|
||||||
|
if (e instanceof SQLQueryCompleteEvent) {
|
||||||
|
return new String[]{((SQLQueryCompleteEvent) e).getQuery()};
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isSingle() {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Class<? extends String> getReturnType() {
|
||||||
|
return String.class;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString(Event e, boolean debug) {
|
||||||
|
return "sql query";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean init(final Expression<?>[] expressions, final int matchedPattern, final Kleenean isDelayed, final SkriptParser.ParseResult parseResult) {
|
||||||
|
if (!ScriptLoader.isCurrentEvent(SQLQueryCompleteEvent.class)) {
|
||||||
|
Skript.error("Cannot use 'sql query' outside of a complete of sql query event", ErrorQuality.SEMANTIC_ERROR);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,13 +1,12 @@
|
|||||||
package com.btk5h.skriptdb.skript;
|
package com.btk5h.skriptdb.skript;
|
||||||
|
|
||||||
import org.bukkit.event.Event;
|
|
||||||
|
|
||||||
import ch.njol.skript.Skript;
|
import ch.njol.skript.Skript;
|
||||||
import ch.njol.skript.lang.Expression;
|
import ch.njol.skript.lang.Expression;
|
||||||
import ch.njol.skript.lang.ExpressionType;
|
import ch.njol.skript.lang.ExpressionType;
|
||||||
import ch.njol.skript.lang.SkriptParser;
|
import ch.njol.skript.lang.SkriptParser;
|
||||||
import ch.njol.skript.lang.util.SimpleExpression;
|
import ch.njol.skript.lang.util.SimpleExpression;
|
||||||
import ch.njol.util.Kleenean;
|
import ch.njol.util.Kleenean;
|
||||||
|
import org.bukkit.event.Event;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Opts out of automatic SQL injection protection for a specific expression in a statement.
|
* Opts out of automatic SQL injection protection for a specific expression in a statement.
|
||||||
@@ -20,44 +19,44 @@ import ch.njol.util.Kleenean;
|
|||||||
* @since 0.1.0
|
* @since 0.1.0
|
||||||
*/
|
*/
|
||||||
public class ExprUnsafe extends SimpleExpression<String> {
|
public class ExprUnsafe extends SimpleExpression<String> {
|
||||||
static {
|
static {
|
||||||
Skript.registerExpression(ExprUnsafe.class, String.class, ExpressionType.COMBINED,
|
Skript.registerExpression(ExprUnsafe.class, String.class, ExpressionType.COMBINED,
|
||||||
"unsafe %string%");
|
"unsafe %string%");
|
||||||
}
|
}
|
||||||
|
|
||||||
private Expression<String> stringExpression;
|
private Expression<String> stringExpression;
|
||||||
private String rawExpression;
|
private String rawExpression;
|
||||||
|
|
||||||
public String getRawExpression() {
|
public String getRawExpression() {
|
||||||
return rawExpression;
|
return rawExpression;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected String[] get(Event e) {
|
protected String[] get(Event e) {
|
||||||
return stringExpression.getArray(e);
|
return stringExpression.getArray(e);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean isSingle() {
|
public boolean isSingle() {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Class<? extends String> getReturnType() {
|
public Class<? extends String> getReturnType() {
|
||||||
return String.class;
|
return String.class;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String toString(Event e, boolean debug) {
|
public String toString(Event e, boolean debug) {
|
||||||
return "unsafe " + stringExpression.toString(e, debug);
|
return "unsafe " + stringExpression.toString(e, debug);
|
||||||
}
|
}
|
||||||
|
|
||||||
@SuppressWarnings("unchecked")
|
@SuppressWarnings("unchecked")
|
||||||
@Override
|
@Override
|
||||||
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed,
|
||||||
SkriptParser.ParseResult parseResult) {
|
SkriptParser.ParseResult parseResult) {
|
||||||
stringExpression = (Expression<String>) exprs[0];
|
stringExpression = (Expression<String>) exprs[0];
|
||||||
rawExpression = parseResult.expr.substring("unsafe".length()).trim();
|
rawExpression = parseResult.expr.substring("unsafe".length()).trim();
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,58 +11,58 @@ import com.zaxxer.hikari.HikariDataSource;
|
|||||||
import java.io.StreamCorruptedException;
|
import java.io.StreamCorruptedException;
|
||||||
|
|
||||||
public class Types {
|
public class Types {
|
||||||
static {
|
static {
|
||||||
Classes.registerClass(new ClassInfo<>(HikariDataSource.class, "datasource")
|
Classes.registerClass(new ClassInfo<>(HikariDataSource.class, "datasource")
|
||||||
.user("datasources?")
|
.user("datasources?")
|
||||||
.parser(new Parser<HikariDataSource>() {
|
.parser(new Parser<HikariDataSource>() {
|
||||||
@Override
|
@Override
|
||||||
public HikariDataSource parse(String s, ParseContext context) {
|
public HikariDataSource parse(String s, ParseContext context) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String toString(HikariDataSource o, int flags) {
|
public String toString(HikariDataSource o, int flags) {
|
||||||
return o.getJdbcUrl();
|
return o.getJdbcUrl();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String toVariableNameString(HikariDataSource o) {
|
public String toVariableNameString(HikariDataSource o) {
|
||||||
return o.getJdbcUrl();
|
return o.getJdbcUrl();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getVariableNamePattern() {
|
public String getVariableNamePattern() {
|
||||||
return "jdbc:.+";
|
return "jdbc:.+";
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
.serializer(new Serializer<HikariDataSource>() {
|
.serializer(new Serializer<HikariDataSource>() {
|
||||||
@Override
|
@Override
|
||||||
public Fields serialize(HikariDataSource o) {
|
public Fields serialize(HikariDataSource o) {
|
||||||
Fields fields = new Fields();
|
Fields fields = new Fields();
|
||||||
fields.putObject("jdbcurl", o.getJdbcUrl());
|
fields.putObject("jdbcurl", o.getJdbcUrl());
|
||||||
return fields;
|
return fields;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void deserialize(HikariDataSource o, Fields f) {
|
public void deserialize(HikariDataSource o, Fields f) {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected HikariDataSource deserialize(Fields fields) throws StreamCorruptedException {
|
protected HikariDataSource deserialize(Fields fields) throws StreamCorruptedException {
|
||||||
HikariDataSource ds = new HikariDataSource();
|
HikariDataSource ds = new HikariDataSource();
|
||||||
ds.setJdbcUrl((String) fields.getObject("jdbcurl"));
|
ds.setJdbcUrl((String) fields.getObject("jdbcurl"));
|
||||||
return ds;
|
return ds;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean mustSyncDeserialization() {
|
public boolean mustSyncDeserialization() {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected boolean canBeInstantiated() {
|
protected boolean canBeInstantiated() {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
name: skript-db
|
name: skript-db
|
||||||
version: 1.3.1
|
version: 1.3.5
|
||||||
main: com.btk5h.skriptdb.SkriptDB
|
main: com.btk5h.skriptdb.SkriptDB
|
||||||
depend: [Skript]
|
depend: [Skript]
|
||||||
authors: [btk5h, FranKusmiruk, Govindas]
|
authors: [btk5h, FranKusmiruk, Govindas, TPGamesNL]
|
||||||
api-version: 1.13
|
api-version: 1.13
|
||||||
|
|||||||
Reference in New Issue
Block a user