Files
zeekayandHanzo Dev 679ae61163 feat(transport): address the engine and the SDK, so both can use a socket
The engine took a PORT and the SDK an Options.HostPort, so neither end
could use a unix socket for traffic that never leaves the host. The
field name was not the only obstacle: luxfi/zap bound net.Listen("tcp",
":port") and reached a peer with net.DialTimeout("tcp", addr), so a path
was refused at the dialer with "missing port in address" and could not
be bound at all. That is fixed upstream (luxfi/zap Address + Network);
this pins the commit until it is tagged.

EmbedConfig.Address, Embedded.Address, ServeGated(addr) and
client.Options.Address each take ONE address, because a port is a detail
of one transport rather than a thing every listener has. A path binds or
dials a unix socket; anything else is host:port.

TestDialRequiresHostPort became TestDialRequiresAddress: Dial needs an
address, and asserts a socket path is refused for being absent rather
than for being portless. TestE2E_UnixSocket runs a real workflow over a
socket end to end, and TestE2E_UnixSocketRebind proves the socket is
unlinked on stop so a restart binds cleanly.

Co-authored-by: Hanzo Dev <[email protected]>
2026-07-28 16:27:19 -07:00

88 lines
2.2 KiB
Go

// Copyright © 2026 Hanzo AI. MIT License.
package examples
import (
"context"
"fmt"
"net"
"time"
"github.com/hanzoai/tasks/pkg/sdk/client"
tasks "github.com/hanzoai/tasks/pkg/tasks"
)
// EmbedResult captures the lifecycle of an in-process daemon dial.
type EmbedResult struct {
Port int
HealthService string
HealthStatus string
StartedID string
StartedRunID string
ListedCount int
}
// Embed boots tasks.Embed in-process on an ephemeral port, dials it via
// pkg/sdk/client, and round-trips Health → ExecuteWorkflow →
// ListWorkflows. This is the canonical "single binary" pattern Base
// apps use when TASKS_EMBED=true; the sample exercises the same code
// path so a regression in either the Embed boot or the SDK dial would
// trip the test suite immediately.
func Embed(ctx context.Context) (*EmbedResult, error) {
// Pick a free port. Do it before Embed so we can pass it through.
l, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
return nil, fmt.Errorf("listen: %w", err)
}
port := l.Addr().(*net.TCPAddr).Port
_ = l.Close()
srv, err := tasks.Embed(ctx, tasks.EmbedConfig{
Address: fmt.Sprintf(":%d", port),
Namespace: "default",
})
if err != nil {
return nil, fmt.Errorf("embed: %w", err)
}
defer func() { _ = srv.Stop(context.Background()) }()
c, err := client.Dial(client.Options{
Address: fmt.Sprintf("127.0.0.1:%d", port),
Namespace: "default",
DialTimeout: 3 * time.Second,
CallTimeout: 3 * time.Second,
})
if err != nil {
return nil, fmt.Errorf("dial: %w", err)
}
defer c.Close()
res := &EmbedResult{Port: port}
svc, status, err := c.Health(ctx)
if err != nil {
return nil, fmt.Errorf("health: %w", err)
}
res.HealthService, res.HealthStatus = svc, status
run, err := c.ExecuteWorkflow(ctx, client.StartWorkflowOptions{
ID: fmt.Sprintf("embed-%d", port),
TaskQueue: "default",
}, "EmbedDemo")
if err != nil {
return nil, fmt.Errorf("execute: %w", err)
}
res.StartedID, res.StartedRunID = run.GetID(), run.GetRunID()
list, err := c.ListWorkflows(ctx, "", 10, nil)
if err != nil {
return nil, fmt.Errorf("list: %w", err)
}
for _, e := range list.Executions {
if e.WorkflowID == res.StartedID {
res.ListedCount++
}
}
return res, nil
}