-
-
Notifications
You must be signed in to change notification settings - Fork 155
/
headers.go
69 lines (58 loc) · 1.76 KB
/
headers.go
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package jws
import (
"context"
"github.com/lestrrat-go/iter/mapiter"
"github.com/lestrrat-go/jwx/internal/iter"
"github.com/pkg/errors"
)
// Iterate returns a channel that successively returns all the
// header name and values.
func (h *stdHeaders) Iterate(ctx context.Context) Iterator {
pairs := h.makePairs()
ch := make(chan *HeaderPair, len(pairs))
go func(ctx context.Context, ch chan *HeaderPair, pairs []*HeaderPair) {
defer close(ch)
for _, pair := range pairs {
select {
case <-ctx.Done():
return
case ch <- pair:
}
}
}(ctx, ch, pairs)
return mapiter.New(ch)
}
func (h *stdHeaders) Walk(ctx context.Context, visitor Visitor) error {
return iter.WalkMap(ctx, h, visitor)
}
func (h *stdHeaders) AsMap(ctx context.Context) (map[string]interface{}, error) {
return iter.AsMap(ctx, h)
}
func (h *stdHeaders) Copy(ctx context.Context, dst Headers) error {
for _, pair := range h.makePairs() {
if err := dst.Set(pair.Key.(string), pair.Value); err != nil {
return errors.Wrapf(err, `failed to set header`)
}
}
return nil
}
// mergeHeaders merges two headers, and works even if the first Header
// object is nil. This is not exported because ATM it felt like this
// function is not frequently used, and MergeHeaders seemed a clunky name
func mergeHeaders(ctx context.Context, h1, h2 Headers) (Headers, error) {
h3 := NewHeaders()
if h1 != nil {
if err := h1.Copy(ctx, h3); err != nil {
return nil, errors.Wrap(err, `failed to copy headers from first Header`)
}
}
if h2 != nil {
if err := h2.Copy(ctx, h3); err != nil {
return nil, errors.Wrap(err, `failed to copy headers from second Header`)
}
}
return h3, nil
}
func (h *stdHeaders) Merge(ctx context.Context, h2 Headers) (Headers, error) {
return mergeHeaders(ctx, h, h2)
}