A Flight server that compares the given ticket to an expected value.
| 333 | |
| 334 | |
| 335 | class CheckTicketFlightServer(FlightServerBase): |
| 336 | """A Flight server that compares the given ticket to an expected value.""" |
| 337 | |
| 338 | def __init__(self, expected_ticket, location=None, **kwargs): |
| 339 | super().__init__(location, **kwargs) |
| 340 | self.expected_ticket = expected_ticket |
| 341 | |
| 342 | def do_get(self, context, ticket): |
| 343 | assert self.expected_ticket == ticket.ticket |
| 344 | data1 = [pa.array([-10, -5, 0, 5, 10], type=pa.int32())] |
| 345 | table = pa.Table.from_arrays(data1, names=['a']) |
| 346 | return flight.RecordBatchStream(table) |
| 347 | |
| 348 | def do_put(self, context, descriptor, reader): |
| 349 | self.last_message = reader.read_all() |
| 350 | |
| 351 | |
| 352 | class InvalidStreamFlightServer(FlightServerBase): |
no outgoing calls