Migrate new API to use NIOFileSystem

vapor · Apr 24, 2024 · 863207d · 863207d
1 parent f2c439a
commit 863207d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/Sources/Vapor/Utilities/FileIO.swift b/Sources/Vapor/Utilities/FileIO.swift
@@ -304,13 +304,10 @@ public struct FileIO: Sendable {
     ///     - path: Path to file on the disk.
     /// - returns: `ByteBuffer` containing the file data.
     public func collectFile(at path: String) async throws -> ByteBuffer {
-        guard
-            let attributes = try? FileManager.default.attributesOfItem(atPath: path),
-            let fileSize = attributes[.size] as? NSNumber
-        else {
+        guard let fileSize = try await FileSystem.shared.info(forFileAt: .init(path))?.size else {
             throw Abort(.internalServerError)
         }
-        return try await self.read(path: path, fromOffset: 0, byteCount: fileSize.intValue)
+        return try await self.read(path: path, fromOffset: 0, byteCount: Int(fileSize))
     }
 
     /// Wrapper around `NIOFileSystem.FileChunks`.

diff --git a/Tests/VaporTests/FileTests.swift b/Tests/VaporTests/FileTests.swift
@@ -486,7 +486,7 @@ final class FileTests: XCTestCase {
     }
 
     func testAsyncFileRead() async throws {
-        let app = Application(.testing)
+        let app = await Application(.testing)
         defer { app.shutdown() }
 
         let request = Request(application: app, on: app.eventLoopGroup.next())