發送字符串消息給多個線程

Question

我有一個IRC客戶端接收套接字上的消息。

從這個客戶端我創建了幾個機器人，連接到其他人在twitch上的聊天頻道。 （這些是授權的而不是垃圾郵件機器人！）。

每個bot都是在一個單獨的線程中創建的，該線程將通道名稱和其他一些參數一起使用。

我的問題是我的IRC套接字只能綁定到一個端口，並處理所有的IRC消息，每個消息都有一個#channel字符串作爲第三個字符串，將它指向特定的通道。這些消息可以在每個機器人內部處理，因爲每個人都知道其頻道的名稱。

我的問題是;如何將通過套接字接收的字符串發送到多個線程？

import time 
import socket 
import threading 
import string 
import sys 
import os 

class IRCBetBot: 
    #irc ref 
    irc = None 

    def __init__(self,IRCRef,playerName,channelName,currencyName): 

     #assign variables 
     self.irc = IRCRef 
     self.channel = '#' + channelName 

     self.irc.send(('JOIN ' + self.channel + '\r\n') .encode("utf8")) 

     #create readbuffer to hold strings from IRC 
     readbuffer = "" 

     # This is the main loop 
     while 1: 

      ##readbuffer## <- need to send message from IRC to this variable 

      for line in temp: 
       line=str.rstrip(line) 
       line=str.split(line) 

       if (len(line) >= 4) and ("PRIVMSG" == line[1]) and (self.channel == line[2]) and not ("jtv" in line[0]): 
        #call function to handle user message 
       if(line[0]=="PING"): 
        self.irc.send(("PONG %s\r\n" % line[0]).encode("utf8")) 




def runAsThread(ircref,userName, channelName, currencyPrefix): 
    print("Got to runAsThread with : " + str(userName) + " " + str(channelName) + " " + str(currencyPrefix)) 
    IRCBetBot(ircref,userName,channelName,currencyPrefix) 

# Here we create the IRC connection 
#IRC connection variables 
nick = 'mybot'     #alter this value with the username used to connect to IRC eg: "username". 
password = "oauth:mykey"  #alter this value with the password used to connect to IRC from the username above. 
server = 'irc.twitch.tv' 
port = 6667 

#create IRC socket 
irc = socket.socket() 

irc.connect((server, port)) 

#sends variables for connection to twitch chat 
irc.send(('PASS ' + password + '\r\n').encode("utf8")) 
irc.send(('USER ' + nick + '\r\n').encode("utf8")) 
irc.send(('NICK ' + nick + '\r\n').encode("utf8")) 

# Array to hold all the new threads 
threads = [] 
# authorised Channels loaded from file in real program 
authorisedChannels = [["user1","#channel1","coin1"],["user2","#channel2","coin2"],["user3","#channel3","coin3"]] 

for item in authorisedChannels: 
    try: 
     userName = item[0] 
     channelName = item[1] 
     currencyPrefix = item [2] 
     myTuple = (irc,userName,channelName,currencyPrefix) 
     thread = threading.Thread(target=runAsThread,args = myTuple,) 
     thread.start() 
     threads.append(thread) 
     time.sleep(5) # wait to avoid too many connections to IRC at once from same IP 
    except Exception as e: 
     print("An error occurred while creating threads.") 
     print(str(e)) 

#create readbuffer to hold strings from IRC 
readbuffer = "" 

# This is the main loop 
while 1: 
    readbuffer= readbuffer+self.irc.recv(1024).decode("utf-8") 
    temp=str.split(readbuffer, "\n") 
    readbuffer=temp.pop() 
    # 
    #Need to send readbuffer to each IRCBetBot() created in runAsThread that contains a while 1: loop to listen for strings in its __init__() method. 
    # 

print ("Waiting...") 

for thread in threads: 
    thread.join() 

print ("Complete.") 

我需要以某種方式得到主迴路的readbuffer成單獨的線程創建的每個IRCBetBot對象？有任何想法嗎？

Answer 1

下面是一個示例，顯示如何使用每個線程的隊列來完成此操作。我們不是隻創建一個線程列表，而是創建一個以通道爲關鍵字的線程的字典，並存儲線程對象和可用於與字典中的線程交談的隊列。

#!/usr/bin/python3 

import threading 
from queue import Queue 


class IRCBetBot(threading.Thread): 
    def __init__(self, q, playerName, channelName, currencyName): 
     super().__init__() 
     self.channel = channelName 
     self.playerName = playerName 
     self.currencyName = currencyName 
     self.queue = q 

    def run(self): 
     readbuffer = "" 
     while 1: 
      readbuffer = self.queue.get() # This will block until a message is sent to the queue. 
      print("{} got msg {}".format(self.channel, readbuffer)) 

if __name__ == "__main__": 

    authorisedChannels = [["user1","#channel1","coin1"], 
          ["user2","#channel2","coin2"], 
          ["user3","#channel3","coin3"]] 

threads = {} 
for item in authorisedChannels: 
    try: 
     userName = item[0] 
     channelName = item[1] 
     currencyPrefix = item [2] 
     myTuple = (userName,channelName,currencyPrefix) 
     q = Queue() 
     thread = IRCBetBot(q, *myTuple) 
     thread.start() 
     threads[channelName] = (q, thread) 
    except Exception as e: 
     print("An error occurred while creating threads.") 
     print(str(e)) 

while 1: 
    a = input("Input your message (channel: msg): ") 
    channel, msg = a.split(":") 
    threads[channel][0].put(msg) # Sends a message using the queue object 

正如你可以看到，當郵件進入插座，我們解析通道出（你的代碼已經這樣做了），然後就傳遞給我們的線程字典合適的隊列中的消息。

樣本輸出（略調整，這樣輸出不炒因併發print話費）：

dan@dantop:~$ ./test.py 
Input your message (channel: msg): #channel1: hi there 
#channel1 got msg hi there 
Input your message (channel: msg): #channel2: another one 
#channel2 got msg another one 

Answer 2

那麼一種方法可能是讓readBuffers的數組類似於線程數組。然後每個線程基本上都在等待它的特定readbuffer上的數據。

當您獲取數據時，您可以將它傳遞給您感興趣的線程，或者將數據複製到所有讀緩衝區，並讓線程處理它們（如果他們感興趣）。在這種情況下，觀察者模式最適合。