From 9a6b04412a7f1a94b43e8036e04028324fb3ade0 Mon Sep 17 00:00:00 2001
From: "Daniel J. Summers" <daniel@bitbadger.solutions>
Date: Tue, 19 Apr 2022 20:43:16 -0400
Subject: [PATCH] Fix env check

---
 .../DistributedRethinkDBCache.fs                    | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/RethinkDB.DistributedCache/DistributedRethinkDBCache.fs b/src/RethinkDB.DistributedCache/DistributedRethinkDBCache.fs
index d31b14c..b544536 100644
--- a/src/RethinkDB.DistributedCache/DistributedRethinkDBCache.fs
+++ b/src/RethinkDB.DistributedCache/DistributedRethinkDBCache.fs
@@ -67,11 +67,8 @@ type DistributedRethinkDBCache (options : IOptions<DistributedRethinkDBCacheOpti
         if log.IsEnabled LogLevel.Debug then log.LogDebug $"[{table}] %s{text ()}" 
 
     /// Make sure the RethinkDB database, table, expiration index exist
-    let checkEnvironment (_ : CancellationToken) =
+    let environmentCheck (_ : CancellationToken) =
         backgroundTask {
-            if environmentChecked then
-                dbug <| fun () -> "Skipping environment check because it has already been performed"
-                return ()
             dbug <| fun () -> "|> Checking for proper RethinkDB cache environment"
             // Database
             match db with
@@ -109,6 +106,14 @@ type DistributedRethinkDBCache (options : IOptions<DistributedRethinkDBCacheOpti
             environmentChecked <- true
         }
 
+    /// Make sure the RethinkDB database, table, expiration index exist
+    let checkEnvironment (cnxToken : CancellationToken) =
+        backgroundTask {
+            match environmentChecked with
+            | true -> dbug <| fun () -> "Skipping environment check because it has already been performed"
+            | false -> do! environmentCheck cnxToken
+        }
+    
     /// Remove entries from the cache that are expired
     let purgeExpired (_ : CancellationToken) =
         backgroundTask {