/
websockets.py
executable file
·116 lines (104 loc) · 4.36 KB
/
websockets.py
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#!/usr/bin/env python
#-*-Mode:python;coding:utf-8;tab-width:4;c-basic-offset:4;indent-tabs-mode:()-*-
# ex: set ft=python fenc=utf-8 sts=4 ts=4 sw=4 et nomod:
#
# MIT License
#
# Copyright (c) 2013-2020 Michael Truog <mjtruog at protonmail dot com>
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
# DEALINGS IN THE SOFTWARE.
#
"""
Websockets Integration Test with Python/C
"""
from __future__ import print_function
import sys
import threading
import time
import traceback
from cloudi_c import API, TerminateException
class _Task(threading.Thread):
def __init__(self, api):
threading.Thread.__init__(self)
self.__api = api
def run(self):
try:
self.__api.subscribe('bounce/get', self.__request)
self.__api.subscribe('bounce/delay', self.__delay)
self.__api.subscribe('bounce/websocket/connect',
self.__connect)
self.__api.subscribe('bounce/websocket/disconnect',
self.__disconnect)
result = self.__api.poll()
assert result is False
except TerminateException:
pass
except Exception:
traceback.print_exc(file=sys.stderr)
print('terminate websockets python_c')
def __connect(self, request_type, name, pattern, request_info, request,
timeout, priority, trans_id, pid):
# pylint: disable=unused-argument
# pylint: disable=too-many-arguments
assert request == b'CONNECT'
print('connect: %s' %
str(API.info_key_value_parse(request_info)))
return 'got connect! yay!'
def __disconnect(self, request_type, name, pattern, request_info, request,
timeout, priority, trans_id, pid):
# pylint: disable=unused-argument
# pylint: disable=too-many-arguments
assert request == b'DISCONNECT'
print('disconnect: %s' %
str(API.info_key_value_parse(request_info)))
return ''
def __request(self, request_type, name, pattern, request_info, request,
timeout, priority, trans_id, pid):
# pylint: disable=unused-argument
# pylint: disable=too-many-arguments
# send the request to self
self.__api.send_async(self.__api.prefix() + 'bounce/delay',
request)
return request
def __delay(self, request_type, name, pattern, request_info, request,
timeout, priority, trans_id, pid):
# pylint: disable=unused-argument
# pylint: disable=too-many-arguments
time.sleep(1.0)
assert name[-6:] == '/delay'
trans_ids = self.__api.mcast_async(name[:-6] + '/websocket',
b'notification: got "' +
request + b'" 1 second ago')
if trans_ids == []:
print('websockets: (no websockets connected?)')
else:
for check in trans_ids:
(_, response, _) = self.__api.recv_async(trans_id=check)
print('websockets: %s' % str(response))
def _main():
thread_count = API.thread_count()
assert thread_count >= 1
threads = [_Task(API(thread_index))
for thread_index in range(thread_count)]
for thread in threads:
thread.start()
for thread in threads:
thread.join()
if __name__ == '__main__':
_main()