Проблема с сохранением данных с нескольких Клиентов в один файл. TCP Python - PullRequest
0 голосов
/ 20 апреля 2020

Моя главная цель - получить данные от нескольких клиентов в один текстовый файл на моем сервере. Все работает отлично, клиенты подключаются к серверу, сервер распознает, что они являются разными клиентами, более того, они отправляют данные на сервер (и это сохраняет в файл data_r.txt), однако они перезаписывают файл. Поэтому я получаю данные только от последнего клиента, который отправил данные. Имя файла должно быть одинаковым для всех клиентов. Понятия не имею, что делать и как это исправить.

Сервер TCP

import socket
from threading import Thread
from socketserver import ThreadingMixIn

from threading import Thread
from socketserver import ThreadingMixIn

HOST = '192.168.0.108'
TCP_PORT = 60001
BUFFER_SIZE = 1024
class ClientThread(Thread):

   def __init__(self, ip, port, sock):
       Thread.__init__(self)
       self.ip = ip
       self.port = port
       self.sock = sock

   def run(self):
       filename = 'data_r.txt'
       f = open(filename, 'wb')
       while True:

          data = self.sock.recv(1024)
          if not data:
             f.close()
             #self.sock.close()
             break
          f.write(data)
    self.sock.sendall("File received")
    self.sock.close()


tcpsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
print('Creating socket...')
tcpsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
print('Socket created successfully')
print('Binding socket to the port...')
tcpsock.bind((HOST, TCP_PORT))
print(f'Socket binded successfully to port {TCP_PORT} ')
print('Waiting for clients to join....')

threads = []

while True:
   tcpsock.listen(5)
   (conn, (ip, port)) = tcpsock.accept()
   #conn.settimeout(7)
   newthread = ClientThread(ip, port, conn)
   print(newthread)
   newthread.start()
   threads.append(newthread)

for t in threads:
   t.join()

Клиент TCP

import socket

# TCP_IP = 'localhost'
HOST = '192.168.0.108'
TCP_PORT = 60001

BUFFER_SIZE = 1024

socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
print('Connecting to socket')
socket.connect((HOST, TCP_PORT))
print(f'Connected to host: {HOST}   on   port: {TCP_PORT}')

print('')
filename='data_s.txt'
f = open(filename, 'rb')

while True:
   chunk = f.read(BUFFER_SIZE)
   if not chunk:
      print ("File transfer completed")
      f.close()
      break
   socket.send(chunk)

c = socket.recv(BUFFER_SIZE)
print (c)
socket.close()
print('connection closed')

1 Ответ

0 голосов
/ 20 апреля 2020

Первое, что нужно исправить, это использовать open(filename, 'ab') вместо 'wb', потому что вы хотите добавлять к файлу, а не обрезать его.

Во-вторых, у вас есть условие гонки и, как комментатор сказал, что вы должны использовать что-то вроде блокировки для синхронизации доступа. В противном случае данные будут потеряны рано или поздно.

...