stackable/headers.go at main · saryginrodion/stackable · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package stackable

import "iter"

type HeadersContainer struct {
    headers map[string][]string
}

func NewHeadersContainer() HeadersContainer {
    return HeadersContainer{
        headers: make(map[string][]string),
    }
}

func (s *HeadersContainer) Contains(key string) bool {
    _, present := s.headers[key]

    return present
}

func (s *HeadersContainer) Set(key string, value string) {
    s.headers[key] = []string{value}
}

func (s *HeadersContainer) Add(key string, value string) {
    if !s.Contains(key) {
        s.Set(key, value)
        return
    }

    s.headers[key] = append(s.headers[key], value)
}

func (s *HeadersContainer) Get(key string) []string {
    return s.headers[key]
}

func (s *HeadersContainer) Delete(key string) []string {
    deleted := s.headers[key]

    delete(s.headers, key)

    return deleted
}

func (s *HeadersContainer) Entries() iter.Seq2[string, []string] {
    return func (yield func(string, []string) bool) {
        for k, v := range s.headers {
            if !yield(k, v) {
                return
            }
        }
    }
}