forked from mock-server/mockserver-client-node
-
Notifications
You must be signed in to change notification settings - Fork 0
/
webSocketClient.js
80 lines (69 loc) · 2.85 KB
/
webSocketClient.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/*
* mockserver
* http://mock-server.com
*
* Copyright (c) 2014 James Bloom
* Licensed under the Apache License, Version 2.0
*/
(function () {
"use strict";
if (module && require) {
var WebSocketClient = require('websocket').client;
var webSocketClient = function (host, port, contextPath) {
var clientId;
var clientIdHandler;
var requestHandler;
var webSocketLocation = "ws://" + host + ":" + port + contextPath + "/_mockserver_callback_websocket";
var client = new WebSocketClient({
maxReceivedFrameSize: 64 * 1024 * 1024, // 64MiB
maxReceivedMessageSize: 64 * 1024 * 1024, // 64MiB
fragmentOutgoingMessages: false
});
client.on('connectFailed', function (error) {
console.log('Connect Error: ' + error.toString());
});
client.on('connect', function (connection) {
connection.on('error', function (error) {
console.log('Connection Error: ' + error.toString());
});
connection.on('message', function (message) {
if (message.type === 'utf8') {
var payload = JSON.parse(message.utf8Data);
if (payload.type === "org.mockserver.model.HttpRequest") {
var request = JSON.parse(payload.value);
var response = requestHandler(request);
connection.sendUTF(JSON.stringify(response));
} else if (payload.type === "org.mockserver.client.serialization.model.WebSocketClientIdDTO") {
var registration = JSON.parse(payload.value);
if (registration.clientId) {
clientId = registration.clientId;
if (clientIdHandler) {
clientIdHandler(clientId);
}
}
}
} else {
console.log('Incorrect message format: ' + JSON.parse(message));
}
});
});
client.connect(webSocketLocation, []);
function requestCallback(callback) {
requestHandler = callback;
}
function clientIdCallback(callback) {
clientIdHandler = callback;
if (clientId) {
clientIdHandler(clientId);
}
}
return {
requestCallback: requestCallback,
clientIdCallback: clientIdCallback
};
};
module.exports = {
webSocketClient: webSocketClient
};
}
})();