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

• Type: Improvement
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Fix Version/s: 6.0.0
• Component/s: FlightRPC, Python
• Labels: pull-request-available
• External issue URL: https://github.com/apache/arrow/issues/29224

Language

C++ Python

• Assignee: lixiang li
• Reporter: lixiang li
• Created: 06/Aug/21 09:29
• Updated: 11/Jan/23 08:34
• Resolved: 11/Aug/21 12:39

Description

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.

 

 

Issue Links

• links to: GitHub Pull Request #10893