ce0a97c0af
- Updated benchmark functions to utilize TaiID instead of pool names for improved consistency and accuracy in tests. - Refactored test cases across various files to ensure compatibility with the new TaiID structure. - Enhanced setup functions to accept pointers to poolConfig for better memory management. - Removed deprecated config struct and adjusted related documentation to reflect the changes in the sandbox architecture. Made-with: Cursor
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package sandbox
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/yaoapp/yao/config"
|
|
)
|
|
|
|
// BuildGRPCEnv builds the gRPC environment variables for a sandbox container
|
|
// based on the Tai node's mode and address from the registry.
|
|
//
|
|
// mode is the TaiNode.Mode ("local", "direct", "tunnel").
|
|
// addr is the TaiNode.Addr (e.g. "tai://host:port" for direct mode).
|
|
// sandboxID is the container's sandbox identifier.
|
|
//
|
|
// The Yao gRPC port is read from config.Conf.GRPC.Port.
|
|
func BuildGRPCEnv(mode, addr, sandboxID string) map[string]string {
|
|
grpcPort := config.Conf.GRPC.Port
|
|
if grpcPort == 0 {
|
|
grpcPort = 9099
|
|
}
|
|
portStr := strconv.Itoa(grpcPort)
|
|
|
|
env := map[string]string{
|
|
"YAO_SANDBOX_ID": sandboxID,
|
|
}
|
|
|
|
switch mode {
|
|
case "local":
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("host.docker.internal:%s", portStr)
|
|
|
|
case "tunnel":
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("127.0.0.1:%s", portStr)
|
|
|
|
case "direct":
|
|
u, err := url.Parse(addr)
|
|
if err != nil || u.Hostname() == "" {
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("host.docker.internal:%s", portStr)
|
|
return env
|
|
}
|
|
taiHost := u.Hostname()
|
|
taiPort := u.Port()
|
|
if taiPort == "" {
|
|
taiPort = "19100"
|
|
}
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%s", taiHost, taiPort)
|
|
|
|
default:
|
|
env["YAO_GRPC_ADDR"] = fmt.Sprintf("host.docker.internal:%s", portStr)
|
|
}
|
|
return env
|
|
}
|