[patch] ✨ Add DNS resolver to resolve Server IPs

Signed-off-by: Rintaro Okamura <rintaro.okamura@gmail.com>
rinx · Mar 5, 2021 · 62d7134 · 62d7134
1 parent 893fb78
commit 62d7134
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/alvd/agent/service/tunnel/tunnel.go b/pkg/alvd/agent/service/tunnel/tunnel.go
@@ -57,6 +57,17 @@ func (t *tunnel) Start(ctx context.Context) <-chan error {
 				}
 				return
 			case addr := <-t.connectCh:
+				host, port, err := net.SplitHostPort(addr)
+				if err == nil {
+					ips, err := net.DefaultResolver.LookupIPAddr(ctx, host)
+					if err == nil {
+						for _, ip := range ips {
+							t.connect(ctx, net.JoinHostPort(ip.String(), port))
+						}
+						continue
+					}
+				}
+
 				t.connect(ctx, addr)
 			case addr := <-t.disconnectCh:
 				t.disconnect(ctx, addr)