1 // Copyright 2022 Chaos Mesh Authors. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 // 15 16 package tasks 17 18 import "sync" 19 20 type LockMap[K comparable] struct { 21 sync.Map 22 } 23 24 func NewLockMap[K comparable]() LockMap[K] { 25 return LockMap[K]{ 26 sync.Map{}, 27 } 28 } 29 30 func (l *LockMap[K]) Lock(key K) func() { 31 value, _ := l.LoadOrStore(key, &sync.Mutex{}) 32 mtx := value.(*sync.Mutex) 33 mtx.Lock() 34 35 return func() { 36 if mtx != nil { 37 mtx.Unlock() 38 } 39 } 40 } 41 42 // Del :TODO: Fix bug on deleting a using value 43 func (l *LockMap[K]) Del(key K) { 44 l.Delete(key) 45 } 46