utilizzando bytearray con socket.recv_into

Question

Sto facendo un po 'di socket IO e usando un oggetto bytearray come buffer. Vorrei ricevere dati con un offset in questo buffer usando csock.recv_into come mostrato di seguito per evitare di creare oggetti stringa intermediari. Sfortunatamente, sembra che non sia possibile utilizzare bytearrays in questo modo, e il codice sottostante non funziona.

buf = bytearray(b" " * toread) 
read = 0 
while(toread): 
    nbytes = csock.recv_into(buf[read:],toread) 
    toread -= nbytes 
    read += nbytes 

Così, invece io sto usando il codice qui sotto, che fa usare una stringa temporanea (e lavora) ...

buf = bytearray(b" " * toread) 
read = 0 
while(toread): 
    tmp = csock.recv(toread) 
    nbytes = len(tmp) 
    buf[read:] = tmp 
    toread -= nbytes 
    read += nbytes 

c'è un più elegante modo per farlo che doesn' t richiedono di copiare stringhe intermedie in giro?

Answer 1

Utilizzare un memoryview per avvolgere la bytearray:

buf = bytearray(toread) 
view = memoryview(buf) 
while toread: 
    nbytes = sock.recv_into(view, toread) 
    view = view[nbytes:] # slicing views is cheap 
    toread -= nbytes