X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityLoader.java;h=0917130f2fa279b589995334c10c54e4d3cf02ae;hb=c6c4802b4e389b4457adf0827efafca1cd7d1e9c;hp=550079b0c604479954771b26deb4fbeed0d7f3df;hpb=4c4b77eff97fd247c6b1c8bbb30aeb6ea3d2c172;p=Sone.git
diff --git a/src/main/java/net/pterodactylus/sone/freenet/wot/IdentityLoader.java b/src/main/java/net/pterodactylus/sone/freenet/wot/IdentityLoader.java
index 550079b..0917130 100644
--- a/src/main/java/net/pterodactylus/sone/freenet/wot/IdentityLoader.java
+++ b/src/main/java/net/pterodactylus/sone/freenet/wot/IdentityLoader.java
@@ -1,5 +1,5 @@
/*
- * Sone - IdentityLoader.java - Copyright © 2013 David Roden
+ * Sone - IdentityLoader.java - Copyright © 2013â2019 David Roden
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
@@ -17,57 +17,64 @@
package net.pterodactylus.sone.freenet.wot;
-import static com.google.common.collect.HashMultimap.create;
+import static java.util.concurrent.TimeUnit.*;
+import static net.pterodactylus.sone.freenet.wot.Context.*;
-import java.util.Collection;
-import java.util.Set;
+import java.util.*;
+import java.util.logging.*;
-import net.pterodactylus.sone.freenet.plugin.PluginException;
+import net.pterodactylus.sone.freenet.plugin.*;
import com.google.common.base.Optional;
-import com.google.common.collect.Multimap;
+import com.google.common.base.*;
+import com.google.inject.*;
/**
* Loads {@link OwnIdentity}s and the {@link Identity}s they trust.
- *
- * @author David âBombeâ Roden
*/
public class IdentityLoader {
+ private final Logger logger = Logger.getLogger(IdentityLoader.class.getName());
private final WebOfTrustConnector webOfTrustConnector;
- private final Optional context;
+ private final Optional context;
public IdentityLoader(WebOfTrustConnector webOfTrustConnector) {
- this(webOfTrustConnector, Optional.absent());
+ this(webOfTrustConnector, Optional.absent());
}
- public IdentityLoader(WebOfTrustConnector webOfTrustConnector, Optional context) {
+ @Inject
+ public IdentityLoader(WebOfTrustConnector webOfTrustConnector, Optional context) {
this.webOfTrustConnector = webOfTrustConnector;
this.context = context;
}
- public Multimap loadIdentities() throws WebOfTrustException {
+ public Map> loadIdentities() throws WebOfTrustException {
+ Stopwatch stopwatch = Stopwatch.createStarted();
Collection currentOwnIdentities = webOfTrustConnector.loadAllOwnIdentities();
+ logger.fine("Loaded " + currentOwnIdentities.size() + " own identities in " + (stopwatch.elapsed(MILLISECONDS) / 1000.0) + "s.");
return loadTrustedIdentitiesForOwnIdentities(currentOwnIdentities);
}
- private Multimap loadTrustedIdentitiesForOwnIdentities(Collection ownIdentities) throws PluginException {
- Multimap currentIdentities = create();
+ private Map> loadTrustedIdentitiesForOwnIdentities(Collection ownIdentities) throws PluginException {
+ Map> currentIdentities = new HashMap<>();
for (OwnIdentity ownIdentity : ownIdentities) {
if (identityDoesNotHaveTheCorrectContext(ownIdentity)) {
+ currentIdentities.put(ownIdentity, Collections.emptySet());
continue;
}
- Set trustedIdentities = webOfTrustConnector.loadTrustedIdentities(ownIdentity, context.orNull());
- currentIdentities.putAll(ownIdentity, trustedIdentities);
+ Stopwatch stopwatch = Stopwatch.createStarted();
+ Set trustedIdentities = webOfTrustConnector.loadTrustedIdentities(ownIdentity, context.transform(extractContext));
+ logger.fine("Loaded " + trustedIdentities.size() + " identities for " + ownIdentity.getNickname() + " in " + (stopwatch.elapsed(MILLISECONDS) / 1000.0) + "s.");
+ currentIdentities.put(ownIdentity, trustedIdentities);
}
return currentIdentities;
}
private boolean identityDoesNotHaveTheCorrectContext(OwnIdentity ownIdentity) {
- return context.isPresent() && !ownIdentity.hasContext(context.get());
+ return context.isPresent() && !ownIdentity.hasContext(context.transform(extractContext).get());
}
}