mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 09:50:47 +01:00
32 lines
588 B
Go
32 lines
588 B
Go
|
//go:build !ios
|
||
|
|
|
||
|
|
package net
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Conn wraps a net.Conn to override the Close method
|
||
|
|
type Conn struct {
|
||
|
|
net.Conn
|
||
|
|
ID ConnectionID
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close overrides the net.Conn Close method to execute all registered hooks after closing the connection
|
||
|
|
func (c *Conn) Close() error {
|
||
|
|
err := c.Conn.Close()
|
||
|
|
|
||
|
|
dialerCloseHooksMutex.RLock()
|
||
|
|
defer dialerCloseHooksMutex.RUnlock()
|
||
|
|
|
||
|
|
for _, hook := range dialerCloseHooks {
|
||
|
|
if err := hook(c.ID, &c.Conn); err != nil {
|
||
|
|
log.Errorf("Error executing dialer close hook: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|