簡體   English   中英

使用 QThreadPool 時如何殺死工作線程?

[英]How to kill a worker thread when using QThreadPool?

我有這台服務器 class 子類 QRunnable ,它首先廣播 UDP 然后在客戶端連接時切換到 TCP 。 服務器 class 有一個事件循環來殺死自己和一個隊列來彈出然后發送到客戶端。

然后我有一個簡單的 GUI,我想輸入文本,然后在按下按鈕時通過服務器將其發送出去。

但是當我關閉 GUI 時,我的線程 class 不會停止我通過 QThreadpool 調用的,有沒有辦法在我關閉 GUI 后正確停止工作線程? 我已經有一個事件循環,但是當我調用它時,它什么也沒做,但是當我使用服務器 class 作為普通的 python 線程 class 時,它可以正確關閉。 任何幫助表示贊賞,在此先感謝!

服務器代碼:

class Server(QRunnable):

def __init__(self, ip_port):
    super(Server, self).__init__()
    self._stop_event = threading.Event()

    self.ip_port = ip_port
    self.ip_addr = ""

    self.sckt = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self.broadcast = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)

    self.client_sock = None
    self.q_to_send = queue.Queue()

def get_ip_addr(self):
    pattern = r'inet (?:addr:)?(?!127\.0\.0\.1)((?:\d+\.){3}\d+)'
    p = subprocess.Popen(['ifconfig'], stdout=subprocess.PIPE)
    self.ip_addr = re.search(pattern, p.stdout.read().decode()).group(1)
    print("server ip", self.ip_addr)

def set_up_tcp(self):
    self.sckt.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    self.sckt.settimeout(0.5)
    self.sckt.bind((self.ip_addr, self.ip_port))
    self.sckt.listen(5)

def broadcast_udp(self):
    self.broadcast.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
    self.broadcast.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
    self.broadcast.sendto(bytes(self.ip_addr, encoding='utf-8'), ("<broadcast>", self.ip_port))
    print("UDP broadcast sent!")

def connected(self):
    self.broadcast.close()
    self.client_sock.setblocking(False)

def stop(self):
    self._stop_event.set()
    print("Stopping server thread...")

def stopped(self):
    return self._stop_event.is_set()

def run(self):
    self.get_ip_addr()
    self.set_up_tcp()
    while not self.stopped():

        while True:
            self.broadcast_udp()
            time.sleep(1)
            try:
                self.client_sock, _ = self.sckt.accept()
                print("\nA client has connected!")
                break
            except socket.timeout:
                print("socket timeout")
                pass
        self.connected()
        print("Switched to TCP")

        while True:
            try:
                time.sleep(0.01)
                if self.q_to_send.empty():
                    pass
                else:
                    msg = self.q_to_send.get()
                    self.client_sock.send(bytes(msg, "utf8"))
                    print("'" + msg + "'" + " sent")
            except BrokenPipeError:
                print("\nClient disconnected! Now broadcasting UDP")
                break

圖形用戶界面代碼:

class Ui_MainWindow(QMainWindow):
def __init__(self):
    super(Ui_MainWindow, self).__init__()

    MainWindow.setObjectName("MainWindow")
    MainWindow.resize(470, 319)
    self.centralwidget = QtWidgets.QWidget(MainWindow)

    self.centralwidget.setObjectName("centralwidget")
    self.sendButton = QtWidgets.QPushButton(self.centralwidget)
    self.sendButton.setGeometry(QtCore.QRect(180, 200, 112, 32))

    self.sendButton.setObjectName("sendButton")
    self.label = QtWidgets.QLabel(self.centralwidget)
    self.label.setGeometry(QtCore.QRect(140, 30, 241, 20))

    self.label.setObjectName("label")
    self.textEditBox = QtWidgets.QTextEdit(self.centralwidget)
    self.textEditBox.setGeometry(QtCore.QRect(120, 90, 231, 71))

    self.textEditBox.setObjectName("textEditBox")
    MainWindow.setCentralWidget(self.centralwidget)
    self.menubar = QtWidgets.QMenuBar(MainWindow)
    self.menubar.setGeometry(QtCore.QRect(0, 0, 470, 22))

    self.menubar.setObjectName("menubar")
    MainWindow.setMenuBar(self.menubar)
    self.statusbar = QtWidgets.QStatusBar(MainWindow)
    self.statusbar.setObjectName("statusbar")
    MainWindow.setStatusBar(self.statusbar)

    self.retranslateUi(MainWindow)
    QtCore.QMetaObject.connectSlotsByName(MainWindow)

    #init UI
    self.initUI()

    #start server
    self.server = Server(10000)
    self.threadpool = QThreadPool()
    self.threadpool.start(self.server)

def initUI(self):
    self.sendButton.clicked.connect(self.send)

def retranslateUi(self, MainWindow):
    _translate = QtCore.QCoreApplication.translate
    MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
    self.sendButton.setText(_translate("MainWindow", "Send"))
    self.label.setText(_translate("MainWindow", "Enter command below to send"))

def send(self):
    self.server.q_to_send.put(self.textEditBox.toPlainText())

if __name__ == "__main__":
import sys

app = QtWidgets.QApplication(sys.argv)
MainWindow = QtWidgets.QMainWindow()
ui = Ui_MainWindow()
MainWindow.show()
app.exec_()

為什么不呢,在你離開的時候,比如在 QMainWindow.closeEvent() 中,告訴你服務器停止:self.server.q_to_send.put(TIME_IS_UP_LET_US_GET_OFF_WORK) 或 self.server.stop()

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM