fix websocket

[vitaliysemenko]

Fix websocket function definition and check param availability

[zentai]

for this function, I think they calling wrong api,

def on_close(original_connection):
    websocket_connection = websocket_connection_handler[original_connection]
    websocket_connection.close().   <<< should be close() instead of on_close()