This repository has been archived by the owner on Dec 29, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 5
/
bytemap.go
68 lines (61 loc) · 1.93 KB
/
bytemap.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
// Copyright 2020 Google LLC
//
// Licensed 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 triemap
// ByteSliceMap emulates `map[[]byte]interface{}`, implemented as a Trie.
//
// It seems to perform worse than `map[string]interface{}` even when casting `string([]byte)`
type ByteSliceMap struct {
value interface{}
children map[byte]*ByteSliceMap
}
// PutString is a convenience method to insert a value using a string key.
func (n *ByteSliceMap) PutString(s string, v interface{}) *ByteSliceMap {
return n.Put([]byte(s), v)
}
// Put inserts a value into the `ByteMap` using `[]byte` as a key
func (n *ByteSliceMap) Put(s []byte, v interface{}) *ByteSliceMap {
for _, r := range s {
n = n.put(r)
}
n.value = v
return n
}
func (n *ByteSliceMap) put(r byte) *ByteSliceMap {
if child, ok := n.children[r]; ok {
return child
}
var child ByteSliceMap
if n.children == nil {
n.children = map[byte]*ByteSliceMap{r: &child}
} else {
n.children[r] = &child
}
return &child
}
// Get returns a value as mapped by the `[]byte` key and a boolean of whether the value exists in the map.
func (n *ByteSliceMap) Get(s []byte) (interface{}, bool) {
for _, r := range s {
var ok bool
if n, ok = n.children[r]; !ok {
return nil, false
}
}
return n.value, true
}
// GetString is a convenience method to get a value using a string key.
//
// See: `Get`
func (n *ByteSliceMap) GetString(s string) (interface{}, bool) {
return n.Get([]byte(s))
}