Upstream/Plan
1
0
Fork 0
mirror of https://github.com/plan-player-analytics/Plan.git synced 2024-07-03 01:55:09 +02:00
Code Issues Projects Releases Wiki Activity

Add possibility to query directly via service

Browse Source
This commit is contained in:
Risto Lahtela 2020-11-17 18:11:38 +02:00
parent 24b6669c7d
commit 35ca5a8560
2 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Plan/common/src/main/java/com/djrapitops/plan/DataService.java
View File

@ -16,6 +16,8 @@
*/
package com.djrapitops.plan;
import com.djrapitops.plan.storage.database.queries.Query;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Function;
@ -73,4 +75,6 @@ public interface DataService {
<P, S> DataService registerSupplier(Class<S> type, Class<P> parameterType, Function<P, S> supplierWithParameter);
<P, S> DataService registerDBSupplier(Class<S> type, Class<P> parameterType, Function<P, Query<S>> supplierWithParameter);
}

16
Plan/common/src/main/java/com/djrapitops/plan/DataSvc.java
View File

@ -16,6 +16,10 @@
*/
package com.djrapitops.plan;
import com.djrapitops.plan.storage.database.DBSystem;
import com.djrapitops.plan.storage.database.queries.Query;
import dagger.Lazy;
import javax.inject.Inject;
import javax.inject.Singleton;
import java.util.*;
@ -33,8 +37,13 @@ public class DataSvc implements DataService {
private final Map<ClassPair, Function> suppliersWithParameter;
private final MultiHashMap<Class, Consumer> consumers;
private final Lazy<DBSystem> dbSystem;
@Inject
public DataSvc() {
public DataSvc(
Lazy<DBSystem> dbSystem
) {
this.dbSystem = dbSystem;
mappers = new MultiHashMap<>();
mappersReverse = new MultiHashMap<>();
suppliers = new ConcurrentHashMap<>();
@ -122,6 +131,11 @@ public class DataSvc implements DataService {
return this;
}
@Override
public <P, S> DataService registerDBSupplier(Class<S> type, Class<P> parameterType, Function<P, Query<S>> queryVisitor) {
return registerSupplier(type, parameterType, parameter -> dbSystem.get().getDatabase().query(queryVisitor.apply(parameter)));
}
private static class ClassPair<A, B> {
final Class<A> a;
final Class<B> b;