Add the channel id to the label

openziti · Dec 19, 2024 · e05804b · e05804b
1 parent 98cffae
commit e05804b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/classic_impl.go b/classic_impl.go
@@ -105,7 +105,7 @@ func (impl *classicImpl) Certificates() []*x509.Certificate {
 }
 
 func (impl *classicImpl) Label() string {
-	return fmt.Sprintf("u{%s}->i{%s}", impl.LogicalName(), impl.ConnectionId())
+	return fmt.Sprintf("u{%s}->i{%s/%s}", impl.LogicalName(), impl.id, impl.ConnectionId())
 }
 
 func (impl *classicImpl) Close() error {

diff --git a/reconnecting_dialer.go b/reconnecting_dialer.go
@@ -165,8 +165,6 @@ func (dialer *reconnectingDialer) sendHello(impl *reconnectingImpl) error {
 
 	if id, ok := response.GetStringHeader(IdHeader); ok {
 		impl.id = &identity.TokenId{Token: id}
-	} else if certs := impl.Certificates(); len(certs) > 0 {
-		impl.id = &identity.TokenId{Token: certs[0].Subject.CommonName}
 	}
 
 	impl.headers.Store(response.Headers)

diff --git a/reconnecting_impl.go b/reconnecting_impl.go
@@ -107,7 +107,7 @@ func (impl *reconnectingImpl) Certificates() []*x509.Certificate {
 }
 
 func (impl *reconnectingImpl) Label() string {
-	return fmt.Sprintf("u{%s}->i{%s}", impl.LogicalName(), impl.ConnectionId())
+	return fmt.Sprintf("u{%s}->i{%s/%s}", impl.LogicalName(), impl.id.Token, impl.ConnectionId())
 }
 
 func (impl *reconnectingImpl) Close() error {
@@ -122,7 +122,13 @@ func (impl *reconnectingImpl) IsClosed() bool {
 }
 
 func newReconnectingImpl(peer transport.Conn, reconnectionHandler reconnectionHandler, timeout time.Duration) *reconnectingImpl {
+	id := &identity.TokenId{Token: "unknown"}
+	if certs := peer.PeerCertificates(); len(certs) > 0 {
+		id = &identity.TokenId{Token: certs[0].Subject.CommonName}
+	}
+
 	return &reconnectingImpl{
+		id:                  id,
 		peer:                peer,
 		reconnectionHandler: reconnectionHandler,
 		readF:               ReadV2,