apps/kubevpn
1
0
Fork 0
mirror of https://github.com/kubenetworks/kubevpn.git synced 2025-10-21 14:29:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: use grpc stream send cancel operation (#629)

Browse Source 
* refactor: use grpc stream send cancel operation

* refactor: ssh jump remove flags
This commit is contained in:
naison
2025-06-07 12:24:28 +08:00
committed by GitHub
parent 4f3e443bca
commit bb991fc1d7
47 changed files with 1565 additions and 2760 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/dev/options.go
View File

@@ -122,7 +122,11 @@ func (option *Options) Connect(ctx context.Context, sshConfig *pkgssh.SshConfig,
ManagerNamespace: managerNamespace,
}
option.AddRollbackFunc(func() error {
resp, err := cli.Disconnect(ctx, &rpc.DisconnectRequest{
resp, err := cli.Disconnect(context.Background())
if err != nil {
return err
}
err = resp.Send(&rpc.DisconnectRequest{
KubeconfigBytes: ptr.To(string(kubeConfigBytes)),
Namespace: ptr.To(ns),
SshJump: sshConfig.ToRPC(),
@@ -130,16 +134,21 @@ func (option *Options) Connect(ctx context.Context, sshConfig *pkgssh.SshConfig,
if err != nil {
return err
}
_ = util.PrintGRPCStream[rpc.DisconnectResponse](resp)
_ = util.PrintGRPCStream[rpc.DisconnectResponse](ctx, resp)
return nil
})
var resp rpc.Daemon_ProxyClient
resp, err = cli.Proxy(ctx, req)
resp, err = cli.Proxy(context.Background())
if err != nil {
plog.G(ctx).Errorf("Connect to cluster error: %s", err.Error())
return err
}
err = util.PrintGRPCStream[rpc.CloneResponse](resp)
err = resp.Send(req)
if err != nil {
plog.G(ctx).Errorf("Connect to cluster error: %s", err.Error())
return err
}
err = util.PrintGRPCStream[rpc.CloneResponse](ctx, resp)
return err
}