philipp/uncloud
1
0
Fork 0
mirror of https://github.com/psviderski/uncloud.git synced 2026-08-26 19:13:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 'uc image push' closed network connection error if takes >10s (fixes #141)

Browse Source
This commit is contained in:
Pasha Sviderski
2025-10-13 20:19:46 +10:00
parent 75b932b66e
commit 6ba98e4576
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/proxy/proxy.go
+3 -2
View File
@@ -81,10 +81,11 @@ func (p *Proxy) handleConnection(ctx context.Context, localConn net.Conn) {
defer p.activeConns.Done() defer p.activeConns.Done()
defer localConn.Close() defer localConn.Close()
ctx, cancel := context.WithTimeout(ctx, 10*time.Second) // Use a separate context with timeout for dialing the remote address.
dialCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel() defer cancel()
remoteConn, err := p.DialContext(ctx, "tcp", p.RemoteAddr) remoteConn, err := p.DialContext(dialCtx, "tcp", p.RemoteAddr)
if err != nil { if err != nil {
if p.OnError != nil { if p.OnError != nil {
p.OnError(fmt.Errorf("connect remote address '%s': %w", p.RemoteAddr, err)) p.OnError(fmt.Errorf("connect remote address '%s': %w", p.RemoteAddr, err))