chalkpy 2.90.1__py3-none-any.whl → 2.95.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- chalk/__init__.py +2 -1
- chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
- chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
- chalk/_gen/chalk/artifacts/v1/chart_pb2.py +16 -16
- chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +4 -0
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
- chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
- chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
- chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
- chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
- chalk/_gen/chalk/dataframe/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
- chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
- chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
- chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
- chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
- chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
- chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
- chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
- chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
- chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
- chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
- chalk/_gen/chalk/server/v1/builder_pb2.py +358 -288
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +360 -10
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +225 -0
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +60 -0
- chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
- chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2.py +141 -119
- chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +106 -4
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +52 -38
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +62 -1
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
- chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
- chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/deployment_pb2.py +6 -6
- chalk/_gen/chalk/server/v1/deployment_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/environment_pb2.py +14 -12
- chalk/_gen/chalk/server/v1/environment_pb2.pyi +19 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2.py +4 -2
- chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
- chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/server/v1/graph_pb2.py +38 -26
- chalk/_gen/chalk/server/v1/graph_pb2.pyi +58 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +47 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +18 -0
- chalk/_gen/chalk/server/v1/incident_pb2.py +23 -21
- chalk/_gen/chalk/server/v1/incident_pb2.pyi +15 -1
- chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
- chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
- chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
- chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
- chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
- chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
- chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
- chalk/_gen/chalk/server/v1/queries_pb2.pyi +32 -2
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -12
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +16 -3
- chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
- chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2.py +15 -3
- chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +22 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
- chalk/_gen/chalk/server/v1/team_pb2.py +154 -141
- chalk/_gen/chalk/server/v1/team_pb2.pyi +30 -2
- chalk/_gen/chalk/server/v1/team_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
- chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
- chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
- chalk/_gen/chalk/server/v1/trace_pb2.pyi +20 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +16 -10
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +52 -1
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
- chalk/_lsp/error_builder.py +11 -0
- chalk/_version.py +1 -1
- chalk/client/client.py +128 -43
- chalk/client/client_async.py +149 -0
- chalk/client/client_async_impl.py +22 -0
- chalk/client/client_grpc.py +539 -104
- chalk/client/client_impl.py +449 -122
- chalk/client/dataset.py +7 -1
- chalk/client/models.py +98 -0
- chalk/client/serialization/model_serialization.py +92 -9
- chalk/df/LazyFramePlaceholder.py +1154 -0
- chalk/features/_class_property.py +7 -0
- chalk/features/_embedding/embedding.py +1 -0
- chalk/features/_encoding/converter.py +83 -2
- chalk/features/feature_field.py +40 -30
- chalk/features/feature_set_decorator.py +1 -0
- chalk/features/feature_wrapper.py +42 -3
- chalk/features/hooks.py +81 -10
- chalk/features/inference.py +33 -31
- chalk/features/resolver.py +224 -24
- chalk/functions/__init__.py +65 -3
- chalk/gitignore/gitignore_parser.py +5 -1
- chalk/importer.py +142 -68
- chalk/ml/__init__.py +2 -0
- chalk/ml/model_hooks.py +194 -26
- chalk/ml/model_reference.py +56 -8
- chalk/ml/model_version.py +24 -15
- chalk/ml/utils.py +20 -17
- chalk/operators/_utils.py +10 -3
- chalk/parsed/_proto/export.py +22 -0
- chalk/parsed/duplicate_input_gql.py +3 -0
- chalk/parsed/json_conversions.py +20 -14
- chalk/parsed/to_proto.py +16 -4
- chalk/parsed/user_types_to_json.py +31 -10
- chalk/parsed/validation_from_registries.py +182 -0
- chalk/queries/named_query.py +16 -6
- chalk/queries/scheduled_query.py +9 -1
- chalk/serialization/parsed_annotation.py +24 -11
- chalk/sql/__init__.py +18 -0
- chalk/sql/_internal/integrations/databricks.py +55 -17
- chalk/sql/_internal/integrations/mssql.py +127 -62
- chalk/sql/_internal/integrations/redshift.py +4 -0
- chalk/sql/_internal/sql_file_resolver.py +53 -9
- chalk/sql/_internal/sql_source.py +35 -2
- chalk/streams/_kafka_source.py +5 -1
- chalk/streams/_windows.py +15 -2
- chalk/utils/_otel_version.py +13 -0
- chalk/utils/async_helpers.py +2 -2
- chalk/utils/missing_dependency.py +5 -4
- chalk/utils/tracing.py +185 -95
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/METADATA +4 -6
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/RECORD +202 -146
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
|
@@ -18,14 +18,23 @@ class PubSubTopic(_message.Message):
|
|
|
18
18
|
topic_id: str
|
|
19
19
|
def __init__(self, project_id: _Optional[str] = ..., topic_id: _Optional[str] = ...) -> None: ...
|
|
20
20
|
|
|
21
|
+
class InMemoryTopic(_message.Message):
|
|
22
|
+
__slots__ = ("name",)
|
|
23
|
+
NAME_FIELD_NUMBER: _ClassVar[int]
|
|
24
|
+
name: str
|
|
25
|
+
def __init__(self, name: _Optional[str] = ...) -> None: ...
|
|
26
|
+
|
|
21
27
|
class Topic(_message.Message):
|
|
22
|
-
__slots__ = ("sqs_topic", "pubsub_topic")
|
|
28
|
+
__slots__ = ("sqs_topic", "pubsub_topic", "in_memory_topic")
|
|
23
29
|
SQS_TOPIC_FIELD_NUMBER: _ClassVar[int]
|
|
24
30
|
PUBSUB_TOPIC_FIELD_NUMBER: _ClassVar[int]
|
|
31
|
+
IN_MEMORY_TOPIC_FIELD_NUMBER: _ClassVar[int]
|
|
25
32
|
sqs_topic: SQSTopic
|
|
26
33
|
pubsub_topic: PubSubTopic
|
|
34
|
+
in_memory_topic: InMemoryTopic
|
|
27
35
|
def __init__(
|
|
28
36
|
self,
|
|
29
37
|
sqs_topic: _Optional[_Union[SQSTopic, _Mapping]] = ...,
|
|
30
38
|
pubsub_topic: _Optional[_Union[PubSubTopic, _Mapping]] = ...,
|
|
39
|
+
in_memory_topic: _Optional[_Union[InMemoryTopic, _Mapping]] = ...,
|
|
31
40
|
) -> None: ...
|
|
@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
|
|
|
19
19
|
|
|
20
20
|
|
|
21
21
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
22
|
-
b'\n\x1b\x63halk/server/v1/trace.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\
|
|
22
|
+
b'\n\x1b\x63halk/server/v1/trace.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xe1\x06\n\tChalkSpan\x12\x17\n\x07span_id\x18\x01 \x01(\tR\x06spanId\x12\x19\n\x08trace_id\x18\x02 \x01(\tR\x07traceId\x12)\n\x0eparent_span_id\x18\x03 \x01(\tH\x00R\x0cparentSpanId\x88\x01\x01\x12%\n\x0eoperation_name\x18\x04 \x01(\tR\roperationName\x12\x39\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12\x35\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndTime\x12\x35\n\x08\x64uration\x18\x07 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12\x38\n\x06status\x18\x08 \x01(\x0b\x32 .chalk.server.v1.ChalkSpanStatusR\x06status\x12J\n\nattributes\x18\t \x03(\x0b\x32*.chalk.server.v1.ChalkSpan.AttributesEntryR\nattributes\x12\x37\n\x06\x65vents\x18\n \x03(\x0b\x32\x1f.chalk.server.v1.ChalkSpanEventR\x06\x65vents\x12\x34\n\x05links\x18\x0b \x03(\x0b\x32\x1e.chalk.server.v1.ChalkSpanLinkR\x05links\x12\x32\n\x04kind\x18\x0c \x01(\x0e\x32\x1e.chalk.server.v1.ChalkSpanKindR\x04kind\x12\x63\n\x13resource_attributes\x18\r \x03(\x0b\x32\x32.chalk.server.v1.ChalkSpan.ResourceAttributesEntryR\x12resourceAttributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a\x45\n\x17ResourceAttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_parent_span_id"~\n\x0f\x43halkSpanStatus\x12\x34\n\x04\x63ode\x18\x01 \x01(\x0e\x32 .chalk.server.v1.ChalkStatusCodeR\x04\x63ode\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x00R\x0b\x64\x65scription\x88\x01\x01\x42\x0e\n\x0c_description"\xee\x01\n\x0e\x43halkSpanEvent\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x38\n\ttimestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\ttimestamp\x12O\n\nattributes\x18\x03 \x03(\x0b\x32/.chalk.server.v1.ChalkSpanEvent.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xd2\x01\n\rChalkSpanLink\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x17\n\x07span_id\x18\x02 \x01(\tR\x06spanId\x12N\n\nattributes\x18\x03 \x03(\x0b\x32..chalk.server.v1.ChalkSpanLink.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xe1\x02\n\nChalkTrace\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x30\n\x05spans\x18\x02 \x03(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x05spans\x12%\n\x0croot_span_id\x18\x03 \x01(\tH\x00R\nrootSpanId\x88\x01\x01\x12!\n\x0cservice_name\x18\x04 \x01(\tR\x0bserviceName\x12\x64\n\x13resource_attributes\x18\x05 \x03(\x0b\x32\x33.chalk.server.v1.ChalkTrace.ResourceAttributesEntryR\x12resourceAttributes\x1a\x45\n\x17ResourceAttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0f\n\r_root_span_id"w\n\x0fGetTraceRequest\x12&\n\x0coperation_id\x18\x01 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x1e\n\x08trace_id\x18\x02 \x01(\tH\x01R\x07traceId\x88\x01\x01\x42\x0f\n\r_operation_idB\x0b\n\t_trace_id"E\n\x10GetTraceResponse\x12\x31\n\x05trace\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.ChalkTraceR\x05trace"\xeb\x02\n\x10ListTraceRequest\x12>\n\nstart_time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tstartTime\x88\x01\x01\x12:\n\x08\x65nd_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x07\x65ndTime\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12&\n\x0cservice_name\x18\x04 \x01(\tH\x03R\x0bserviceName\x88\x01\x01\x12 \n\tspan_name\x18\x05 \x01(\tH\x04R\x08spanName\x88\x01\x01\x12"\n\npage_token\x18\x06 \x01(\tH\x05R\tpageToken\x88\x01\x01\x42\r\n\x0b_start_timeB\x0b\n\t_end_timeB\x08\n\x06_limitB\x0f\n\r_service_nameB\x0c\n\n_span_nameB\r\n\x0b_page_token"\x89\x01\n\x11ListTraceResponse\x12\x33\n\x06traces\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.ChalkTraceR\x06traces\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token"D\n\x0eGetSpanRequest\x12\x17\n\x07span_id\x18\x01 \x01(\tR\x06spanId\x12\x19\n\x08trace_id\x18\x02 \x01(\tR\x07traceId"A\n\x0fGetSpanResponse\x12.\n\x04span\x18\x01 \x01(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x04span"\xbe\x05\n\x0fListSpanRequest\x12\x1e\n\x08trace_id\x18\x01 \x01(\tH\x00R\x07traceId\x88\x01\x01\x12>\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tstartTime\x88\x01\x01\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x07\x65ndTime\x88\x01\x01\x12\x19\n\x05limit\x18\x04 \x01(\x05H\x03R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x05 \x01(\tH\x04R\tpageToken\x88\x01\x01\x12)\n\x0eparent_span_id\x18\x06 \x01(\tH\x05R\x0cparentSpanId\x88\x01\x01\x12*\n\x0eoperation_name\x18\x07 \x01(\tH\x06R\roperationName\x88\x01\x01\x12&\n\x0cservice_name\x18\x08 \x01(\tH\x07R\x0bserviceName\x88\x01\x01\x12\x46\n\x0bstatus_code\x18\t \x01(\x0e\x32 .chalk.server.v1.ChalkStatusCodeH\x08R\nstatusCode\x88\x01\x01\x12+\n\x0fmin_duration_us\x18\n \x01(\x03H\tR\rminDurationUs\x88\x01\x01\x12+\n\x0fmax_duration_us\x18\x0b \x01(\x03H\nR\rmaxDurationUs\x88\x01\x01\x42\x0b\n\t_trace_idB\r\n\x0b_start_timeB\x0b\n\t_end_timeB\x08\n\x06_limitB\r\n\x0b_page_tokenB\x11\n\x0f_parent_span_idB\x11\n\x0f_operation_nameB\x0f\n\r_service_nameB\x0e\n\x0c_status_codeB\x12\n\x10_min_duration_usB\x12\n\x10_max_duration_us"\x85\x01\n\x10ListSpanResponse\x12\x30\n\x05spans\x18\x01 \x03(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x05spans\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token*k\n\x0f\x43halkStatusCode\x12!\n\x1d\x43HALK_STATUS_CODE_UNSPECIFIED\x10\x00\x12\x18\n\x14\x43HALK_STATUS_CODE_OK\x10\x01\x12\x1b\n\x17\x43HALK_STATUS_CODE_ERROR\x10\x02*\xc2\x01\n\rChalkSpanKind\x12\x1f\n\x1b\x43HALK_SPAN_KIND_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43HALK_SPAN_KIND_SERVER\x10\x01\x12\x1a\n\x16\x43HALK_SPAN_KIND_CLIENT\x10\x02\x12\x1c\n\x18\x43HALK_SPAN_KIND_PRODUCER\x10\x03\x12\x1c\n\x18\x43HALK_SPAN_KIND_CONSUMER\x10\x04\x12\x1c\n\x18\x43HALK_SPAN_KIND_INTERNAL\x10\x05\x32\xf2\x02\n\x0cTraceService\x12W\n\x08GetTrace\x12 .chalk.server.v1.GetTraceRequest\x1a!.chalk.server.v1.GetTraceResponse"\x06\x90\x02\x01\x80}\x06\x12Z\n\tListTrace\x12!.chalk.server.v1.ListTraceRequest\x1a".chalk.server.v1.ListTraceResponse"\x06\x90\x02\x01\x80}\x06\x12T\n\x07GetSpan\x12\x1f.chalk.server.v1.GetSpanRequest\x1a .chalk.server.v1.GetSpanResponse"\x06\x90\x02\x01\x80}\x06\x12W\n\x08ListSpan\x12 .chalk.server.v1.ListSpanRequest\x1a!.chalk.server.v1.ListSpanResponse"\x06\x90\x02\x01\x80}\x06\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nTraceProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
|
|
23
23
|
)
|
|
24
24
|
|
|
25
25
|
_globals = globals()
|
|
@@ -32,6 +32,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
32
32
|
]._serialized_options = b"\n\023com.chalk.server.v1B\nTraceProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
|
|
33
33
|
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._options = None
|
|
34
34
|
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_options = b"8\001"
|
|
35
|
+
_globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._options = None
|
|
36
|
+
_globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_options = b"8\001"
|
|
35
37
|
_globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._options = None
|
|
36
38
|
_globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_options = b"8\001"
|
|
37
39
|
_globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._options = None
|
|
@@ -46,44 +48,46 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
46
48
|
_globals["_TRACESERVICE"].methods_by_name["GetSpan"]._serialized_options = b"\220\002\001\200}\006"
|
|
47
49
|
_globals["_TRACESERVICE"].methods_by_name["ListSpan"]._options = None
|
|
48
50
|
_globals["_TRACESERVICE"].methods_by_name["ListSpan"]._serialized_options = b"\220\002\001\200}\006"
|
|
49
|
-
_globals["_CHALKSTATUSCODE"]._serialized_start =
|
|
50
|
-
_globals["_CHALKSTATUSCODE"]._serialized_end =
|
|
51
|
-
_globals["_CHALKSPANKIND"]._serialized_start =
|
|
52
|
-
_globals["_CHALKSPANKIND"]._serialized_end =
|
|
51
|
+
_globals["_CHALKSTATUSCODE"]._serialized_start = 3628
|
|
52
|
+
_globals["_CHALKSTATUSCODE"]._serialized_end = 3735
|
|
53
|
+
_globals["_CHALKSPANKIND"]._serialized_start = 3738
|
|
54
|
+
_globals["_CHALKSPANKIND"]._serialized_end = 3932
|
|
53
55
|
_globals["_CHALKSPAN"]._serialized_start = 147
|
|
54
|
-
_globals["_CHALKSPAN"]._serialized_end =
|
|
55
|
-
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_start =
|
|
56
|
-
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_end =
|
|
57
|
-
_globals["
|
|
58
|
-
_globals["
|
|
59
|
-
_globals["
|
|
60
|
-
_globals["
|
|
61
|
-
_globals["
|
|
62
|
-
_globals["
|
|
63
|
-
_globals["
|
|
64
|
-
_globals["
|
|
65
|
-
_globals["
|
|
66
|
-
_globals["
|
|
67
|
-
_globals["
|
|
68
|
-
_globals["
|
|
69
|
-
_globals["
|
|
70
|
-
_globals["
|
|
71
|
-
_globals["
|
|
72
|
-
_globals["
|
|
73
|
-
_globals["
|
|
74
|
-
_globals["
|
|
75
|
-
_globals["
|
|
76
|
-
_globals["
|
|
77
|
-
_globals["
|
|
78
|
-
_globals["
|
|
79
|
-
_globals["
|
|
80
|
-
_globals["
|
|
81
|
-
_globals["
|
|
82
|
-
_globals["
|
|
83
|
-
_globals["
|
|
84
|
-
_globals["
|
|
85
|
-
_globals["
|
|
86
|
-
_globals["
|
|
87
|
-
_globals["
|
|
88
|
-
_globals["
|
|
56
|
+
_globals["_CHALKSPAN"]._serialized_end = 1012
|
|
57
|
+
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_start = 861
|
|
58
|
+
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_end = 922
|
|
59
|
+
_globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_start = 924
|
|
60
|
+
_globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_end = 993
|
|
61
|
+
_globals["_CHALKSPANSTATUS"]._serialized_start = 1014
|
|
62
|
+
_globals["_CHALKSPANSTATUS"]._serialized_end = 1140
|
|
63
|
+
_globals["_CHALKSPANEVENT"]._serialized_start = 1143
|
|
64
|
+
_globals["_CHALKSPANEVENT"]._serialized_end = 1381
|
|
65
|
+
_globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_start = 861
|
|
66
|
+
_globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_end = 922
|
|
67
|
+
_globals["_CHALKSPANLINK"]._serialized_start = 1384
|
|
68
|
+
_globals["_CHALKSPANLINK"]._serialized_end = 1594
|
|
69
|
+
_globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_start = 861
|
|
70
|
+
_globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_end = 922
|
|
71
|
+
_globals["_CHALKTRACE"]._serialized_start = 1597
|
|
72
|
+
_globals["_CHALKTRACE"]._serialized_end = 1950
|
|
73
|
+
_globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_start = 924
|
|
74
|
+
_globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_end = 993
|
|
75
|
+
_globals["_GETTRACEREQUEST"]._serialized_start = 1952
|
|
76
|
+
_globals["_GETTRACEREQUEST"]._serialized_end = 2071
|
|
77
|
+
_globals["_GETTRACERESPONSE"]._serialized_start = 2073
|
|
78
|
+
_globals["_GETTRACERESPONSE"]._serialized_end = 2142
|
|
79
|
+
_globals["_LISTTRACEREQUEST"]._serialized_start = 2145
|
|
80
|
+
_globals["_LISTTRACEREQUEST"]._serialized_end = 2508
|
|
81
|
+
_globals["_LISTTRACERESPONSE"]._serialized_start = 2511
|
|
82
|
+
_globals["_LISTTRACERESPONSE"]._serialized_end = 2648
|
|
83
|
+
_globals["_GETSPANREQUEST"]._serialized_start = 2650
|
|
84
|
+
_globals["_GETSPANREQUEST"]._serialized_end = 2718
|
|
85
|
+
_globals["_GETSPANRESPONSE"]._serialized_start = 2720
|
|
86
|
+
_globals["_GETSPANRESPONSE"]._serialized_end = 2785
|
|
87
|
+
_globals["_LISTSPANREQUEST"]._serialized_start = 2788
|
|
88
|
+
_globals["_LISTSPANREQUEST"]._serialized_end = 3490
|
|
89
|
+
_globals["_LISTSPANRESPONSE"]._serialized_start = 3493
|
|
90
|
+
_globals["_LISTSPANRESPONSE"]._serialized_end = 3626
|
|
91
|
+
_globals["_TRACESERVICE"]._serialized_start = 3935
|
|
92
|
+
_globals["_TRACESERVICE"]._serialized_end = 4305
|
|
89
93
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -54,6 +54,7 @@ class ChalkSpan(_message.Message):
|
|
|
54
54
|
"events",
|
|
55
55
|
"links",
|
|
56
56
|
"kind",
|
|
57
|
+
"resource_attributes",
|
|
57
58
|
)
|
|
58
59
|
class AttributesEntry(_message.Message):
|
|
59
60
|
__slots__ = ("key", "value")
|
|
@@ -63,6 +64,14 @@ class ChalkSpan(_message.Message):
|
|
|
63
64
|
value: str
|
|
64
65
|
def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
|
|
65
66
|
|
|
67
|
+
class ResourceAttributesEntry(_message.Message):
|
|
68
|
+
__slots__ = ("key", "value")
|
|
69
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
70
|
+
VALUE_FIELD_NUMBER: _ClassVar[int]
|
|
71
|
+
key: str
|
|
72
|
+
value: str
|
|
73
|
+
def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
|
|
74
|
+
|
|
66
75
|
SPAN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
67
76
|
TRACE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
68
77
|
PARENT_SPAN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -75,6 +84,7 @@ class ChalkSpan(_message.Message):
|
|
|
75
84
|
EVENTS_FIELD_NUMBER: _ClassVar[int]
|
|
76
85
|
LINKS_FIELD_NUMBER: _ClassVar[int]
|
|
77
86
|
KIND_FIELD_NUMBER: _ClassVar[int]
|
|
87
|
+
RESOURCE_ATTRIBUTES_FIELD_NUMBER: _ClassVar[int]
|
|
78
88
|
span_id: str
|
|
79
89
|
trace_id: str
|
|
80
90
|
parent_span_id: str
|
|
@@ -87,6 +97,7 @@ class ChalkSpan(_message.Message):
|
|
|
87
97
|
events: _containers.RepeatedCompositeFieldContainer[ChalkSpanEvent]
|
|
88
98
|
links: _containers.RepeatedCompositeFieldContainer[ChalkSpanLink]
|
|
89
99
|
kind: ChalkSpanKind
|
|
100
|
+
resource_attributes: _containers.ScalarMap[str, str]
|
|
90
101
|
def __init__(
|
|
91
102
|
self,
|
|
92
103
|
span_id: _Optional[str] = ...,
|
|
@@ -101,6 +112,7 @@ class ChalkSpan(_message.Message):
|
|
|
101
112
|
events: _Optional[_Iterable[_Union[ChalkSpanEvent, _Mapping]]] = ...,
|
|
102
113
|
links: _Optional[_Iterable[_Union[ChalkSpanLink, _Mapping]]] = ...,
|
|
103
114
|
kind: _Optional[_Union[ChalkSpanKind, str]] = ...,
|
|
115
|
+
resource_attributes: _Optional[_Mapping[str, str]] = ...,
|
|
104
116
|
) -> None: ...
|
|
105
117
|
|
|
106
118
|
class ChalkSpanStatus(_message.Message):
|
|
@@ -261,6 +273,8 @@ class ListSpanRequest(_message.Message):
|
|
|
261
273
|
"operation_name",
|
|
262
274
|
"service_name",
|
|
263
275
|
"status_code",
|
|
276
|
+
"min_duration_us",
|
|
277
|
+
"max_duration_us",
|
|
264
278
|
)
|
|
265
279
|
TRACE_ID_FIELD_NUMBER: _ClassVar[int]
|
|
266
280
|
START_TIME_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -271,6 +285,8 @@ class ListSpanRequest(_message.Message):
|
|
|
271
285
|
OPERATION_NAME_FIELD_NUMBER: _ClassVar[int]
|
|
272
286
|
SERVICE_NAME_FIELD_NUMBER: _ClassVar[int]
|
|
273
287
|
STATUS_CODE_FIELD_NUMBER: _ClassVar[int]
|
|
288
|
+
MIN_DURATION_US_FIELD_NUMBER: _ClassVar[int]
|
|
289
|
+
MAX_DURATION_US_FIELD_NUMBER: _ClassVar[int]
|
|
274
290
|
trace_id: str
|
|
275
291
|
start_time: _timestamp_pb2.Timestamp
|
|
276
292
|
end_time: _timestamp_pb2.Timestamp
|
|
@@ -280,6 +296,8 @@ class ListSpanRequest(_message.Message):
|
|
|
280
296
|
operation_name: str
|
|
281
297
|
service_name: str
|
|
282
298
|
status_code: ChalkStatusCode
|
|
299
|
+
min_duration_us: int
|
|
300
|
+
max_duration_us: int
|
|
283
301
|
def __init__(
|
|
284
302
|
self,
|
|
285
303
|
trace_id: _Optional[str] = ...,
|
|
@@ -291,6 +309,8 @@ class ListSpanRequest(_message.Message):
|
|
|
291
309
|
operation_name: _Optional[str] = ...,
|
|
292
310
|
service_name: _Optional[str] = ...,
|
|
293
311
|
status_code: _Optional[_Union[ChalkStatusCode, str]] = ...,
|
|
312
|
+
min_duration_us: _Optional[int] = ...,
|
|
313
|
+
max_duration_us: _Optional[int] = ...,
|
|
294
314
|
) -> None: ...
|
|
295
315
|
|
|
296
316
|
class ListSpanResponse(_message.Message):
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: chalk/streaming/v1/debug_service.proto
|
|
4
|
+
# Protobuf Python Version: 4.25.3
|
|
5
|
+
"""Generated protocol buffer code."""
|
|
6
|
+
|
|
7
|
+
from google.protobuf import descriptor as _descriptor
|
|
8
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
|
9
|
+
from google.protobuf import symbol_database as _symbol_database
|
|
10
|
+
from google.protobuf.internal import builder as _builder
|
|
11
|
+
# @@protoc_insertion_point(imports)
|
|
12
|
+
|
|
13
|
+
_sym_db = _symbol_database.Default()
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
|
|
17
|
+
from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
21
|
+
b'\n&chalk/streaming/v1/debug_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"`\n\x16\x45nableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId"n\n\x17\x45nableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12\x39\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tenabledAt"<\n\x17\x44isableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"4\n\x18\x44isableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled">\n\x19GetDebugModeStatusRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"\xe8\x01\n\x1aGetDebugModeStatusResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12>\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tenabledAt\x88\x01\x01\x12%\n\x0estorage_bucket\x18\x03 \x01(\tR\rstorageBucket\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x42\r\n\x0b_enabled_atB\x10\n\x0e_deployment_id"<\n\x17GetDebugMessagesRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"Y\n\x18GetDebugMessagesResponse\x12\x18\n\x07parquet\x18\x01 \x01(\x0cR\x07parquet\x12\x19\n\x05\x65rror\x18\x02 \x01(\tH\x00R\x05\x65rror\x88\x01\x01\x42\x08\n\x06_error2\xf0\x03\n\x15StreamingDebugService\x12o\n\x0f\x45nableDebugMode\x12*.chalk.streaming.v1.EnableDebugModeRequest\x1a+.chalk.streaming.v1.EnableDebugModeResponse"\x03\x80}\x0c\x12r\n\x10\x44isableDebugMode\x12+.chalk.streaming.v1.DisableDebugModeRequest\x1a,.chalk.streaming.v1.DisableDebugModeResponse"\x03\x80}\x0c\x12{\n\x12GetDebugModeStatus\x12-.chalk.streaming.v1.GetDebugModeStatusRequest\x1a..chalk.streaming.v1.GetDebugModeStatusResponse"\x06\x90\x02\x01\x80}\x0b\x12u\n\x10GetDebugMessages\x12+.chalk.streaming.v1.GetDebugMessagesRequest\x1a,.chalk.streaming.v1.GetDebugMessagesResponse"\x06\x90\x02\x01\x80}\x0b\x42\x95\x01\n\x16\x63om.chalk.streaming.v1B\x11\x44\x65\x62ugServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
|
|
22
|
+
)
|
|
23
|
+
|
|
24
|
+
_globals = globals()
|
|
25
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
|
26
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.streaming.v1.debug_service_pb2", _globals)
|
|
27
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
|
28
|
+
_globals["DESCRIPTOR"]._options = None
|
|
29
|
+
_globals[
|
|
30
|
+
"DESCRIPTOR"
|
|
31
|
+
]._serialized_options = b"\n\026com.chalk.streaming.v1B\021DebugServiceProtoP\001\242\002\003CSX\252\002\022Chalk.Streaming.V1\312\002\022Chalk\\Streaming\\V1\342\002\036Chalk\\Streaming\\V1\\GPBMetadata\352\002\024Chalk::Streaming::V1"
|
|
32
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._options = None
|
|
33
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._serialized_options = b"\200}\014"
|
|
34
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._options = None
|
|
35
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._serialized_options = b"\200}\014"
|
|
36
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugModeStatus"]._options = None
|
|
37
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
|
|
38
|
+
"GetDebugModeStatus"
|
|
39
|
+
]._serialized_options = b"\220\002\001\200}\013"
|
|
40
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugMessages"]._options = None
|
|
41
|
+
_globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
|
|
42
|
+
"GetDebugMessages"
|
|
43
|
+
]._serialized_options = b"\220\002\001\200}\013"
|
|
44
|
+
_globals["_ENABLEDEBUGMODEREQUEST"]._serialized_start = 128
|
|
45
|
+
_globals["_ENABLEDEBUGMODEREQUEST"]._serialized_end = 224
|
|
46
|
+
_globals["_ENABLEDEBUGMODERESPONSE"]._serialized_start = 226
|
|
47
|
+
_globals["_ENABLEDEBUGMODERESPONSE"]._serialized_end = 336
|
|
48
|
+
_globals["_DISABLEDEBUGMODEREQUEST"]._serialized_start = 338
|
|
49
|
+
_globals["_DISABLEDEBUGMODEREQUEST"]._serialized_end = 398
|
|
50
|
+
_globals["_DISABLEDEBUGMODERESPONSE"]._serialized_start = 400
|
|
51
|
+
_globals["_DISABLEDEBUGMODERESPONSE"]._serialized_end = 452
|
|
52
|
+
_globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_start = 454
|
|
53
|
+
_globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_end = 516
|
|
54
|
+
_globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_start = 519
|
|
55
|
+
_globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_end = 751
|
|
56
|
+
_globals["_GETDEBUGMESSAGESREQUEST"]._serialized_start = 753
|
|
57
|
+
_globals["_GETDEBUGMESSAGESREQUEST"]._serialized_end = 813
|
|
58
|
+
_globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_start = 815
|
|
59
|
+
_globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_end = 904
|
|
60
|
+
_globals["_STREAMINGDEBUGSERVICE"]._serialized_start = 907
|
|
61
|
+
_globals["_STREAMINGDEBUGSERVICE"]._serialized_end = 1403
|
|
62
|
+
# @@protoc_insertion_point(module_scope)
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
|
|
2
|
+
from google.protobuf import timestamp_pb2 as _timestamp_pb2
|
|
3
|
+
from google.protobuf import descriptor as _descriptor
|
|
4
|
+
from google.protobuf import message as _message
|
|
5
|
+
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
6
|
+
|
|
7
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
8
|
+
|
|
9
|
+
class EnableDebugModeRequest(_message.Message):
|
|
10
|
+
__slots__ = ("resolver_fqn", "deployment_id")
|
|
11
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
12
|
+
DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
13
|
+
resolver_fqn: str
|
|
14
|
+
deployment_id: str
|
|
15
|
+
def __init__(self, resolver_fqn: _Optional[str] = ..., deployment_id: _Optional[str] = ...) -> None: ...
|
|
16
|
+
|
|
17
|
+
class EnableDebugModeResponse(_message.Message):
|
|
18
|
+
__slots__ = ("enabled", "enabled_at")
|
|
19
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
20
|
+
ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
21
|
+
enabled: bool
|
|
22
|
+
enabled_at: _timestamp_pb2.Timestamp
|
|
23
|
+
def __init__(
|
|
24
|
+
self, enabled: bool = ..., enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...
|
|
25
|
+
) -> None: ...
|
|
26
|
+
|
|
27
|
+
class DisableDebugModeRequest(_message.Message):
|
|
28
|
+
__slots__ = ("resolver_fqn",)
|
|
29
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
30
|
+
resolver_fqn: str
|
|
31
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
32
|
+
|
|
33
|
+
class DisableDebugModeResponse(_message.Message):
|
|
34
|
+
__slots__ = ("enabled",)
|
|
35
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
36
|
+
enabled: bool
|
|
37
|
+
def __init__(self, enabled: bool = ...) -> None: ...
|
|
38
|
+
|
|
39
|
+
class GetDebugModeStatusRequest(_message.Message):
|
|
40
|
+
__slots__ = ("resolver_fqn",)
|
|
41
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
42
|
+
resolver_fqn: str
|
|
43
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
44
|
+
|
|
45
|
+
class GetDebugModeStatusResponse(_message.Message):
|
|
46
|
+
__slots__ = ("enabled", "enabled_at", "storage_bucket", "deployment_id")
|
|
47
|
+
ENABLED_FIELD_NUMBER: _ClassVar[int]
|
|
48
|
+
ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
|
|
49
|
+
STORAGE_BUCKET_FIELD_NUMBER: _ClassVar[int]
|
|
50
|
+
DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
|
51
|
+
enabled: bool
|
|
52
|
+
enabled_at: _timestamp_pb2.Timestamp
|
|
53
|
+
storage_bucket: str
|
|
54
|
+
deployment_id: str
|
|
55
|
+
def __init__(
|
|
56
|
+
self,
|
|
57
|
+
enabled: bool = ...,
|
|
58
|
+
enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
59
|
+
storage_bucket: _Optional[str] = ...,
|
|
60
|
+
deployment_id: _Optional[str] = ...,
|
|
61
|
+
) -> None: ...
|
|
62
|
+
|
|
63
|
+
class GetDebugMessagesRequest(_message.Message):
|
|
64
|
+
__slots__ = ("resolver_fqn",)
|
|
65
|
+
RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
|
|
66
|
+
resolver_fqn: str
|
|
67
|
+
def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
|
|
68
|
+
|
|
69
|
+
class GetDebugMessagesResponse(_message.Message):
|
|
70
|
+
__slots__ = ("parquet", "error")
|
|
71
|
+
PARQUET_FIELD_NUMBER: _ClassVar[int]
|
|
72
|
+
ERROR_FIELD_NUMBER: _ClassVar[int]
|
|
73
|
+
parquet: bytes
|
|
74
|
+
error: str
|
|
75
|
+
def __init__(self, parquet: _Optional[bytes] = ..., error: _Optional[str] = ...) -> None: ...
|
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
|
2
|
+
"""Client and server classes corresponding to protobuf-defined services."""
|
|
3
|
+
|
|
4
|
+
import grpc
|
|
5
|
+
|
|
6
|
+
from chalk._gen.chalk.streaming.v1 import debug_service_pb2 as chalk_dot_streaming_dot_v1_dot_debug__service__pb2
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
class StreamingDebugServiceStub(object):
|
|
10
|
+
"""Streaming Debug Service
|
|
11
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
def __init__(self, channel):
|
|
15
|
+
"""Constructor.
|
|
16
|
+
|
|
17
|
+
Args:
|
|
18
|
+
channel: A grpc.Channel.
|
|
19
|
+
"""
|
|
20
|
+
self.EnableDebugMode = channel.unary_unary(
|
|
21
|
+
"/chalk.streaming.v1.StreamingDebugService/EnableDebugMode",
|
|
22
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.SerializeToString,
|
|
23
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.FromString,
|
|
24
|
+
)
|
|
25
|
+
self.DisableDebugMode = channel.unary_unary(
|
|
26
|
+
"/chalk.streaming.v1.StreamingDebugService/DisableDebugMode",
|
|
27
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.SerializeToString,
|
|
28
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.FromString,
|
|
29
|
+
)
|
|
30
|
+
self.GetDebugModeStatus = channel.unary_unary(
|
|
31
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugModeStatus",
|
|
32
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.SerializeToString,
|
|
33
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.FromString,
|
|
34
|
+
)
|
|
35
|
+
self.GetDebugMessages = channel.unary_unary(
|
|
36
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugMessages",
|
|
37
|
+
request_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.SerializeToString,
|
|
38
|
+
response_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.FromString,
|
|
39
|
+
)
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
class StreamingDebugServiceServicer(object):
|
|
43
|
+
"""Streaming Debug Service
|
|
44
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
45
|
+
"""
|
|
46
|
+
|
|
47
|
+
def EnableDebugMode(self, request, context):
|
|
48
|
+
"""Enable debug mode for a streaming resolver"""
|
|
49
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
50
|
+
context.set_details("Method not implemented!")
|
|
51
|
+
raise NotImplementedError("Method not implemented!")
|
|
52
|
+
|
|
53
|
+
def DisableDebugMode(self, request, context):
|
|
54
|
+
"""Disable debug mode for a streaming resolver"""
|
|
55
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
56
|
+
context.set_details("Method not implemented!")
|
|
57
|
+
raise NotImplementedError("Method not implemented!")
|
|
58
|
+
|
|
59
|
+
def GetDebugModeStatus(self, request, context):
|
|
60
|
+
"""Get the current debug mode status for a resolver"""
|
|
61
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
62
|
+
context.set_details("Method not implemented!")
|
|
63
|
+
raise NotImplementedError("Method not implemented!")
|
|
64
|
+
|
|
65
|
+
def GetDebugMessages(self, request, context):
|
|
66
|
+
"""Get recent debug messages as parquet"""
|
|
67
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
68
|
+
context.set_details("Method not implemented!")
|
|
69
|
+
raise NotImplementedError("Method not implemented!")
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
def add_StreamingDebugServiceServicer_to_server(servicer, server):
|
|
73
|
+
rpc_method_handlers = {
|
|
74
|
+
"EnableDebugMode": grpc.unary_unary_rpc_method_handler(
|
|
75
|
+
servicer.EnableDebugMode,
|
|
76
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.FromString,
|
|
77
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.SerializeToString,
|
|
78
|
+
),
|
|
79
|
+
"DisableDebugMode": grpc.unary_unary_rpc_method_handler(
|
|
80
|
+
servicer.DisableDebugMode,
|
|
81
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.FromString,
|
|
82
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.SerializeToString,
|
|
83
|
+
),
|
|
84
|
+
"GetDebugModeStatus": grpc.unary_unary_rpc_method_handler(
|
|
85
|
+
servicer.GetDebugModeStatus,
|
|
86
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.FromString,
|
|
87
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.SerializeToString,
|
|
88
|
+
),
|
|
89
|
+
"GetDebugMessages": grpc.unary_unary_rpc_method_handler(
|
|
90
|
+
servicer.GetDebugMessages,
|
|
91
|
+
request_deserializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.FromString,
|
|
92
|
+
response_serializer=chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.SerializeToString,
|
|
93
|
+
),
|
|
94
|
+
}
|
|
95
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
|
96
|
+
"chalk.streaming.v1.StreamingDebugService", rpc_method_handlers
|
|
97
|
+
)
|
|
98
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
|
99
|
+
|
|
100
|
+
|
|
101
|
+
# This class is part of an EXPERIMENTAL API.
|
|
102
|
+
class StreamingDebugService(object):
|
|
103
|
+
"""Streaming Debug Service
|
|
104
|
+
Provides APIs for managing and querying streaming resolver debug mode
|
|
105
|
+
"""
|
|
106
|
+
|
|
107
|
+
@staticmethod
|
|
108
|
+
def EnableDebugMode(
|
|
109
|
+
request,
|
|
110
|
+
target,
|
|
111
|
+
options=(),
|
|
112
|
+
channel_credentials=None,
|
|
113
|
+
call_credentials=None,
|
|
114
|
+
insecure=False,
|
|
115
|
+
compression=None,
|
|
116
|
+
wait_for_ready=None,
|
|
117
|
+
timeout=None,
|
|
118
|
+
metadata=None,
|
|
119
|
+
):
|
|
120
|
+
return grpc.experimental.unary_unary(
|
|
121
|
+
request,
|
|
122
|
+
target,
|
|
123
|
+
"/chalk.streaming.v1.StreamingDebugService/EnableDebugMode",
|
|
124
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeRequest.SerializeToString,
|
|
125
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.EnableDebugModeResponse.FromString,
|
|
126
|
+
options,
|
|
127
|
+
channel_credentials,
|
|
128
|
+
insecure,
|
|
129
|
+
call_credentials,
|
|
130
|
+
compression,
|
|
131
|
+
wait_for_ready,
|
|
132
|
+
timeout,
|
|
133
|
+
metadata,
|
|
134
|
+
)
|
|
135
|
+
|
|
136
|
+
@staticmethod
|
|
137
|
+
def DisableDebugMode(
|
|
138
|
+
request,
|
|
139
|
+
target,
|
|
140
|
+
options=(),
|
|
141
|
+
channel_credentials=None,
|
|
142
|
+
call_credentials=None,
|
|
143
|
+
insecure=False,
|
|
144
|
+
compression=None,
|
|
145
|
+
wait_for_ready=None,
|
|
146
|
+
timeout=None,
|
|
147
|
+
metadata=None,
|
|
148
|
+
):
|
|
149
|
+
return grpc.experimental.unary_unary(
|
|
150
|
+
request,
|
|
151
|
+
target,
|
|
152
|
+
"/chalk.streaming.v1.StreamingDebugService/DisableDebugMode",
|
|
153
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeRequest.SerializeToString,
|
|
154
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.DisableDebugModeResponse.FromString,
|
|
155
|
+
options,
|
|
156
|
+
channel_credentials,
|
|
157
|
+
insecure,
|
|
158
|
+
call_credentials,
|
|
159
|
+
compression,
|
|
160
|
+
wait_for_ready,
|
|
161
|
+
timeout,
|
|
162
|
+
metadata,
|
|
163
|
+
)
|
|
164
|
+
|
|
165
|
+
@staticmethod
|
|
166
|
+
def GetDebugModeStatus(
|
|
167
|
+
request,
|
|
168
|
+
target,
|
|
169
|
+
options=(),
|
|
170
|
+
channel_credentials=None,
|
|
171
|
+
call_credentials=None,
|
|
172
|
+
insecure=False,
|
|
173
|
+
compression=None,
|
|
174
|
+
wait_for_ready=None,
|
|
175
|
+
timeout=None,
|
|
176
|
+
metadata=None,
|
|
177
|
+
):
|
|
178
|
+
return grpc.experimental.unary_unary(
|
|
179
|
+
request,
|
|
180
|
+
target,
|
|
181
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugModeStatus",
|
|
182
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusRequest.SerializeToString,
|
|
183
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugModeStatusResponse.FromString,
|
|
184
|
+
options,
|
|
185
|
+
channel_credentials,
|
|
186
|
+
insecure,
|
|
187
|
+
call_credentials,
|
|
188
|
+
compression,
|
|
189
|
+
wait_for_ready,
|
|
190
|
+
timeout,
|
|
191
|
+
metadata,
|
|
192
|
+
)
|
|
193
|
+
|
|
194
|
+
@staticmethod
|
|
195
|
+
def GetDebugMessages(
|
|
196
|
+
request,
|
|
197
|
+
target,
|
|
198
|
+
options=(),
|
|
199
|
+
channel_credentials=None,
|
|
200
|
+
call_credentials=None,
|
|
201
|
+
insecure=False,
|
|
202
|
+
compression=None,
|
|
203
|
+
wait_for_ready=None,
|
|
204
|
+
timeout=None,
|
|
205
|
+
metadata=None,
|
|
206
|
+
):
|
|
207
|
+
return grpc.experimental.unary_unary(
|
|
208
|
+
request,
|
|
209
|
+
target,
|
|
210
|
+
"/chalk.streaming.v1.StreamingDebugService/GetDebugMessages",
|
|
211
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesRequest.SerializeToString,
|
|
212
|
+
chalk_dot_streaming_dot_v1_dot_debug__service__pb2.GetDebugMessagesResponse.FromString,
|
|
213
|
+
options,
|
|
214
|
+
channel_credentials,
|
|
215
|
+
insecure,
|
|
216
|
+
call_credentials,
|
|
217
|
+
compression,
|
|
218
|
+
wait_for_ready,
|
|
219
|
+
timeout,
|
|
220
|
+
metadata,
|
|
221
|
+
)
|