gf/g/os/gmlock/gmlock_mutex.go

// Copyright 2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.

package gmlock

import (
    "sync"
    "gitee.com/johng/gf/g/container/gtype"
)

// 互斥锁对象
type Mutex struct {
    mu     sync.RWMutex
    rcount *gtype.Int          // RLock次数
    wcount *gtype.Int          // Lock次数
}

// 创建一把内存锁使用的底层RWMutex
func NewMutex() *Mutex {
    return &Mutex{
        rcount : gtype.NewInt(),
        wcount : gtype.NewInt(),
    }
}

// 不阻塞Lock
func (l *Mutex) TryLock() bool {
    if l.wcount.Val() == 0 && l.rcount.Val() == 0 {
        l.Lock()
        return true
    }
    return false
}

func (l *Mutex) Lock() {
    l.wcount.Add(1)
    l.mu.Lock()
}

// 安全的Unlock
func (l *Mutex) Unlock() {
    if l.wcount.Val() > 0 {
        l.mu.Unlock()
        l.wcount.Add(-1)
    }
}

// 不阻塞RLock
func (l *Mutex) TryRLock() bool {
    if l.wcount.Val() == 0 {
        l.RLock()
        return true
    }
    return false
}

func (l *Mutex) RLock() {
    l.rcount.Add(1)
    l.mu.RLock()
}

// 安全的RUnlock
func (l *Mutex) RUnlock() {
    if l.rcount.Val() > 0 {
        l.mu.RUnlock()
        l.rcount.Add(-1)
    }
}