mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
57292f0fe8
* feat: idle sleep for proxmox LXCs * refactor: replace deprecated docker api types * chore(api): remove debug task list endpoint * refactor: move servemux to gphttp/servemux; favicon.go to v1/favicon * refactor: introduce Pool interface, move agent_pool to agent module * refactor: simplify api code * feat: introduce debug api * refactor: remove net.URL and net.CIDR types, improved unmarshal handling * chore: update Makefile for debug build tag, update README * chore: add gperr.Unwrap method * feat: relative time and duration formatting * chore: add ROOT_DIR environment variable, refactor * migration: move homepage override and icon cache to $BASE_DIR/data, add migration code * fix: nil dereference on marshalling service health * fix: wait for route deletion * chore: enhance tasks debuggability * feat: stdout access logger and MultiWriter * fix(agent): remove agent properly on verify error * fix(metrics): disk exclusion logic and added corresponding tests * chore: update schema and prettify, fix package.json and Makefile * fix: I/O buffer not being shrunk before putting back to pool * feat: enhanced error handling module * chore: deps upgrade * feat: better value formatting and handling --------- Co-authored-by: yusing <yusing@6uo.me>
103 lines
2 KiB
Go
103 lines
2 KiB
Go
package functional
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/puzpuzpuz/xsync/v3"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Map[KT comparable, VT any] struct {
|
|
*xsync.MapOf[KT, VT]
|
|
}
|
|
|
|
func NewMapOf[KT comparable, VT any](options ...func(*xsync.MapConfig)) Map[KT, VT] {
|
|
return Map[KT, VT]{xsync.NewMapOf[KT, VT](options...)}
|
|
}
|
|
|
|
func NewMapFrom[KT comparable, VT any](m map[KT]VT) (res Map[KT, VT]) {
|
|
res = NewMapOf[KT, VT](xsync.WithPresize(len(m)))
|
|
for k, v := range m {
|
|
res.Store(k, v)
|
|
}
|
|
return
|
|
}
|
|
|
|
func NewMap[MapType Map[KT, VT], KT comparable, VT any]() Map[KT, VT] {
|
|
return NewMapOf[KT, VT]()
|
|
}
|
|
|
|
// RangeAll calls the given function for each key-value pair in the map.
|
|
//
|
|
// Parameters:
|
|
//
|
|
// do: function to call for each key-value pair
|
|
//
|
|
// Returns:
|
|
//
|
|
// nothing
|
|
func (m Map[KT, VT]) RangeAll(do func(k KT, v VT)) {
|
|
m.Range(func(k KT, v VT) bool {
|
|
do(k, v)
|
|
return true
|
|
})
|
|
}
|
|
|
|
// RangeAllParallel calls the given function for each key-value pair in the map,
|
|
// in parallel. The map is not safe for modification from within the function.
|
|
//
|
|
// Parameters:
|
|
//
|
|
// do: function to call for each key-value pair
|
|
//
|
|
// Returns:
|
|
//
|
|
// nothing
|
|
func (m Map[KT, VT]) RangeAllParallel(do func(k KT, v VT)) {
|
|
if m.Size() < minParallelSize {
|
|
m.RangeAll(do)
|
|
return
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
m.Range(func(k KT, v VT) bool {
|
|
wg.Add(1)
|
|
go func() {
|
|
do(k, v)
|
|
wg.Done()
|
|
}()
|
|
return true
|
|
})
|
|
wg.Wait()
|
|
}
|
|
|
|
// CollectErrors calls the given function for each key-value pair in the map,
|
|
// then returns a slice of errors collected.
|
|
func (m Map[KT, VT]) CollectErrors(do func(k KT, v VT) error) []error {
|
|
errs := make([]error, 0)
|
|
m.Range(func(k KT, v VT) bool {
|
|
if err := do(k, v); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
return true
|
|
})
|
|
return errs
|
|
}
|
|
|
|
func (m Map[KT, VT]) Has(k KT) bool {
|
|
_, ok := m.Load(k)
|
|
return ok
|
|
}
|
|
|
|
func (m Map[KT, VT]) String() string {
|
|
tmp := make(map[KT]VT, m.Size())
|
|
m.RangeAll(func(k KT, v VT) {
|
|
tmp[k] = v
|
|
})
|
|
data, err := yaml.Marshal(tmp)
|
|
if err != nil {
|
|
return err.Error()
|
|
}
|
|
return string(data)
|
|
}
|