mirror of
https://github.com/xHyroM/lighteco.git
synced 2024-11-10 01:18:07 +01:00
feat(sql): impl load user
This commit is contained in:
parent
2a5a7133bb
commit
e340b32e86
4 changed files with 43 additions and 5 deletions
|
@ -2,6 +2,7 @@ package dev.xhyrom.lighteco.api.model.user;
|
|||
|
||||
import dev.xhyrom.lighteco.api.model.currency.Currency;
|
||||
import org.checkerframework.checker.nullness.qual.NonNull;
|
||||
import org.checkerframework.checker.nullness.qual.Nullable;
|
||||
|
||||
import java.math.BigDecimal;
|
||||
import java.util.UUID;
|
||||
|
@ -19,7 +20,7 @@ public interface User {
|
|||
*
|
||||
* @return the username
|
||||
*/
|
||||
@NonNull String getUsername();
|
||||
@Nullable String getUsername();
|
||||
|
||||
/**
|
||||
* Get the balance of this user for the specified currency.
|
||||
|
|
|
@ -2,12 +2,17 @@ package dev.xhyrom.lighteco.common.storage.provider.sql;
|
|||
|
||||
import dev.xhyrom.lighteco.api.model.user.User;
|
||||
import dev.xhyrom.lighteco.api.storage.StorageProvider;
|
||||
import dev.xhyrom.lighteco.common.model.currency.Currency;
|
||||
import dev.xhyrom.lighteco.common.plugin.LightEcoPlugin;
|
||||
import dev.xhyrom.lighteco.common.storage.provider.sql.connection.ConnectionFactory;
|
||||
import org.checkerframework.checker.nullness.qual.NonNull;
|
||||
|
||||
import java.math.BigDecimal;
|
||||
import java.sql.Connection;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.util.UUID;
|
||||
import java.util.function.Function;
|
||||
|
||||
public class SqlStorageProvider implements StorageProvider {
|
||||
private static final String SAVE_USER_LOCAL_CURRENCY = "";
|
||||
|
@ -19,29 +24,52 @@ public class SqlStorageProvider implements StorageProvider {
|
|||
(
|
||||
SELECT currency_identifier, balance
|
||||
FROM {prefix}_users
|
||||
WHERE uuid=?
|
||||
WHERE uuid=?1
|
||||
UNION ALL
|
||||
SELECT currency_identifier, balance
|
||||
FROM {prefix}_{context}_users
|
||||
WHERE uuid=?
|
||||
WHERE uuid=?1
|
||||
);
|
||||
""";
|
||||
|
||||
private final LightEcoPlugin plugin;
|
||||
private final ConnectionFactory connectionFactory;
|
||||
private final Function<String, String> statementProcessor;
|
||||
|
||||
public SqlStorageProvider(LightEcoPlugin plugin, ConnectionFactory connectionFactory) {
|
||||
this.plugin = plugin;
|
||||
this.connectionFactory = connectionFactory;
|
||||
this.statementProcessor = connectionFactory.getStatementProcessor().compose(
|
||||
s -> s
|
||||
.replace("{prefix}", plugin.getConfig().storage.tablePrefix)
|
||||
.replace("{context}", plugin.getConfig().server)
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public @NonNull User loadUser(@NonNull UUID uniqueId) throws Exception {
|
||||
try (Connection c = this.connectionFactory.getConnection()) {
|
||||
String uniqueIdString = uniqueId.toString();
|
||||
dev.xhyrom.lighteco.common.model.user.User user = this.plugin.getUserManager().getOrMake(uniqueId);
|
||||
|
||||
try (Connection c = this.connectionFactory.getConnection()) {
|
||||
try (PreparedStatement ps = c.prepareStatement(LOAD_WHOLE_USER)) {
|
||||
ps.setString(1, uniqueIdString);
|
||||
ps.setString(2, uniqueIdString);
|
||||
|
||||
ResultSet rs = ps.executeQuery();
|
||||
|
||||
while (rs.next()) {
|
||||
String currencyIdentifier = rs.getString("currency_identifier");
|
||||
Currency currency = this.plugin.getCurrencyManager().getIfLoaded(currencyIdentifier);
|
||||
|
||||
BigDecimal balance = rs.getBigDecimal("balance");
|
||||
|
||||
user.setBalance(currency, balance);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
return user.getProxy();
|
||||
}
|
||||
|
||||
@Override
|
||||
|
|
|
@ -3,10 +3,13 @@ package dev.xhyrom.lighteco.common.storage.provider.sql.connection;
|
|||
import dev.xhyrom.lighteco.common.plugin.LightEcoPlugin;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.util.function.Function;
|
||||
|
||||
public interface ConnectionFactory {
|
||||
void init(LightEcoPlugin plugin);
|
||||
void shutdown() throws Exception;
|
||||
|
||||
Function<String, String> getStatementProcessor();
|
||||
|
||||
Connection getConnection() throws Exception;
|
||||
}
|
||||
|
|
|
@ -7,6 +7,7 @@ import java.lang.reflect.Constructor;
|
|||
import java.sql.Connection;
|
||||
import java.sql.SQLException;
|
||||
import java.util.Properties;
|
||||
import java.util.function.Function;
|
||||
|
||||
public class H2ConnectionFactory extends FileConnectionFactory {
|
||||
private Constructor<?> connectionConstructor;
|
||||
|
@ -36,4 +37,9 @@ public class H2ConnectionFactory extends FileConnectionFactory {
|
|||
throw new SQLException("Failed to create connection", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public Function<String, String> getStatementProcessor() {
|
||||
return s -> s.replace('\'', '`');
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue