dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improvements after review

Browse Source
This commit is contained in:
ShvaykaD 2020-12-21 14:50:07 +02:00 committed by Andrew Shvayka
parent 0de5868bc5
commit aede1af6f9
2 changed files with 8 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
dao/src/main/java/org/thingsboard/server/dao/timeseries/CassandraBaseTimeseriesDao.java
View File

@ -405,33 +405,29 @@ public class CassandraBaseTimeseriesDao extends CassandraAbstractAsyncDao implem
} else {
CassandraPartitionCacheKey partitionSearchKey = new CassandraPartitionCacheKey(entityId, key, partition);
CompletableFuture<Boolean> hasFuture = cassandraTsPartitionsCache.has(partitionSearchKey);
SettableFuture<Boolean> listenableFuture = SettableFuture.create();
SettableFuture<Void> listenableFuture = SettableFuture.create();
if (hasFuture == null) {
return processDoSavePartition(tenantId, entityId, key, partition, partitionSearchKey, ttl);
} else {
long finalTtl = ttl;
hasFuture.whenComplete((result, throwable) -> {
if (throwable != null) {
listenableFuture.setException(throwable);
} else if (result) {
listenableFuture.set(null);
} else {
listenableFuture.set(result);
listenableFuture.setFuture(processDoSavePartition(tenantId, entityId, key, partition, partitionSearchKey, finalTtl));
}
});
long finalTtl = ttl;
return Futures.transformAsync(listenableFuture, result -> {
if (result) {
return Futures.immediateFuture(null);
} else {
return processDoSavePartition(tenantId, entityId, key, partition, partitionSearchKey, finalTtl);
}
}, readResultsProcessingExecutor);
return listenableFuture;
}
}
}
private ListenableFuture<Void> processDoSavePartition(TenantId tenantId, EntityId entityId, String key, long partition, CassandraPartitionCacheKey partitionSearchKey, long ttl) {
return Futures.transformAsync(doSavePartition(tenantId, entityId, key, ttl, partition), input -> {
return Futures.transform(doSavePartition(tenantId, entityId, key, ttl, partition), input -> {
cassandraTsPartitionsCache.put(partitionSearchKey);
return Futures.immediateFuture(input);
return input;
}, readResultsProcessingExecutor);
}

7
dao/src/test/java/org/thingsboard/server/dao/nosql/CassandraPartitionsCacheTest.java
View File

@ -103,15 +103,12 @@ public class CassandraPartitionsCacheTest {
ReflectionTestUtils.setField(cassandraBaseTimeseriesDao, "cluster", cluster);
doReturn(Futures.immediateFuture(null)).when(cassandraBaseTimeseriesDao).getFuture(any(ResultSetFuture.class), any());
}
@Test
public void testPartitionSave() throws Exception {
cassandraBaseTimeseriesDao.init();
UUID id = UUID.randomUUID();
TenantId tenantId = new TenantId(id);
long tsKvEntryTs = System.currentTimeMillis();
@ -119,14 +116,10 @@ public class CassandraPartitionsCacheTest {
for (int i = 0; i < 50000; i++) {
cassandraBaseTimeseriesDao.savePartition(tenantId, tenantId, tsKvEntryTs, "test" + i, 0);
}
for (int i = 0; i < 60000; i++) {
cassandraBaseTimeseriesDao.savePartition(tenantId, tenantId, tsKvEntryTs, "test" + i, 0);
}
verify(cassandraBaseTimeseriesDao, times(60000)).executeAsyncWrite(any(TenantId.class), any(Statement.class));
}
}