Shutdown notification to achieve a clean shutdown.

Otherwise the client always restarts the resource because of the lost connection.
We currently require this in tests to be able to delete the db, but eventually
we likely want a "disable akonadi" function that shuts resources down,
and keeps clients from restarting them (e.g. via configuration).

3 files changed, 24 insertions, 0 deletions

diff --git a/common/commands.h b/common/commands.h
index 26729dc..0007ffc 100644
--- a/common/commands.h
+++ b/common/commands.h
@@ -43,6 +43,7 @@ enum CommandIds {
     CreateEntityCommand,
     SearchSourceCommand, // need a buffer definition for this, but relies on Query API
     ShutdownCommand,
+    NotificationCommand,
     CustomCommand = 0xffff
 };
 
diff --git a/common/commands/notification.fbs b/common/commands/notification.fbs
new file mode 100644
index 0000000..6684472
--- /dev/null
+++ b/common/commands/notification.fbs
@@ -0,0 +1,9 @@
+namespace Akonadi2;
+
+enum NotificationType : byte { Shutdown = 1, Status, Warning, Progress }
+
+table Notification {
+    type: NotificationType = Status;
+}
+
+root_type Notification;
diff --git a/common/resourceaccess.cpp b/common/resourceaccess.cpp
index 59cbece..20e1e8c 100644
--- a/common/resourceaccess.cpp
+++ b/common/resourceaccess.cpp
@@ -25,6 +25,7 @@
 #include "common/handshake_generated.h"
 #include "common/revisionupdate_generated.h"
 #include "common/synchronize_generated.h"
+#include "common/notification_generated.h"
 #include "log.h"
 
 #include <QCoreApplication>
@@ -325,6 +326,19 @@ bool ResourceAccess::processMessageBuffer()
             QMetaObject::invokeMethod(this, "callCallbacks", Qt::QueuedConnection, QGenericReturnArgument(), Q_ARG(int, buffer->id()));
             break;
         }
+        case Commands::NotificationCommand: {
+            auto buffer = GetNotification(d->partialMessageBuffer.constData() + headerSize);
+            switch (buffer->type()) {
+                case Akonadi2::NotificationType::NotificationType_Shutdown:
+                    Log() << "Received shutdown notification.";
+                    close();
+                    break;
+                default:
+                    Warning() << "Received unknown notification: " << buffer->type();
+                    break;
+            }
+            break;
+        }
         default:
             break;
     }