Refactored websocket message.

1 files changed, 19 insertions, 12 deletions

diff --git a/zenserver/testing/httptest.cpp b/zenserver/testing/httptest.cpp
index 41a4f064b..10b69c469 100644
--- a/zenserver/testing/httptest.cpp
+++ b/zenserver/testing/httptest.cpp
@@ -8,6 +8,8 @@
 
 namespace zen {
 
+using namespace std::literals;
+
 HttpTestingService::HttpTestingService()
 {
 	m_Router.RegisterRoute(
@@ -136,29 +138,34 @@ HttpTestingService::HandlePackageRequest(HttpServerRequest& HttpServiceRequest)
 	return (InsertResult.first->second = new PackageHandler(*this, RequestId)).Get();
 }
 
-bool
-HttpTestingService::HandleMessage(WebSocketId SocketId, const CbPackage& Msg)
+void
+HttpTestingService::RegisterHandlers(WebSocketServer& Server)
 {
-	using namespace std::literals;
+	Server.RegisterRequestHandler("SayHello"sv, *this);
+}
 
-	CbObject Request = Msg.GetObject();
+bool
+HttpTestingService::HandleRequest(const WebSocketMessage& RequestMsg)
+{
+	CbObjectView Request = RequestMsg.Body().GetObject();
 
-	CbObjectView	 Header		   = Request["Header"sv].AsObjectView();
-	std::string_view Method		   = Header["Method"].AsString();
-	const uint64_t	 CorrelationId = Header["CorrelationId"].AsUInt64();
+	std::string_view Method = Request["Method"].AsString();
 
-	if (Method != "TestHelloZen"sv)
+	if (Method != "SayHello"sv)
 	{
 		return false;
 	}
 
 	CbObjectWriter Response;
-	Response.BeginObject("Header");
-	Response << "CorrelationId"sv << CorrelationId;
-	Response.EndObject();
 	Response.AddString("Result"sv, "Hello Friend!!");
 
-	PublishMessage(SocketId, Response.Save());
+	WebSocketMessage ResponseMsg;
+	ResponseMsg.SetMessageType(WebSocketMessageType::kResponse);
+	ResponseMsg.SetCorrelationId(RequestMsg.CorrelationId());
+	ResponseMsg.SetSocketId(RequestMsg.SocketId());
+	ResponseMsg.SetBody(Response.Save());
+
+	SocketServer().SendResponse(std::move(ResponseMsg));
 
 	return true;
 }