def do_get(self, context, ticket):
dataset_name = ticket.ticket.decode('utf-8')
print("Ticket:", dataset_name)
if dataset_name not in self.datasets:
raise fl.FlightUnavailableError
dataset = self.datasets[dataset_name]
return fl.RecordBatchStream(dataset)
def get_data(self, _, request) -> Any:
"""Return time series data as Arrow data."""
selector = SeriesSelector(request["selector"]["source"],
request["selector"]["name"])
start_date = parse_date(request["start_date"])
end_date = parse_date(request["end_date"])
data = self.__source.get_data(selector, start_date, end_date)
return fl.RecordBatchStream(data)
def do_get(self, context, ticket):
"""Fetch a dataset
Parameters
----------
context: unknown
ticket: unknown
"""
table = self.tables[ticket.ticket]
return fl.RecordBatchStream(table)
def do_get(self, context, ticket):
assert self.expected_ticket == ticket.ticket
data1 = [pa.array([-10, -5, 0, 5, 10], type=pa.int32())]
table = pa.Table.from_arrays(data1, names=['a'])
return flight.RecordBatchStream(table)
def do_get(self, context, ticket):
return flight.RecordBatchStream(self.last_message)
def do_get(self, context, ticket):
# Return a fresh table, so that Flight is the only one keeping a
# reference.
table = self.table_factories[ticket.ticket]()
return flight.RecordBatchStream(table)
def do_get(self, context, ticket):
table = self.tables[ticket.ticket]
return fl.RecordBatchStream(table)
def do_get(self, context, ticket):
table = self._cache[ticket.ticket]
return flight.RecordBatchStream(table)
