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)