Method GetOrSet

tools/store/store.go:176–200

GetOrSet retrieves a single existing value for the provided key or stores a new one if it doesn't exist.

(key K, setFunc func() T)

// GetOrSet retrieves a single existing value for the provided key
// or stores a new one if it doesn't exist.
func (s *Store[K, T]) GetOrSet(key K, setFunc func() T) T {
	// lock only reads to minimize locks contention
	s.mu.RLock()
	v, ok := s.data[key]
	s.mu.RUnlock()
	if ok {
		return v
	}

	// lock again for write
	s.mu.Lock()
	defer s.mu.Unlock()

	// check again in case it was set between the 2 locks
	v, ok = s.data[key]
	if !ok {
		v = setFunc()
		if s.data == nil {
			s.data = make(map[K]T)
		}
		s.data[key] = v
	}

	return v
}

// SetIfLessThanLimit sets (or overwrite if already exist) a new value for key.
//