From e34100e120a51ea203e141193e14011cea55534c Mon Sep 17 00:00:00 2001
From: Tommy van der Vorst <tommy@pixelspark.nl>
Date: Sun, 3 Dec 2017 19:49:35 +0100
Subject: [PATCH] Add a way to obtain current database

---
 Sources/PostgresWireServer/QueryConnection.swift | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Sources/PostgresWireServer/QueryConnection.swift b/Sources/PostgresWireServer/QueryConnection.swift
index 9844b36..d7bc9b7 100644
--- a/Sources/PostgresWireServer/QueryConnection.swift
+++ b/Sources/PostgresWireServer/QueryConnection.swift
@@ -11,6 +11,7 @@ public final class QueryClientConnection<PreparedStatementType: PreparedStatemen
 		case closed
 	}
 
+	public private(set) var database: String? = nil
 	public private(set) var username: String? = nil
 	public private(set) var password: String? = nil
 	public private(set) var majorVersion: UInt16? = nil
@@ -662,6 +663,7 @@ public final class QueryClientConnection<PreparedStatementType: PreparedStatemen
 								// Read parameters
 								if let p = try s.readParameters(length: len - UInt32(8)) {
 									s.username = p["user"]
+									s.database = p["database"]
 
 									// Send authentication request
 									let buf = Data(bytes: [UInt8(Character("R").codePoint), 0, 0, 0, 8, 0, 0, 0, 3])