git rid of NPE during store closing

bifromqio · Sep 14, 2024 · 8cb12a9 · 8cb12a9
1 parent 86cb78f
commit 8cb12a9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/build-cov.yaml b/.github/workflows/build-cov.yaml
@@ -11,9 +11,9 @@ jobs:
   build:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Set up JDK 17
-        uses: actions/setup-java@v3
+        uses: actions/setup-java@v4
         with:
           java-version: '17'
           distribution: 'temurin'

diff --git a/base-kv/base-kv-store-server/src/main/java/com/baidu/bifromq/basekv/store/KVRangeStore.java b/base-kv/base-kv-store-server/src/main/java/com/baidu/bifromq/basekv/store/KVRangeStore.java
@@ -351,6 +351,9 @@ private void receive(StoreMessage storeMessage) {
             if (payload.hasEnsureRange()) {
                 EnsureRange request = storeMessage.getPayload().getEnsureRange();
                 mgmtTaskRunner.add(() -> {
+                    if (status.get() != Status.STARTED) {
+                        return CompletableFuture.completedFuture(null);
+                    }
                     KVRangeId rangeId = payload.getRangeId();
                     IKVRangeFSM range = kvRangeMap.get(rangeId);
                     try {