micromegas 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

micromegas/__init__.py

@@ -6,5 +6,4 @@ from . import flightsql
 
 def connect():
     "connect to the analytics service using default values"
-    channel_cred = grpc.local_channel_credentials()
-    return flightsql.client.FlightSQLClient("localhost:50051", channel_cred)
+    return flightsql.client.FlightSQLClient("grpc://localhost:50051")

micromegas/flightsql/__init__.py

@@ -1,10 +1,4 @@
 # protobufs
-from . import Flight_pb2
-from . import Flight_pb2_grpc
 from . import FlightSql_pb2
-from . import FlightSql_pb2_grpc
-# arrow deserialization
-from . import arrow_flatbuffers
-from . import arrow_ipc_reader
 # client API
 from . import client

micromegas/flightsql/client.py

@@ -1,227 +1,116 @@
+import certifi
 import pyarrow
-import grpc
+from pyarrow import flight
+from typing import Any
+import sys
 from google.protobuf import any_pb2
-from . import Flight_pb2_grpc
-from . import FlightSql_pb2_grpc
 from . import FlightSql_pb2
-from . import Flight_pb2
-from . import arrow_flatbuffers
-from . import arrow_ipc_reader
 from . import time
 
 
-def fb_time_unit_to_string(fb_time_unit):
-    time_unit_enum = arrow_flatbuffers.TimeUnit
-    if fb_time_unit == time_unit_enum.SECOND:
-        return "s"
-    if fb_time_unit == time_unit_enum.MILLISECOND:
-        return "ms"
-    if fb_time_unit == time_unit_enum.MICROSECOND:
-        return "us"
-    if fb_time_unit == time_unit_enum.NANOSECOND:
-        return "ns"
-    raise RuntimeError("unsupported time unit {}".format(fb_time_unit))
-
-
-def fb_field_type_to_arrow(fb_field):
-    fb_type = fb_field.TypeType()
-    fb_type_enum = arrow_flatbuffers.Type
-    assert fb_type != fb_type_enum.NONE
-    if fb_type == fb_type_enum.Null:
-        return pyarrow.null()
-    elif fb_type == fb_type_enum.Int:
-        type_int = arrow_flatbuffers.Int()
-        field_type_table = fb_field.Type()
-        type_int.Init(field_type_table.Bytes, field_type_table.Pos)
-        if type_int.IsSigned():
-            if type_int.BitWidth() == 8:
-                return pyarrow.int8()
-            elif type_int.BitWidth() == 16:
-                return pyarrow.int16()
-            elif type_int.BitWidth() == 32:
-                return pyarrow.int32()
-            elif type_int.BitWidth() == 64:
-                return pyarrow.int64()
-            else:
-                raise RuntimeError(
-                    "unsupported int size {}".format(type_int.BitWidth())
-                )
-        else:
-            if type_int.BitWidth() == 8:
-                return pyarrow.uint8()
-            elif type_int.BitWidth() == 16:
-                return pyarrow.uint16()
-            elif type_int.BitWidth() == 32:
-                return pyarrow.uint32()
-            elif type_int.BitWidth() == 64:
-                return pyarrow.uint64()
-            else:
-                raise RuntimeError(
-                    "unsupported uint size {}".format(type_int.BitWidth())
-                )
-    elif fb_type == fb_type_enum.FloatingPoint:
-        return pyarrow.float64()
-    elif fb_type == fb_type_enum.Binary:
-        return pyarrow.binary()
-    elif fb_type == fb_type_enum.Utf8:
-        return pyarrow.utf8()
-    elif fb_type == fb_type_enum.Bool:
-        return pyarrow.bool()
-    elif fb_type == fb_type_enum.Timestamp:
-        ts_type = arrow_flatbuffers.Timestamp()
-        field_type_table = fb_field.Type()
-        ts_type.Init(field_type_table.Bytes, field_type_table.Pos)
-        return pyarrow.timestamp(
-            fb_time_unit_to_string(ts_type.Unit()), ts_type.Timezone()
-        )
-    elif fb_type == fb_type_enum.List:
-        assert 1 == fb_field.ChildrenLength()
-        child_field = fb_field_to_arrow(fb_field.Children(0))
-        return pyarrow.list_(child_field)
-    elif fb_type == fb_type_enum.Struct_:
-        struct_fields = []
-        for child_index in range(fb_field.ChildrenLength()):
-            child = fb_field_to_arrow(fb_field.Children(child_index))
-            struct_fields.append(child)
-        return pyarrow.struct(struct_fields)
-    raise RuntimeError("unknown flatbuffer type {}".format(fb_type))
-
-
-def fb_field_to_arrow(fb_field):
-    arrow_type = fb_field_type_to_arrow(fb_field)
-    return pyarrow.field(fb_field.Name(), arrow_type)
-
-
-def make_query_flight_descriptor(sql):
-    command_query = FlightSql_pb2.CommandStatementQuery(query=sql)
-    any_cmd = any_pb2.Any()
-    any_cmd.Pack(command_query)
-    desc = Flight_pb2.FlightDescriptor()
-    desc.type = Flight_pb2.FlightDescriptor.DescriptorType.CMD
-    desc.cmd = any_cmd.SerializeToString()
-    return desc
-
-
-def read_schema_from_flight_data(flight_data):
-    msg = arrow_flatbuffers.Message.GetRootAs(flight_data.data_header, 0)
-    assert msg.Version() == arrow_flatbuffers.MetadataVersion.V5
-    header = msg.Header()
-    assert msg.HeaderType() == arrow_flatbuffers.MessageHeader.Schema
-
-    schema = arrow_flatbuffers.Schema()
-    schema.Init(header.Bytes, header.Pos)
-    nb_fields = schema.FieldsLength()
-    arrow_fields = []
-    for x in range(nb_fields):
-        field = schema.Fields(x)
-        arrow_f = fb_field_to_arrow(field)
-        arrow_fields.append(arrow_f)
-    arrow_schema = pyarrow.schema(arrow_fields)
-    return arrow_schema
-
-
-def read_record_batch_from_flight_data(arrow_schema, flight_data):
-    msg = arrow_flatbuffers.Message.GetRootAs(flight_data.data_header, 0)
-    assert msg.HeaderType() == arrow_flatbuffers.MessageHeader.RecordBatch
-    header = msg.Header()
-    fb_record_batch = arrow_flatbuffers.RecordBatch()
-    fb_record_batch.Init(header.Bytes, header.Pos)
-    nodes = []
-    for node_index in range(fb_record_batch.NodesLength()):
-        node = fb_record_batch.Nodes(node_index)
-        nodes.append(node)
-
-    buffers = []
-    for buffer_index in range(fb_record_batch.BuffersLength()):
-        buffer = fb_record_batch.Buffers(buffer_index)
-        buffers.append(buffer)
+class MicromegasMiddleware(flight.ClientMiddleware):
+    def __init__(self, headers):
+        self.headers = headers
 
