@@ -162,16 +162,20 @@ def _is_sock_write_buff_empty(self):
162
162
163
163
# ----------------------------------------------------------------------
164
164
def _try_to_close_socket (self ):
165
+ self ._shutdown_socket ()
165
166
try :
166
- self ._sock .shutdown (socket .SHUT_WR )
167
167
self ._sock .close ()
168
168
except Exception as exc :
169
- self ._log_close_socket_error (exc )
169
+ message = f'Error on closing the transport socket: { exc } '
170
+ self ._log_close_socket_error (message )
171
+
172
+ # ----------------------------------------------------------------------
173
+ def _shutdown_socket (self ):
174
+ pass # not necessary for UDP sockets
170
175
171
176
# ----------------------------------------------------------------------
172
- def _log_close_socket_error (self , exc ):
173
- msg = f'Error on closing the transport socket: { exc } '
174
- safe_log_via_print ('warning' , msg )
177
+ def _log_close_socket_error (self , message ):
178
+ safe_log_via_print ('warning' , message )
175
179
176
180
# ----------------------------------------------------------------------
177
181
def close (self ):
@@ -235,6 +239,14 @@ def _send_via_socket(self, data):
235
239
data_to_send = self ._convert_data_to_send (data )
236
240
self ._sock .sendall (data_to_send )
237
241
242
+ # ----------------------------------------------------------------------
243
+ def _shutdown_socket (self ):
244
+ try :
245
+ self ._sock .shutdown (socket .SHUT_WR )
246
+ except Exception as exc :
247
+ message = f'Error on shutting down the transport socket: { exc } '
248
+ self ._log_close_socket_error (message )
249
+
238
250
239
251
class BeatsTransport :
240
252
0 commit comments