git/work
1
0
Fork 0
Code Issues Pull requests 2 Projects Releases 5 Packages Wiki Activity Actions

feat: implement ready chanel to wait for established connection

Browse source
This commit is contained in:
Patryk Hegenberg 2026-01-11 11:44:45 +01:00
parent 4ed6a61b1d
commit 5de9ff7961
3 changed files with 41 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
internal/ssh/forwarder.go
View file

@ -28,7 +28,7 @@ func NewForwarder(client *ssh.Client, localPort, remotePort, remoteHost string)
}
}
func (f *Forwarder) Start(ctx context.Context) error {
func (f *Forwarder) Start(ctx context.Context, ready chan<- struct{}) error {
localAddr := "127.0.0.1:" + f.localPort
remoteAddr := net.JoinHostPort(f.remoteHost, f.remotePort)
@ -37,6 +37,10 @@ func (f *Forwarder) Start(ctx context.Context) error {
return fmt.Errorf("failed to listen on %s: %w", localAddr, err)
}
if ready != nil {
close(ready)
}
go func() {
<-ctx.Done()
listener.Close()