Go SDK: coordinator-protocol execution server and task runner#67154
Draft
jason810496 wants to merge 1 commit into
Draft
Go SDK: coordinator-protocol execution server and task runner#67154jason810496 wants to merge 1 commit into
jason810496 wants to merge 1 commit into
Conversation
- Add `messages_test.go` to test message decoding and encoding functionalities. - Introduce `serde.go` for serialization of various data types to Airflow's format. - Create `serde_test.go` to validate serialization logic and ensure correctness. - Implement `server.go` to handle communication with the supervisor and manage task execution. - Add `task_runner.go` to execute tasks based on received startup details and handle success/failure.
This was referenced May 19, 2026
This was referenced May 19, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Why
The Go SDK bundle binary currently only speaks the go-plugin protocol
used by the Edge Worker. To let the same binary run under the
ExecutableCoordinator path on the Python side (per ADR 0003), the
bundle needs to be able to act as a coordinator-protocol server:
accept startup details from the supervisor over stdio, drive a task
through its lifecycle, and stream logs / comms frames back. This PR
adds that runtime as a new
pkg/executionpackage and wires thebundle server to invoke it.
How
go-sdk/pkg/executionpackage implements the coordinator-sideruntime: msgpack framing (
frames.go), the message catalogue(
messages.go), Airflow-shaped serialisation (serde.go), thecomms decoder/encoder pair (
comms.go), an stdio-backed logger(
logger.go), aClientfor outbound supervisor calls(
client.go), theServerthat owns the stdio loop (server.go),and a
TaskRunnerthat drives a task from startup-details throughto completion (
task_runner.go).dag_parser.goreuses the existing registry to enumerate DAGs/tasksfor the supervisor without spawning the legacy go-plugin path.
bundlev1server/server.go,bundlev1/registry.go,and
bundlev1/task.golearn to hand a task off to the new runner;sdkcontext/keys.goexposes the new context keys;sdk/client.goand
sdk/connection.goget the small adjustments required for thenew entry path.
What
go-sdk/pkg/execution/{client,comms,dag_parser,frames,logger,messages,serde,server,task_runner}.goplus matching unit and integration tests.
go-sdk/bundle/bundlev1/{bundlev1server/server.go,registry.go,task.go}to route coordinator-protocol invocations through the new runtime.
sdkcontextkeys consumed by the runner and small SDKclient/connection tweaks.
go.mod/go.sumfor the new transport/codec dependencies.Next
Was generative AI tooling used to co-author this PR?