gopkg/easylock/lock.go

// Package easylock ...
//
// Description : 包装各种姿势的锁
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2021-02-24 6:26 下午
package easylock

import "sync"

// NewLock获取普通锁实例,因为只有一把锁,flag没有意义,传空即可
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 10:42 下午 2021/2/24
func NewLock() EasyLock {
	return &lock{
		sc: &sync.RWMutex{},
		lockCnt: &LockCnt{
			Write: 0,
			Read:  0,
		},
	}
}

type lock struct {
	sc      *sync.RWMutex
	lockCnt *LockCnt
	base
}

func (l *lock) Lock(optionFuncList ...OptionFunc) error {
	defer l.AddLockCnt(l.lockCnt)
	l.sc.Lock()
	return nil
}

func (l *lock) Unlock(optionFuncList ...OptionFunc) error {
	defer l.DecreaseLockCnt(l.lockCnt)
	l.sc.Unlock()
	return nil
}

func (l *lock) RLock(optionFuncList ...OptionFunc) error {
	defer l.AddRLockCnt(l.lockCnt)
	l.sc.RLock()
	return nil
}

func (l *lock) RUnlock(optionFuncList ...OptionFunc) error {
	defer l.DecreaseRLockCnt(l.lockCnt)
	l.sc.RUnlock()
	return nil
}

func (l *lock) GetLockCnt() *LockCnt {
	return l.lockCnt
}