-    body = pyarrow.py_buffer(flight_data.data_body)
-    arrow_buffers = []
-    for b in buffers:
-        s = body.slice(b.Offset(), b.Length())
-        arrow_buffers.append(s)
-    rb = arrow_ipc_reader.read_record_batch(arrow_schema, nodes, arrow_buffers)
-    return rb
+    def call_completed(self, exception):
+        if exception is not None:
+            print(exception, file=sys.stderr)
 
+    def received_headers(self, headers):
+        pass
 
-def channel_creds_from_token(token):
-    call_credentials = grpc.access_token_call_credentials(token)
-    channel_cred = grpc.composite_channel_credentials(
-        grpc.ssl_channel_credentials(), call_credentials
-    )
-    return channel_cred
+    def sending_headers(self):
+        return self.headers
 
 
-class FlightSQLAuthMetadataPlugin(grpc.AuthMetadataPlugin):
+class MicromegasMiddlewareFactory(flight.ClientMiddlewareFactory):
     def __init__(self, headers):
-        # we transform the keys into lowercase to avoid illegal grpc metadata (like 'Authorization', for example)
-        self.__headers = [(k.lower(), v) for (k, v) in headers.items()]
-
-    def __call__(self, context, callback):
-        callback(self.__headers, None)
-
+        self.headers = headers
+
+    def start_call(self, info):
+        return MicromegasMiddleware(self.headers)
+
+def make_call_headers( begin, end ):
+    call_headers = []
+    if begin is not None:
+        call_headers.append(
+            (
+                "query_range_begin".encode("utf8"),
+                time.format_datetime(begin).encode("utf8"),
+            )
+        )
+    if end is not None:
+        call_headers.append(
+            (
+                "query_range_end".encode("utf8"),
+                time.format_datetime(end).encode("utf8"),
+            )
+        )
+    return call_headers
 
