mirror of
https://github.com/kubernetes/client-go.git
synced 2026-07-17 18:26:55 +00:00
23c1f4e30f
When debugging, it helps to keep output from different connections separate. This can be done with contextual logging and using different loggers for each connection. Cancellation is handled separately for requests. Therefore the new APIs only add support for passing a logger instance. Kubernetes-commit: a325a4223395dfa71005b8e5dd8ea60bd91d9329
90 lines
3.4 KiB
Go
90 lines
3.4 KiB
Go
/*
|
|
Copyright 2024 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package portforward
|
|
|
|
import (
|
|
"k8s.io/apimachinery/pkg/util/httpstream"
|
|
"k8s.io/klog/v2"
|
|
streamhttp "k8s.io/streaming/pkg/httpstream"
|
|
)
|
|
|
|
var _ httpstream.Dialer = &FallbackDialer{}
|
|
var _ streamhttp.Dialer = &StreamingFallbackDialer{}
|
|
|
|
// FallbackDialer encapsulates a primary and secondary dialer, including
|
|
// the boolean function to determine if the primary dialer failed. Implements
|
|
// the httpstream.Dialer interface.
|
|
type FallbackDialer struct {
|
|
primary httpstream.Dialer
|
|
secondary httpstream.Dialer
|
|
shouldFallback func(error) bool
|
|
}
|
|
|
|
// NewFallbackDialer creates the FallbackDialer with the primary and secondary dialers,
|
|
// as well as the boolean function to determine if the primary dialer failed.
|
|
func NewFallbackDialer(primary, secondary httpstream.Dialer, shouldFallback func(error) bool) httpstream.Dialer {
|
|
return &FallbackDialer{
|
|
primary: primary,
|
|
secondary: secondary,
|
|
shouldFallback: shouldFallback,
|
|
}
|
|
}
|
|
|
|
// StreamingFallbackDialer encapsulates a primary and secondary streaming dialer
|
|
// with fallback behavior.
|
|
type StreamingFallbackDialer struct {
|
|
primary streamhttp.Dialer
|
|
secondary streamhttp.Dialer
|
|
shouldFallback func(error) bool
|
|
}
|
|
|
|
// NewFallbackDialerForStreaming creates a fallback dialer for in-tree callers
|
|
// that use k8s.io/streaming/pkg/httpstream types.
|
|
func NewFallbackDialerForStreaming(primary, secondary streamhttp.Dialer, shouldFallback func(error) bool) streamhttp.Dialer {
|
|
return &StreamingFallbackDialer{
|
|
primary: primary,
|
|
secondary: secondary,
|
|
shouldFallback: shouldFallback,
|
|
}
|
|
}
|
|
|
|
// Dial is the single function necessary to implement the "httpstream.Dialer" interface.
|
|
// It takes the protocol version strings to request, returning an the upgraded
|
|
// httstream.Connection and the negotiated protocol version accepted. If the initial
|
|
// primary dialer fails, this function attempts the secondary dialer. Returns an error
|
|
// if one occurs.
|
|
func (f *FallbackDialer) Dial(protocols ...string) (httpstream.Connection, string, error) {
|
|
conn, version, err := f.primary.Dial(protocols...)
|
|
if err != nil && f.shouldFallback(err) {
|
|
//nolint:logcheck // This code is only used by kubectl where contextual logging is not that useful.
|
|
klog.V(4).Infof("fallback to secondary dialer from primary dialer err: %v", err)
|
|
return f.secondary.Dial(protocols...)
|
|
}
|
|
return conn, version, err
|
|
}
|
|
|
|
// Dial is the single function necessary to implement the
|
|
// "k8s.io/streaming/pkg/httpstream.Dialer" interface.
|
|
func (f *StreamingFallbackDialer) Dial(protocols ...string) (streamhttp.Connection, string, error) {
|
|
conn, version, err := f.primary.Dial(protocols...)
|
|
if err != nil && f.shouldFallback(err) {
|
|
klog.V(4).Infof("fallback to secondary dialer from primary dialer err: %v", err)
|
|
return f.secondary.Dial(protocols...)
|
|
}
|
|
return conn, version, err
|
|
}
|