78 lines
1.3 KiB
Go
78 lines
1.3 KiB
Go
package server
|
|
|
|
import "sync"
|
|
|
|
type Collection struct {
|
|
items []*Server
|
|
sync.RWMutex
|
|
}
|
|
|
|
// Create a new collection from a slice of servers.
|
|
func NewCollection(servers []*Server) *Collection {
|
|
return &Collection{
|
|
items: servers,
|
|
}
|
|
}
|
|
|
|
// Return all of the items in the collection.
|
|
func (c *Collection) All() []*Server {
|
|
c.RLock()
|
|
defer c.RUnlock()
|
|
|
|
return c.items
|
|
}
|
|
|
|
// Adds an item to the collection store.
|
|
func (c *Collection) Add(s *Server) {
|
|
c.Lock()
|
|
c.items = append(c.items, s)
|
|
c.Unlock()
|
|
}
|
|
|
|
// Returns only those items matching the filter criteria.
|
|
func (c *Collection) Filter(filter func(*Server) bool) []*Server {
|
|
c.RLock()
|
|
defer c.RUnlock()
|
|
|
|
r := make([]*Server, 0)
|
|
for _, v := range c.items {
|
|
if filter(v) {
|
|
r = append(r, v)
|
|
}
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
// Returns a single element from the collection matching the filter. If nothing is
|
|
// found a nil result is returned.
|
|
func (c *Collection) Find(filter func(*Server) bool) *Server {
|
|
c.RLock()
|
|
defer c.RUnlock()
|
|
|
|
for _, v := range c.items {
|
|
if filter(v) {
|
|
return v
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Removes all items from the collection that match the filter function.
|
|
//
|
|
// TODO: cancel the context?
|
|
func (c *Collection) Remove(filter func(*Server) bool) {
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
|
|
r := make([]*Server, 0)
|
|
for _, v := range c.items {
|
|
if !filter(v) {
|
|
r = append(r, v)
|
|
}
|
|
}
|
|
|
|
c.items = r
|
|
}
|