philipp/uncloud
1
0
Fork 0
mirror of https://github.com/psviderski/uncloud.git synced 2026-08-28 12:03:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix proxy Director usage and close on shutdown

Browse Source
This commit is contained in:
Pavel Sviderski
2024-11-08 15:50:28 +10:00
parent 4493bb3c27
commit 1316dd3ffc
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/machine/machine.go
+3 -1
View File
@@ -148,7 +148,7 @@ func NewMachine(config *Config) (*Machine, error) {
dockerServer := docker.NewServer(dockerCli) dockerServer := docker.NewServer(dockerCli)
// Init a local gRPC proxy server that proxies requests to the local or remote machine API servers. // Init a local gRPC proxy server that proxies requests to the local or remote machine API servers.
proxyDirector := apiproxy.NewDirector(config.MachineSockPath) proxyDirector := apiproxy.NewDirector(config.MachineSockPath, APIPort)
localProxyServer := grpc.NewServer( localProxyServer := grpc.NewServer(
grpc.ForceServerCodecV2(proxy.Codec()), grpc.ForceServerCodecV2(proxy.Codec()),
grpc.UnknownServiceHandler( grpc.UnknownServiceHandler(
@@ -329,6 +329,8 @@ func (m *Machine) Run(ctx context.Context) error {
slog.Info("Stopping local API proxy server.") slog.Info("Stopping local API proxy server.")
// TODO: implement timeout for graceful shutdown. // TODO: implement timeout for graceful shutdown.
m.localProxyServer.GracefulStop() m.localProxyServer.GracefulStop()
// Close the proxy director to close all backend connections.
m.proxyDirector.Close()
slog.Info("Local API proxy server stopped.") slog.Info("Local API proxy server stopped.")
return nil return nil
}, },