mirror of
https://github.com/impact-eintr/netstack.git
synced 2025-10-06 21:32:59 +08:00
52 lines
991 B
Go
52 lines
991 B
Go
package tmutex
|
||
|
||
import (
|
||
"sync/atomic"
|
||
)
|
||
|
||
type Mutex struct {
|
||
v int32
|
||
ch chan struct{}
|
||
}
|
||
|
||
func (m *Mutex) Init() {
|
||
m.v = 1
|
||
m.ch = make(chan struct{}, 1)
|
||
}
|
||
|
||
func (m *Mutex) Lock() {
|
||
// ==0时 只有一个锁持有者
|
||
if atomic.AddInt32(&m.v, -1) == 0 {
|
||
return
|
||
}
|
||
// !=0时 有多个想持有锁者
|
||
for {
|
||
if v := atomic.LoadInt32(&m.v);v >= 0 && atomic.SwapInt32(&m.v, -1) == 1 {
|
||
return
|
||
}
|
||
<-m.ch // 排队阻塞 等待锁释放
|
||
}
|
||
}
|
||
|
||
func (m *Mutex) TryLock() bool {
|
||
v := atomic.LoadInt32(&m.v)
|
||
if v <= 0 {
|
||
return false
|
||
}
|
||
// CAS操作需要输入两个数值，一个旧值（期望操作前的值）和一个新值，
|
||
// 在操作期间先比较下旧值有没有发生变化，
|
||
// 如果没有发生变化，才交换成新值，发生了变化则不交换。
|
||
return atomic.CompareAndSwapInt32(&m.v, 1, 0)
|
||
}
|
||
|
||
func (m *Mutex) Unlock() {
|
||
if atomic.SwapInt32(&m.v, 1) == 0 { // 没有任何持有者
|
||
return
|
||
}
|
||
|
||
select {
|
||
case m.ch <- struct{}{}:
|
||
default:
|
||
}
|
||
}
|