From 5ece0a13c45341b8a0ca1015a29c0ceed0c33c31 Mon Sep 17 00:00:00 2001
From: Timothy Flynn <trflynn89@pm.me>
Date: Fri, 9 Dec 2022 11:18:54 -0500
Subject: [PATCH] SQLServer: Add a hook to inform owners of disconnected SQL
 clients

---
 Userland/Services/SQLServer/ConnectionFromClient.cpp | 3 +++
 Userland/Services/SQLServer/ConnectionFromClient.h   | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/Userland/Services/SQLServer/ConnectionFromClient.cpp b/Userland/Services/SQLServer/ConnectionFromClient.cpp
index 6e1156f946..f45a3a3125 100644
--- a/Userland/Services/SQLServer/ConnectionFromClient.cpp
+++ b/Userland/Services/SQLServer/ConnectionFromClient.cpp
@@ -38,6 +38,9 @@ ConnectionFromClient::ConnectionFromClient(NonnullOwnPtr<Core::Stream::LocalSock
 void ConnectionFromClient::die()
 {
     s_connections.remove(client_id());
+
+    if (on_disconnect)
+        on_disconnect();
 }
 
 Messages::SQLServer::ConnectResponse ConnectionFromClient::connect(DeprecatedString const& database_name)
diff --git a/Userland/Services/SQLServer/ConnectionFromClient.h b/Userland/Services/SQLServer/ConnectionFromClient.h
index e28947e8e5..d41ba1451c 100644
--- a/Userland/Services/SQLServer/ConnectionFromClient.h
+++ b/Userland/Services/SQLServer/ConnectionFromClient.h
@@ -7,6 +7,7 @@
 #pragma once
 
 #include <AK/DeprecatedString.h>
+#include <AK/Function.h>
 #include <AK/HashMap.h>
 #include <AK/Vector.h>
 #include <LibIPC/ConnectionFromClient.h>
@@ -28,6 +29,7 @@ public:
     static RefPtr<ConnectionFromClient> client_connection_for(int client_id);
 
     void set_database_path(DeprecatedString);
+    Function<void()> on_disconnect;
 
 private:
     explicit ConnectionFromClient(NonnullOwnPtr<Core::Stream::LocalSocket>, int client_id);