Урок 91 · Задача 1

SafeMap: защита карты RWMutex

Запись в общий map из нескольких горутин — гонка. Реализуйте SafeMap: Set под Lock, Get и Len под RLock.

Показать решение

func (s *SafeMap) Set(k string, v int) {
	s.mu.Lock()
	defer s.mu.Unlock()
	if s.m == nil {
		s.m = make(map[string]int)
	}
	s.m[k] = v
}

func (s *SafeMap) Get(k string) (int, bool) {
	s.mu.RLock()
	defer s.mu.RUnlock()
	v, ok := s.m[k]
	return v, ok
}

func (s *SafeMap) Len() int {
	s.mu.RLock()
	defer s.mu.RUnlock()
	return len(s.m)
}

package main

import "sync"

type SafeMap struct {
	mu sync.RWMutex
	m  map[string]int
}

// Set записывает под Lock.
func (s *SafeMap) Set(k string, v int) {
	// ваш код
}

// Get читает под RLock.
func (s *SafeMap) Get(k string) (int, bool) {
	// ваш код
	return 0, false
}

// Len возвращает размер под RLock.
func (s *SafeMap) Len() int {
	// ваш код
	return 0
}