-
Notifications
You must be signed in to change notification settings - Fork 798
/
types.go
112 lines (94 loc) · 2.91 KB
/
types.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package protocol
import (
"mosn.io/mosn/pkg/types"
)
// Protocol type definition
const (
Auto types.Protocol = "Auto"
SofaRPC types.Protocol = "SofaRpc"
HTTP1 types.Protocol = "Http1"
HTTP2 types.Protocol = "Http2"
Xprotocol types.Protocol = "X"
)
// header direction definition
const (
Request = "Request"
Response = "Response"
)
// Host key for routing in MOSN Header
const (
MosnHeaderDirection = "x-mosn-direction" // for protocol convert
MosnHeaderHostKey = "x-mosn-host"
MosnHeaderPathKey = "x-mosn-path"
MosnHeaderQueryStringKey = "x-mosn-querystring"
MosnHeaderMethod = "x-mosn-method"
MosnOriginalHeaderPathKey = "x-mosn-original-path"
)
// Hseader with special meaning in istio
// todo maybe use ":authority"
const (
IstioHeaderHostKey = "authority"
)
// TODO: move CommonHeader to common, not only in protocol
// CommonHeader wrapper for map[string]string
type CommonHeader map[string]string
// Get value of key
func (h CommonHeader) Get(key string) (value string, ok bool) {
value, ok = h[key]
return
}
// Set key-value pair in header map, the previous pair will be replaced if exists
func (h CommonHeader) Set(key string, value string) {
h[key] = value
}
// Add value for given key.
// Multiple headers with the same key may be added with this function.
// Use Set for setting a single header for the given key.
func (h CommonHeader) Add(key string, value string) {
panic("not supported")
}
// Del delete pair of specified key
func (h CommonHeader) Del(key string) {
delete(h, key)
}
// Range calls f sequentially for each key and value present in the map.
// If f returns false, range stops the iteration.
func (h CommonHeader) Range(f func(key, value string) bool) {
for k, v := range h {
// stop if f return false
if !f(k, v) {
break
}
}
}
// Clone used to deep copy header's map
func (h CommonHeader) Clone() types.HeaderMap {
copy := make(map[string]string)
for k, v := range h {
copy[k] = v
}
return CommonHeader(copy)
}
func (h CommonHeader) ByteSize() uint64 {
var size uint64
for k, v := range h {
size += uint64(len(k) + len(v))
}
return size
}