6d5ae17d2f
- Modified BuildGRPCEnv to accept the Tai node's gRPC port directly, enhancing flexibility for different modes. - Updated test cases to reflect changes in gRPC address construction, ensuring accurate environment variable settings for local, direct, and tunnel modes. - Added a new test for handling unknown modes, improving test coverage and robustness. - Adjusted Docker configuration to ensure proper host resolution for gRPC communication. Made-with: Cursor
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package sandbox
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/yao/config"
|
|
)
|
|
|
|
const taiHost = "host.tai.internal"
|
|
|
|
// BuildGRPCEnv builds the gRPC environment variables for a sandbox container.
|
|
//
|
|
// All containers reach the host via "host.tai.internal" (injected by Tai at
|
|
// container creation). The port depends on the mode:
|
|
//
|
|
// - local: Yao gRPC port (Tai and Yao on the same machine)
|
|
// - tunnel/direct: Tai gRPC port (Tai Gateway forwards to Yao)
|
|
//
|
|
// taiGRPCPort is the Tai node's gRPC port from registration (Ports.GRPC).
|
|
func BuildGRPCEnv(mode string, taiGRPCPort int, sandboxID string) map[string]string {
|
|
env := map[string]string{
|
|
"YAO_SANDBOX_ID": sandboxID,
|
|
}
|
|
|
|
switch mode {
|
|
case "local":
|
|
port := config.Conf.GRPC.Port
|
|
if port == 0 {
|
|
port = 9099
|
|
}
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)
|
|
|
|
case "tunnel", "direct":
|
|
port := taiGRPCPort
|
|
if port == 0 {
|
|
port = 19100
|
|
}
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)
|
|
|
|
default:
|
|
port := config.Conf.GRPC.Port
|
|
if port == 0 {
|
|
port = 9099
|
|
}
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)
|
|
}
|
|
return env
|
|
}
|