|
4 | 4 | import com.zaxxer.hikari.HikariDataSource; |
5 | 5 | import lombok.extern.log4j.Log4j2; |
6 | 6 | import me.hteppl.data.DataManager; |
7 | | -import org.jdbi.v3.core.Jdbi; |
| 7 | +import org.sql2o.Sql2o; |
8 | 8 |
|
9 | 9 | import javax.sql.DataSource; |
10 | 10 | import java.io.IOException; |
|
14 | 14 | @Log4j2 |
15 | 15 | public class Create { |
16 | 16 |
|
17 | | - public static Jdbi createSQLite(String database) { |
18 | | - return createSQLite(DataManager.getSettings().getSqliteDirectory(), database); |
| 17 | + /** |
| 18 | + * Helper method for PostgreSQL |
| 19 | + */ |
| 20 | + public static Sql2o createPostgreSQL(String host, int port, String database, String user, String password, String properties) { |
| 21 | + StringBuilder jdbcUrl = new StringBuilder("jdbc:postgresql://") |
| 22 | + .append(host).append(":").append(port).append("/").append(database); |
| 23 | + if (properties != null && !properties.isBlank()) { |
| 24 | + jdbcUrl.append("?").append(properties); |
| 25 | + } |
| 26 | + return createDataSource("org.postgresql.Driver", jdbcUrl.toString(), user, password); |
| 27 | + } |
| 28 | + |
| 29 | + /** |
| 30 | + * Helper method for MariaDB |
| 31 | + */ |
| 32 | + public static Sql2o createMariaDB(String host, int port, String database, String user, String password, String properties) { |
| 33 | + StringBuilder jdbcUrl = new StringBuilder("jdbc:mariadb://") |
| 34 | + .append(host).append(":").append(port).append("/").append(database); |
| 35 | + if (properties != null && !properties.isBlank()) { |
| 36 | + jdbcUrl.append("?").append(properties); |
| 37 | + } |
| 38 | + return createDataSource("org.mariadb.jdbc.Driver", jdbcUrl.toString(), user, password); |
19 | 39 | } |
20 | 40 |
|
21 | | - public static Jdbi createSQLite(String folder, String database) { |
| 41 | + /** |
| 42 | + * Helper method for SQLite |
| 43 | + */ |
| 44 | + public static Sql2o createSQLite(String folder, String database) { |
22 | 45 | try { |
23 | 46 | Class.forName("org.sqlite.JDBC"); |
24 | 47 | Files.createDirectories(Paths.get(folder)); |
25 | | - } catch (ClassNotFoundException ex) { |
26 | | - throw new RuntimeException("Error while trying to load SQLite driver", ex); |
27 | | - } catch (IOException ex) { |
28 | | - throw new RuntimeException("Error while trying to create " + folder, ex); |
| 48 | + } catch (IOException | ClassNotFoundException ex) { |
| 49 | + throw new RuntimeException(ex); |
29 | 50 | } |
30 | 51 |
|
31 | | - return Jdbi.create("jdbc:sqlite:" + folder + "/" + database + ".db"); |
32 | | - } |
33 | | - |
34 | | - public static Jdbi createMySQL(String host, int port, String database, String user, String password) { |
35 | | - return createMySQL(host, port, database, user, password, null); |
| 52 | + return new Sql2o("jdbc:sqlite:" + folder + "/" + database + ".db", null, null); |
36 | 53 | } |
37 | 54 |
|
38 | | - public static Jdbi createMySQL(String host, int port, String database, String user, String password, String properties) { |
| 55 | + public static Sql2o createDataSource(String driverClass, String jdbcUrl, String user, String password) { |
39 | 56 | try { |
40 | | - Class.forName("org.mariadb.jdbc.Driver"); |
41 | | - } catch (ClassNotFoundException ex) { |
42 | | - throw new RuntimeException("Error while trying to load MariaDB driver", ex); |
| 57 | + Class.forName(driverClass); |
| 58 | + } catch (ClassNotFoundException e) { |
| 59 | + throw new RuntimeException("Driver not found: " + driverClass, e); |
43 | 60 | } |
| 61 | + HikariConfig config = new HikariConfig(); |
| 62 | + Settings.HikariSettings hikari = DataManager.getSettings().getHikari(); |
44 | 63 |
|
45 | | - try { |
46 | | - var settings = DataManager.getSettings(); |
47 | | - var hikari = settings.getHikari(); |
48 | | - var config = new HikariConfig(); |
49 | | - |
50 | | - properties = properties != null && !properties.trim().isEmpty() ? "?" + properties : settings.getMysqlProperties(); |
51 | | - |
52 | | - config.setJdbcUrl("jdbc:mariadb://" + host + ":" + port + "/" + database + properties); |
53 | | - config.setUsername(user); |
54 | | - config.setPassword(password); |
55 | | - config.setAutoCommit(hikari.autoCommit); |
56 | | - config.setConnectionTimeout(hikari.connectionTimeout); |
57 | | - config.setIdleTimeout(hikari.idleTimeout); |
58 | | - config.setKeepaliveTime(hikari.keepaliveTime); |
59 | | - config.setMaxLifetime(hikari.maxLifetime); |
60 | | - config.setMaximumPoolSize(hikari.maximumPoolSize); |
61 | | - |
62 | | - return createByDataSource(new HikariDataSource(config)); |
63 | | - } catch (Exception ex) { |
64 | | - throw new RuntimeException(ex); |
65 | | - } |
| 64 | + // jdbcUrl |
| 65 | + config.setJdbcUrl(jdbcUrl); |
| 66 | + // credentials |
| 67 | + config.setUsername(user); |
| 68 | + config.setPassword(password); |
| 69 | + // HikariConfig variables |
| 70 | + config.setMinimumIdle(hikari.minIdle); |
| 71 | + config.setMaximumPoolSize(hikari.maximumPoolSize); |
| 72 | + config.setMaxLifetime(hikari.maxLifetime); |
| 73 | + config.setConnectionTimeout(hikari.connectionTimeout); |
| 74 | + config.setValidationTimeout(hikari.validationTimeout); |
| 75 | + config.setIdleTimeout(hikari.idleTimeout); |
| 76 | + config.setAutoCommit(hikari.autoCommit); |
| 77 | + config.setKeepaliveTime(hikari.keepaliveTime); |
| 78 | + return new Sql2o(new HikariDataSource(config)); |
66 | 79 | } |
67 | 80 |
|
68 | | - public static Jdbi createByDataSource(DataSource ds) { |
69 | | - return Jdbi.create(ds); |
| 81 | + public static Sql2o createFromDataSource(DataSource ds) { |
| 82 | + return new Sql2o(ds); |
70 | 83 | } |
71 | 84 | } |
0 commit comments