PostgreSQLDemo

Author: Andrea-Scuderi

Examples/PostgreSQLDemo/Package.resolved

@@ -0,0 +1,25 @@
+// swift-tools-version:5.1
+// The swift-tools-version declares the minimum version of Swift required to build this package.
+
+import PackageDescription
+
+let package = Package(
+    name: "PostgreSQLDemo",
+    dependencies: [
+        // Dependencies declare other packages that this package depends on.
+        // .package(url: /* package url */, from: "1.0.0"),
+         .package(url: "https://github.com/swift-sprinter/aws-lambda-swift-sprinter-nio-plugin", from: "1.0.0-alpha.3"),
+        .package(url: "https://github.com/vapor/postgres-nio.git", from: "1.0.0-alpha.1.6"),
+        .package(url: "https://github.com/apple/swift-log.git", from: "1.0.0"),
+    ],
+    targets: [
+        // Targets are the basic building blocks of a package. A target can define a module or a test suite.
+        // Targets can depend on other targets in this package, and on products in packages which this package depends on.
+        .target(
+            name: "PostgreSQLDemo",
+            dependencies: ["LambdaSwiftSprinterNioPlugin", "Logging", "PostgresNIO"]),
+        .testTarget(
+            name: "PostgreSQLDemoTests",
+            dependencies: ["PostgreSQLDemo"]),
+    ]
+)

Examples/PostgreSQLDemo/Package.swift

@@ -0,0 +1,3 @@
+# PostgreSQLDemo
+
+A description of this package.

Examples/PostgreSQLDemo/README.md

@@ -0,0 +1,59 @@
+import AsyncHTTPClient
+import Foundation
+#if canImport(FoundationNetworking)
+import FoundationNetworking
+#endif
+import LambdaSwiftSprinter
+import LambdaSwiftSprinterNioPlugin
+import Logging
+import NIO
+import NIOFoundationCompat
+import PostgresNIO
+
+struct Event: Codable {
+    let query: String
+}
+
+struct Response: Codable {
+    let value: String
+}
+
+enum LambdaError: Error {
+    case connectionFailed
+}
+
+let logger = Logger(label: "AWS.Lambda.Redis")
+let endpoint = "<yourdb>.rds.amazonaws.com"
+do {
+    let eventLoop = httpClient.eventLoopGroup.next()
+    let connection = try PostgresConnection.connect(
+        to: try .makeAddressResolvingHost(endpoint,
+                                          port: 5432),
+        on: eventLoop
+    ).wait()
+    
+    logger.error("after connection")
+    
+    try connection.authenticate(username: "<username>",
+                                database: "<db>",
+                                password: "<password>").wait()
+    
+    
+    let syncCodableNIOLambda: SyncCodableNIOLambda<Event, Response> = { (event, context) throws -> EventLoopFuture<Response> in
+        
+        let future = connection.query(event.query).map { (rows) -> Response in
+            return Response(value: "\(rows)")
+            
+        }
+        return future
+    }
+    
+    
+    let sprinter = try SprinterNIO()
+    //Note amend this line in case if it's required to use a different lambda handler.
+    sprinter.register(handler: "query", lambda: syncCodableNIOLambda)
+    
+    try sprinter.run()
+} catch {
+    logger.error("\(String(describing: error))")
+}

Examples/PostgreSQLDemo/Sources/PostgreSQLDemo/main.swift

@@ -0,0 +1,7 @@
+import XCTest
+
+import PostgreSQLDemoTests
+
+var tests = [XCTestCaseEntry]()
+tests += PostgreSQLDemoTests.allTests()
+XCTMain(tests)

Examples/PostgreSQLDemo/Tests/LinuxMain.swift

@@ -0,0 +1,47 @@
+import XCTest
+import class Foundation.Bundle
+
+final class PostgreSQLDemoTests: XCTestCase {
+    func testExample() throws {
+        // This is an example of a functional test case.
+        // Use XCTAssert and related functions to verify your tests produce the correct
+        // results.
+
+        // Some of the APIs that we use below are available in macOS 10.13 and above.
+        guard #available(macOS 10.13, *) else {
+            return
+        }
+
+        let fooBinary = productsDirectory.appendingPathComponent("PostgreSQLDemo")
+
+        let process = Process()
+        process.executableURL = fooBinary
+
+        let pipe = Pipe()
+        process.standardOutput = pipe
+
+        try process.run()
+        process.waitUntilExit()
+
+        let data = pipe.fileHandleForReading.readDataToEndOfFile()
+        let output = String(data: data, encoding: .utf8)
+
+        XCTAssertEqual(output, "Hello, world!\n")
+    }
+
+    /// Returns path to the built products directory.
+    var productsDirectory: URL {
+      #if os(macOS)
+        for bundle in Bundle.allBundles where bundle.bundlePath.hasSuffix(".xctest") {
+            return bundle.bundleURL.deletingLastPathComponent()
+        }
+        fatalError("couldn't find the products directory")
+      #else
+        return Bundle.main.bundleURL
+      #endif
+    }
+
+    static var allTests = [
+        ("testExample", testExample),
+    ]
+}

Examples/PostgreSQLDemo/Tests/PostgreSQLDemoTests/PostgreSQLDemoTests.swift

@@ -0,0 +1,9 @@
+import XCTest
+
+#if !canImport(ObjectiveC)
+public func allTests() -> [XCTestCaseEntry] {
+    return [
+        testCase(PostgreSQLDemoTests.allTests),
+    ]
+}
+#endif

Examples/PostgreSQLDemo/Tests/PostgreSQLDemoTests/XCTestManifests.swift

@@ -0,0 +1,3 @@
+{
+    "query": "SELECT version()"
+}

Examples/PostgreSQLDemo/event.json

@@ -36,6 +36,12 @@ LAMBDA_HANDLER?=$(SWIFT_EXECUTABLE).helloWorld
 # LAMBDA_FUNCTION_NAME?=RedisDemo
 # LAMBDA_HANDLER?=$(SWIFT_EXECUTABLE).setGet
 
+# PostgreSQLDemo Example Configuration
+# SWIFT_EXECUTABLE=PostgreSQLDemo
+# SWIFT_PROJECT_PATH=Examples/PostgreSQLDemo
+# LAMBDA_FUNCTION_NAME=PostgreSQLDemo
+# LAMBDA_HANDLER=$(SWIFT_EXECUTABLE).query
+
 # AWS Configuration
 IAM_ROLE_NAME?=lambda_sprinter_basic_execution
 AWS_PROFILE?=default