120 lines
2.1 KiB
Go
120 lines
2.1 KiB
Go
|
package concurrent
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"sync"
|
|||
|
|
|
|||
|
|
. "gitea.zaclys.com/bvaudour/gob/option"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Map fournit une solution pour ajouter des éléments dans un map de façon concurrente.
|
|||
|
|
type Map[K comparable, V any] struct {
|
|||
|
|
data map[K]V
|
|||
|
|
mtx sync.Mutex
|
|||
|
|
closed bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (m *Map[K, V]) checkOpened() {
|
|||
|
|
if m.closed {
|
|||
|
|
panic("The map cannot be modified anymore.")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (m *Map[K, V]) checkClosed() {
|
|||
|
|
if !m.closed {
|
|||
|
|
panic("The Map needs to be opened.")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (m *Map[K, V]) changeState(closed bool) {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
m.closed = closed
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (m *Map[K, V]) get() (out map[K]V) {
|
|||
|
|
out = make(map[K]V)
|
|||
|
|
for k, v := range m.data {
|
|||
|
|
out[k] = v
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Clear vide le map.
|
|||
|
|
// Cette méthode nécessite que le map soit ouvert.
|
|||
|
|
func (m *Map[K, V]) Clear() {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
m.checkOpened()
|
|||
|
|
clear(m.data)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Close empêche les modifications du map.
|
|||
|
|
func (m *Map[K, V]) Close() {
|
|||
|
|
m.changeState(true)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Open permet les modifications du map.
|
|||
|
|
func (m *Map[K, V]) Open() {
|
|||
|
|
m.changeState(false)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Set associe la clé et la valeur données dans le map.
|
|||
|
|
// Cette méthode nécessite que le map soit ouvert.
|
|||
|
|
func (m *Map[K, V]) Set(key K, value V) {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
m.checkOpened()
|
|||
|
|
m.data[key] = value
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Get retourne la valeur du map à la clé donnée.
|
|||
|
|
func (m *Map[K, V]) Get(key K) (value Option[V]) {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
if v, ok := m.data[key]; ok {
|
|||
|
|
value = Some(v)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Data retourne le map.
|
|||
|
|
// Cette méthode nécessite que le map soit fermé.
|
|||
|
|
func (m *Map[K, V]) Data() map[K]V {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
m.checkClosed()
|
|||
|
|
|
|||
|
|
return m.get()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CloseData retourne le map après l’avoir préalablement fermé en écriture.
|
|||
|
|
func (m *Map[K, V]) CloseData() map[K]V {
|
|||
|
|
m.mtx.Lock()
|
|||
|
|
defer m.mtx.Unlock()
|
|||
|
|
|
|||
|
|
m.closed = true
|
|||
|
|
|
|||
|
|
return m.get()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewMap retourne un map initialisé avec les valeurs fournies.
|
|||
|
|
func NewMap[K comparable, V any](elems ...map[K]V) *Map[K, V] {
|
|||
|
|
out := make(map[K]V)
|
|||
|
|
for _, m := range elems {
|
|||
|
|
for k, v := range m {
|
|||
|
|
out[k] = v
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &Map[K, V]{
|
|||
|
|
data: out,
|
|||
|
|
}
|
|||
|
|
}
|