xref: /aosp_15_r20/external/cronet/net/data/websocket/send-message-every-quarter-second_wsh.py (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1*6777b538SAndroid Build Coastguard Worker# Copyright 2023 The Chromium Authors
2*6777b538SAndroid Build Coastguard Worker# Use of this source code is governed by a BSD-style license that can be
3*6777b538SAndroid Build Coastguard Worker# found in the LICENSE file.
4*6777b538SAndroid Build Coastguard Worker
5*6777b538SAndroid Build Coastguard Workerimport six
6*6777b538SAndroid Build Coastguard Workerimport time
7*6777b538SAndroid Build Coastguard Worker
8*6777b538SAndroid Build Coastguard Worker
9*6777b538SAndroid Build Coastguard Workerdef web_socket_do_extra_handshake(request):
10*6777b538SAndroid Build Coastguard Worker  request.ws_extension_processors = []
11*6777b538SAndroid Build Coastguard Worker
12*6777b538SAndroid Build Coastguard Worker
13*6777b538SAndroid Build Coastguard Workerdef web_socket_transfer_data(request):
14*6777b538SAndroid Build Coastguard Worker  while True:
15*6777b538SAndroid Build Coastguard Worker    message = request.ws_stream.receive_message()
16*6777b538SAndroid Build Coastguard Worker
17*6777b538SAndroid Build Coastguard Worker    if message is None:
18*6777b538SAndroid Build Coastguard Worker      return
19*6777b538SAndroid Build Coastguard Worker
20*6777b538SAndroid Build Coastguard Worker    # Send <message> count messages to the web socket, each a quarter second
21*6777b538SAndroid Build Coastguard Worker    # apart.
22*6777b538SAndroid Build Coastguard Worker    for i in range(int(message)):
23*6777b538SAndroid Build Coastguard Worker      request.ws_stream.send_message('ping', binary=False)
24*6777b538SAndroid Build Coastguard Worker      time.sleep(0.25)
25