在函数内调用时不使用多个工作进程的ThreadPoolExecutor

Question

我正在尝试创建一个模块，该模块扫描给定的IP地址，并根据端口的当前状态返回每个端口的true或false。当上下文管理器被自己调用时，它工作得很好，但是当它在一个函数中被调用时，它会停止使用所有分配给它的线程。下面是我的代码：

import socket
import concurrent.futures

def _scan(ip, port):
    scanner = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    scanner.settimeout(1)
    try:
        scanner.connect((ip, port))
        scanner.close()
        return True
    except:
        return False

def portScan(ip, workers, portNum):
    with concurrent.futures.ThreadPoolExecutor(max_workers=workers) as executor:
        for port in range(portNum):
            future = executor.submit(_scan, ip, port + 1)
            print(future.result())

portScan("1.1.1.1", 100, 1000)

Answer 1

问题是，在提交下一个之前，您需要等待每个将来都完成。您可以改用map。它将工作分散到所有线程，并以与提交的参数相同的顺序迭代结果。

import socket
import concurrent.futures

def _scan(params):
    ip, port = params
    scanner = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    scanner.settimeout(1)
    try:
        scanner.connect((ip, port))
        scanner.close()
        return True
    except:
        return False


def portScan(ip, workers, portNum):
    with concurrent.futures.ThreadPoolExecutor(max_workers=workers) as executor:
        for result in executor.map(_scan, ((ip, port) for port in range(portNum))):
            print(result)

portScan("127.0.0.1", 5, 22)