apps/kubevpn
1
0
Fork 0
mirror of https://github.com/kubenetworks/kubevpn.git synced 2025-12-24 11:51:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: refactor print GRPC message (#386)

Browse Source
This commit is contained in:
naison
2024-12-06 19:29:11 +08:00
committed by GitHub
parent d9a978d330
commit 81f62eab31
19 changed files with 168 additions and 306 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
pkg/daemon/action/proxy.go
View File

@@ -99,19 +99,15 @@ func (svr *Server) Proxy(req *rpc.ConnectRequest, resp rpc.Daemon_ProxyServer) (
if err != nil {
return err
}
var recv *rpc.DisconnectResponse
for {
recv, err = disconnectResp.Recv()
if err == io.EOF {
break
} else if err != nil {
log.Errorf("Receive from disconnect failed: %v", err)
return err
}
err = resp.Send(&rpc.ConnectResponse{Message: recv.Message})
if err != nil {
return err
}
err = util.CopyAndConvertGRPCStream[rpc.DisconnectResponse, rpc.ConnectResponse](
disconnectResp,
resp,
func(response *rpc.DisconnectResponse) *rpc.ConnectResponse {
return &rpc.ConnectResponse{Message: response.Message}
},
)
if err != nil {
return err
}
util.InitLoggerForClient(config.Debug)
log.SetOutput(out)
@@ -125,18 +121,9 @@ func (svr *Server) Proxy(req *rpc.ConnectRequest, resp rpc.Daemon_ProxyServer) (
if err != nil {
return err
}
var recv *rpc.ConnectResponse
for {
recv, err = connResp.Recv()
if err == io.EOF {
break
} else if err != nil {
return err
}
err = resp.Send(recv)
if err != nil {
return err
}
err = util.CopyGRPCStream[rpc.ConnectResponse](connResp, resp)
if err != nil {
return err
}
util.InitLoggerForClient(config.Debug)
log.SetOutput(out)