hyro/lighteco
1
0
Fork 0
mirror of https://github.com/xHyroM/lighteco.git synced 2024-11-24 15:51:06 +01:00
Code Issues Projects Releases Wiki Activity

feat: sqlite connection factory

Browse source
This commit is contained in:
Jozef Steinhübl 2023-08-28 12:18:26 +02:00
parent e340b32e86
commit 0578cdf3f4
3 changed files with 47 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
common/src/main/java/dev/xhyrom/lighteco/common/storage/provider/sql/connection/file/FileConnectionFactory.java
View file

@ -5,6 +5,7 @@ import dev.xhyrom.lighteco.common.storage.provider.sql.connection.ConnectionFact
import java.io.File;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.function.Function;
abstract class FileConnectionFactory implements ConnectionFactory {
private Connection connection;
@ -33,4 +34,9 @@ abstract class FileConnectionFactory implements ConnectionFactory {
return connection;
}
@Override
public Function<String, String> getStatementProcessor() {
return s -> s.replace('\'', '`');
}
}

6
common/src/main/java/dev/xhyrom/lighteco/common/storage/provider/sql/connection/file/H2ConnectionFactory.java
View file

@ -18,6 +18,7 @@ public class H2ConnectionFactory extends FileConnectionFactory {
@Override
public void init(LightEcoPlugin plugin) {
// TODO: implement
//ClassLoader classLoader = plugin
}
@ -37,9 +38,4 @@ public class H2ConnectionFactory extends FileConnectionFactory {
throw new SQLException("Failed to create connection", e);
}
}
@Override
public Function<String, String> getStatementProcessor() {
return s -> s.replace('\'', '`');
}
}

40
common/src/main/java/dev/xhyrom/lighteco/common/storage/provider/sql/connection/file/SqliteConnectionFactory.java Normal file
View file

@ -0,0 +1,40 @@
package dev.xhyrom.lighteco.common.storage.provider.sql.connection.file;
import dev.xhyrom.lighteco.common.plugin.LightEcoPlugin;
import java.io.File;
import java.lang.reflect.Constructor;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
import java.util.function.Function;
public class SqliteConnectionFactory extends FileConnectionFactory {
private Constructor<?> connectionConstructor;
public SqliteConnectionFactory(File file) {
super(file);
}
@Override
public void init(LightEcoPlugin plugin) {
// TODO: implement
}
@Override
protected Connection createConnection(File file) throws SQLException {
try {
return (Connection) this.connectionConstructor.newInstance(
"jdbc:sqlite:" + file.getAbsolutePath(),
new Properties(),
null, null, false
);
} catch (Exception e) {
if (e.getCause() instanceof SQLException) {
throw (SQLException) e.getCause();
}
throw new SQLException("Failed to create connection", e);
}
}
}