pyfect-dev 0.1.0__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.
- pyfect/__init__.py +729 -0
- pyfect/_all.py +242 -0
- pyfect/_background.py +28 -0
- pyfect/_binary_codecs.py +124 -0
- pyfect/_cached.py +165 -0
- pyfect/_catch.py +128 -0
- pyfect/_catch_reason.py +574 -0
- pyfect/_concurrency.py +291 -0
- pyfect/_context_effects.py +163 -0
- pyfect/_context_managers.py +76 -0
- pyfect/_core.py +368 -0
- pyfect/_digits.py +71 -0
- pyfect/_effect_args.py +46 -0
- pyfect/_effect_base.py +71 -0
- pyfect/_effect_concurrency.py +133 -0
- pyfect/_effect_eager.py +103 -0
- pyfect/_effect_errors.py +344 -0
- pyfect/_effect_folding.py +110 -0
- pyfect/_effect_hash.py +140 -0
- pyfect/_effect_logging.py +57 -0
- pyfect/_effect_metrics.py +102 -0
- pyfect/_effect_resources.py +179 -0
- pyfect/_effect_services.py +25 -0
- pyfect/_effect_timing.py +199 -0
- pyfect/_effect_tracing.py +99 -0
- pyfect/_equal.py +195 -0
- pyfect/_extras.py +43 -0
- pyfect/_fiber.py +969 -0
- pyfect/_fiber_all.py +134 -0
- pyfect/_fiber_collections.py +106 -0
- pyfect/_fn.py +706 -0
- pyfect/_hamt.py +174 -0
- pyfect/_interrupt_reason.py +33 -0
- pyfect/_isaac.py +149 -0
- pyfect/_iterables.py +231 -0
- pyfect/_js.py +263 -0
- pyfect/_js_date.py +87 -0
- pyfect/_json_text.py +286 -0
- pyfect/_json_value.py +9 -0
- pyfect/_layer_core.py +403 -0
- pyfect/_layer_for_life.py +33 -0
- pyfect/_layer_tracing.py +70 -0
- pyfect/_locked_ref.py +147 -0
- pyfect/_loop_thread.py +152 -0
- pyfect/_metric_ref.py +29 -0
- pyfect/_one_or_many.py +14 -0
- pyfect/_os_wait.py +118 -0
- pyfect/_path_posix.py +129 -0
- pyfect/_primitives.py +434 -0
- pyfect/_public.py +36 -0
- pyfect/_reasons.py +64 -0
- pyfect/_released.py +67 -0
- pyfect/_root.py +87 -0
- pyfect/_round_robin.py +73 -0
- pyfect/_some.py +17 -0
- pyfect/_stack.py +52 -0
- pyfect/_stack_frame.py +114 -0
- pyfect/_string_hash.py +35 -0
- pyfect/_subscriber_queue.py +47 -0
- pyfect/_tags.py +14 -0
- pyfect/_tracer_ref.py +21 -0
- pyfect/_tx_ops.py +77 -0
- pyfect/_unsubscripted.py +23 -0
- pyfect/_url.py +54 -0
- pyfect/_waiters.py +127 -0
- pyfect/_wake_tracking.py +98 -0
- pyfect/_wire_codec.py +123 -0
- pyfect/ai/__init__.py +73 -0
- pyfect/ai/_approvals.py +249 -0
- pyfect/ai/_decision_answers.py +185 -0
- pyfect/ai/_generated_client.py +298 -0
- pyfect/ai/_generated_support.py +84 -0
- pyfect/ai/_mcp/__init__.py +1 -0
- pyfect/ai/_mcp/batch.py +68 -0
- pyfect/ai/_mcp/client_transport.py +412 -0
- pyfect/ai/_mcp/client_wire.py +284 -0
- pyfect/ai/_mcp/core.py +490 -0
- pyfect/ai/_mcp/dispatch.py +1048 -0
- pyfect/ai/_mcp/http.py +462 -0
- pyfect/ai/_mcp/http_admission.py +269 -0
- pyfect/ai/_mcp/http_status.py +27 -0
- pyfect/ai/_mcp/input_wire.py +226 -0
- pyfect/ai/_mcp/log_levels.py +55 -0
- pyfect/ai/_mcp/param_headers.py +63 -0
- pyfect/ai/_mcp/projection.py +248 -0
- pyfect/ai/_mcp/register.py +683 -0
- pyfect/ai/_mcp/reverse_policy.py +155 -0
- pyfect/ai/_mcp/routing_header.py +24 -0
- pyfect/ai/_mcp/sessions.py +114 -0
- pyfect/ai/_mcp/stdio.py +110 -0
- pyfect/ai/_mcp/subscriptions.py +134 -0
- pyfect/ai/_mcp/tool_schema.py +35 -0
- pyfect/ai/_noul_answers.py +64 -0
- pyfect/ai/_openai_all_of.py +137 -0
- pyfect/ai/_openai_errors.py +233 -0
- pyfect/ai/_provider.py +439 -0
- pyfect/ai/_provider_tools.py +129 -0
- pyfect/ai/_stream_resolution.py +231 -0
- pyfect/ai/_structured_output_checks.py +138 -0
- pyfect/ai/_structured_output_codec.py +229 -0
- pyfect/ai/_structured_output_json.py +109 -0
- pyfect/ai/_tool_call_ids.py +80 -0
- pyfect/ai/ai_error.py +785 -0
- pyfect/ai/anthropic/__init__.py +28 -0
- pyfect/ai/anthropic/_errors.py +164 -0
- pyfect/ai/anthropic/_prompt.py +375 -0
- pyfect/ai/anthropic/_response.py +443 -0
- pyfect/ai/anthropic/_schema.py +206 -0
- pyfect/ai/anthropic/_stream.py +377 -0
- pyfect/ai/anthropic/_tools.py +248 -0
- pyfect/ai/anthropic/client.py +289 -0
- pyfect/ai/anthropic/config.py +32 -0
- pyfect/ai/anthropic/error.py +27 -0
- pyfect/ai/anthropic/language_model.py +366 -0
- pyfect/ai/anthropic/telemetry.py +51 -0
- pyfect/ai/anthropic/tool.py +570 -0
- pyfect/ai/anthropic_structured_output.py +138 -0
- pyfect/ai/chat.py +548 -0
- pyfect/ai/cloudflare/__init__.py +15 -0
- pyfect/ai/cloudflare/_errors.py +108 -0
- pyfect/ai/cloudflare/client.py +174 -0
- pyfect/ai/cloudflare/decision_model.py +117 -0
- pyfect/ai/cloudflare/schema.py +144 -0
- pyfect/ai/decision.py +290 -0
- pyfect/ai/decision_model.py +283 -0
- pyfect/ai/embedding_model.py +177 -0
- pyfect/ai/id_generator.py +79 -0
- pyfect/ai/language_model.py +1077 -0
- pyfect/ai/mcp_client.py +682 -0
- pyfect/ai/mcp_client_error.py +223 -0
- pyfect/ai/mcp_protocol.py +205 -0
- pyfect/ai/mcp_schema.py +1300 -0
- pyfect/ai/mcp_server.py +404 -0
- pyfect/ai/model.py +56 -0
- pyfect/ai/openai/__init__.py +49 -0
- pyfect/ai/openai/_errors.py +89 -0
- pyfect/ai/openai/_prompt.py +527 -0
- pyfect/ai/openai/_requests.py +34 -0
- pyfect/ai/openai/_response.py +569 -0
- pyfect/ai/openai/_schema.py +769 -0
- pyfect/ai/openai/_stream.py +771 -0
- pyfect/ai/openai/_tools.py +128 -0
- pyfect/ai/openai/_web_socket.py +295 -0
- pyfect/ai/openai/client.py +296 -0
- pyfect/ai/openai/client_generated.py +123 -0
- pyfect/ai/openai/config.py +34 -0
- pyfect/ai/openai/decision_model.py +177 -0
- pyfect/ai/openai/embedding_model.py +116 -0
- pyfect/ai/openai/error.py +8 -0
- pyfect/ai/openai/generated/__init__.py +7213 -0
- pyfect/ai/openai/generated/_recursive.py +18 -0
- pyfect/ai/openai/generated/_schemas_0.py +1015 -0
- pyfect/ai/openai/generated/_schemas_1.py +1016 -0
- pyfect/ai/openai/generated/_schemas_2.py +1017 -0
- pyfect/ai/openai/generated/_schemas_3.py +615 -0
- pyfect/ai/openai/generated/client.py +5298 -0
- pyfect/ai/openai/generated/types.py +8123 -0
- pyfect/ai/openai/language_model.py +304 -0
- pyfect/ai/openai/telemetry.py +19 -0
- pyfect/ai/openai/tool.py +391 -0
- pyfect/ai/openai_compat/__init__.py +25 -0
- pyfect/ai/openai_compat/_prompt.py +191 -0
- pyfect/ai/openai_compat/_request.py +153 -0
- pyfect/ai/openai_compat/_response.py +407 -0
- pyfect/ai/openai_compat/_schema.py +97 -0
- pyfect/ai/openai_compat/client.py +234 -0
- pyfect/ai/openai_compat/config.py +32 -0
- pyfect/ai/openai_compat/embedding_model.py +139 -0
- pyfect/ai/openai_compat/error.py +7 -0
- pyfect/ai/openai_compat/language_model.py +245 -0
- pyfect/ai/openai_compat/telemetry.py +51 -0
- pyfect/ai/openai_structured_output.py +118 -0
- pyfect/ai/openrouter/__init__.py +36 -0
- pyfect/ai/openrouter/_errors.py +186 -0
- pyfect/ai/openrouter/_prompt.py +272 -0
- pyfect/ai/openrouter/_response.py +226 -0
- pyfect/ai/openrouter/_stream.py +320 -0
- pyfect/ai/openrouter/client.py +283 -0
- pyfect/ai/openrouter/config.py +36 -0
- pyfect/ai/openrouter/decision_model.py +161 -0
- pyfect/ai/openrouter/error.py +23 -0
- pyfect/ai/openrouter/generated/__init__.py +3029 -0
- pyfect/ai/openrouter/generated/_recursive.py +15 -0
- pyfect/ai/openrouter/generated/_schemas_0.py +1015 -0
- pyfect/ai/openrouter/generated/_schemas_1.py +522 -0
- pyfect/ai/openrouter/generated/client.py +1445 -0
- pyfect/ai/openrouter/generated/types.py +4769 -0
- pyfect/ai/openrouter/language_model.py +345 -0
- pyfect/ai/openrouter/schema.py +165 -0
- pyfect/ai/prompt.py +664 -0
- pyfect/ai/response.py +768 -0
- pyfect/ai/response_id_tracker.py +82 -0
- pyfect/ai/telemetry.py +126 -0
- pyfect/ai/tokenizer.py +61 -0
- pyfect/ai/tool.py +536 -0
- pyfect/ai/toolkit.py +490 -0
- pyfect/ai/typesafe/__init__.py +17 -0
- pyfect/ai/typesafe/_errors.py +120 -0
- pyfect/ai/typesafe/client.py +146 -0
- pyfect/ai/typesafe/config.py +34 -0
- pyfect/ai/typesafe/decision_model.py +90 -0
- pyfect/ai/typesafe/schema.py +140 -0
- pyfect/batching/__init__.py +58 -0
- pyfect/batching/_api.py +71 -0
- pyfect/batching/_engine.py +229 -0
- pyfect/batching/_from_effect.py +76 -0
- pyfect/batching/_persisted.py +98 -0
- pyfect/batching/_request.py +133 -0
- pyfect/batching/_resolver.py +477 -0
- pyfect/batching/_with_cache.py +90 -0
- pyfect/big_decimal.py +500 -0
- pyfect/brand.py +114 -0
- pyfect/byte_size.py +354 -0
- pyfect/cache/__init__.py +29 -0
- pyfect/cache/_cache.py +426 -0
- pyfect/cache/_expiry.py +69 -0
- pyfect/cache/_scoped_cache.py +546 -0
- pyfect/case.py +174 -0
- pyfect/cause.py +462 -0
- pyfect/channel/__init__.py +155 -0
- pyfect/channel/_base.py +92 -0
- pyfect/channel/_catch.py +158 -0
- pyfect/channel/_catch_reason.py +572 -0
- pyfect/channel/_channel.py +158 -0
- pyfect/channel/_combinators.py +220 -0
- pyfect/channel/_concurrency_methods.py +217 -0
- pyfect/channel/_concurrent_map.py +118 -0
- pyfect/channel/_constructors.py +349 -0
- pyfect/channel/_context.py +232 -0
- pyfect/channel/_error_methods.py +336 -0
- pyfect/channel/_errors.py +180 -0
- pyfect/channel/_filters.py +225 -0
- pyfect/channel/_lifecycle.py +71 -0
- pyfect/channel/_lifecycle_methods.py +206 -0
- pyfect/channel/_loop.py +100 -0
- pyfect/channel/_merge.py +246 -0
- pyfect/channel/_repeat.py +154 -0
- pyfect/channel/_run.py +149 -0
- pyfect/channel/_run_methods.py +341 -0
- pyfect/channel/_sinks.py +88 -0
- pyfect/channel/_sources.py +238 -0
- pyfect/channel/_text.py +176 -0
- pyfect/channel/_transform_methods.py +415 -0
- pyfect/channel/_upstream.py +67 -0
- pyfect/channel_schema.py +131 -0
- pyfect/cli/__init__.py +79 -0
- pyfect/cli/_ansi.py +82 -0
- pyfect/cli/_command.py +469 -0
- pyfect/cli/_config.py +164 -0
- pyfect/cli/_environment.py +10 -0
- pyfect/cli/_help.py +195 -0
- pyfect/cli/_lexer.py +81 -0
- pyfect/cli/_parser.py +465 -0
- pyfect/cli/_settings.py +85 -0
- pyfect/cli/_suggest.py +33 -0
- pyfect/cli/_width.py +134 -0
- pyfect/cli/_wizard.py +317 -0
- pyfect/cli/argument.py +212 -0
- pyfect/cli/cli_config.py +40 -0
- pyfect/cli/cli_error.py +242 -0
- pyfect/cli/cli_output.py +305 -0
- pyfect/cli/command.py +440 -0
- pyfect/cli/completions/__init__.py +50 -0
- pyfect/cli/completions/_completer.ps1 +190 -0
- pyfect/cli/completions/_descriptor.py +182 -0
- pyfect/cli/completions/bash.py +300 -0
- pyfect/cli/completions/fish.py +164 -0
- pyfect/cli/completions/powershell.py +156 -0
- pyfect/cli/completions/zsh.py +189 -0
- pyfect/cli/flag.py +246 -0
- pyfect/cli/global_flag.py +99 -0
- pyfect/cli/help_doc.py +63 -0
- pyfect/cli/param.py +789 -0
- pyfect/cli/primitive.py +308 -0
- pyfect/cli/prompt/__init__.py +75 -0
- pyfect/cli/prompt/_auto_complete.py +205 -0
- pyfect/cli/prompt/_choices.py +104 -0
- pyfect/cli/prompt/_confirm.py +195 -0
- pyfect/cli/prompt/_core.py +512 -0
- pyfect/cli/prompt/_date.py +173 -0
- pyfect/cli/prompt/_date_parts.py +387 -0
- pyfect/cli/prompt/_file.py +372 -0
- pyfect/cli/prompt/_multi_select.py +204 -0
- pyfect/cli/prompt/_number.py +257 -0
- pyfect/cli/prompt/_select.py +117 -0
- pyfect/cli/prompt/_text.py +239 -0
- pyfect/cli/terminal_capabilities.py +118 -0
- pyfect/clock.py +150 -0
- pyfect/cluster/__init__.py +155 -0
- pyfect/cluster/_cluster_abandon.py +36 -0
- pyfect/cluster/_cluster_layer.py +122 -0
- pyfect/cluster/_entity_activation.py +16 -0
- pyfect/cluster/_entity_manager/__init__.py +22 -0
- pyfect/cluster/_entity_manager/_activation.py +540 -0
- pyfect/cluster/_entity_manager/_core.py +186 -0
- pyfect/cluster/_entity_manager/_decode.py +60 -0
- pyfect/cluster/_entity_manager/_manager.py +423 -0
- pyfect/cluster/_entity_manager/_state.py +113 -0
- pyfect/cluster/_entity_reaper.py +94 -0
- pyfect/cluster/_hash.py +13 -0
- pyfect/cluster/_interruptors.py +88 -0
- pyfect/cluster/_resource_map.py +179 -0
- pyfect/cluster/_resource_ref.py +266 -0
- pyfect/cluster/_shard_lock.py +14 -0
- pyfect/cluster/cluster_cron.py +230 -0
- pyfect/cluster/cluster_error.py +81 -0
- pyfect/cluster/cluster_http.py +210 -0
- pyfect/cluster/cluster_metrics.py +54 -0
- pyfect/cluster/cluster_schema.py +137 -0
- pyfect/cluster/cluster_socket.py +197 -0
- pyfect/cluster/cluster_workflow_engine/__init__.py +24 -0
- pyfect/cluster/cluster_workflow_engine/_engine.py +1099 -0
- pyfect/cluster/cluster_workflow_engine/_entities.py +298 -0
- pyfect/cluster/deliver_at.py +31 -0
- pyfect/cluster/entity/__init__.py +39 -0
- pyfect/cluster/entity/_entity.py +332 -0
- pyfect/cluster/entity/_handlers.py +168 -0
- pyfect/cluster/entity/_keep_alive.py +74 -0
- pyfect/cluster/entity/_queue.py +103 -0
- pyfect/cluster/entity/_request.py +34 -0
- pyfect/cluster/entity/_services.py +23 -0
- pyfect/cluster/entity/_test_client.py +237 -0
- pyfect/cluster/entity_address.py +25 -0
- pyfect/cluster/entity_id.py +20 -0
- pyfect/cluster/entity_proxy.py +161 -0
- pyfect/cluster/entity_proxy_server.py +103 -0
- pyfect/cluster/entity_resource.py +122 -0
- pyfect/cluster/entity_type.py +20 -0
- pyfect/cluster/envelope.py +307 -0
- pyfect/cluster/http_runner.py +254 -0
- pyfect/cluster/k8s_http_client.py +281 -0
- pyfect/cluster/k8s_types.py +1394 -0
- pyfect/cluster/machine_id.py +21 -0
- pyfect/cluster/message.py +219 -0
- pyfect/cluster/message_storage/__init__.py +67 -0
- pyfect/cluster/message_storage/_encoded.py +393 -0
- pyfect/cluster/message_storage/_memory.py +386 -0
- pyfect/cluster/message_storage/_noop.py +109 -0
- pyfect/cluster/message_storage/_service.py +388 -0
- pyfect/cluster/reply.py +236 -0
- pyfect/cluster/runner.py +32 -0
- pyfect/cluster/runner_address.py +23 -0
- pyfect/cluster/runner_health.py +93 -0
- pyfect/cluster/runner_server.py +286 -0
- pyfect/cluster/runner_storage.py +212 -0
- pyfect/cluster/runners/__init__.py +55 -0
- pyfect/cluster/runners/_local.py +460 -0
- pyfect/cluster/runners/_noop.py +57 -0
- pyfect/cluster/runners/_rpc.py +290 -0
- pyfect/cluster/runners/_service.py +113 -0
- pyfect/cluster/shard_id.py +39 -0
- pyfect/cluster/sharding/__init__.py +12 -0
- pyfect/cluster/sharding/_acquisition.py +490 -0
- pyfect/cluster/sharding/_clients.py +271 -0
- pyfect/cluster/sharding/_make.py +388 -0
- pyfect/cluster/sharding/_routing.py +296 -0
- pyfect/cluster/sharding/_runners_sync.py +198 -0
- pyfect/cluster/sharding/_service.py +91 -0
- pyfect/cluster/sharding/_singletons.py +183 -0
- pyfect/cluster/sharding/_state.py +296 -0
- pyfect/cluster/sharding/_storage_loop.py +421 -0
- pyfect/cluster/sharding_config.py +230 -0
- pyfect/cluster/sharding_registration_event.py +28 -0
- pyfect/cluster/single_runner.py +49 -0
- pyfect/cluster/singleton.py +36 -0
- pyfect/cluster/singleton_address.py +20 -0
- pyfect/cluster/snowflake.py +133 -0
- pyfect/cluster/socket_runner.py +67 -0
- pyfect/cluster/sql_message_storage.py +1001 -0
- pyfect/cluster/sql_runner_storage.py +841 -0
- pyfect/cluster/test_runner.py +23 -0
- pyfect/config/__init__.py +93 -0
- pyfect/config/_all.py +96 -0
- pyfect/config/_collections.py +291 -0
- pyfect/config/_config.py +261 -0
- pyfect/config/_constructors.py +106 -0
- pyfect/config/_error.py +27 -0
- pyfect/config/_leaf.py +92 -0
- pyfect/config/_scalars.py +193 -0
- pyfect/config/_schema.py +340 -0
- pyfect/config_provider/__init__.py +60 -0
- pyfect/config_provider/_dir.py +84 -0
- pyfect/config_provider/_dotenv.py +127 -0
- pyfect/config_provider/_env.py +79 -0
- pyfect/config_provider/_layers.py +80 -0
- pyfect/config_provider/_node.py +90 -0
- pyfect/config_provider/_provider.py +118 -0
- pyfect/config_provider/_unknown.py +62 -0
- pyfect/console/__init__.py +63 -0
- pyfect/console/_effects.py +134 -0
- pyfect/console/_live.py +156 -0
- pyfect/console/_service.py +49 -0
- pyfect/console/_table.py +90 -0
- pyfect/context.py +414 -0
- pyfect/cron/__init__.py +30 -0
- pyfect/cron/_model.py +183 -0
- pyfect/cron/_search.py +268 -0
- pyfect/cron/_text.py +202 -0
- pyfect/cron/_wall.py +42 -0
- pyfect/crypto.py +257 -0
- pyfect/date_time/__init__.py +77 -0
- pyfect/date_time/_calendar.py +209 -0
- pyfect/date_time/_current.py +118 -0
- pyfect/date_time/_epoch.py +22 -0
- pyfect/date_time/_iso.py +80 -0
- pyfect/date_time/_zones.py +135 -0
- pyfect/date_time/intl.py +144 -0
- pyfect/deferred.py +145 -0
- pyfect/devtools/__init__.py +24 -0
- pyfect/devtools/client.py +323 -0
- pyfect/devtools/layers.py +32 -0
- pyfect/devtools/schema.py +301 -0
- pyfect/devtools/server.py +72 -0
- pyfect/duration.py +295 -0
- pyfect/effect.py +565 -0
- pyfect/effectable.py +37 -0
- pyfect/encoding/__init__.py +62 -0
- pyfect/encoding/_archive.py +197 -0
- pyfect/encoding/_json_array_scanner.py +477 -0
- pyfect/encoding/base64.py +23 -0
- pyfect/encoding/base64_url.py +19 -0
- pyfect/encoding/encoding_error.py +17 -0
- pyfect/encoding/hex.py +28 -0
- pyfect/encoding/ini.py +188 -0
- pyfect/encoding/json_array.py +246 -0
- pyfect/encoding/ndjson.py +461 -0
- pyfect/encoding/schema_binary/__init__.py +257 -0
- pyfect/encoding/schema_binary/_channels.py +173 -0
- pyfect/encoding/schema_binary/_compile.py +584 -0
- pyfect/encoding/schema_binary/_decode.py +848 -0
- pyfect/encoding/schema_binary/_dictionary.py +90 -0
- pyfect/encoding/schema_binary/_encode.py +838 -0
- pyfect/encoding/schema_binary/_fingerprint.py +168 -0
- pyfect/encoding/schema_binary/_layout.py +350 -0
- pyfect/encoding/schema_binary/_literals.py +47 -0
- pyfect/encoding/schema_binary/_parser.py +163 -0
- pyfect/encoding/schema_binary/_reader.py +148 -0
- pyfect/encoding/schema_binary/_target.py +310 -0
- pyfect/encoding/schema_binary/_values.py +133 -0
- pyfect/encoding/schema_binary/_wire.py +253 -0
- pyfect/encoding/sse.py +476 -0
- pyfect/encoding/tar/__init__.py +396 -0
- pyfect/encoding/tar/_decoder.py +220 -0
- pyfect/encoding/tar/_encoder.py +83 -0
- pyfect/encoding/tar/_header.py +494 -0
- pyfect/encoding/tar/_model.py +82 -0
- pyfect/encoding/toml.py +375 -0
- pyfect/encoding/xml/__init__.py +222 -0
- pyfect/encoding/xml/_chars.py +119 -0
- pyfect/encoding/xml/_collector.py +106 -0
- pyfect/encoding/xml/_inputs.py +126 -0
- pyfect/encoding/xml/_model.py +114 -0
- pyfect/encoding/xml/_parser.py +694 -0
- pyfect/encoding/xml/_scanner.py +180 -0
- pyfect/encoding/xml/_writer.py +238 -0
- pyfect/encoding/yaml.py +739 -0
- pyfect/encoding/zip/__init__.py +392 -0
- pyfect/encoding/zip/_decoder.py +378 -0
- pyfect/encoding/zip/_format.py +122 -0
- pyfect/encoding/zip/_inflate.py +65 -0
- pyfect/encoding/zip/_model.py +86 -0
- pyfect/encoding/zip/_writer.py +311 -0
- pyfect/error_reporter.py +240 -0
- pyfect/errors.py +321 -0
- pyfect/eventlog/__init__.py +77 -0
- pyfect/eventlog/_conflicts.py +94 -0
- pyfect/eventlog/_crypto.py +64 -0
- pyfect/eventlog/_identity.py +26 -0
- pyfect/eventlog/_identity_derivation.py +79 -0
- pyfect/eventlog/_transferable_bytes.py +55 -0
- pyfect/eventlog/_uuid.py +33 -0
- pyfect/eventlog/event.py +115 -0
- pyfect/eventlog/event_group.py +55 -0
- pyfect/eventlog/event_journal.py +630 -0
- pyfect/eventlog/event_log.py +1039 -0
- pyfect/eventlog/event_log_encryption.py +219 -0
- pyfect/eventlog/event_log_message.py +357 -0
- pyfect/eventlog/event_log_remote.py +465 -0
- pyfect/eventlog/event_log_server.py +442 -0
- pyfect/eventlog/event_log_server_encrypted.py +322 -0
- pyfect/eventlog/event_log_server_unencrypted.py +814 -0
- pyfect/eventlog/event_log_session_auth.py +250 -0
- pyfect/eventlog/event_row.py +423 -0
- pyfect/eventlog/sql_event_journal.py +692 -0
- pyfect/eventlog/sql_event_log_server_encrypted.py +440 -0
- pyfect/eventlog/sql_event_log_server_unencrypted.py +568 -0
- pyfect/execution_plan/__init__.py +42 -0
- pyfect/execution_plan/_events.py +128 -0
- pyfect/execution_plan/_model.py +108 -0
- pyfect/execution_plan/_run.py +176 -0
- pyfect/exit.py +205 -0
- pyfect/fiber_handle.py +305 -0
- pyfect/fiber_map.py +350 -0
- pyfect/fiber_set.py +232 -0
- pyfect/file_system/__init__.py +69 -0
- pyfect/file_system/_copy.py +116 -0
- pyfect/file_system/_errors.py +32 -0
- pyfect/file_system/_file.py +177 -0
- pyfect/file_system/_live.py +351 -0
- pyfect/file_system/_model.py +72 -0
- pyfect/file_system/_noop.py +189 -0
- pyfect/file_system/_service.py +336 -0
- pyfect/file_system/_stat.py +69 -0
- pyfect/file_system/_watch.py +107 -0
- pyfect/filters.py +245 -0
- pyfect/graph/__init__.py +101 -0
- pyfect/graph/_all_pairs.py +150 -0
- pyfect/graph/_combine.py +259 -0
- pyfect/graph/_compact.py +143 -0
- pyfect/graph/_components.py +200 -0
- pyfect/graph/_cycles.py +144 -0
- pyfect/graph/_enumerate.py +168 -0
- pyfect/graph/_errors.py +51 -0
- pyfect/graph/_flow.py +164 -0
- pyfect/graph/_forest.py +109 -0
- pyfect/graph/_graph.py +269 -0
- pyfect/graph/_lowlink.py +98 -0
- pyfect/graph/_matching.py +101 -0
- pyfect/graph/_mutate.py +211 -0
- pyfect/graph/_query.py +138 -0
- pyfect/graph/_records.py +119 -0
- pyfect/graph/_render.py +109 -0
- pyfect/graph/_shortest.py +274 -0
- pyfect/graph/_snapshot.py +50 -0
- pyfect/graph/_store.py +136 -0
- pyfect/graph/_traverse.py +289 -0
- pyfect/graph/_view.py +384 -0
- pyfect/graph/_walker.py +70 -0
- pyfect/graphql/__init__.py +87 -0
- pyfect/graphql/_codec.py +117 -0
- pyfect/graphql/_http.py +309 -0
- pyfect/graphql/_schemas.py +39 -0
- pyfect/graphql/_slices.py +110 -0
- pyfect/graphql/_template.py +111 -0
- pyfect/graphql/_types.py +115 -0
- pyfect/graphql/_ws.py +395 -0
- pyfect/graphql/_ws_messages.py +120 -0
- pyfect/graphql/client.py +428 -0
- pyfect/graphql/client_error.py +285 -0
- pyfect/graphql/group.py +142 -0
- pyfect/graphql/middleware.py +98 -0
- pyfect/graphql/operation.py +409 -0
- pyfect/graphql/protocol.py +119 -0
- pyfect/graphql/resolver.py +574 -0
- pyfect/graphql/template.py +61 -0
- pyfect/hash_map.py +317 -0
- pyfect/hash_ring.py +206 -0
- pyfect/hash_set.py +163 -0
- pyfect/http/__init__.py +125 -0
- pyfect/http/_asgi_listener.py +272 -0
- pyfect/http/_authority.py +63 -0
- pyfect/http/_brotli.py +27 -0
- pyfect/http/_compression.py +52 -0
- pyfect/http/_field.py +18 -0
- pyfect/http/_foreign_path.py +27 -0
- pyfect/http/_form_data.py +34 -0
- pyfect/http/_httpx_options.py +120 -0
- pyfect/http/_in_memory.py +76 -0
- pyfect/http/_request_state.py +59 -0
- pyfect/http/_vary.py +14 -0
- pyfect/http/abort.py +65 -0
- pyfect/http/asgi/__init__.py +59 -0
- pyfect/http/asgi/_app.py +155 -0
- pyfect/http/asgi/_http.py +201 -0
- pyfect/http/asgi/_mount.py +263 -0
- pyfect/http/asgi/_scope.py +86 -0
- pyfect/http/asgi/_serve.py +72 -0
- pyfect/http/asgi/_types.py +13 -0
- pyfect/http/asgi/_web_socket.py +325 -0
- pyfect/http/body.py +437 -0
- pyfect/http/client/__init__.py +111 -0
- pyfect/http/client/_circuit_breaker.py +113 -0
- pyfect/http/client/_cookie_jar.py +120 -0
- pyfect/http/client/_make.py +252 -0
- pyfect/http/client/_rate_limit.py +423 -0
- pyfect/http/client/_redirects.py +93 -0
- pyfect/http/client/_retry.py +115 -0
- pyfect/http/client/_service.py +497 -0
- pyfect/http/cookies.py +444 -0
- pyfect/http/error.py +188 -0
- pyfect/http/etag.py +96 -0
- pyfect/http/file_like.py +49 -0
- pyfect/http/find_my_way/__init__.py +10 -0
- pyfect/http/find_my_way/_nodes.py +186 -0
- pyfect/http/find_my_way/_paths.py +43 -0
- pyfect/http/find_my_way/_query_string.py +67 -0
- pyfect/http/find_my_way/_router.py +401 -0
- pyfect/http/headers.py +343 -0
- pyfect/http/http_effect.py +198 -0
- pyfect/http/httpx_client.py +291 -0
- pyfect/http/hypercorn_server.py +122 -0
- pyfect/http/incoming_message.py +140 -0
- pyfect/http/live_server/__init__.py +191 -0
- pyfect/http/live_server/_body.py +164 -0
- pyfect/http/live_server/_exchange.py +321 -0
- pyfect/http/live_server/_head.py +194 -0
- pyfect/http/live_server/_inbound.py +156 -0
- pyfect/http/live_server/_listener.py +211 -0
- pyfect/http/live_server/_serve.py +225 -0
- pyfect/http/live_server/_timers.py +20 -0
- pyfect/http/live_server/_upgrade.py +108 -0
- pyfect/http/live_server/_websocket.py +286 -0
- pyfect/http/live_server/_ws_frames.py +242 -0
- pyfect/http/method.py +30 -0
- pyfect/http/middleware/__init__.py +61 -0
- pyfect/http/middleware/_compression.py +156 -0
- pyfect/http/middleware/_cors.py +129 -0
- pyfect/http/middleware/_decompression.py +274 -0
- pyfect/http/middleware/_forwarded.py +113 -0
- pyfect/http/middleware/_logger.py +73 -0
- pyfect/http/middleware/_search_params.py +34 -0
- pyfect/http/middleware/_tracer.py +153 -0
- pyfect/http/middleware/_types.py +16 -0
- pyfect/http/mime/__init__.py +69 -0
- pyfect/http/mime/_types.py +361 -0
- pyfect/http/multipart.py +613 -0
- pyfect/http/multipart_parser/__init__.py +348 -0
- pyfect/http/multipart_parser/_content_type.py +109 -0
- pyfect/http/multipart_parser/_headers.py +206 -0
- pyfect/http/multipart_parser/_search.py +85 -0
- pyfect/http/origin.py +146 -0
- pyfect/http/platform.py +593 -0
- pyfect/http/prometheus.py +34 -0
- pyfect/http/request.py +384 -0
- pyfect/http/response.py +474 -0
- pyfect/http/router/__init__.py +82 -0
- pyfect/http/router/_api.py +462 -0
- pyfect/http/router/_in_memory.py +76 -0
- pyfect/http/router/_markers.py +47 -0
- pyfect/http/router/_middleware.py +181 -0
- pyfect/http/router/_middleware_stack.py +51 -0
- pyfect/http/router/_provided.py +31 -0
- pyfect/http/router/_route.py +114 -0
- pyfect/http/router/_router.py +308 -0
- pyfect/http/schema.py +276 -0
- pyfect/http/server.py +199 -0
- pyfect/http/server_error.py +261 -0
- pyfect/http/server_request.py +729 -0
- pyfect/http/server_respondable.py +77 -0
- pyfect/http/server_response.py +569 -0
- pyfect/http/static_server/__init__.py +257 -0
- pyfect/http/static_server/_conditional.py +72 -0
- pyfect/http/static_server/_mime.py +55 -0
- pyfect/http/static_server/_range.py +53 -0
- pyfect/http/status.py +51 -0
- pyfect/http/template.py +161 -0
- pyfect/http/trace_context.py +91 -0
- pyfect/http/url.py +442 -0
- pyfect/http/url_params.py +304 -0
- pyfect/http/uvicorn_server.py +119 -0
- pyfect/http/web.py +171 -0
- pyfect/http/web_handler.py +463 -0
- pyfect/http/wire.py +209 -0
- pyfect/http_api/__init__.py +83 -0
- pyfect/http_api/_assets/scalar.js +8 -0
- pyfect/http_api/_assets/swagger-ui.css +3 -0
- pyfect/http_api/_assets/swagger-ui.js +4 -0
- pyfect/http_api/_client_method.py +220 -0
- pyfect/http_api/_client_payload.py +144 -0
- pyfect/http_api/_client_response.py +377 -0
- pyfect/http_api/_handlers.py +282 -0
- pyfect/http_api/_html.py +38 -0
- pyfect/http_api/_media_type.py +6 -0
- pyfect/http_api/_openapi_responses.py +164 -0
- pyfect/http_api/_openapi_schemas.py +163 -0
- pyfect/http_api/_path.py +102 -0
- pyfect/http_api/_payload.py +260 -0
- pyfect/http_api/_response.py +363 -0
- pyfect/http_api/_security.py +174 -0
- pyfect/http_api/api.py +265 -0
- pyfect/http_api/builder.py +447 -0
- pyfect/http_api/client.py +474 -0
- pyfect/http_api/endpoint.py +969 -0
- pyfect/http_api/error.py +280 -0
- pyfect/http_api/group.py +215 -0
- pyfect/http_api/middleware.py +232 -0
- pyfect/http_api/openapi.py +396 -0
- pyfect/http_api/scalar.py +143 -0
- pyfect/http_api/schema.py +713 -0
- pyfect/http_api/security.py +91 -0
- pyfect/http_api/swagger.py +69 -0
- pyfect/http_api/test.py +105 -0
- pyfect/json_patch.py +255 -0
- pyfect/json_pointer.py +100 -0
- pyfect/json_schema/__init__.py +66 -0
- pyfect/json_schema/_api.py +96 -0
- pyfect/json_schema/_converter.py +258 -0
- pyfect/json_schema/_draft04.py +198 -0
- pyfect/json_schema/_draft07.py +203 -0
- pyfect/json_schema/_keywords.py +202 -0
- pyfect/json_schema/_model.py +201 -0
- pyfect/json_schema/_openapi.py +208 -0
- pyfect/latch.py +127 -0
- pyfect/layer.py +479 -0
- pyfect/layer_map.py +279 -0
- pyfect/layer_ref.py +208 -0
- pyfect/logger/__init__.py +91 -0
- pyfect/logger/_batched.py +49 -0
- pyfect/logger/_console.py +116 -0
- pyfect/logger/_file.py +31 -0
- pyfect/logger/_formats.py +104 -0
- pyfect/logger/_layer.py +50 -0
- pyfect/logger/_level.py +53 -0
- pyfect/logger/_log.py +142 -0
- pyfect/logger/_model.py +163 -0
- pyfect/logger/_refs.py +78 -0
- pyfect/logger/_stdlib.py +54 -0
- pyfect/logger/_text.py +263 -0
- pyfect/logger/_tracer.py +35 -0
- pyfect/managed_runtime.py +393 -0
- pyfect/metric/__init__.py +119 -0
- pyfect/metric/_boundaries.py +27 -0
- pyfect/metric/_kinds.py +417 -0
- pyfect/metric/_metric.py +302 -0
- pyfect/metric/_prometheus.py +150 -0
- pyfect/metric/_refs.py +78 -0
- pyfect/metric/_runtime.py +75 -0
- pyfect/metric/_snapshot.py +97 -0
- pyfect/metric/_state.py +74 -0
- pyfect/msgspec.py +98 -0
- pyfect/mutable_hash_map.py +140 -0
- pyfect/mutable_hash_set.py +95 -0
- pyfect/net/__init__.py +18 -0
- pyfect/net/_errors.py +29 -0
- pyfect/net/address_resolver.py +179 -0
- pyfect/net/dns/__init__.py +92 -0
- pyfect/net/dns/_error.py +43 -0
- pyfect/net/dns/_records.py +335 -0
- pyfect/net/dns/_service.py +287 -0
- pyfect/net/dns/_system.py +89 -0
- pyfect/net/host.py +311 -0
- pyfect/net/ip_interface.py +202 -0
- pyfect/net/ip_network.py +250 -0
- pyfect/net/net_address/__init__.py +245 -0
- pyfect/net/net_address/_inet.py +412 -0
- pyfect/net/net_address/_ip.py +348 -0
- pyfect/net/net_address/_mac.py +119 -0
- pyfect/optic/__init__.py +50 -0
- pyfect/optic/_builtins.py +90 -0
- pyfect/optic/_data.py +97 -0
- pyfect/optic/_optic.py +422 -0
- pyfect/optic/_steps.py +151 -0
- pyfect/option.py +362 -0
- pyfect/otel/__init__.py +22 -0
- pyfect/otel/_attributes.py +67 -0
- pyfect/otel/_producer.py +322 -0
- pyfect/otel/_shutdown.py +26 -0
- pyfect/otel/_span.py +292 -0
- pyfect/otel/_wrapper.py +92 -0
- pyfect/otel/logger.py +154 -0
- pyfect/otel/metrics.py +126 -0
- pyfect/otel/resource.py +111 -0
- pyfect/otel/sdk.py +195 -0
- pyfect/otel/tracer.py +157 -0
- pyfect/partitioned_semaphore.py +211 -0
- pyfect/path.py +304 -0
- pyfect/persistence/__init__.py +77 -0
- pyfect/persistence/_columns.py +32 -0
- pyfect/persistence/_error.py +14 -0
- pyfect/persistence/_seconds.py +68 -0
- pyfect/persistence/_sql_keys.py +47 -0
- pyfect/persistence/circuit_breaker/__init__.py +120 -0
- pyfect/persistence/circuit_breaker/_breaker.py +317 -0
- pyfect/persistence/circuit_breaker/_errors.py +67 -0
- pyfect/persistence/circuit_breaker/_memory.py +243 -0
- pyfect/persistence/circuit_breaker/_redis.py +509 -0
- pyfect/persistence/circuit_breaker/_store.py +60 -0
- pyfect/persistence/circuit_breaker/_types.py +191 -0
- pyfect/persistence/key_value_store/__init__.py +51 -0
- pyfect/persistence/key_value_store/_layers.py +216 -0
- pyfect/persistence/key_value_store/_schema_store.py +91 -0
- pyfect/persistence/key_value_store/_sql.py +197 -0
- pyfect/persistence/key_value_store/_store.py +212 -0
- pyfect/persistence/persistable.py +164 -0
- pyfect/persistence/persisted_cache.py +159 -0
- pyfect/persistence/persisted_queue/__init__.py +75 -0
- pyfect/persistence/persisted_queue/_error.py +42 -0
- pyfect/persistence/persisted_queue/_mailbox.py +101 -0
- pyfect/persistence/persisted_queue/_memory.py +182 -0
- pyfect/persistence/persisted_queue/_queue.py +272 -0
- pyfect/persistence/persisted_queue/_redis.py +455 -0
- pyfect/persistence/persisted_queue/_redis_scripts.py +395 -0
- pyfect/persistence/persisted_queue/_sql.py +334 -0
- pyfect/persistence/persisted_queue/_sql_migrations.py +319 -0
- pyfect/persistence/persisted_queue/_sql_statements.py +441 -0
- pyfect/persistence/persisted_queue/_store.py +147 -0
- pyfect/persistence/persistence/__init__.py +84 -0
- pyfect/persistence/persistence/_kvs.py +135 -0
- pyfect/persistence/persistence/_memory.py +62 -0
- pyfect/persistence/persistence/_redis.py +162 -0
- pyfect/persistence/persistence/_service.py +276 -0
- pyfect/persistence/persistence/_sql.py +499 -0
- pyfect/persistence/rate_limiter/__init__.py +82 -0
- pyfect/persistence/rate_limiter/_errors.py +59 -0
- pyfect/persistence/rate_limiter/_limiter.py +325 -0
- pyfect/persistence/rate_limiter/_memory.py +217 -0
- pyfect/persistence/rate_limiter/_redis.py +485 -0
- pyfect/persistence/rate_limiter/_store.py +89 -0
- pyfect/persistence/redis/__init__.py +52 -0
- pyfect/persistence/redis/_error.py +12 -0
- pyfect/persistence/redis/_make.py +234 -0
- pyfect/persistence/redis/_script.py +78 -0
- pyfect/persistence/redis/_service.py +104 -0
- pyfect/persistence/redis_py.py +303 -0
- pyfect/persistence/user_directories.py +195 -0
- pyfect/platform.py +26 -0
- pyfect/platform_error.py +256 -0
- pyfect/pool/__init__.py +36 -0
- pyfect/pool/_pool.py +707 -0
- pyfect/pool/_state.py +161 -0
- pyfect/pool/_ttl.py +105 -0
- pyfect/pool/_waiters.py +116 -0
- pyfect/primary_key.py +35 -0
- pyfect/process/__init__.py +82 -0
- pyfect/process/_child.py +213 -0
- pyfect/process/_command.py +358 -0
- pyfect/process/_handle.py +77 -0
- pyfect/process/_live.py +455 -0
- pyfect/process/_os.py +78 -0
- pyfect/process/_pipes.py +291 -0
- pyfect/process/_spawner.py +75 -0
- pyfect/pubsub/__init__.py +33 -0
- pyfect/pubsub/_atomic.py +181 -0
- pyfect/pubsub/_hub.py +214 -0
- pyfect/pubsub/_pubsub.py +214 -0
- pyfect/pubsub/_replay.py +107 -0
- pyfect/pubsub/_subscription.py +226 -0
- pyfect/pull.py +132 -0
- pyfect/py.typed +0 -0
- pyfect/pydantic.py +188 -0
- pyfect/queue/__init__.py +33 -0
- pyfect/queue/_queue.py +525 -0
- pyfect/queue/_state.py +111 -0
- pyfect/queue/_views.py +55 -0
- pyfect/random.py +146 -0
- pyfect/rc_map.py +363 -0
- pyfect/rc_ref.py +271 -0
- pyfect/reactivity/__init__.py +53 -0
- pyfect/reactivity/_async_result_schema.py +227 -0
- pyfect/reactivity/_hydration.py +18 -0
- pyfect/reactivity/_live.py +21 -0
- pyfect/reactivity/_reactivity.py +256 -0
- pyfect/reactivity/_result_effect.py +17 -0
- pyfect/reactivity/_time.py +11 -0
- pyfect/reactivity/async_result.py +679 -0
- pyfect/reactivity/atom/__init__.py +143 -0
- pyfect/reactivity/atom/_combinators.py +223 -0
- pyfect/reactivity/atom/_constructors.py +80 -0
- pyfect/reactivity/atom/_conversions.py +79 -0
- pyfect/reactivity/atom/_effect.py +252 -0
- pyfect/reactivity/atom/_fn.py +341 -0
- pyfect/reactivity/atom/_kvs.py +101 -0
- pyfect/reactivity/atom/_optimistic.py +162 -0
- pyfect/reactivity/atom/_runtime.py +323 -0
- pyfect/reactivity/atom/_serializable.py +65 -0
- pyfect/reactivity/atom/_subscription_ref.py +102 -0
- pyfect/reactivity/atom/_swr.py +69 -0
- pyfect/reactivity/atom/_types.py +435 -0
- pyfect/reactivity/atom_http_api.py +667 -0
- pyfect/reactivity/atom_ref.py +260 -0
- pyfect/reactivity/atom_registry/__init__.py +219 -0
- pyfect/reactivity/atom_registry/_batch.py +96 -0
- pyfect/reactivity/atom_registry/_dispatcher.py +60 -0
- pyfect/reactivity/atom_registry/_lifetime.py +202 -0
- pyfect/reactivity/atom_registry/_node.py +423 -0
- pyfect/reactivity/atom_registry/_registry.py +380 -0
- pyfect/reactivity/atom_rpc.py +283 -0
- pyfect/reactivity/hydration.py +200 -0
- pyfect/redacted.py +83 -0
- pyfect/ref.py +129 -0
- pyfect/resource.py +98 -0
- pyfect/result.py +306 -0
- pyfect/rpc/__init__.py +64 -0
- pyfect/rpc/_encoded_schema.py +86 -0
- pyfect/rpc/_json_rpc.py +246 -0
- pyfect/rpc/_methods.py +177 -0
- pyfect/rpc/_rpc.py +498 -0
- pyfect/rpc/client/__init__.py +100 -0
- pyfect/rpc/client/_core.py +449 -0
- pyfect/rpc/client/_http.py +257 -0
- pyfect/rpc/client/_protocol.py +171 -0
- pyfect/rpc/client/_serialized.py +321 -0
- pyfect/rpc/client/_transports.py +328 -0
- pyfect/rpc/client/_worker.py +283 -0
- pyfect/rpc/client_error.py +35 -0
- pyfect/rpc/group.py +296 -0
- pyfect/rpc/message.py +299 -0
- pyfect/rpc/middleware.py +182 -0
- pyfect/rpc/schema.py +89 -0
- pyfect/rpc/serialization.py +366 -0
- pyfect/rpc/server/__init__.py +181 -0
- pyfect/rpc/server/_core.py +549 -0
- pyfect/rpc/server/_protocol.py +123 -0
- pyfect/rpc/server/_serialized.py +363 -0
- pyfect/rpc/server/_stdio.py +101 -0
- pyfect/rpc/server/_transports.py +427 -0
- pyfect/rpc/server/_worker.py +108 -0
- pyfect/rpc/test.py +74 -0
- pyfect/rpc/worker.py +92 -0
- pyfect/run.py +393 -0
- pyfect/schedule/__init__.py +91 -0
- pyfect/schedule/_combinators.py +168 -0
- pyfect/schedule/_constructors.py +223 -0
- pyfect/schedule/_cron.py +49 -0
- pyfect/schedule/_drive.py +191 -0
- pyfect/schedule/_model.py +405 -0
- pyfect/scheduler.py +575 -0
- pyfect/schema/__init__.py +866 -0
- pyfect/schema/_adapters.py +39 -0
- pyfect/schema/_arbitrary_constraint.py +242 -0
- pyfect/schema/_arbitrary_graph.py +325 -0
- pyfect/schema/_arbitrary_links.py +179 -0
- pyfect/schema/_arbitrary_number.py +112 -0
- pyfect/schema/_arrays.py +287 -0
- pyfect/schema/_assignable.py +118 -0
- pyfect/schema/_boolean_literals.py +31 -0
- pyfect/schema/_checks_number.py +216 -0
- pyfect/schema/_checks_order.py +140 -0
- pyfect/schema/_checks_size.py +277 -0
- pyfect/schema/_checks_string.py +241 -0
- pyfect/schema/_class.py +569 -0
- pyfect/schema/_codec_derivation.py +227 -0
- pyfect/schema/_codec_links.py +221 -0
- pyfect/schema/_codecs.py +316 -0
- pyfect/schema/_collections.py +172 -0
- pyfect/schema/_constructor_default.py +89 -0
- pyfect/schema/_data.py +440 -0
- pyfect/schema/_data_codecs.py +145 -0
- pyfect/schema/_decimal_regexp.py +243 -0
- pyfect/schema/_declare.py +68 -0
- pyfect/schema/_derive.py +292 -0
- pyfect/schema/_duration_codecs.py +27 -0
- pyfect/schema/_ecma_pattern.py +416 -0
- pyfect/schema/_error.py +26 -0
- pyfect/schema/_error_instance.py +222 -0
- pyfect/schema/_graph.py +271 -0
- pyfect/schema/_instances.py +240 -0
- pyfect/schema/_jsonschema/__init__.py +90 -0
- pyfect/schema/_jsonschema/_annotations.py +114 -0
- pyfect/schema/_jsonschema/_checks.py +253 -0
- pyfect/schema/_jsonschema/_compile.py +346 -0
- pyfect/schema/_jsonschema/_exact.py +19 -0
- pyfect/schema/_jsonschema/_merge.py +211 -0
- pyfect/schema/_jsonschema/_objects.py +231 -0
- pyfect/schema/_jsonschema/_references.py +221 -0
- pyfect/schema/_jsonschema/_template.py +69 -0
- pyfect/schema/_literals.py +147 -0
- pyfect/schema/_middleware.py +209 -0
- pyfect/schema/_net.py +522 -0
- pyfect/schema/_options.py +49 -0
- pyfect/schema/_parse.py +351 -0
- pyfect/schema/_parsed_text.py +37 -0
- pyfect/schema/_primitives.py +48 -0
- pyfect/schema/_record.py +41 -0
- pyfect/schema/_refine.py +81 -0
- pyfect/schema/_schema.py +158 -0
- pyfect/schema/_semver.py +144 -0
- pyfect/schema/_strategy/__init__.py +258 -0
- pyfect/schema/_strategy/_big_decimal.py +203 -0
- pyfect/schema/_strategy/_composites.py +119 -0
- pyfect/schema/_strategy/_leaves.py +94 -0
- pyfect/schema/_strategy/_net.py +133 -0
- pyfect/schema/_struct.py +87 -0
- pyfect/schema/_struct_keys.py +128 -0
- pyfect/schema/_tagged.py +254 -0
- pyfect/schema/_template_literal.py +172 -0
- pyfect/schema/_to_codec_array_from_single.py +115 -0
- pyfect/schema/_to_codec_go_json.py +174 -0
- pyfect/schema/_to_codec_json.py +73 -0
- pyfect/schema/_to_codec_string_tree.py +115 -0
- pyfect/schema/_to_differ_json_patch.py +44 -0
- pyfect/schema/_to_encoder_xml.py +119 -0
- pyfect/schema/_to_equivalence.py +186 -0
- pyfect/schema/_to_formatter.py +180 -0
- pyfect/schema/_to_iso.py +73 -0
- pyfect/schema/_to_strategy.py +31 -0
- pyfect/schema/_transform.py +210 -0
- pyfect/schema/_tree.py +27 -0
- pyfect/schema/_type_walk.py +53 -0
- pyfect/schema/_undefined.py +58 -0
- pyfect/schema/_value_codecs.py +84 -0
- pyfect/schema/_values.py +123 -0
- pyfect/schema/_void.py +86 -0
- pyfect/schema/ast/__init__.py +227 -0
- pyfect/schema/ast/_annotations.py +43 -0
- pyfect/schema/ast/_arrays.py +210 -0
- pyfect/schema/ast/_base.py +295 -0
- pyfect/schema/ast/_candidates.py +309 -0
- pyfect/schema/ast/_check.py +181 -0
- pyfect/schema/ast/_class.py +159 -0
- pyfect/schema/ast/_constructors.py +65 -0
- pyfect/schema/ast/_declaration.py +97 -0
- pyfect/schema/ast/_encoding.py +111 -0
- pyfect/schema/ast/_json.py +129 -0
- pyfect/schema/ast/_keywords.py +125 -0
- pyfect/schema/ast/_literal.py +157 -0
- pyfect/schema/ast/_objects.py +356 -0
- pyfect/schema/ast/_record.py +36 -0
- pyfect/schema/ast/_result.py +116 -0
- pyfect/schema/ast/_steps.py +209 -0
- pyfect/schema/ast/_suspend.py +65 -0
- pyfect/schema/ast/_template_literal.py +256 -0
- pyfect/schema/ast/_union.py +172 -0
- pyfect/schema/getter/__init__.py +136 -0
- pyfect/schema/getter/_binary.py +104 -0
- pyfect/schema/getter/_coerce.py +196 -0
- pyfect/schema/getter/_getter.py +243 -0
- pyfect/schema/getter/_make.py +151 -0
- pyfect/schema/getter/_text.py +141 -0
- pyfect/schema/getter/_tree_record.py +170 -0
- pyfect/schema/issue.py +420 -0
- pyfect/schema/model/__init__.py +120 -0
- pyfect/schema/model/_class.py +245 -0
- pyfect/schema/model/_dates.py +103 -0
- pyfect/schema/model/_fields.py +77 -0
- pyfect/schema/model/_toolkit.py +114 -0
- pyfect/schema/model/_uuid.py +69 -0
- pyfect/schema/representation/__init__.py +424 -0
- pyfect/schema/representation/_builtin_revivers.py +331 -0
- pyfect/schema/representation/_code.py +106 -0
- pyfect/schema/representation/_code_builtins.py +281 -0
- pyfect/schema/representation/_code_document.py +563 -0
- pyfect/schema/representation/_code_sort.py +111 -0
- pyfect/schema/representation/_from.py +278 -0
- pyfect/schema/representation/_import/__init__.py +78 -0
- pyfect/schema/representation/_import/_finalize.py +121 -0
- pyfect/schema/representation/_import/_intersect.py +467 -0
- pyfect/schema/representation/_import/_js_order.py +66 -0
- pyfect/schema/representation/_import/_keywords.py +224 -0
- pyfect/schema/representation/_import/_nodes.py +199 -0
- pyfect/schema/representation/_import/_translate.py +455 -0
- pyfect/schema/representation/_json.py +520 -0
- pyfect/schema/representation/_json_schema.py +178 -0
- pyfect/schema/representation/_model.py +284 -0
- pyfect/schema/representation/_revivers.py +89 -0
- pyfect/schema/representation/_to.py +249 -0
- pyfect/schema/transformation/__init__.py +99 -0
- pyfect/schema/transformation/_builtins.py +314 -0
- pyfect/schema/transformation/_durations.py +150 -0
- pyfect/schema/transformation/_transformation.py +177 -0
- pyfect/schema/variant_schema/__init__.py +69 -0
- pyfect/schema/variant_schema/_class.py +188 -0
- pyfect/schema/variant_schema/_defaults.py +30 -0
- pyfect/schema/variant_schema/_field.py +32 -0
- pyfect/schema/variant_schema/_struct.py +130 -0
- pyfect/schema/variant_schema/_toolkit.py +160 -0
- pyfect/scope.py +374 -0
- pyfect/scoped_ref.py +124 -0
- pyfect/semaphore.py +276 -0
- pyfect/semver.py +403 -0
- pyfect/semver_range.py +420 -0
- pyfect/sink/__init__.py +131 -0
- pyfect/sink/_combinators.py +151 -0
- pyfect/sink/_constructors.py +187 -0
- pyfect/sink/_context.py +87 -0
- pyfect/sink/_folds.py +336 -0
- pyfect/sink/_for_each.py +87 -0
- pyfect/sink/_sink.py +265 -0
- pyfect/sink/_steps.py +34 -0
- pyfect/sink/_taking.py +183 -0
- pyfect/socket/__init__.py +146 -0
- pyfect/socket/_accepted.py +65 -0
- pyfect/socket/_connection.py +384 -0
- pyfect/socket/_errors.py +135 -0
- pyfect/socket/_loop.py +66 -0
- pyfect/socket/_net.py +281 -0
- pyfect/socket/_socket.py +245 -0
- pyfect/socket/_tls.py +88 -0
- pyfect/socket/_tls_session.py +117 -0
- pyfect/socket/_web_socket.py +555 -0
- pyfect/socket/datagram/__init__.py +110 -0
- pyfect/socket/datagram/_core.py +460 -0
- pyfect/socket/datagram/_errors.py +102 -0
- pyfect/socket/datagram/_live.py +675 -0
- pyfect/socket/datagram/_model.py +217 -0
- pyfect/socket/server.py +258 -0
- pyfect/socket/ws.py +484 -0
- pyfect/sql/__init__.py +60 -0
- pyfect/sql/_loop.py +134 -0
- pyfect/sql/clickhouse/__init__.py +74 -0
- pyfect/sql/clickhouse/_connection.py +237 -0
- pyfect/sql/clickhouse/_rows.py +87 -0
- pyfect/sql/clickhouse/client.py +322 -0
- pyfect/sql/clickhouse/config.py +241 -0
- pyfect/sql/clickhouse/driver.py +434 -0
- pyfect/sql/clickhouse/error.py +106 -0
- pyfect/sql/clickhouse/migrator.py +47 -0
- pyfect/sql/clickhouse/params.py +168 -0
- pyfect/sql/client/__init__.py +35 -0
- pyfect/sql/client/_client.py +309 -0
- pyfect/sql/client/_sqlite_acquirers.py +130 -0
- pyfect/sql/client/_transaction.py +410 -0
- pyfect/sql/connection.py +87 -0
- pyfect/sql/d1/__init__.py +48 -0
- pyfect/sql/d1/api.py +270 -0
- pyfect/sql/d1/client.py +344 -0
- pyfect/sql/d1/config.py +55 -0
- pyfect/sql/error.py +311 -0
- pyfect/sql/libsql/__init__.py +58 -0
- pyfect/sql/libsql/_connection.py +110 -0
- pyfect/sql/libsql/_hrana.py +258 -0
- pyfect/sql/libsql/_local.py +251 -0
- pyfect/sql/libsql/_targets.py +28 -0
- pyfect/sql/libsql/_values.py +46 -0
- pyfect/sql/libsql/client.py +225 -0
- pyfect/sql/libsql/config.py +170 -0
- pyfect/sql/libsql/migrator.py +43 -0
- pyfect/sql/libsql/sdk.py +123 -0
- pyfect/sql/migrator.py +434 -0
- pyfect/sql/model.py +324 -0
- pyfect/sql/mssql/__init__.py +79 -0
- pyfect/sql/mssql/_connection.py +169 -0
- pyfect/sql/mssql/_driver.py +103 -0
- pyfect/sql/mssql/_names.py +9 -0
- pyfect/sql/mssql/_pool.py +125 -0
- pyfect/sql/mssql/_tds.py +447 -0
- pyfect/sql/mssql/_tds_params.py +155 -0
- pyfect/sql/mssql/_worker.py +155 -0
- pyfect/sql/mssql/client.py +344 -0
- pyfect/sql/mssql/config.py +118 -0
- pyfect/sql/mssql/error.py +111 -0
- pyfect/sql/mssql/migrator.py +43 -0
- pyfect/sql/mssql/parameter.py +44 -0
- pyfect/sql/mssql/procedure.py +119 -0
- pyfect/sql/mssql/types.py +137 -0
- pyfect/sql/mysql/__init__.py +60 -0
- pyfect/sql/mysql/_asyncmy.py +168 -0
- pyfect/sql/mysql/_bind.py +76 -0
- pyfect/sql/mysql/_connection.py +156 -0
- pyfect/sql/mysql/_pool.py +63 -0
- pyfect/sql/mysql/_session.py +262 -0
- pyfect/sql/mysql/client.py +244 -0
- pyfect/sql/mysql/config.py +108 -0
- pyfect/sql/mysql/error.py +104 -0
- pyfect/sql/mysql/migrator.py +44 -0
- pyfect/sql/pg/__init__.py +82 -0
- pyfect/sql/pg/_resolve.py +284 -0
- pyfect/sql/pg/_session.py +380 -0
- pyfect/sql/pg/client.py +398 -0
- pyfect/sql/pg/config.py +142 -0
- pyfect/sql/pg/connection.py +585 -0
- pyfect/sql/pg/error.py +131 -0
- pyfect/sql/pg/migrator.py +139 -0
- pyfect/sql/pg/pool.py +265 -0
- pyfect/sql/resolver.py +294 -0
- pyfect/sql/result.py +24 -0
- pyfect/sql/schema.py +116 -0
- pyfect/sql/sqlite/__init__.py +26 -0
- pyfect/sql/sqlite/_connection.py +230 -0
- pyfect/sql/sqlite/_worker.py +71 -0
- pyfect/sql/sqlite/client.py +248 -0
- pyfect/sql/sqlite/migrator.py +43 -0
- pyfect/sql/statement/__init__.py +144 -0
- pyfect/sql/statement/_compiler.py +384 -0
- pyfect/sql/statement/_segments.py +487 -0
- pyfect/sql/statement/_statement.py +526 -0
- pyfect/sql/statement/_transforms.py +93 -0
- pyfect/sql/stream.py +88 -0
- pyfect/stdio.py +206 -0
- pyfect/stream/__init__.py +151 -0
- pyfect/stream/_async_iterable.py +129 -0
- pyfect/stream/_base.py +69 -0
- pyfect/stream/_bridge_methods.py +139 -0
- pyfect/stream/_catch.py +135 -0
- pyfect/stream/_catch_reason.py +558 -0
- pyfect/stream/_channel_methods.py +61 -0
- pyfect/stream/_chunk_methods.py +169 -0
- pyfect/stream/_chunking.py +310 -0
- pyfect/stream/_concat.py +65 -0
- pyfect/stream/_constructors.py +341 -0
- pyfect/stream/_context.py +147 -0
- pyfect/stream/_error_methods.py +254 -0
- pyfect/stream/_execution_plan.py +199 -0
- pyfect/stream/_fanout.py +154 -0
- pyfect/stream/_fanout_methods.py +183 -0
- pyfect/stream/_filter_methods.py +203 -0
- pyfect/stream/_grouping.py +168 -0
- pyfect/stream/_many.py +65 -0
- pyfect/stream/_map_methods.py +233 -0
- pyfect/stream/_mapping.py +126 -0
- pyfect/stream/_run.py +114 -0
- pyfect/stream/_run_methods.py +114 -0
- pyfect/stream/_service_methods.py +36 -0
- pyfect/stream/_sink_methods.py +151 -0
- pyfect/stream/_sinking.py +277 -0
- pyfect/stream/_sources.py +124 -0
- pyfect/stream/_stream.py +162 -0
- pyfect/stream/_taking.py +330 -0
- pyfect/stream/_text.py +43 -0
- pyfect/stream/_timing.py +294 -0
- pyfect/stream/_timing_methods.py +252 -0
- pyfect/stream/_zip_methods.py +279 -0
- pyfect/stream/_zipping.py +213 -0
- pyfect/subscription_ref.py +66 -0
- pyfect/synchronized_ref.py +38 -0
- pyfect/take.py +36 -0
- pyfect/terminal/__init__.py +41 -0
- pyfect/terminal/_keys.py +113 -0
- pyfect/terminal/_live.py +223 -0
- pyfect/terminal/_service.py +73 -0
- pyfect/testing/__init__.py +122 -0
- pyfect/testing/_assertions.py +92 -0
- pyfect/testing/_clock.py +309 -0
- pyfect/testing/_console.py +164 -0
- pyfect/testing/_crypto.py +44 -0
- pyfect/testing/_decorators.py +211 -0
- pyfect/testing/_flaky.py +34 -0
- pyfect/testing/_layers.py +153 -0
- pyfect/testing/_prop.py +255 -0
- pyfect/testing/_runners.py +84 -0
- pyfect/testing/_runtime.py +174 -0
- pyfect/testing/_schema.py +238 -0
- pyfect/testing/plugin.py +59 -0
- pyfect/tracer/__init__.py +107 -0
- pyfect/tracer/_model.py +176 -0
- pyfect/tracer/_native.py +172 -0
- pyfect/tracer/_refs.py +87 -0
- pyfect/tracer/_spans.py +269 -0
- pyfect/transaction.py +340 -0
- pyfect/trie.py +317 -0
- pyfect/tx_chunk.py +127 -0
- pyfect/tx_deferred.py +75 -0
- pyfect/tx_hash_map.py +265 -0
- pyfect/tx_hash_set.py +151 -0
- pyfect/tx_priority_queue.py +224 -0
- pyfect/tx_pubsub.py +194 -0
- pyfect/tx_queue/__init__.py +27 -0
- pyfect/tx_queue/_queue.py +359 -0
- pyfect/tx_queue/_views.py +60 -0
- pyfect/tx_reentrant_lock.py +201 -0
- pyfect/tx_ref.py +79 -0
- pyfect/tx_semaphore.py +136 -0
- pyfect/tx_subscription_ref.py +118 -0
- pyfect/workers/__init__.py +62 -0
- pyfect/workers/_frames.py +82 -0
- pyfect/workers/_mailbox.py +51 -0
- pyfect/workers/_port.py +106 -0
- pyfect/workers/_process.py +155 -0
- pyfect/workers/_thread.py +100 -0
- pyfect/workers/python_worker.py +133 -0
- pyfect/workers/python_worker_runner.py +112 -0
- pyfect/workers/transferable.py +103 -0
- pyfect/workers/worker.py +273 -0
- pyfect/workers/worker_error.py +86 -0
- pyfect/workers/worker_runner.py +85 -0
- pyfect/workflow/__init__.py +78 -0
- pyfect/workflow/_crypto.py +23 -0
- pyfect/workflow/_encoded.py +381 -0
- pyfect/workflow/_memory.py +424 -0
- pyfect/workflow/_result.py +43 -0
- pyfect/workflow/_services.py +241 -0
- pyfect/workflow/_suspension.py +153 -0
- pyfect/workflow/activity.py +427 -0
- pyfect/workflow/durable_clock.py +72 -0
- pyfect/workflow/durable_deferred.py +403 -0
- pyfect/workflow/durable_queue.py +259 -0
- pyfect/workflow/workflow.py +509 -0
- pyfect/workflow/workflow_engine.py +42 -0
- pyfect/workflow/workflow_proxy.py +161 -0
- pyfect/workflow/workflow_proxy_server.py +146 -0
- pyfect_dev-0.1.0.dist-info/METADATA +296 -0
- pyfect_dev-0.1.0.dist-info/RECORD +1265 -0
- pyfect_dev-0.1.0.dist-info/WHEEL +4 -0
- pyfect_dev-0.1.0.dist-info/entry_points.txt +3 -0
- pyfect_dev-0.1.0.dist-info/licenses/LICENSE +21 -0
pyfect/pool/_pool.py
ADDED
|
@@ -0,0 +1,707 @@
|
|
|
1
|
+
"""``Pool[A, E]``: borrow, invalidate and reserve pooled items."""
|
|
2
|
+
# pyright: reportPrivateUsage=false
|
|
3
|
+
|
|
4
|
+
from __future__ import annotations
|
|
5
|
+
|
|
6
|
+
import math
|
|
7
|
+
from collections.abc import Callable, Generator
|
|
8
|
+
from typing import Any, Final, Literal, Never, overload
|
|
9
|
+
|
|
10
|
+
from pyfect._core import (
|
|
11
|
+
Effect,
|
|
12
|
+
Frame,
|
|
13
|
+
Restore,
|
|
14
|
+
Step,
|
|
15
|
+
_Callback,
|
|
16
|
+
_FailCause,
|
|
17
|
+
_FlatMap,
|
|
18
|
+
_FromExit,
|
|
19
|
+
_Gen,
|
|
20
|
+
_Locally,
|
|
21
|
+
_SetInterruptible,
|
|
22
|
+
_Succeed,
|
|
23
|
+
_Suspend,
|
|
24
|
+
_Sync,
|
|
25
|
+
_WithFiber,
|
|
26
|
+
uninterruptible_mask,
|
|
27
|
+
)
|
|
28
|
+
from pyfect._effect_args import require_effect
|
|
29
|
+
from pyfect._fiber import Fiber
|
|
30
|
+
from pyfect._interrupt_reason import closed
|
|
31
|
+
from pyfect._unsubscripted import Unsubscripted
|
|
32
|
+
from pyfect.cause import Cause
|
|
33
|
+
from pyfect.cause import interrupt as interrupt_cause
|
|
34
|
+
from pyfect.clock import Clock, ClockService
|
|
35
|
+
from pyfect.context import Context
|
|
36
|
+
from pyfect.duration import Duration, to_seconds
|
|
37
|
+
from pyfect.exit import Exit, Failure, Success
|
|
38
|
+
from pyfect.logger._log import log_unhandled
|
|
39
|
+
from pyfect.pool._state import PoolConfig, PoolItem, PoolState, Strategy
|
|
40
|
+
from pyfect.pool._waiters import Waiter
|
|
41
|
+
from pyfect.scope import CloseableScope, Scope, acquire_release, fork_in
|
|
42
|
+
from pyfect.semaphore import Semaphore
|
|
43
|
+
|
|
44
|
+
_VOID: Final[Effect[None]] = _Succeed(None)
|
|
45
|
+
|
|
46
|
+
type TimeToLiveStrategy = Literal["creation", "usage"]
|
|
47
|
+
|
|
48
|
+
# Hands a borrower an item it may use: ``(item, value, fiber, restore)``.
|
|
49
|
+
type _Lease[A, E, X, R] = Callable[[PoolItem[A, E], A, Fiber[Any, Any], Restore], Effect[X, Any, R]]
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
class Pool[A = Unsubscripted, E = Never]:
|
|
53
|
+
"""Items acquired with a scoped effect and lent to fibers. Make one with
|
|
54
|
+
``Pool.make`` / ``make_with_ttl`` / ``make_with_strategy`` inside a
|
|
55
|
+
scope; closing that scope shuts the pool down, waiting for borrowed
|
|
56
|
+
items to come back before releasing everything."""
|
|
57
|
+
|
|
58
|
+
__slots__ = ("_acquire", "_clock", "_context", "config", "state")
|
|
59
|
+
|
|
60
|
+
def __init__(
|
|
61
|
+
self,
|
|
62
|
+
acquire: Effect[A, E, Any],
|
|
63
|
+
context: Context,
|
|
64
|
+
clock: ClockService,
|
|
65
|
+
config: PoolConfig[A, E],
|
|
66
|
+
state: PoolState[A, E],
|
|
67
|
+
) -> None:
|
|
68
|
+
self._acquire = acquire
|
|
69
|
+
self._context = context
|
|
70
|
+
self._clock = clock
|
|
71
|
+
self.config = config
|
|
72
|
+
self.state = state
|
|
73
|
+
|
|
74
|
+
# --- constructors ---------------------------------------------------------
|
|
75
|
+
|
|
76
|
+
@overload
|
|
77
|
+
@classmethod
|
|
78
|
+
def make[T, X = Never, R = Never](
|
|
79
|
+
cls: type[Pool[Unsubscripted, Any]],
|
|
80
|
+
acquire: Effect[T, X, R],
|
|
81
|
+
*,
|
|
82
|
+
size: int,
|
|
83
|
+
concurrency: int = 1,
|
|
84
|
+
target_utilization: float = 1.0,
|
|
85
|
+
) -> Effect[Pool[T, X], Never, Scope | R]: ...
|
|
86
|
+
@overload
|
|
87
|
+
@classmethod
|
|
88
|
+
def make[R = Never](
|
|
89
|
+
cls,
|
|
90
|
+
acquire: Effect[A, E, R],
|
|
91
|
+
*,
|
|
92
|
+
size: int,
|
|
93
|
+
concurrency: int = 1,
|
|
94
|
+
target_utilization: float = 1.0,
|
|
95
|
+
) -> Effect[Pool[A, E], Never, Scope | R]: ...
|
|
96
|
+
@classmethod
|
|
97
|
+
def make[T, X = Never, R = Never](
|
|
98
|
+
cls,
|
|
99
|
+
acquire: Effect[T, X, R],
|
|
100
|
+
*,
|
|
101
|
+
size: int,
|
|
102
|
+
concurrency: int = 1,
|
|
103
|
+
target_utilization: float = 1.0,
|
|
104
|
+
) -> Effect[Pool[T, X], Never, Scope | R]:
|
|
105
|
+
"""A pool of ``size`` items, acquired in the background at once.
|
|
106
|
+
Each item serves up to ``concurrency`` borrowers at a time."""
|
|
107
|
+
return Pool.make_with_strategy(
|
|
108
|
+
acquire,
|
|
109
|
+
min=size,
|
|
110
|
+
max=size,
|
|
111
|
+
strategy=_NOOP,
|
|
112
|
+
concurrency=concurrency,
|
|
113
|
+
target_utilization=target_utilization,
|
|
114
|
+
)
|
|
115
|
+
|
|
116
|
+
@overload
|
|
117
|
+
@classmethod
|
|
118
|
+
def make_with_ttl[T, X = Never, R = Never](
|
|
119
|
+
cls: type[Pool[Unsubscripted, Any]],
|
|
120
|
+
acquire: Effect[T, X, R],
|
|
121
|
+
*,
|
|
122
|
+
min: int,
|
|
123
|
+
max: int,
|
|
124
|
+
time_to_live: Duration,
|
|
125
|
+
time_to_live_strategy: TimeToLiveStrategy = "usage",
|
|
126
|
+
concurrency: int = 1,
|
|
127
|
+
target_utilization: float = 1.0,
|
|
128
|
+
) -> Effect[Pool[T, X], Never, Scope | R]: ...
|
|
129
|
+
@overload
|
|
130
|
+
@classmethod
|
|
131
|
+
def make_with_ttl[R = Never](
|
|
132
|
+
cls,
|
|
133
|
+
acquire: Effect[A, E, R],
|
|
134
|
+
*,
|
|
135
|
+
min: int,
|
|
136
|
+
max: int,
|
|
137
|
+
time_to_live: Duration,
|
|
138
|
+
time_to_live_strategy: TimeToLiveStrategy = "usage",
|
|
139
|
+
concurrency: int = 1,
|
|
140
|
+
target_utilization: float = 1.0,
|
|
141
|
+
) -> Effect[Pool[A, E], Never, Scope | R]: ...
|
|
142
|
+
@classmethod
|
|
143
|
+
def make_with_ttl[T, X = Never, R = Never](
|
|
144
|
+
cls,
|
|
145
|
+
acquire: Effect[T, X, R],
|
|
146
|
+
*,
|
|
147
|
+
min: int,
|
|
148
|
+
max: int,
|
|
149
|
+
time_to_live: Duration,
|
|
150
|
+
time_to_live_strategy: TimeToLiveStrategy = "usage",
|
|
151
|
+
concurrency: int = 1,
|
|
152
|
+
target_utilization: float = 1.0,
|
|
153
|
+
) -> Effect[Pool[T, X], Never, Scope | R]:
|
|
154
|
+
"""A pool between ``min`` and ``max`` items, growing with demand.
|
|
155
|
+
Items above ``min`` are released once they have been idle
|
|
156
|
+
(``"usage"``) or alive (``"creation"``) for ``time_to_live``."""
|
|
157
|
+
from pyfect.pool._ttl import creation_ttl, usage_ttl
|
|
158
|
+
|
|
159
|
+
require_effect(acquire, "acquire")
|
|
160
|
+
ttl = to_seconds(time_to_live)
|
|
161
|
+
make_strategy = creation_ttl if time_to_live_strategy == "creation" else usage_ttl
|
|
162
|
+
strategy: Effect[Strategy[T, X]] = make_strategy(ttl)
|
|
163
|
+
return strategy.flat_map(
|
|
164
|
+
lambda s: Pool.make_with_strategy(
|
|
165
|
+
acquire,
|
|
166
|
+
min=min,
|
|
167
|
+
max=max,
|
|
168
|
+
strategy=s,
|
|
169
|
+
concurrency=concurrency,
|
|
170
|
+
target_utilization=target_utilization,
|
|
171
|
+
)
|
|
172
|
+
)
|
|
173
|
+
|
|
174
|
+
@overload
|
|
175
|
+
@classmethod
|
|
176
|
+
def make_with_strategy[T, X = Never, R = Never](
|
|
177
|
+
cls: type[Pool[Unsubscripted, Any]],
|
|
178
|
+
acquire: Effect[T, X, R],
|
|
179
|
+
*,
|
|
180
|
+
min: int,
|
|
181
|
+
max: int,
|
|
182
|
+
strategy: Strategy[T, X],
|
|
183
|
+
concurrency: int = 1,
|
|
184
|
+
target_utilization: float = 1.0,
|
|
185
|
+
) -> Effect[Pool[T, X], Never, Scope | R]: ...
|
|
186
|
+
@overload
|
|
187
|
+
@classmethod
|
|
188
|
+
def make_with_strategy[R = Never](
|
|
189
|
+
cls,
|
|
190
|
+
acquire: Effect[A, E, R],
|
|
191
|
+
*,
|
|
192
|
+
min: int,
|
|
193
|
+
max: int,
|
|
194
|
+
strategy: Strategy[A, E],
|
|
195
|
+
concurrency: int = 1,
|
|
196
|
+
target_utilization: float = 1.0,
|
|
197
|
+
) -> Effect[Pool[A, E], Never, Scope | R]: ...
|
|
198
|
+
@classmethod
|
|
199
|
+
def make_with_strategy[T, X = Never, R = Never](
|
|
200
|
+
cls,
|
|
201
|
+
acquire: Effect[T, X, R],
|
|
202
|
+
*,
|
|
203
|
+
min: int,
|
|
204
|
+
max: int,
|
|
205
|
+
strategy: Strategy[T, X],
|
|
206
|
+
concurrency: int = 1,
|
|
207
|
+
target_utilization: float = 1.0,
|
|
208
|
+
) -> Effect[Pool[T, X], Never, Scope | R]:
|
|
209
|
+
"""A pool between ``min`` and ``max`` items, resized by
|
|
210
|
+
``strategy``."""
|
|
211
|
+
require_effect(acquire, "acquire")
|
|
212
|
+
bounds = (min, max)
|
|
213
|
+
|
|
214
|
+
def body(restore: Restore) -> Effect[Pool[T, X]]:
|
|
215
|
+
def with_fiber(fiber: Fiber[Any, Any]) -> Effect[Pool[T, X]]:
|
|
216
|
+
context = fiber._context
|
|
217
|
+
scope = context.get_unsafe(Scope)
|
|
218
|
+
config = PoolConfig(
|
|
219
|
+
acquire=_Locally(acquire, context.merge),
|
|
220
|
+
concurrency=concurrency,
|
|
221
|
+
is_fixed=bounds[0] == bounds[1],
|
|
222
|
+
min_size=bounds[0],
|
|
223
|
+
max_size=bounds[1],
|
|
224
|
+
strategy=strategy,
|
|
225
|
+
target_utilization=_clamp(target_utilization, 0.1, 1.0),
|
|
226
|
+
)
|
|
227
|
+
state: PoolState[T, X] = PoolState(scope, Semaphore(1))
|
|
228
|
+
pool = Pool(acquire, context, context.lookup(Clock)[1], config, state)
|
|
229
|
+
return _Gen(lambda: pool._start(scope, restore))
|
|
230
|
+
|
|
231
|
+
return _WithFiber(with_fiber)
|
|
232
|
+
|
|
233
|
+
return uninterruptible_mask(body)
|
|
234
|
+
|
|
235
|
+
def _start(
|
|
236
|
+
self, scope: CloseableScope, restore: Restore
|
|
237
|
+
) -> Generator[Effect[Any, Any, Any], Any, Pool[A, E]]:
|
|
238
|
+
yield from scope.add_finalizer(lambda _: self._shutdown())
|
|
239
|
+
if self.config.min_size > 0:
|
|
240
|
+
resizing = yield from restore(self._resize()).fork_daemon(start_immediately=True)
|
|
241
|
+
yield from scope.add_finalizer(lambda _: resizing.interrupt())
|
|
242
|
+
if self.config.strategy is not _NOOP:
|
|
243
|
+
running = yield from restore(self.config.strategy.run(self)).fork_daemon()
|
|
244
|
+
yield from scope.add_finalizer(lambda _: running.interrupt())
|
|
245
|
+
return self
|
|
246
|
+
|
|
247
|
+
# --- borrowing ------------------------------------------------------------
|
|
248
|
+
|
|
249
|
+
def get(self) -> Effect[A, E, Scope]:
|
|
250
|
+
"""Borrow an item until the current scope closes, waiting for one
|
|
251
|
+
if none is free. Fails with the acquisition's error if acquiring
|
|
252
|
+
the item failed; interrupted once the pool is shutting down."""
|
|
253
|
+
|
|
254
|
+
def body(fiber: Fiber[Any, Any]) -> Effect[A, E, Scope]:
|
|
255
|
+
fast = self._fast_item()
|
|
256
|
+
if fast is not None:
|
|
257
|
+
item, value = fast
|
|
258
|
+
return self._lease_scoped(item, value, fiber)
|
|
259
|
+
return self._get_slow(
|
|
260
|
+
lambda item, value, fiber, _restore: self._lease_scoped(item, value, fiber)
|
|
261
|
+
)
|
|
262
|
+
|
|
263
|
+
return _WithFiber(body)
|
|
264
|
+
|
|
265
|
+
def use[B, E2 = Never, R2 = Never](
|
|
266
|
+
self, f: Callable[[A], Effect[B, E2, R2]]
|
|
267
|
+
) -> Effect[B, E | E2, R2]:
|
|
268
|
+
"""Borrow an item for as long as ``f(item)`` runs."""
|
|
269
|
+
|
|
270
|
+
def body(fiber: Fiber[Any, Any]) -> Effect[B, E | E2, R2]:
|
|
271
|
+
fast = self._fast_item()
|
|
272
|
+
if fast is not None:
|
|
273
|
+
item, value = fast
|
|
274
|
+
return self._use_item(item, value, f, fiber, None)
|
|
275
|
+
return self._get_slow(
|
|
276
|
+
lambda item, value, fiber, restore: self._use_item(item, value, f, fiber, restore)
|
|
277
|
+
)
|
|
278
|
+
|
|
279
|
+
return _WithFiber(body)
|
|
280
|
+
|
|
281
|
+
def invalidate(self, item: A) -> Effect[None]:
|
|
282
|
+
"""Release ``item`` once nobody borrows it, and acquire a
|
|
283
|
+
replacement. The item is found by identity (``is``), as Effect's
|
|
284
|
+
``===``."""
|
|
285
|
+
|
|
286
|
+
def body() -> Effect[None]:
|
|
287
|
+
if self.state.is_shutting_down:
|
|
288
|
+
return _VOID
|
|
289
|
+
pool_item = self._find(item)
|
|
290
|
+
if pool_item is None:
|
|
291
|
+
return _VOID
|
|
292
|
+
pool_item.disable_reclaim = True
|
|
293
|
+
return self._invalidate_item(pool_item).uninterruptible()
|
|
294
|
+
|
|
295
|
+
return _Suspend(body)
|
|
296
|
+
|
|
297
|
+
def reserve(self, item: A) -> Effect[None, Never, Scope]:
|
|
298
|
+
"""Keep ``item`` from new borrowers until the current scope closes,
|
|
299
|
+
counting it as fully used, without extending existing leases. A
|
|
300
|
+
no-op when each item serves one borrower. The item is found by
|
|
301
|
+
identity (``is``)."""
|
|
302
|
+
state = self.state
|
|
303
|
+
concurrency = self.config.concurrency
|
|
304
|
+
|
|
305
|
+
def take() -> PoolItem[A, E] | None:
|
|
306
|
+
if concurrency == 1:
|
|
307
|
+
return None
|
|
308
|
+
pool_item = self._find(item)
|
|
309
|
+
if pool_item is None:
|
|
310
|
+
return None
|
|
311
|
+
if pool_item.reservations == 0:
|
|
312
|
+
state.usage += concurrency - 1
|
|
313
|
+
pool_item.reservations += 1
|
|
314
|
+
state._remove_available(pool_item)
|
|
315
|
+
return pool_item
|
|
316
|
+
|
|
317
|
+
def give_back(pool_item: PoolItem[A, E] | None) -> Effect[None]:
|
|
318
|
+
def run(fiber: Fiber[Any, Any]) -> Effect[None]:
|
|
319
|
+
if pool_item is None:
|
|
320
|
+
return _VOID
|
|
321
|
+
pool_item.reservations -= 1
|
|
322
|
+
if pool_item.reservations > 0:
|
|
323
|
+
return _VOID
|
|
324
|
+
state.usage -= concurrency - 1
|
|
325
|
+
if (
|
|
326
|
+
not state.is_shutting_down
|
|
327
|
+
and pool_item in state.items
|
|
328
|
+
and pool_item not in state.invalidated
|
|
329
|
+
and pool_item.ref_count < concurrency
|
|
330
|
+
):
|
|
331
|
+
state._add_available_front(pool_item)
|
|
332
|
+
self._wake_waiters(fiber, concurrency - pool_item.ref_count)
|
|
333
|
+
return _VOID
|
|
334
|
+
|
|
335
|
+
return _WithFiber(run)
|
|
336
|
+
|
|
337
|
+
return acquire_release(_Sync(take), lambda pool_item, _: give_back(pool_item)).as_void()
|
|
338
|
+
|
|
339
|
+
def _find(self, value: A) -> PoolItem[A, E] | None:
|
|
340
|
+
"""The pool item holding ``value`` itself; an equal value is another item."""
|
|
341
|
+
for pool_item in self.state.items:
|
|
342
|
+
if isinstance(pool_item.exit, Success) and pool_item.exit.value is value:
|
|
343
|
+
return pool_item
|
|
344
|
+
return None
|
|
345
|
+
|
|
346
|
+
# --- leases ---------------------------------------------------------------
|
|
347
|
+
|
|
348
|
+
def _fast_item(self) -> tuple[PoolItem[A, E], A] | None:
|
|
349
|
+
"""The item a borrower takes without waiting, counted in ``usage``,
|
|
350
|
+
if one is free, nobody waits and the pool need not grow first."""
|
|
351
|
+
state = self.state
|
|
352
|
+
head = state.available_head
|
|
353
|
+
if state.is_shutting_down or state.waiters or head is None:
|
|
354
|
+
return None
|
|
355
|
+
if not isinstance(head.exit, Success):
|
|
356
|
+
return None
|
|
357
|
+
state.usage += 1
|
|
358
|
+
if self.config.is_fixed or self._target_size() <= self._active_size():
|
|
359
|
+
return head, head.exit.value
|
|
360
|
+
state.usage -= 1
|
|
361
|
+
return None
|
|
362
|
+
|
|
363
|
+
def _bookkeep(self, item: PoolItem[A, E]) -> None:
|
|
364
|
+
item.ref_count += 1
|
|
365
|
+
if item.ref_count >= self.config.concurrency:
|
|
366
|
+
self.state._remove_available(item)
|
|
367
|
+
|
|
368
|
+
def _lease_scoped(self, item: PoolItem[A, E], value: A, fiber: Fiber[Any, Any]) -> Effect[A]:
|
|
369
|
+
self._bookkeep(item)
|
|
370
|
+
scope = fiber._context.get_unsafe(Scope)
|
|
371
|
+
if scope.is_closed:
|
|
372
|
+
fiber._push(_Release(self._release_item(item)))
|
|
373
|
+
else:
|
|
374
|
+
scope._add(object(), lambda _: self._release_item(item))
|
|
375
|
+
return _Succeed(value)
|
|
376
|
+
|
|
377
|
+
def _use_item[B, E2, R2](
|
|
378
|
+
self,
|
|
379
|
+
item: PoolItem[A, E],
|
|
380
|
+
value: A,
|
|
381
|
+
f: Callable[[A], Effect[B, E2, R2]],
|
|
382
|
+
fiber: Fiber[Any, Any],
|
|
383
|
+
restore: Restore | None,
|
|
384
|
+
) -> Effect[B, E2, R2]:
|
|
385
|
+
# Count the lease and install its release in the same step.
|
|
386
|
+
self._bookkeep(item)
|
|
387
|
+
fiber._push(_Release(self._release_item(item)))
|
|
388
|
+
body: Effect[B, E2, R2] = _Suspend(lambda: f(value))
|
|
389
|
+
return body if restore is None else restore(body)
|
|
390
|
+
|
|
391
|
+
def _get_slow[X, R](self, lease: _Lease[A, E, X, R]) -> Effect[X, Any, R]:
|
|
392
|
+
def body(restore: Restore) -> Effect[X, Any, R]:
|
|
393
|
+
state = self.state
|
|
394
|
+
state.usage += 1
|
|
395
|
+
waiter = Waiter()
|
|
396
|
+
|
|
397
|
+
def leave(fiber: Fiber[Any, Any]) -> Effect[None]:
|
|
398
|
+
state.usage -= 1
|
|
399
|
+
# Only a woken waiter is still registered here: one cancelled
|
|
400
|
+
# while asleep already left. It passes its wake on, so the
|
|
401
|
+
# capacity it was woken for is not stranded.
|
|
402
|
+
if state.waiters.leave(waiter):
|
|
403
|
+
self._wake_waiters(fiber, 1)
|
|
404
|
+
return _VOID
|
|
405
|
+
|
|
406
|
+
wait: Effect[X, Any, R] = (
|
|
407
|
+
restore(self._wait_for_item(waiter))
|
|
408
|
+
.on_error(lambda _: _WithFiber(leave))
|
|
409
|
+
.flat_map(lambda _: loop)
|
|
410
|
+
)
|
|
411
|
+
|
|
412
|
+
def attempt(fiber: Fiber[Any, Any]) -> Effect[X, Any, R]:
|
|
413
|
+
if state.is_shutting_down:
|
|
414
|
+
state.usage -= 1
|
|
415
|
+
return _FailCause(interrupt_cause(fiber.id).annotate(closed("Pool")))
|
|
416
|
+
# A borrower that has not waited yet queues behind those
|
|
417
|
+
# already waiting, who keep their place until they take an
|
|
418
|
+
# item.
|
|
419
|
+
if state.waiters and waiter not in state.waiters:
|
|
420
|
+
return wait
|
|
421
|
+
# Replace failed slots only when the pool cannot grow.
|
|
422
|
+
item = state._first_healthy_available()
|
|
423
|
+
if item is None and self._target_size() <= self._active_size():
|
|
424
|
+
item = state.available_head
|
|
425
|
+
if item is None:
|
|
426
|
+
return wait
|
|
427
|
+
if isinstance(item.exit, Failure):
|
|
428
|
+
state._remove(item)
|
|
429
|
+
return loop
|
|
430
|
+
state.waiters.leave(waiter)
|
|
431
|
+
return lease(item, item.exit.value, fiber, restore)
|
|
432
|
+
|
|
433
|
+
step: Effect[X, Any, R] = _WithFiber(attempt)
|
|
434
|
+
|
|
435
|
+
def grow(fiber: Fiber[Any, Any]) -> Effect[X, Any, R]:
|
|
436
|
+
if state.is_shutting_down:
|
|
437
|
+
state.usage -= 1
|
|
438
|
+
return _FailCause(interrupt_cause(fiber.id).annotate(closed("Pool")))
|
|
439
|
+
if self._target_size() <= self._active_size():
|
|
440
|
+
return step
|
|
441
|
+
resize = fork_in(self._resize().interruptible(), state.scope)
|
|
442
|
+
return state.resize_semaphore.with_permits_if_available(resize).and_then(step)
|
|
443
|
+
|
|
444
|
+
loop: Effect[X, Any, R] = _WithFiber(grow)
|
|
445
|
+
return loop
|
|
446
|
+
|
|
447
|
+
return uninterruptible_mask(body)
|
|
448
|
+
|
|
449
|
+
def _wait_for_item(self, waiter: Waiter) -> Effect[None, Any]:
|
|
450
|
+
state = self.state
|
|
451
|
+
|
|
452
|
+
def register(resume: Callable[[Effect[None, Any]], None]) -> Effect[None] | None:
|
|
453
|
+
if state.is_shutting_down or (
|
|
454
|
+
not state.waiters and state._first_healthy_available() is not None
|
|
455
|
+
):
|
|
456
|
+
resume(_VOID)
|
|
457
|
+
return None
|
|
458
|
+
state.waiters.park(waiter, resume)
|
|
459
|
+
return _Sync(lambda: state.waiters.cancel(waiter))
|
|
460
|
+
|
|
461
|
+
return _Callback(register)
|
|
462
|
+
|
|
463
|
+
def _release_item(self, item: PoolItem[A, E]) -> Effect[None]:
|
|
464
|
+
def run(fiber: Fiber[Any, Any]) -> Effect[None]:
|
|
465
|
+
state = self.state
|
|
466
|
+
item.ref_count -= 1
|
|
467
|
+
state.usage -= 1
|
|
468
|
+
if item.ref_count == 0:
|
|
469
|
+
item.idle_since = self._clock.now_nanos_unsafe()
|
|
470
|
+
if item in state.invalidated:
|
|
471
|
+
return self._invalidate_item(item)
|
|
472
|
+
# Every release frees one slot, so it admits one waiter.
|
|
473
|
+
if item.ref_count < self.config.concurrency:
|
|
474
|
+
state._add_available_front(item)
|
|
475
|
+
self._wake_waiters(fiber, 1)
|
|
476
|
+
return _VOID
|
|
477
|
+
|
|
478
|
+
return _WithFiber(run)
|
|
479
|
+
|
|
480
|
+
def _wake_waiters(self, fiber: Fiber[Any, Any], count: int) -> None:
|
|
481
|
+
state = self.state
|
|
482
|
+
state.waiters.wake(fiber._scheduler, count, lambda: state.available_head is not None)
|
|
483
|
+
|
|
484
|
+
# --- sizing ---------------------------------------------------------------
|
|
485
|
+
|
|
486
|
+
def _target_size(self) -> float:
|
|
487
|
+
"""Effect's ``targetSize``; ``max_size`` wins over ``min_size``. A
|
|
488
|
+
concurrency of 0 divides as JavaScript does: to Infinity, or to NaN
|
|
489
|
+
with no usage."""
|
|
490
|
+
if self.state.is_shutting_down:
|
|
491
|
+
return 0
|
|
492
|
+
config = self.config
|
|
493
|
+
if config.is_fixed:
|
|
494
|
+
return config.min_size
|
|
495
|
+
utilization = self.state.usage / config.target_utilization
|
|
496
|
+
if config.concurrency == 0:
|
|
497
|
+
return config.max_size if utilization > 0 else math.nan
|
|
498
|
+
target = math.ceil(utilization / config.concurrency)
|
|
499
|
+
return min(max(config.min_size, target), config.max_size)
|
|
500
|
+
|
|
501
|
+
def _active_size(self) -> int:
|
|
502
|
+
return len(self.state.items) - len(self.state.invalidated)
|
|
503
|
+
|
|
504
|
+
def _invalidate_item(self, item: PoolItem[A, E]) -> Effect[None]:
|
|
505
|
+
state = self.state
|
|
506
|
+
|
|
507
|
+
def body() -> Effect[None]:
|
|
508
|
+
if item not in state.items:
|
|
509
|
+
return _VOID
|
|
510
|
+
resize = fork_in(
|
|
511
|
+
self._resize().interruptible(), state.scope, start_immediately=True
|
|
512
|
+
).as_void()
|
|
513
|
+
if item.ref_count == 0:
|
|
514
|
+
state._remove(item)
|
|
515
|
+
return item.finalizer.and_then(resize)
|
|
516
|
+
# An invalidated item stops counting towards the active size, so
|
|
517
|
+
# the pool tops itself up now: borrowers already queued would
|
|
518
|
+
# otherwise wait for an item that is never coming back.
|
|
519
|
+
state.invalidated[item] = None
|
|
520
|
+
state._remove_available(item)
|
|
521
|
+
return resize
|
|
522
|
+
|
|
523
|
+
return _Suspend(body)
|
|
524
|
+
|
|
525
|
+
def _resize(self) -> Effect[None]:
|
|
526
|
+
resize = self.state.resize_semaphore.with_permits(self._resize_loop())
|
|
527
|
+
return resize.and_then(_WithFiber(self._after_resize))
|
|
528
|
+
|
|
529
|
+
def _after_resize(self, fiber: Fiber[Any, Any]) -> Effect[None]:
|
|
530
|
+
# A borrower that found this resize running parked without starting
|
|
531
|
+
# one. Effect resumes the resize as its acquisitions finish, so it is
|
|
532
|
+
# done by then; here it resumes later, so if the pool still needs to
|
|
533
|
+
# grow, the borrowers retry and start the next resize.
|
|
534
|
+
if self._target_size() > self._active_size():
|
|
535
|
+
self.state.waiters.wake_all(fiber._scheduler)
|
|
536
|
+
return _VOID
|
|
537
|
+
|
|
538
|
+
def _resize_loop(self) -> Effect[None]:
|
|
539
|
+
def body() -> Effect[None]:
|
|
540
|
+
active = self._active_size()
|
|
541
|
+
target = self._target_size()
|
|
542
|
+
if math.isnan(target):
|
|
543
|
+
# Effect's loop acquires nothing and goes round for good here,
|
|
544
|
+
# holding the resize permit until it is interrupted.
|
|
545
|
+
return _Callback(lambda _: None)
|
|
546
|
+
if active >= target:
|
|
547
|
+
return _VOID
|
|
548
|
+
to_acquire = int(target - active)
|
|
549
|
+
acquire_one = self._acquire_one()
|
|
550
|
+
fixed = self.config.is_fixed
|
|
551
|
+
if to_acquire == 1:
|
|
552
|
+
if fixed:
|
|
553
|
+
return acquire_one.as_void()
|
|
554
|
+
return acquire_one.flat_map(
|
|
555
|
+
lambda item: _VOID if isinstance(item.exit, Failure) else self._resize_loop()
|
|
556
|
+
)
|
|
557
|
+
acquired = acquire_one.replicate_effect(to_acquire, concurrency=to_acquire)
|
|
558
|
+
if fixed:
|
|
559
|
+
return acquired.as_void()
|
|
560
|
+
return acquired.flat_map(
|
|
561
|
+
lambda items: (
|
|
562
|
+
_VOID
|
|
563
|
+
if any(isinstance(i.exit, Failure) for i in items)
|
|
564
|
+
else self._resize_loop()
|
|
565
|
+
)
|
|
566
|
+
)
|
|
567
|
+
|
|
568
|
+
return _Suspend(body)
|
|
569
|
+
|
|
570
|
+
def _acquire_one(self) -> Effect[PoolItem[A, E]]:
|
|
571
|
+
"""One item, reclaimed or acquired. Waiters are woken as each item
|
|
572
|
+
arrives, not once the whole batch is in."""
|
|
573
|
+
strategy = self.config.strategy
|
|
574
|
+
state = self.state
|
|
575
|
+
|
|
576
|
+
def rejoin(item: PoolItem[A, E] | None) -> Effect[PoolItem[A, E]]:
|
|
577
|
+
if item is None:
|
|
578
|
+
return self._allocate()
|
|
579
|
+
# A reclaimed item rejoins circulation like a released one.
|
|
580
|
+
if item in state.items and item not in state.invalidated:
|
|
581
|
+
state._add_available_front(item)
|
|
582
|
+
return _Succeed(item)
|
|
583
|
+
|
|
584
|
+
item = self._allocate() if strategy is _NOOP else strategy.reclaim(self).flat_map(rejoin)
|
|
585
|
+
|
|
586
|
+
def wake(fiber: Fiber[Any, Any], item: PoolItem[A, E]) -> Effect[None]:
|
|
587
|
+
# A failed acquisition wakes everyone to retry it. A new item
|
|
588
|
+
# serves as many waiters as it has slots.
|
|
589
|
+
if isinstance(item.exit, Failure):
|
|
590
|
+
self.state.waiters.wake_all(fiber._scheduler)
|
|
591
|
+
else:
|
|
592
|
+
self._wake_waiters(fiber, self.config.concurrency)
|
|
593
|
+
return _VOID
|
|
594
|
+
|
|
595
|
+
return item.tap(lambda i: _WithFiber(lambda fiber: wake(fiber, i)))
|
|
596
|
+
|
|
597
|
+
def _allocate(self) -> Effect[PoolItem[A, E]]:
|
|
598
|
+
def body(restore: Restore) -> Effect[PoolItem[A, E]]:
|
|
599
|
+
scope = CloseableScope()
|
|
600
|
+
context = self._context.add(Scope, scope)
|
|
601
|
+
acquire: Effect[A, E] = _Locally(self._acquire, lambda _: context)
|
|
602
|
+
acquired = acquire.exit().flat_map(lambda ex: self._track(ex, scope))
|
|
603
|
+
return restore(acquired).on_exit(
|
|
604
|
+
lambda ex: scope.close(ex) if isinstance(ex, Failure) else _VOID
|
|
605
|
+
)
|
|
606
|
+
|
|
607
|
+
return uninterruptible_mask(body)
|
|
608
|
+
|
|
609
|
+
def _track(self, ex: Exit[A, E], scope: CloseableScope) -> Effect[PoolItem[A, E]]:
|
|
610
|
+
state = self.state
|
|
611
|
+
strategy = self.config.strategy
|
|
612
|
+
item = PoolItem(
|
|
613
|
+
ex,
|
|
614
|
+
scope.close(ex).catch_cause(_report_finalizer_failure),
|
|
615
|
+
self._clock.now_nanos_unsafe(),
|
|
616
|
+
)
|
|
617
|
+
|
|
618
|
+
def on_acquire() -> Effect[None]:
|
|
619
|
+
# A borrower may have removed the item before this runs.
|
|
620
|
+
return strategy.on_acquire(item) if item in state.items else _VOID
|
|
621
|
+
|
|
622
|
+
if isinstance(ex, Success):
|
|
623
|
+
state.items[item] = None
|
|
624
|
+
state._add_available(item)
|
|
625
|
+
return _Succeed(item) if strategy is _NOOP else _Suspend(on_acquire).as_(item)
|
|
626
|
+
# Hand a waiting borrower its own failure; otherwise keep the slot.
|
|
627
|
+
waiter = (
|
|
628
|
+
state.waiters.first_sleeping() if state._first_healthy_available() is None else None
|
|
629
|
+
)
|
|
630
|
+
if waiter is not None:
|
|
631
|
+
waiter(_FailCause(ex.cause))
|
|
632
|
+
else:
|
|
633
|
+
state.items[item] = None
|
|
634
|
+
state._add_available(item)
|
|
635
|
+
# Cleanup must not delay a replacement acquisition.
|
|
636
|
+
cleanup = item.finalizer.and_then(_Suspend(on_acquire))
|
|
637
|
+
return fork_in(cleanup, state.scope, start_immediately=True).as_(item)
|
|
638
|
+
|
|
639
|
+
# --- shutdown -------------------------------------------------------------
|
|
640
|
+
|
|
641
|
+
def _shutdown(self) -> Effect[None]:
|
|
642
|
+
def body() -> Generator[Effect[Any, Any, Any], Any]:
|
|
643
|
+
state = self.state
|
|
644
|
+
if state.is_shutting_down:
|
|
645
|
+
return
|
|
646
|
+
state.is_shutting_down = True
|
|
647
|
+
size = len(state.items)
|
|
648
|
+
returned = Semaphore(size)
|
|
649
|
+
for item in list(state.items):
|
|
650
|
+
if item.ref_count > 0:
|
|
651
|
+
item.finalizer = item.finalizer.and_then(returned.release(1).as_void())
|
|
652
|
+
state.invalidated[item] = None
|
|
653
|
+
yield from returned.take(1)
|
|
654
|
+
else:
|
|
655
|
+
state._remove(item)
|
|
656
|
+
yield from item.finalizer
|
|
657
|
+
state.waiters.release_all()
|
|
658
|
+
# Wait for every borrowed item to come back and be released.
|
|
659
|
+
yield from returned.take(size)
|
|
660
|
+
|
|
661
|
+
return _Gen(body)
|
|
662
|
+
|
|
663
|
+
|
|
664
|
+
class _Release(Frame):
|
|
665
|
+
"""Below a lease's body: returns the item however the body ends. The
|
|
666
|
+
release runs uninterruptibly, then the borrower ends as its body did."""
|
|
667
|
+
|
|
668
|
+
__slots__ = ("release",)
|
|
669
|
+
|
|
670
|
+
def __init__(self, release: Effect[None]) -> None:
|
|
671
|
+
self.release = release
|
|
672
|
+
|
|
673
|
+
def _on_success(self, value: Any, fiber: Fiber[Any, Any]) -> Step:
|
|
674
|
+
return self._run(Success(value), fiber)
|
|
675
|
+
|
|
676
|
+
def _on_failure(self, cause: Cause[Any], fiber: Fiber[Any, Any]) -> Step:
|
|
677
|
+
return self._run(Failure(cause), fiber)
|
|
678
|
+
|
|
679
|
+
def _run(self, ex: Exit[Any, Any], fiber: Fiber[Any, Any]) -> Step:
|
|
680
|
+
# Stepped here, not returned: an interrupted fiber would turn a
|
|
681
|
+
# returned effect into a failure before it could run.
|
|
682
|
+
return _SetInterruptible(_FlatMap(self.release, lambda _: _FromExit(ex)), False)._step(
|
|
683
|
+
fiber
|
|
684
|
+
)
|
|
685
|
+
|
|
686
|
+
|
|
687
|
+
def _report_finalizer_failure(cause: Cause[Any]) -> Effect[None]:
|
|
688
|
+
return log_unhandled("Unhandled error in pool finalizer", cause)
|
|
689
|
+
|
|
690
|
+
|
|
691
|
+
def _clamp[N: (int, float)](value: N, low: N, high: N) -> N:
|
|
692
|
+
return low if value < low else high if value > high else value
|
|
693
|
+
|
|
694
|
+
|
|
695
|
+
def _noop_run(_: Pool[Any, Any]) -> Effect[None]:
|
|
696
|
+
return _VOID
|
|
697
|
+
|
|
698
|
+
|
|
699
|
+
def _noop_on_acquire(_: PoolItem[Any, Any]) -> Effect[None]:
|
|
700
|
+
return _VOID
|
|
701
|
+
|
|
702
|
+
|
|
703
|
+
def _noop_reclaim(_: Pool[Any, Any]) -> Effect[PoolItem[Any, Any] | None]:
|
|
704
|
+
return _Succeed(None)
|
|
705
|
+
|
|
706
|
+
|
|
707
|
+
_NOOP: Final[Strategy[Any, Any]] = Strategy(_noop_run, _noop_on_acquire, _noop_reclaim)
|