mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-01 02:30:18 +00:00
58 lines
1.9 KiB
Go
58 lines
1.9 KiB
Go
/*
|
|
Copyright 2023 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 remotecommand
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
var _ Executor = &FallbackExecutor{}
|
|
|
|
type FallbackExecutor struct {
|
|
primary Executor
|
|
secondary Executor
|
|
shouldFallback func(error) bool
|
|
}
|
|
|
|
// NewFallbackExecutor creates an Executor that first attempts to use the
|
|
// WebSocketExecutor, falling back to the legacy SPDYExecutor if the initial
|
|
// websocket "StreamWithContext" call fails.
|
|
// func NewFallbackExecutor(config *restclient.Config, method string, url *url.URL) (Executor, error) {
|
|
func NewFallbackExecutor(primary, secondary Executor, shouldFallback func(error) bool) (Executor, error) {
|
|
return &FallbackExecutor{
|
|
primary: primary,
|
|
secondary: secondary,
|
|
shouldFallback: shouldFallback,
|
|
}, nil
|
|
}
|
|
|
|
// Stream is deprecated. Please use "StreamWithContext".
|
|
func (f *FallbackExecutor) Stream(options StreamOptions) error {
|
|
return f.StreamWithContext(context.Background(), options)
|
|
}
|
|
|
|
// StreamWithContext initially attempts to call "StreamWithContext" using the
|
|
// primary executor, falling back to calling the secondary executor if the
|
|
// initial primary call to upgrade to a websocket connection fails.
|
|
func (f *FallbackExecutor) StreamWithContext(ctx context.Context, options StreamOptions) error {
|
|
err := f.primary.StreamWithContext(ctx, options)
|
|
if f.shouldFallback(err) {
|
|
return f.secondary.StreamWithContext(ctx, options)
|
|
}
|
|
return err
|
|
}
|