import socket
import json,struct
from concurrent.futures import ThreadPoolExecutor
debug = True

s = socket.socket()
s.bind(("127.0.0.1",8848))
s.listen()
clients = {}

pool = ThreadPoolExecutor(100)

def send_msg(soc, msg):
    l = len(msg.encode("utf-8"))
    soc.send(struct.pack("q",l))

    # 發(fā)數(shù)據(jù)
    soc.send(msg.encode("utf-8"))

'''用于監(jiān)聽(tīng)已經(jīng)建立連接的客戶(hù)端發(fā)來(lái)的消息'''
def task(ip,c):
    while True:
        l = c.recv(8)
        ls = struct.unpack("q",l)[0]
        data = json.loads(c.recv(ls).decode("utf-8"))   #接收到來(lái)自客戶(hù)端的消息。 eg:    data = {'to_addr':'msg':''}
        print('來(lái)自%s的消息：%s'%(ip,data['msg']))
        # 客戶(hù)端發(fā)過(guò)來(lái)的數(shù)據(jù)
        # 數(shù)據(jù)有兩種情況 一種是發(fā)給所有人的 另一種單獨(dú)發(fā)給某一個(gè)人的
        if data.get('to_addr'): #傳過(guò)來(lái)一個(gè)json格式的字典，如果這個(gè)to_addr不是空的話(huà)，就私發(fā)給目標(biāo)客戶(hù)
            target_ip = data["to_addr"]        #從data字典里中獲取目標(biāo)ip
            target_conn = clients.get(target_ip)   #獲取目標(biāo)客戶(hù)的conn鏈接
            send_msg(target_conn,data['msg'])    #發(fā)送訊息
        else:
            for ip,conn in clients.items():
                # if c != target_conn:
                send_msg(conn,data['msg'])#data['msg']


        #     # 從所有客戶(hù)端列表中找到這一個(gè)  發(fā)給它
        #     to_addr = data["to_addr"]
        #     # print(data["to_addr"],"_______________")
        #     soc = clients.get(to_addr)
        #     send_msg(soc,data["msg"])
        # else:
        #     # 遍歷所有客戶(hù)端 發(fā)給每一個(gè)人
        #     for k,soc in clients.items():
        #         # if soc != c:
        #             send_msg(soc,data["msg"])

while True:
    '''以下代碼用于檢測(cè)是否有客戶(hù)端連接'''
    c,add = s.accept()
    print("%s" % add[0],"連接到服務(wù)器!")
    clients[add[0]] = c  # 把ip地址作為key，conn作為value存入clients字典中 ,clients = {'192.188.3.4':conn鏈接}
    if debug:
        print('clients=%s'%clients)

    '''以下代碼用于監(jiān)聽(tīng)已經(jīng)建立連接的客戶(hù)端發(fā)來(lái)的消息'''
    pool.submit(task,add,c)

import socket,json,struct
from threading import Thread

c = socket.socket()
c.connect(("127.0.0.1",8848))
print("連接服務(wù)器成功!")
def recver():
    while True:
        lens_bytes = c.recv(8)
        lens = struct.unpack("q", lens_bytes)[0]
        recv_msg = c.recv(lens).decode("utf-8")
        print(recv_msg)

# 開(kāi)啟線(xiàn)程處理接收數(shù)據(jù)
Thread(target=recver).start()


while True:
    msg = input(">>>>:").strip() # 阻塞
    if "@" in msg:
        info = {"msg":msg.split("@")[0],"to_addr":msg.split("@")[1]}
    else:
        info = {"msg": msg}

    data = json.dumps(info).encode("utf-8")
    c.send(struct.pack("q",len(data)))
    c.send(data)