WebSocketEx.lua

local WebSocket = cc.WebSocket

local events = {
    open  = cc.WEBSOCKET_OPEN,
    data  = cc.WEBSOCKET_MESSAGE,
    close = cc.WEBSOCKET_CLOSE,
    error = cc.WEBSOCKET_ERROR,
}

function WebSocket:on( eventName, callback )
    local code = events[eventName]
    if code then
        self:registerScriptHandler(callback, events[eventName])
    else
        printf("WebSocket:on eventName=%s does not exists", eventName)
    end
    return self
end

WebSocket.send = WebSocket.sendString