Only close socket once.

linqs · Aug 24, 2023 · 58b996a · 58b996a
1 parent 26c2753
commit 58b996a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/psl-python/pslpython/deeppsl/server.py b/psl-python/pslpython/deeppsl/server.py
@@ -33,6 +33,7 @@
 ENCODING = 'utf-8'
 MAX_MESSAGE_SIZE_BYTES = 2 ** 14
 
+
 class ConnectionHandler(object):
     def __init__(self):
         self._model = None
@@ -206,8 +207,6 @@ def main(port):
 
         keep_open = handler.handle_request(connection, data)
         if not keep_open:
-            connection.shutdown(socket.SHUT_RDWR)
-            sock.shutdown(socket.SHUT_RDWR)
             break
 
     connection.close()
@@ -224,5 +223,6 @@ def _load_args(args):
 
     return int(args.pop(0))
 
+
 if __name__ == '__main__':
-    main(_load_args(list(sys.argv)))
+    main(_load_args(list(sys.argv)))