[Python][FlightRPC] pyarrow client do_put close method after write_table did not throw flight error

[asfimport]

hi, when I implentend flight server to do_put data, and throw exceptions proactively, and client will get OSError when close writer, but I expect FlightError.

  

//代码占位符

def do_put(self, context, descriptor, reader, writer):
    table = reader.read_all()
    ....
    raise fl.FlightUnavailableError()
    ....

//代码占位符
df = ...
table = pa.Table.from_pandas(df)
client = fl.connect('...')
writer, _ = client.do_put(descriptor, table.schema, options=options)
writer.write_table(table)
write.close()

also, when I throw exception before read_all, the write_table throw OSError.

 

and i guess,  the FlightStreamWriter   should add write_table and close method, and check_flight_status  according to write_batch had checked. 

thx.

 

 

Reporter: lixiang li / @lilixiang
Assignee: lixiang li / @lilixiang

Original Issue Attachments:

• image-2021-08-06-17-19-27-413.png
• image-2021-08-06-17-21-12-181.png

PRs and other links:

• GitHub Pull Request #10893

_{Note: This issue was originally created as ARROW-13577. Please see the migration documentation for further details.}

[asfimport]

David Li / @lidavidm:
Issue resolved by pull request 10893
#10893