bytes (vapor#38)

* bytes

* update linuxmain

Author: tanner0101

Sources/PostgresNIO/Data/PostgresData+Bytes.swift

@@ -0,0 +1,38 @@
+import struct Foundation.Data
+
+extension PostgresData {
+    public init<Bytes>(bytes: Bytes)
+        where Bytes: Sequence, Bytes.Element == UInt8
+    {
+        var buffer = ByteBufferAllocator().buffer(capacity: 1)
+        buffer.writeBytes(bytes)
+        self.init(type: .bytea, formatCode: .binary, value: buffer)
+    }
+
+    public var bytes: [UInt8]? {
+        guard var value = self.value else {
+            return nil
+        }
+        guard let bytes = value.readBytes(length: value.readableBytes) else {
+            return nil
+        }
+        return bytes
+    }
+}
+
+extension Data: PostgresDataConvertible {
+    public static var postgresDataType: PostgresDataType {
+        return .bytea
+    }
+
+    public var postgresData: PostgresData? {
+        return .init(bytes: self)
+    }
+
+    public init?(postgresData: PostgresData) {
+        guard let bytes = postgresData.bytes else {
+            return nil
+        }
+        self.init(bytes)
+    }
+}

Tests/PostgresNIOTests/NIOPostgresTests.swift

@@ -386,6 +386,15 @@ final class NIOPostgresTests: XCTestCase {
             XCTAssertEqual(rows[0].column("bool")?.bool, false)
         }
     }
+
+    func testBytesSerialize() throws {
+        let conn = try PostgresConnection.test(on: eventLoop).wait()
+        defer { try! conn.close().wait() }
+        let rows = try conn.query("select $1::bytea as bytes", [
+            PostgresData(bytes: [1, 2, 3])
+        ]).wait()
+        XCTAssertEqual(rows[0].column("bytes")?.bytes, [1, 2, 3])
+    }
 
     func testJSONBSerialize() throws {
         struct Object: Codable {

Tests/PostgresNIOTests/XCTestManifests.swift

@@ -9,8 +9,7 @@ extension NIOPostgresTests {
         ("testAverageLengthNumeric", testAverageLengthNumeric),
         ("testBindInteger", testBindInteger),
         ("testBoolSerialize", testBoolSerialize),
-        ("testJSONBSerialize", testJSONBSerialize),
-        ("testJSONBConvertible", testJSONBConvertible),
+        ("testBytesSerialize", testBytesSerialize),
         ("testColumnsInJoin", testColumnsInJoin),
         ("testConnectAndClose", testConnectAndClose),
         ("testDates", testDates),
@@ -20,6 +19,8 @@ extension NIOPostgresTests {
         ("testIntegerArraySerialize", testIntegerArraySerialize),
         ("testIntegers", testIntegers),
         ("testInvalidPassword", testInvalidPassword),
+        ("testJSONBConvertible", testJSONBConvertible),
+        ("testJSONBSerialize", testJSONBSerialize),
         ("testMoney", testMoney),
         ("testNullIntegerArrayParse", testNullIntegerArrayParse),
         ("testNumericParsing", testNumericParsing),