forked from mirror/pkger
100 lines
2.0 KiB
Go
100 lines
2.0 KiB
Go
|
// Code generated by github.com/gobuffalo/mapgen. DO NOT EDIT.
|
||
|
|
||
|
package pkger
|
||
|
|
||
|
import (
|
||
|
"sort"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/gobuffalo/here"
|
||
|
)
|
||
|
|
||
|
// infosMap wraps sync.Map and uses the following types:
|
||
|
// key: string
|
||
|
// value: here.Info
|
||
|
type infosMap struct {
|
||
|
data *sync.Map
|
||
|
init sync.Once
|
||
|
}
|
||
|
|
||
|
func (m *infosMap) Data() *sync.Map {
|
||
|
m.init.Do(func() {
|
||
|
if m.data == nil {
|
||
|
m.data = &sync.Map{}
|
||
|
}
|
||
|
})
|
||
|
return m.data
|
||
|
}
|
||
|
|
||
|
// Delete the key from the map
|
||
|
func (m *infosMap) Delete(key string) {
|
||
|
m.Data().Delete(key)
|
||
|
}
|
||
|
|
||
|
// Load the key from the map.
|
||
|
// Returns here.Info or bool.
|
||
|
// A false return indicates either the key was not found
|
||
|
// or the value is not of type here.Info
|
||
|
func (m *infosMap) Load(key string) (here.Info, bool) {
|
||
|
i, ok := m.Data().Load(key)
|
||
|
if !ok {
|
||
|
return here.Info{}, false
|
||
|
}
|
||
|
s, ok := i.(here.Info)
|
||
|
return s, ok
|
||
|
}
|
||
|
|
||
|
// LoadOrStore will return an existing key or
|
||
|
// store the value if not already in the map
|
||
|
func (m *infosMap) LoadOrStore(key string, value here.Info) (here.Info, bool) {
|
||
|
i, _ := m.Data().LoadOrStore(key, value)
|
||
|
s, ok := i.(here.Info)
|
||
|
return s, ok
|
||
|
}
|
||
|
|
||
|
// LoadOr will return an existing key or
|
||
|
// run the function and store the results
|
||
|
func (m *infosMap) LoadOr(key string, fn func(*infosMap) (here.Info, bool)) (here.Info, bool) {
|
||
|
i, ok := m.Load(key)
|
||
|
if ok {
|
||
|
return i, ok
|
||
|
}
|
||
|
i, ok = fn(m)
|
||
|
if ok {
|
||
|
m.Store(key, i)
|
||
|
return i, ok
|
||
|
}
|
||
|
return i, false
|
||
|
}
|
||
|
|
||
|
// Range over the here.Info values in the map
|
||
|
func (m *infosMap) Range(f func(key string, value here.Info) bool) {
|
||
|
m.Data().Range(func(k, v interface{}) bool {
|
||
|
key, ok := k.(string)
|
||
|
if !ok {
|
||
|
return false
|
||
|
}
|
||
|
value, ok := v.(here.Info)
|
||
|
if !ok {
|
||
|
return false
|
||
|
}
|
||
|
return f(key, value)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// Store a here.Info in the map
|
||
|
func (m *infosMap) Store(key string, value here.Info) {
|
||
|
m.Data().Store(key, value)
|
||
|
}
|
||
|
|
||
|
// Keys returns a list of keys in the map
|
||
|
func (m *infosMap) Keys() []string {
|
||
|
var keys []string
|
||
|
m.Range(func(key string, value here.Info) bool {
|
||
|
keys = append(keys, key)
|
||
|
return true
|
||
|
})
|
||
|
sort.Strings(keys)
|
||
|
return keys
|
||
|
}
|