We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 30aee1b commit 559533aCopy full SHA for 559533a
1 file changed
src/altertable_flightsql/client.py
@@ -242,14 +242,18 @@ def execute(
242
243
# Execute via DoPut
244
writer, reader = self._client.do_put(descriptor, pa.schema([]))
245
- writer.close()
+ # 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()
249
250
# Read result from metadata
251
result = sql_pb2.DoPutUpdateResult()
252
metadata = reader.read()
253
if metadata:
- result.ParseFromString(metadata)
254
+ result.ParseFromString(bytes(metadata))
255
256
+ writer.close()
257
return result.record_count
258
259
def ingest(
0 commit comments