package locker

import (
	"errors"
	"time"
)

type ChannelLocker struct {
	ch  chan struct{}
	lockerWait bool  //是否锁等待，false则直接退出
}

func(c *ChannelLocker)AddLock()error{

	select{
	  case c.ch<- struct{}{} :

	  case <-time.After(time.Millisecond *1):
	     if !c.lockerWait {
			 return errors.New("timeout")
		 }

	}
	return nil
}

func(c *ChannelLocker)UnLock()error{
	<-c.ch
	return nil
}