From c69a92d4bdbc131ab570f11b94434d14107ce6ef Mon Sep 17 00:00:00 2001
From: Thomas Bartelmess <tbartelmess@marketcircle.com>
Date: Sat, 9 Mar 2019 15:40:32 +0100
Subject: [PATCH] Update for latest NIO API's

---
 Sources/NIOPostgres/Connection/PostgresConnection+Connect.swift | 2 +-
 Sources/NIOPostgres/Message/PostgresMessageEncoder.swift        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Sources/NIOPostgres/Connection/PostgresConnection+Connect.swift b/Sources/NIOPostgres/Connection/PostgresConnection+Connect.swift
index 7d7908b0..1f00f0f9 100644
--- a/Sources/NIOPostgres/Connection/PostgresConnection+Connect.swift
+++ b/Sources/NIOPostgres/Connection/PostgresConnection+Connect.swift
@@ -8,7 +8,7 @@ extension PostgresConnection {
         return bootstrap.connect(to: socketAddress).flatMap { channel in
             return channel.pipeline.addHandlers([
                 ByteToMessageHandler(PostgresMessageDecoder()),
-                PostgresMessageEncoder(),
+                MessageToByteHandler(PostgresMessageEncoder()),
                 PostgresConnectionHandler(),
             ]).map {
                 return .init(channel: channel)
diff --git a/Sources/NIOPostgres/Message/PostgresMessageEncoder.swift b/Sources/NIOPostgres/Message/PostgresMessageEncoder.swift
index e1523432..afbd824e 100644
--- a/Sources/NIOPostgres/Message/PostgresMessageEncoder.swift
+++ b/Sources/NIOPostgres/Message/PostgresMessageEncoder.swift
@@ -5,7 +5,7 @@ public final class PostgresMessageEncoder: MessageToByteEncoder {
     public typealias OutboundIn = PostgresMessage
     
     /// See `MessageToByteEncoder`.
-    public func encode(context: ChannelHandlerContext, data message: PostgresMessage, out: inout ByteBuffer) throws {
+    public func encode(data message: PostgresMessage, out: inout ByteBuffer) throws {
         // print("PostgresMessage.ChannelEncoder.encode(\(data))")
         
         // serialize identifier