mirror of
https://github.com/netbirdio/netbird.git
synced 2025-06-21 18:22:37 +02:00
Optimisation for sonar
This commit is contained in:
parent
7369f4bc38
commit
19103031ee
@ -320,9 +320,10 @@ func (c *Client) readLoop(relayConn net.Conn) {
|
|||||||
func (c *Client) handleMsg(msgType messages.MsgType, buf []byte, bufPtr *[]byte, hc *healthcheck.Receiver, internallyStoppedFlag *internalStopFlag) (continueLoop bool) {
|
func (c *Client) handleMsg(msgType messages.MsgType, buf []byte, bufPtr *[]byte, hc *healthcheck.Receiver, internallyStoppedFlag *internalStopFlag) (continueLoop bool) {
|
||||||
switch msgType {
|
switch msgType {
|
||||||
case messages.MsgTypeHealthCheck:
|
case messages.MsgTypeHealthCheck:
|
||||||
c.handleHealthCheck(buf, bufPtr, hc, internallyStoppedFlag)
|
c.handleHealthCheck(hc, internallyStoppedFlag)
|
||||||
|
c.bufPool.Put(bufPtr)
|
||||||
case messages.MsgTypeTransport:
|
case messages.MsgTypeTransport:
|
||||||
return c.handleTrasnportMsg(buf, bufPtr, internallyStoppedFlag)
|
return c.handleTransportMsg(buf, bufPtr, internallyStoppedFlag)
|
||||||
case messages.MsgTypeClose:
|
case messages.MsgTypeClose:
|
||||||
log.Debugf("relay connection close by server")
|
log.Debugf("relay connection close by server")
|
||||||
c.bufPool.Put(bufPtr)
|
c.bufPool.Put(bufPtr)
|
||||||
@ -332,8 +333,7 @@ func (c *Client) handleMsg(msgType messages.MsgType, buf []byte, bufPtr *[]byte,
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Client) handleHealthCheck(buf []byte, bufPtr *[]byte, hc *healthcheck.Receiver, internallyStoppedFlag *internalStopFlag) {
|
func (c *Client) handleHealthCheck(hc *healthcheck.Receiver, internallyStoppedFlag *internalStopFlag) {
|
||||||
c.handleHealthCheck(buf, bufPtr, hc, internallyStoppedFlag)
|
|
||||||
msg := messages.MarshalHealthcheck()
|
msg := messages.MarshalHealthcheck()
|
||||||
_, wErr := c.relayConn.Write(msg)
|
_, wErr := c.relayConn.Write(msg)
|
||||||
if wErr != nil {
|
if wErr != nil {
|
||||||
@ -342,10 +342,9 @@ func (c *Client) handleHealthCheck(buf []byte, bufPtr *[]byte, hc *healthcheck.R
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
hc.Heartbeat()
|
hc.Heartbeat()
|
||||||
c.bufPool.Put(bufPtr)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Client) handleTrasnportMsg(buf []byte, bufPtr *[]byte, internallyStoppedFlag *internalStopFlag) bool {
|
func (c *Client) handleTransportMsg(buf []byte, bufPtr *[]byte, internallyStoppedFlag *internalStopFlag) bool {
|
||||||
peerID, payload, err := messages.UnmarshalTransportMsg(buf)
|
peerID, payload, err := messages.UnmarshalTransportMsg(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if c.serviceIsRunning && !internallyStoppedFlag.isSet() {
|
if c.serviceIsRunning && !internallyStoppedFlag.isSet() {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user