-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
dcc7014
commit 6ab3d15
Showing
3 changed files
with
292 additions
and
50 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| // | ||
| // FastlyCache.swift | ||
| // | ||
| // | ||
| // Created by Andrew Barba on 9/13/23. | ||
| // | ||
|
|
||
| import ComputeRuntime | ||
|
|
||
| extension Fastly { | ||
|
|
||
| public struct Cache: Sendable { | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> FetchResponse) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let body = try await handler().body.body | ||
| return HandlerData.body(body) | ||
| } | ||
| } | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> [UInt8]) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let bytes = try await handler() | ||
| return HandlerData.bytes(bytes) | ||
| } | ||
| } | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> String) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let text = try await handler() | ||
| return HandlerData.bytes(.init(text.utf8)) | ||
| } | ||
| } | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> Data) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let data = try await handler() | ||
| return HandlerData.bytes(data.bytes) | ||
| } | ||
| } | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> [String: Sendable]) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let json = try await handler() | ||
| let data = try JSONSerialization.data(withJSONObject: json) | ||
| return HandlerData.bytes(data.bytes) | ||
| } | ||
| } | ||
|
|
||
| public static func getOrSet(_ key: String, _ handler: () async throws -> [Sendable]) async throws -> ReadableBody { | ||
| return try await getOrSet(key) { | ||
| let json = try await handler() | ||
| let data = try JSONSerialization.data(withJSONObject: json) | ||
| return HandlerData.bytes(data.bytes) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| extension Fastly.Cache { | ||
|
|
||
| internal enum HandlerData { | ||
| case body(_ body: Fastly.Body) | ||
| case bytes(_ bytes: [UInt8]) | ||
| } | ||
|
|
||
| internal static func getOrSet(_ key: String, _ handler: () async throws -> HandlerData) async throws -> ReadableBody { | ||
| // Open the transaction | ||
| let trx = try await Transaction.lookup(key) | ||
|
|
||
| // Get current transaction state | ||
| let state = try await trx.getState() | ||
|
|
||
| // If state is usable then return the body from cache | ||
| if state == .found || state == .usable { | ||
| let body = try await trx.getBody() | ||
| return ReadableWasiBody(body) | ||
| } | ||
|
|
||
| // If its not usable then begin executing handler with new value | ||
| let data = try await handler() | ||
|
|
||
| // Get an instance to the insert handle | ||
| var writer = try await trx.insertAndStreamBack() | ||
|
|
||
| // Append bytes from handler to writeable body | ||
| switch data { | ||
| case .body(let body): | ||
| try writer.body.append(body) | ||
| case .bytes(let bytes): | ||
| try writer.body.write(bytes) | ||
| } | ||
|
|
||
| // Get handle to reabable body out from cache | ||
| return try await ReadableWasiBody(writer.transaction.getBody()) | ||
| } | ||
| } | ||
|
|
||
| extension Fastly.Cache { | ||
| public struct Transaction: Sendable { | ||
|
|
||
| internal let handle: WasiHandle | ||
|
|
||
| internal init(_ handle: WasiHandle) { | ||
| self.handle = handle | ||
| } | ||
|
|
||
| internal static func lookup(_ key: String) async throws -> Transaction { | ||
| var handle: WasiHandle = 0 | ||
| let options = CacheLookupOptions.reserved | ||
| var config = CacheLookupConfig() | ||
| try wasi(fastly_cache__cache_transaction_lookup(key, key.utf8.count, options.rawValue, &config, &handle)) | ||
| return Transaction(handle) | ||
| } | ||
|
|
||
| internal func insertAndStreamBack() async throws -> (body: Fastly.Body, transaction: Transaction) { | ||
| var bodyHandle: WasiHandle = 0 | ||
| var cacheHandle: WasiHandle = 0 | ||
| let options = CacheWriteOptions.reserved | ||
| var config = CacheWriteConfig() | ||
| try wasi(fastly_cache__cache_transaction_insert_and_stream_back(handle, options.rawValue, &config, &bodyHandle, &cacheHandle)) | ||
| return (.init(bodyHandle), .init(cacheHandle)) | ||
| } | ||
|
|
||
| internal func getState() async throws -> CacheState { | ||
| var value: UInt8 = 0 | ||
| try wasi(fastly_cache__cache_get_state(handle, &value)) | ||
| return CacheState(rawValue: value) | ||
| } | ||
|
|
||
| internal func getBody() async throws -> Fastly.Body { | ||
| var bodyHandle: WasiHandle = 0 | ||
| let options = CacheGetBodyOptions.reserved | ||
| var config = CacheGetBodyConfig() | ||
| try wasi(fastly_cache__cache_get_body(handle, options.rawValue, &config, &bodyHandle)) | ||
| return .init(bodyHandle) | ||
| } | ||
|
|
||
| internal func cancel() async throws { | ||
| try wasi(fastly_cache__cache_transaction_cancel(handle)) | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters