mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-19 08:46:38 +00: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
|
|
}
|