-def channel_creds_from_headers(headers):
-    auth_plugin = FlightSQLAuthMetadataPlugin(headers)
-    call_credentials = grpc.metadata_call_credentials(auth_plugin)
-    channel_cred = grpc.composite_channel_credentials(
-        grpc.ssl_channel_credentials(), call_credentials
+def make_query_ticket(sql):
+    ticket_statement_query = FlightSql_pb2.TicketStatementQuery(
+        statement_handle=sql.encode("utf8")
     )
-    return channel_cred
-
+    any = any_pb2.Any()
+    any.Pack(ticket_statement_query)
+    ticket = flight.Ticket(any.SerializeToString())
+    return ticket
+
+def make_arrow_flight_descriptor(command: Any) -> flight.FlightDescriptor:
+    any = any_pb2.Any()
+    any.Pack(command)
+    return flight.FlightDescriptor.for_command(any.SerializeToString())
+
+def make_ingest_flight_desc(table_name):
+    ingest_statement = FlightSql_pb2.CommandStatementIngest(table=table_name, temporary=False)
+    desc = make_arrow_flight_descriptor(ingest_statement)
+    return desc
 
 class FlightSQLClient:
-    def __init__(self, host_port, channel_creds):
-        self.__host_port = host_port
-        self.__channel_creds = channel_creds
-
-    def make_channel(self):
-        if self.__channel_creds is None:
-            return grpc.insecure_channel(self.__host_port)
-        else:
-            return grpc.secure_channel(self.__host_port, self.__channel_creds)
+    def __init__(self, uri, headers=None):
+        fh = open(certifi.where(), "r")
+        cert = fh.read()
+        fh.close()
+        factory = MicromegasMiddlewareFactory(headers)
+        self.__flight_client = flight.connect(
+            location=uri, tls_root_certs=cert, middleware=[factory]
+        )
 
     def query(self, sql, begin=None, end=None):
-        metadata = []
-        if begin is not None:
-            metadata.append(("query_range_begin", time.format_datetime(begin)))
-        if end is not None:
-            metadata.append(("query_range_end", time.format_datetime(end)))
-
-        channel = self.make_channel()
-        stub = Flight_pb2_grpc.FlightServiceStub(channel)
-        desc = make_query_flight_descriptor(sql)
-        info = stub.GetFlightInfo(desc)
-        grpc_rdv = stub.DoGet(info.endpoint[0].ticket, metadata=metadata)
-        flight_data_list = list(grpc_rdv)
-        if len(flight_data_list) < 1:
-            raise RuntimeError("too few flightdata messages {}", len(flight_data_list))
-        schema_message = flight_data_list[0]
-        data_messages = flight_data_list[1:]
-        schema = read_schema_from_flight_data(schema_message)
+        call_headers = make_call_headers(begin, end)
+        options = flight.FlightCallOptions(headers=call_headers)
+        ticket = make_query_ticket(sql)
+        reader = self.__flight_client.do_get(ticket, options=options)
         record_batches = []
-        for msg in data_messages:
-            record_batches.append(read_record_batch_from_flight_data(schema, msg))
-        table = pyarrow.Table.from_batches(record_batches, schema)
+        for chunk in reader:
+            record_batches.append(chunk.data)
+        table = pyarrow.Table.from_batches(record_batches, reader.schema)
         return table.to_pandas()
 
     def query_stream(self, sql, begin=None, end=None):
-        metadata = []
-        if begin is not None:
-            metadata.append(("query_range_begin", time.format_datetime(begin)))
-        if end is not None:
-            metadata.append(("query_range_end", time.format_datetime(end)))
-
-        channel = self.make_channel()
-        stub = Flight_pb2_grpc.FlightServiceStub(channel)
-        desc = make_query_flight_descriptor(sql)
-        info = stub.GetFlightInfo(desc)
-        grpc_rdv = stub.DoGet(info.endpoint[0].ticket, metadata=metadata)
-        schema_message = grpc_rdv.next()
-        schema = read_schema_from_flight_data(schema_message)
-        for msg in grpc_rdv:
-            yield read_record_batch_from_flight_data(schema, msg)
+        ticket = make_query_ticket(sql)
+        call_headers = make_call_headers(begin, end)
+        options = flight.FlightCallOptions(headers=call_headers)
+        reader = self.__flight_client.do_get(ticket, options=options)
+        record_batches = []
+        for chunk in reader:
+            yield chunk.data
+
+    def bulk_ingest(self, table_name, df):
+        desc = make_ingest_flight_desc(table_name)
+        table = pyarrow.Table.from_pandas(df)
+        writer, reader = self.__flight_client.do_put(desc, table.schema)
+        for rb in table.to_batches():
+            writer.write(rb)
+        writer.done_writing()
+        result = reader.read()
+        if result is not None:
+            update_result = FlightSql_pb2.DoPutUpdateResult()
+            update_result.ParseFromString(result.to_pybytes())
+            return update_result
+        else:
+            return None
 
     def retire_partitions(self, view_set_name, view_instance_id, begin, end):
         sql = """
@@ -291,7 +180,7 @@ class FlightSQLClient:
             WHERE stream_id='{stream_id}'
             LIMIT {limit};
             """.format(
-                limit=limit,stream_id=stream_id
+            limit=limit, stream_id=stream_id
         )
         return self.query(sql, begin, end)
 
@@ -301,7 +190,6 @@ class FlightSQLClient:
             FROM view_instance('thread_spans', '{stream_id}')
             LIMIT {limit};
             """.format(
-                limit=limit,stream_id=stream_id
+            limit=limit, stream_id=stream_id
         )
         return self.query(sql, begin, end)
-    

{micromegas-0.7.0.dist-info → micromegas-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: micromegas
-Version: 0.7.0
+Version: 0.9.0
 Summary: Python analytics client for https://github.com/madesroches/micromegas/
 Author: Marc-Antoine Desroches
 Author-email: madesroches@gmail.com
@@ -9,14 +9,14 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: cbor2 (>=5.6.3,<6.0.0)
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: certifi (>=2025.4.26,<2026.0.0)
 Requires-Dist: crc (>=7.0.0,<8.0.0)
-Requires-Dist: flatbuffers (>=24.12.23,<25.0.0)
 Requires-Dist: grpcio (>=1.69.0,<2.0.0)
+Requires-Dist: numpy (>=2.2.6,<3.0.0)
 Requires-Dist: pandas (>=2.2.2,<3.0.0)
 Requires-Dist: protobuf (>=5.29.1,<6.0.0)
 Requires-Dist: pyarrow (>=18.1.0,<19.0.0)
-Requires-Dist: requests (>=2.31.0,<3.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.5,<5.0.0)
 Description-Content-Type: text/markdown

{micromegas-0.7.0.dist-info → micromegas-0.9.0.dist-info}/RECORD

@@ -1,12 +1,7 @@
-micromegas/__init__.py,sha256=TYFODSzzeVqdR3E1T3kTy73sOAYiQ9uh4rk8yMJ0hyM,284
+micromegas/__init__.py,sha256=oh0BAfNUVpHtFgufYWxvPwRfnqBTXY_nTPrNW1cwn-s,225
 micromegas/flightsql/FlightSql_pb2.py,sha256=s3CVAQ4W77ALQ9QYFlPZnPqVsRHTD5Ma0fzabQv_8Mc,28207
-micromegas/flightsql/FlightSql_pb2_grpc.py,sha256=dCQT-tKxsz3dQgD4Cr0-8F4H0zCwNkXLHzlrbWfMiBA,890
-micromegas/flightsql/Flight_pb2.py,sha256=nPe1utv8n59IdNsiv5sJYY10-K04pDVOhKDgV5t-h1A,12513
-micromegas/flightsql/Flight_pb2_grpc.py,sha256=mj9Nlla3wpf5xFAGhMZpLDX8vLBClWfZ4w03AbBvrgs,23483
-micromegas/flightsql/__init__.py,sha256=SRq2X59uKG-iuwFnSyT7wVfiTMSSryAAEWnQVjGWOM8,249
-micromegas/flightsql/arrow_flatbuffers.py,sha256=egpmS59sNFwWmtG2wMeE92MMIyGyZbsm9o24mUnY1MQ,100142
-micromegas/flightsql/arrow_ipc_reader.py,sha256=3rxyEgqo5100e0TT9ZKZxNe7lX6Lk0mS6yRoiIJtH6Q,3163
-micromegas/flightsql/client.py,sha256=v4igU4nHZdaqBAw5qDtDNl6UBYiLvVnCJk1qDeO8LjE,11241
+micromegas/flightsql/__init__.py,sha256=VWuldsAd-N7tjSdaN_bSKefFJB26zrJzO4aKtO6y60g,74
+micromegas/flightsql/client.py,sha256=2-twcMglhxDjVN2dJVdpkMH8can1y3eOweLDcXUZXug,6425
 micromegas/flightsql/time.py,sha256=EH3SUEpFvY0lNMj9mOcvfUJuSgrQ3YX4aJnwteK2qhk,582
 micromegas/perfetto.py,sha256=mXtkQMZ5y9LhyanKWjf3BwehEW_vVR_MtTPrUgsehZ4,7449
 micromegas/thirdparty/perfetto/protos/perfetto/common/android_energy_consumer_descriptor_pb2.py,sha256=l8QNXqnB-mJIkuFr2s1YoLQXHm3G-ZcOGp_OW_hQ0TE,1887
@@ -215,6 +210,6 @@ micromegas/thirdparty/perfetto/protos/perfetto/trace/translation/translation_tab
 micromegas/thirdparty/perfetto/protos/perfetto/trace/trigger_pb2.py,sha256=We7Yi8o3cEcrSNxY1zLUUO6tEWnD36C2f3O_s8_qv0I,1435
 micromegas/thirdparty/perfetto/protos/perfetto/trace/ui_state_pb2.py,sha256=Af-SXwhroNhRXMrtw6e2eU1liCImMRxSdmkt_AuSHf8,1752
 micromegas/time.py,sha256=eD9fWF2UHxaf-92yd1X2SEgUcpKypqPsvjBosLdpnQA,1026
-micromegas-0.7.0.dist-info/METADATA,sha256=nnIGli-hVt5vkZjz7nSu1M7vATYEoDJAgirGmsCRlcQ,30640
-micromegas-0.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-micromegas-0.7.0.dist-info/RECORD,,
+micromegas-0.9.0.dist-info/METADATA,sha256=IQWE9hnWk6UZYUc4Qmx92-pH1RVmVnSCZbL1CxTZVZ4,30648
+micromegas-0.9.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+micromegas-0.9.0.dist-info/RECORD,,

{micromegas-0.7.0.dist-info → micromegas-0.9.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any

micromegas/flightsql/FlightSql_pb2_grpc.py

@@ -1,24 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-import warnings
-
-
-GRPC_GENERATED_VERSION = '1.68.1'
-GRPC_VERSION = grpc.__version__
-_version_not_supported = False
-
-try:
-    from grpc._utilities import first_version_is_lower
-    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
-except ImportError:
-    _version_not_supported = True
-
-if _version_not_supported:
-    raise RuntimeError(
-        f'The grpc package installed is at version {GRPC_VERSION},'
-        + f' but the generated code in FlightSql_pb2_grpc.py depends on'
-        + f' grpcio>={GRPC_GENERATED_VERSION}.'
-        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
-        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
-    )

micromegas/flightsql/Flight_pb2.py

@@ -1,116 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: Flight.proto
-# Protobuf Python Version: 5.29.1
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    5,
-    29,
-    1,
-    '',
-    'Flight.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x46light.proto\x12\x15\x61rrow.flight.protocol\x1a\x1fgoogle/protobuf/timestamp.proto\"=\n\x10HandshakeRequest\x12\x18\n\x10protocol_version\x18\x01 \x01(\x04\x12\x0f\n\x07payload\x18\x02 \x01(\x0c\">\n\x11HandshakeResponse\x12\x18\n\x10protocol_version\x18\x01 \x01(\x04\x12\x0f\n\x07payload\x18\x02 \x01(\x0c\"/\n\tBasicAuth\x12\x10\n\x08username\x18\x02 \x01(\t\x12\x10\n\x08password\x18\x03 \x01(\t\"\x07\n\x05\x45mpty\"/\n\nActionType\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\"\x1e\n\x08\x43riteria\x12\x12\n\nexpression\x18\x01 \x01(\x0c\"$\n\x06\x41\x63tion\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x0c\n\x04\x62ody\x18\x02 \x01(\x0c\"\x16\n\x06Result\x12\x0c\n\x04\x62ody\x18\x01 \x01(\x0c\"\x1e\n\x0cSchemaResult\x12\x0e\n\x06schema\x18\x01 \x01(\x0c\"\xa5\x01\n\x10\x46lightDescriptor\x12\x44\n\x04type\x18\x01 \x01(\x0e\x32\x36.arrow.flight.protocol.FlightDescriptor.DescriptorType\x12\x0b\n\x03\x63md\x18\x02 \x01(\x0c\x12\x0c\n\x04path\x18\x03 \x03(\t\"0\n\x0e\x44\x65scriptorType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04PATH\x10\x01\x12\x07\n\x03\x43MD\x10\x02\"\xec\x01\n\nFlightInfo\x12\x0e\n\x06schema\x18\x01 \x01(\x0c\x12\x42\n\x11\x66light_descriptor\x18\x02 \x01(\x0b\x32\'.arrow.flight.protocol.FlightDescriptor\x12\x37\n\x08\x65ndpoint\x18\x03 \x03(\x0b\x32%.arrow.flight.protocol.FlightEndpoint\x12\x15\n\rtotal_records\x18\x04 \x01(\x03\x12\x13\n\x0btotal_bytes\x18\x05 \x01(\x03\x12\x0f\n\x07ordered\x18\x06 \x01(\x08\x12\x14\n\x0c\x61pp_metadata\x18\x07 \x01(\x0c\"\xd8\x01\n\x08PollInfo\x12/\n\x04info\x18\x01 \x01(\x0b\x32!.arrow.flight.protocol.FlightInfo\x12\x42\n\x11\x66light_descriptor\x18\x02 \x01(\x0b\x32\'.arrow.flight.protocol.FlightDescriptor\x12\x15\n\x08progress\x18\x03 \x01(\x01H\x00\x88\x01\x01\x12\x33\n\x0f\x65xpiration_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x0b\n\t_progress\"J\n\x17\x43\x61ncelFlightInfoRequest\x12/\n\x04info\x18\x01 \x01(\x0b\x32!.arrow.flight.protocol.FlightInfo\"M\n\x16\x43\x61ncelFlightInfoResult\x12\x33\n\x06status\x18\x01 \x01(\x0e\x32#.arrow.flight.protocol.CancelStatus\"\x18\n\x06Ticket\x12\x0e\n\x06ticket\x18\x01 \x01(\x0c\"\x17\n\x08Location\x12\x0b\n\x03uri\x18\x01 \x01(\t\"\xbd\x01\n\x0e\x46lightEndpoint\x12-\n\x06ticket\x18\x01 \x01(\x0b\x32\x1d.arrow.flight.protocol.Ticket\x12\x31\n\x08location\x18\x02 \x03(\x0b\x32\x1f.arrow.flight.protocol.Location\x12\x33\n\x0f\x65xpiration_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0c\x61pp_metadata\x18\x04 \x01(\x0c\"U\n\x1aRenewFlightEndpointRequest\x12\x37\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32%.arrow.flight.protocol.FlightEndpoint\"\x8f\x01\n\nFlightData\x12\x42\n\x11\x66light_descriptor\x18\x01 \x01(\x0b\x32\'.arrow.flight.protocol.FlightDescriptor\x12\x13\n\x0b\x64\x61ta_header\x18\x02 \x01(\x0c\x12\x14\n\x0c\x61pp_metadata\x18\x03 \x01(\x0c\x12\x12\n\tdata_body\x18\xe8\x07 \x01(\x0c\"!\n\tPutResult\x12\x14\n\x0c\x61pp_metadata\x18\x01 \x01(\x0c\"\xfc\x01\n\x12SessionOptionValue\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x14\n\nbool_value\x18\x02 \x01(\x08H\x00\x12\x15\n\x0bint64_value\x18\x03 \x01(\x10H\x00\x12\x16\n\x0c\x64ouble_value\x18\x04 \x01(\x01H\x00\x12V\n\x11string_list_value\x18\x05 \x01(\x0b\x32\x39.arrow.flight.protocol.SessionOptionValue.StringListValueH\x00\x1a!\n\x0fStringListValue\x12\x0e\n\x06values\x18\x01 \x03(\tB\x0e\n\x0coption_value\"\xda\x01\n\x18SetSessionOptionsRequest\x12\\\n\x0fsession_options\x18\x01 \x03(\x0b\x32\x43.arrow.flight.protocol.SetSessionOptionsRequest.SessionOptionsEntry\x1a`\n\x13SessionOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).arrow.flight.protocol.SessionOptionValue:\x02\x38\x01\"\xec\x02\n\x17SetSessionOptionsResult\x12J\n\x06\x65rrors\x18\x01 \x03(\x0b\x32:.arrow.flight.protocol.SetSessionOptionsResult.ErrorsEntry\x1aQ\n\x05\x45rror\x12H\n\x05value\x18\x01 \x01(\x0e\x32\x39.arrow.flight.protocol.SetSessionOptionsResult.ErrorValue\x1a\x63\n\x0b\x45rrorsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x43\n\x05value\x18\x02 \x01(\x0b\x32\x34.arrow.flight.protocol.SetSessionOptionsResult.Error:\x02\x38\x01\"M\n\nErrorValue\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x10\n\x0cINVALID_NAME\x10\x01\x12\x11\n\rINVALID_VALUE\x10\x02\x12\t\n\x05\x45RROR\x10\x03\"\x1a\n\x18GetSessionOptionsRequest\"\xd8\x01\n\x17GetSessionOptionsResult\x12[\n\x0fsession_options\x18\x01 \x03(\x0b\x32\x42.arrow.flight.protocol.GetSessionOptionsResult.SessionOptionsEntry\x1a`\n\x13SessionOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).arrow.flight.protocol.SessionOptionValue:\x02\x38\x01\"\x15\n\x13\x43loseSessionRequest\"\x9d\x01\n\x12\x43loseSessionResult\x12@\n\x06status\x18\x01 \x01(\x0e\x32\x30.arrow.flight.protocol.CloseSessionResult.Status\"E\n\x06Status\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\n\n\x06\x43LOSED\x10\x01\x12\x0b\n\x07\x43LOSING\x10\x02\x12\x11\n\rNOT_CLOSEABLE\x10\x03*\x8b\x01\n\x0c\x43\x61ncelStatus\x12\x1d\n\x19\x43\x41NCEL_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x43\x41NCEL_STATUS_CANCELLED\x10\x01\x12\x1c\n\x18\x43\x41NCEL_STATUS_CANCELLING\x10\x02\x12!\n\x1d\x43\x41NCEL_STATUS_NOT_CANCELLABLE\x10\x03\x32\x85\x07\n\rFlightService\x12\x64\n\tHandshake\x12\'.arrow.flight.protocol.HandshakeRequest\x1a(.arrow.flight.protocol.HandshakeResponse\"\x00(\x01\x30\x01\x12U\n\x0bListFlights\x12\x1f.arrow.flight.protocol.Criteria\x1a!.arrow.flight.protocol.FlightInfo\"\x00\x30\x01\x12]\n\rGetFlightInfo\x12\'.arrow.flight.protocol.FlightDescriptor\x1a!.arrow.flight.protocol.FlightInfo\"\x00\x12\\\n\x0ePollFlightInfo\x12\'.arrow.flight.protocol.FlightDescriptor\x1a\x1f.arrow.flight.protocol.PollInfo\"\x00\x12[\n\tGetSchema\x12\'.arrow.flight.protocol.FlightDescriptor\x1a#.arrow.flight.protocol.SchemaResult\"\x00\x12M\n\x05\x44oGet\x12\x1d.arrow.flight.protocol.Ticket\x1a!.arrow.flight.protocol.FlightData\"\x00\x30\x01\x12R\n\x05\x44oPut\x12!.arrow.flight.protocol.FlightData\x1a .arrow.flight.protocol.PutResult\"\x00(\x01\x30\x01\x12X\n\nDoExchange\x12!.arrow.flight.protocol.FlightData\x1a!.arrow.flight.protocol.FlightData\"\x00(\x01\x30\x01\x12L\n\x08\x44oAction\x12\x1d.arrow.flight.protocol.Action\x1a\x1d.arrow.flight.protocol.Result\"\x00\x30\x01\x12R\n\x0bListActions\x12\x1c.arrow.flight.protocol.Empty\x1a!.arrow.flight.protocol.ActionType\"\x00\x30\x01\x42q\n\x1corg.apache.arrow.flight.implZ2github.com/apache/arrow-go/arrow/flight/gen/flight\xaa\x02\x1c\x41pache.Arrow.Flight.Protocolb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'Flight_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  _globals['DESCRIPTOR']._loaded_options = None
-  _globals['DESCRIPTOR']._serialized_options = b'\n\034org.apache.arrow.flight.implZ2github.com/apache/arrow-go/arrow/flight/gen/flight\252\002\034Apache.Arrow.Flight.Protocol'
-  _globals['_SETSESSIONOPTIONSREQUEST_SESSIONOPTIONSENTRY']._loaded_options = None
-  _globals['_SETSESSIONOPTIONSREQUEST_SESSIONOPTIONSENTRY']._serialized_options = b'8\001'
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORSENTRY']._loaded_options = None
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORSENTRY']._serialized_options = b'8\001'
-  _globals['_GETSESSIONOPTIONSRESULT_SESSIONOPTIONSENTRY']._loaded_options = None
-  _globals['_GETSESSIONOPTIONSRESULT_SESSIONOPTIONSENTRY']._serialized_options = b'8\001'
-  _globals['_CANCELSTATUS']._serialized_start=2998
-  _globals['_CANCELSTATUS']._serialized_end=3137
-  _globals['_HANDSHAKEREQUEST']._serialized_start=72
-  _globals['_HANDSHAKEREQUEST']._serialized_end=133
-  _globals['_HANDSHAKERESPONSE']._serialized_start=135
-  _globals['_HANDSHAKERESPONSE']._serialized_end=197
-  _globals['_BASICAUTH']._serialized_start=199
-  _globals['_BASICAUTH']._serialized_end=246
-  _globals['_EMPTY']._serialized_start=248
-  _globals['_EMPTY']._serialized_end=255
-  _globals['_ACTIONTYPE']._serialized_start=257
-  _globals['_ACTIONTYPE']._serialized_end=304
-  _globals['_CRITERIA']._serialized_start=306
-  _globals['_CRITERIA']._serialized_end=336
-  _globals['_ACTION']._serialized_start=338
-  _globals['_ACTION']._serialized_end=374
-  _globals['_RESULT']._serialized_start=376
-  _globals['_RESULT']._serialized_end=398
-  _globals['_SCHEMARESULT']._serialized_start=400
-  _globals['_SCHEMARESULT']._serialized_end=430
-  _globals['_FLIGHTDESCRIPTOR']._serialized_start=433
-  _globals['_FLIGHTDESCRIPTOR']._serialized_end=598
-  _globals['_FLIGHTDESCRIPTOR_DESCRIPTORTYPE']._serialized_start=550
-  _globals['_FLIGHTDESCRIPTOR_DESCRIPTORTYPE']._serialized_end=598
-  _globals['_FLIGHTINFO']._serialized_start=601
-  _globals['_FLIGHTINFO']._serialized_end=837
-  _globals['_POLLINFO']._serialized_start=840
-  _globals['_POLLINFO']._serialized_end=1056
-  _globals['_CANCELFLIGHTINFOREQUEST']._serialized_start=1058
-  _globals['_CANCELFLIGHTINFOREQUEST']._serialized_end=1132
-  _globals['_CANCELFLIGHTINFORESULT']._serialized_start=1134
-  _globals['_CANCELFLIGHTINFORESULT']._serialized_end=1211
-  _globals['_TICKET']._serialized_start=1213
-  _globals['_TICKET']._serialized_end=1237
-  _globals['_LOCATION']._serialized_start=1239
-  _globals['_LOCATION']._serialized_end=1262
-  _globals['_FLIGHTENDPOINT']._serialized_start=1265
-  _globals['_FLIGHTENDPOINT']._serialized_end=1454
-  _globals['_RENEWFLIGHTENDPOINTREQUEST']._serialized_start=1456
-  _globals['_RENEWFLIGHTENDPOINTREQUEST']._serialized_end=1541
-  _globals['_FLIGHTDATA']._serialized_start=1544
-  _globals['_FLIGHTDATA']._serialized_end=1687
-  _globals['_PUTRESULT']._serialized_start=1689
-  _globals['_PUTRESULT']._serialized_end=1722
-  _globals['_SESSIONOPTIONVALUE']._serialized_start=1725
-  _globals['_SESSIONOPTIONVALUE']._serialized_end=1977
-  _globals['_SESSIONOPTIONVALUE_STRINGLISTVALUE']._serialized_start=1928
-  _globals['_SESSIONOPTIONVALUE_STRINGLISTVALUE']._serialized_end=1961
-  _globals['_SETSESSIONOPTIONSREQUEST']._serialized_start=1980
-  _globals['_SETSESSIONOPTIONSREQUEST']._serialized_end=2198
-  _globals['_SETSESSIONOPTIONSREQUEST_SESSIONOPTIONSENTRY']._serialized_start=2102
-  _globals['_SETSESSIONOPTIONSREQUEST_SESSIONOPTIONSENTRY']._serialized_end=2198
-  _globals['_SETSESSIONOPTIONSRESULT']._serialized_start=2201
-  _globals['_SETSESSIONOPTIONSRESULT']._serialized_end=2565
-  _globals['_SETSESSIONOPTIONSRESULT_ERROR']._serialized_start=2304
-  _globals['_SETSESSIONOPTIONSRESULT_ERROR']._serialized_end=2385
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORSENTRY']._serialized_start=2387
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORSENTRY']._serialized_end=2486
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORVALUE']._serialized_start=2488
-  _globals['_SETSESSIONOPTIONSRESULT_ERRORVALUE']._serialized_end=2565
-  _globals['_GETSESSIONOPTIONSREQUEST']._serialized_start=2567
-  _globals['_GETSESSIONOPTIONSREQUEST']._serialized_end=2593
-  _globals['_GETSESSIONOPTIONSRESULT']._serialized_start=2596
-  _globals['_GETSESSIONOPTIONSRESULT']._serialized_end=2812
-  _globals['_GETSESSIONOPTIONSRESULT_SESSIONOPTIONSENTRY']._serialized_start=2102
-  _globals['_GETSESSIONOPTIONSRESULT_SESSIONOPTIONSENTRY']._serialized_end=2198
-  _globals['_CLOSESESSIONREQUEST']._serialized_start=2814
-  _globals['_CLOSESESSIONREQUEST']._serialized_end=2835
-  _globals['_CLOSESESSIONRESULT']._serialized_start=2838
-  _globals['_CLOSESESSIONRESULT']._serialized_end=2995
-  _globals['_CLOSESESSIONRESULT_STATUS']._serialized_start=2926
-  _globals['_CLOSESESSIONRESULT_STATUS']._serialized_end=2995
-  _globals['_FLIGHTSERVICE']._serialized_start=3140
-  _globals['_FLIGHTSERVICE']._serialized_end=4041
-# @@protoc_insertion_point(module_scope)