From 841045322f62c200e8d08eb642260d5ed4ff9bd6 Mon Sep 17 00:00:00 2001
From: Andrea Cavalli <andrea@cavallium.it>
Date: Wed, 13 Jan 2021 17:40:57 +0100
Subject: [PATCH] Fix registration sink

---
 .../java/it/tdlight/tdlibsession/EventBusFlux.java    | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/main/java/it/tdlight/tdlibsession/EventBusFlux.java b/src/main/java/it/tdlight/tdlibsession/EventBusFlux.java
index a26a63c..6a482c1 100644
--- a/src/main/java/it/tdlight/tdlibsession/EventBusFlux.java
+++ b/src/main/java/it/tdlight/tdlibsession/EventBusFlux.java
@@ -95,22 +95,25 @@ public class EventBusFlux {
 					if (h.succeeded()) {
 						dispose.completionHandler(h2 -> {
 							if (h2.succeeded()) {
-								sink.success();
+								msg.reply((Long) subscriptionId);
 							} else {
-								sink.error(h.cause());
+								logger.error("Failed to register dispose", h.cause());
+								msg.fail(500, "Failed to register dispose");
 							}
 						});
 					} else {
-						sink.error(h.cause());
+						logger.error("Failed to register cancel", h.cause());
+						msg.fail(500, "Failed to register cancel");
 					}
 				});
 
-				msg.reply((Long) subscriptionId);
 			});
 
 			subscribe.completionHandler(h -> {
 				if (h.failed()) {
 					sink.error(h.cause());
+				} else {
+					sink.success();
 				}
 			});
 		});