JavaScript:查找本地WebSocket服务器

时间:2014-11-30 16:59:24

标签: javascript websocket

我正试图在IP范围内尝试检测本地网络websocket服务器(我写的)。我有一个在网络上运行,我希望我的脚本检测它(和其他人在同一个端口上)。

目前我可以使用for循环完成此操作,然后根据我是否可以成功连接到该IP上的服务器来尝试创建列表。然而,这是非常缓慢和低效的。任何想法如何快速有效地做到这一点?

我的代码片段:

var port = 1234;
var ip_range = "192.168.1.";
var servers = [];
var i = 0;

function findServers() {
    if ( i > 255 ) {
        return;
    }
    try {
        var host = "ws://"+ip_range+i+":"+port;
        var socket = new WebSocket(host);
        socket.onopen = function(){
            console.log("Success: " + socket.url);
            servers.push(socket.url);
            i++;
            findServers();
        };
        socket.onerror = function(error){
            console.log("Error: " + socket.url);
            i++;
            findServers();
        };
    } catch (e) {
        console.log("Error: " + socket.url);
        i++;
        findServers();
    }
}

如果它有助于我编写套接字服务器,那么我可以根据需要修改代码。

1 个答案:

答案 0 :(得分:3)

使用试用套接字连接获得快速结果的关键是双重的:

  1. 并行运行多个连接尝试(尽可能多,但不要太多)
  2. 控制不存在的服务器的超时(WebSocket中内置的默认超时很长,因此这会使长时间等待短路,从而导致检查大量套接字的整个过程 - 其中大部分都不存在 - 需要很长时间)。
  3. 您不必使用WebWorkers来完成此任务,事实上,我认为使用WebWorkers效率可能较低,因为不需要为每个套接字创建新线程。 Javascript中的套接字编程完全是异步的,所以你可以让它们中的许多一次完成并且非常有效。

    这是一种完全异步的方式,可以同时运行多个套接字请求,并且应该比使用WebWorkers更有效。可能需要一些实验来查看一次可以打开多少活动WebSocket请求。这个例子将20作为该参数传递,但您可以尝试更多或更少。在您一次遇到活动套接字的浏览器限制之前,越多越好(并行请求越多)。

    获取结果的速度的关键参数是设置等待webSocket响应的时间超时值。在这个实现中,我允许您将其作为参数传递,并再次,您可以尝试适当的值。由于这些都是本地服务器,你正在寻找,我希望只需几秒钟(2000-4000ms)就可能对你有用,并且比普通的WebSocket超时产生更快的结果。

    function findServers(port, ipBase, ipLow, ipHigh, maxInFlight, timeout, cb) {
        var ipCurrent = +ipLow, numInFlight = 0, servers = [];
        ipHigh = +ipHigh;
    
        function tryOne(ip) {
            ++numInFlight;
            var address = "ws://" + ipBase + ip + ":" + port;
            var socket = new WebSocket(address);
            var timer = setTimeout(function() {
                console.log(address + " timeout");
                var s = socket;
                socket = null;
                s.close();
                --numInFlight;
                next();
            }, timeout);
            socket.onopen = function() {
                if (socket) {
                    console.log(address + " success");
                    clearTimeout(timer);
                    servers.push(socket.url);
                    --numInFlight;
                    next();
                }
            };
            socket.onerror = function(err) {
                if (socket) {
                    console.log(address + " error");
                    clearTimeout(timer);
                    --numInFlight;
                    next();
                }
            }
        }
    
        function next() {
            while (ipCurrent <= ipHigh && numInFlight < maxInFlight) {
                tryOne(ipCurrent++);
            }
            // if we get here and there are no requests in flight, then
            // we must be done
            if (numInFlight === 0) {
                console.log(servers);
                cb(servers);
            }
        }
    
        next();
    }
    
    findServers(1234, "192.168.1.", 1, 255, 20, 4000, function(servers) {
        console.log(servers);
    });