aljaz/lnd-manageJ
1
0
Fork 0
mirror of https://github.com/aljazceru/lnd-manageJ.git synced 2026-01-27 01:44:30 +01:00
Code Issues Packages Projects Releases Wiki Activity

cache node information

Browse Source
This commit is contained in:
Carsten Otto
2021-11-05 18:28:02 +01:00
parent 8b4a9796c5
commit 9e6cb5e500
3 changed files with 39 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
grpc-adapter/src/main/java/de/cotto/lndmanagej/grpc/GrpcNodeInfo.java
View File

@@ -1,20 +1,47 @@
package de.cotto.lndmanagej.grpc;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import de.cotto.lndmanagej.model.Node;
import lnrpc.LightningNode;
import lnrpc.NodeInfo;
import org.springframework.stereotype.Component;
import javax.annotation.Nonnull;
import java.util.concurrent.TimeUnit;
@Component
public class GrpcNodeInfo {
private static final int MAXIMUM_SIZE = 500;
private static final int CACHE_EXPIRY_MINUTES = 30;
private final GrpcService grpcService;
private final LoadingCache<String, Node> cache;
public GrpcNodeInfo(GrpcService grpcService) {
this.grpcService = grpcService;
CacheLoader<String, Node> loader = new CacheLoader<>() {
@Override
public Node load(@Nonnull String pubkey) {
return getNodeWithoutCache(pubkey);
}
};
cache = CacheBuilder.newBuilder()
.expireAfterWrite(CACHE_EXPIRY_MINUTES, TimeUnit.MINUTES)
.maximumSize(MAXIMUM_SIZE)
.build(loader);
}
public Node getNode(String pubkey) {
NodeInfo nodeInfo = grpcService.getNodeInfo(pubkey);
return cache.getUnchecked(pubkey);
}
private Node getNodeWithoutCache(String pubkey) {
NodeInfo nodeInfo = grpcService.getNodeInfo(pubkey).orElse(null);
if (nodeInfo == null) {
return Node.builder().withPubkey(pubkey).build();
}
LightningNode node = nodeInfo.getNode();
return Node.builder()
.withPubkey(pubkey)

4
grpc-adapter/src/main/java/de/cotto/lndmanagej/grpc/GrpcService.java
View File

@@ -70,7 +70,7 @@ public class GrpcService {
.orElse(List.of());
}
public NodeInfo getNodeInfo(String pubkey) {
return lightningStub.getNodeInfo(NodeInfoRequest.newBuilder().setPubKey(pubkey).build());
public Optional<NodeInfo> getNodeInfo(String pubkey) {
return get(() -> lightningStub.getNodeInfo(NodeInfoRequest.newBuilder().setPubKey(pubkey).build()));
}
}

10
grpc-adapter/src/test/java/de/cotto/lndmanagej/grpc/GrpcNodeInfoTest.java
View File

@@ -9,6 +9,8 @@ import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import java.util.Optional;
import static de.cotto.lndmanagej.model.NodeFixtures.NODE;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.when;
@@ -27,7 +29,7 @@ class GrpcNodeInfoTest {
.setAlias(NODE.alias())
.setLastUpdate(NODE.lastUpdate())
.build());
when(grpcService.getNodeInfo(NODE.pubkey())).thenReturn(node.build());
when(grpcService.getNodeInfo(NODE.pubkey())).thenReturn(Optional.of(node.build()));
}
@Test
@@ -44,4 +46,10 @@ class GrpcNodeInfoTest {
void getNode_sets_last_update() {
assertThat(grpcNodeInfo.getNode(NODE.pubkey()).lastUpdate()).isEqualTo(NODE.lastUpdate());
}
@Test
void getNode_error() {
when(grpcService.getNodeInfo(NODE.pubkey())).thenReturn(Optional.empty());
assertThat(grpcNodeInfo.getNode(NODE.pubkey()).alias()).isEqualTo(NODE.pubkey());
}
}