Безопасно ли удалять элементы из map во время итерации по нему в Go?

Можно ли удалять ключи из map во время итерации?

Да, это безопасно. В Go спецификация языка явно разрешает удаление элементов из map во время итерации с помощью range. Более того, это идиоматичный способ фильтрации map. Удаляемый элемент просто не будет передан в тело цикла, если итерация до него ещё не дошла.

Как это работает под капотом?

Когда вы вызываете delete(m, k) внутри range, Go runtime корректно обрабатывает ситуацию:

• Если удаляемый ключ уже был обработан — ничего страшного, он просто исчезает из map.
• Если удаляемый ключ ещё не был обработан — итератор пропустит его, и тело цикла для него не выполнится.
• Если удаляемый ключ — это текущий ключ итерации — удаление происходит, и цикл продолжается со следующего элемента.

Важные нюансы

Несмотря на безопасность удаления, есть пара моментов, о которых стоит знать:

• Не добавляйте новые ключи. Добавление новых элементов в map во время итерации может привести к непредсказуемому поведению: новый ключ может быть обработан, а может и нет. Это зависит от внутреннего хэширования и размера map.
• Порядок итерации не гарантирован. В Go map не упорядочен. Даже если вы удаляете элементы, порядок обхода оставшихся ключей случаен.
• Не используйте range для массового удаления всех элементов. Если нужно очистить map, просто присвойте ей новое значение: m = make(map[int]string) — это быстрее и безопаснее.

Пример безопасного удаления

package main

import "fmt"

func main() {
    m := map[int]string{
        1: "apple",
        2: "banana",
        3: "cherry",
        4: "date",
    }

    // Удаляем все элементы, где длина строки больше 5
    for k, v := range m {
        if len(v) > 5 {
            delete(m, k)
        }
    }

    fmt.Println(m) // Вывод: map[3:cherry 4:date] (или другой порядок)
}

Альтернативный подход — копирование ключей

Если вы хотите быть на 100% уверены в поведении (например, для сложной логики), можно сначала собрать ключи для удаления в слайс, а потом удалить их:

package main

import "fmt"

func main() {
    m := map[int]string{
        1: "apple",
        2: "banana",
        3: "cherry",
        4: "date",
    }

    var toDelete []int
    for k, v := range m {
        if len(v) > 5 {
            toDelete = append(toDelete, k)
        }
    }

    for _, k := range toDelete {
        delete(m, k)
    }

    fmt.Println(m) // map[3:cherry 4:date]
}

Этот подход особенно полезен, когда логика удаления зависит от других данных или когда нужно избежать случайных мутаций во время итерации.

Итог

Удаление ключей из map внутри range — это абсолютно безопасная и распространённая практика в Go. Главное — не добавлять новые ключи. Ваш исходный код корректен и не вызовет ошибок.

package main

import (
    "fmt"
    "sync"
)

func main() {
    var mu sync.Mutex
    m := map[int]string{
        1: "apple",
        2: "banana",
    }

    // В одной горутине
    mu.Lock()
    for k, v := range m {
        if v == "banana" {
            delete(m, k)
        }
    }
    mu.Unlock()

    fmt.Println(m)
}