mirror of
https://github.com/xjasonlyu/tun2socks.git
synced 2025-10-05 16:56:54 +08:00
36 lines
592 B
Go
36 lines
592 B
Go
package nat
|
|
|
|
import (
|
|
"net"
|
|
"sync"
|
|
)
|
|
|
|
type Table struct {
|
|
mapping sync.Map
|
|
}
|
|
|
|
func (t *Table) Set(key string, pc net.PacketConn) {
|
|
t.mapping.Store(key, pc)
|
|
}
|
|
|
|
func (t *Table) Get(key string) net.PacketConn {
|
|
item, exist := t.mapping.Load(key)
|
|
if !exist {
|
|
return nil
|
|
}
|
|
return item.(net.PacketConn)
|
|
}
|
|
|
|
func (t *Table) GetOrCreateLock(key string) (*sync.Cond, bool) {
|
|
item, loaded := t.mapping.LoadOrStore(key, sync.NewCond(&sync.Mutex{}))
|
|
return item.(*sync.Cond), loaded
|
|
}
|
|
|
|
func (t *Table) Delete(key string) {
|
|
t.mapping.Delete(key)
|
|
}
|
|
|
|
func NewTable() *Table {
|
|
return &Table{}
|
|
}
|