Add enableCleanSpaceService config for statemachine (#234)

openmessaging · Sep 14, 2022 · 505f438 · 505f438
1 parent 31ce7ef
commit 505f438
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/dledger/src/main/java/io/openmessaging/storage/dledger/DLedgerConfig.java b/dledger/src/main/java/io/openmessaging/storage/dledger/DLedgerConfig.java
@@ -91,6 +91,9 @@ public class DLedgerConfig {
 
  private long leadershipTransferWaitTimeout = 1000;
 
+ // Whether to force log cleanup when the disk reaches a certain space
+ private boolean enableCleanSpaceService = true;
+
  public String getDefaultPath() {
  return storeBaseDir + File.separator + "dledger-" + selfId;
  }
@@ -462,4 +465,12 @@ public String getSelfAddress() {
  public Map<String, String> getPeerAddressMap() {
  return this.peerAddressMap;
  }
+
+ public boolean isEnableCleanSpaceService() {
+ return enableCleanSpaceService;
+ }
+
+ public void setEnableCleanSpaceService(boolean enableCleanSpaceService) {
+ this.enableCleanSpaceService = enableCleanSpaceService;
+ }
 }
diff --git a/dledger/src/main/java/io/openmessaging/storage/dledger/store/file/DLedgerMmapFileStore.java b/dledger/src/main/java/io/openmessaging/storage/dledger/store/file/DLedgerMmapFileStore.java
@@ -85,23 +85,29 @@ public DLedgerMmapFileStore(DLedgerConfig dLedgerConfig, MemberState memberState
  localEntryBuffer = ThreadLocal.withInitial(() -> ByteBuffer.allocate(4 * 1024 * 1024));
  localIndexBuffer = ThreadLocal.withInitial(() -> ByteBuffer.allocate(INDEX_UNIT_SIZE * 2));
  flushDataService = new FlushDataService("DLedgerFlushDataService", logger);
- cleanSpaceService = new CleanSpaceService("DLedgerCleanSpaceService", logger);
+ if (dLedgerConfig.isEnableCleanSpaceService()) {
+ cleanSpaceService = new CleanSpaceService("DLedgerCleanSpaceService", logger);
+ }
  }
 
  @Override
  public void startup() {
  load();
  recover();
  flushDataService.start();
- cleanSpaceService.start();
+ if (cleanSpaceService != null) {
+ cleanSpaceService.start();
+ }
  }
 
  @Override
  public void shutdown() {
  this.dataFileList.flush(0);
  this.indexFileList.flush(0);
  persistCheckPoint();
- cleanSpaceService.shutdown();
+ if (cleanSpaceService != null) {
+ cleanSpaceService.shutdown();
+ }
  flushDataService.shutdown();
  }