Skip to content

Commit 559533a

Browse files
committed
Fix execute result
1 parent 30aee1b commit 559533a

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

src/altertable_flightsql/client.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -242,14 +242,18 @@ def execute(
242242

243243
# Execute via DoPut
244244
writer, reader = self._client.do_put(descriptor, pa.schema([]))
245-
writer.close()
245+
# Signal end of upload while keeping the read side open to receive the
246+
# server's DoPutUpdateResult metadata. writer.close() would close both
247+
# sides prematurely, causing reader.read() to return None.
248+
writer.done_writing()
246249

247250
# Read result from metadata
248251
result = sql_pb2.DoPutUpdateResult()
249252
metadata = reader.read()
250253
if metadata:
251-
result.ParseFromString(metadata)
254+
result.ParseFromString(bytes(metadata))
252255

256+
writer.close()
253257
return result.record_count
254258

255259
def ingest(

0 commit comments

Comments
 (0)