Store file value in cache on read

0xLeif · Apr 19, 2024 · 64115c7 · 64115c7
1 parent 6d47c27
commit 64115c7
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/Sources/AppState/Application/Types/State/Application+FileState.swift b/Sources/AppState/Application/Types/State/Application+FileState.swift
@@ -28,7 +28,36 @@ extension Application {
                 }
 
                 do {
-                    return try fileManager.in(path: path, filename: filename)
+                    let storedValue: Value = try fileManager.in(path: path, filename: filename)
+
+                    defer {
+                        let setValue = {
+                            shared.cache.set(
+                                value: Application.State(
+                                    type: .state,
+                                    initial: storedValue,
+                                    scope: scope
+                                ),
+                                forKey: scope.key
+                            )
+                        }
+
+                        #if (!os(Linux) && !os(Windows))
+                        if NSClassFromString("XCTest") == nil {
+                            Task {
+                                await MainActor.run {
+                                    setValue()
+                                }
+                            }
+                        } else {
+                            setValue()
+                        }
+                        #else
+                        setValue()
+                        #endif
+                    }
+
+                    return storedValue
                 } catch {
                     log(
                         error: error,