github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/dm/tests/_utils/check_port (about)

     1  #!/usr/bin/env python
     2  #coding: utf-8
     3  
     4  import sys
     5  import socket
     6  
     7  
     8  if __name__ == '__main__':
     9      if len(sys.argv) < 3:
    10          sys.exit(2)
    11      s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    12      s.settimeout(1)
    13      # if port is not alive, `socket.error` will be raised and exit with 1
    14      s.connect((sys.argv[1], int(sys.argv[2])))
    15      if s.send('\n'.encode()) != 1:
    16          raise Exception("failed to send")