// fetchAllServicesAndUnits returns a map from service name to service
// and a map from service name to unit name to unit.
func fetchAllServicesAndUnits(st *state.State, unitMatcher unitMatcher) (map[string]*state.Service, map[string]map[string]*state.Unit, error) {
svcMap := make(map[string]*state.Service)
unitMap := make(map[string]map[string]*state.Unit)
services, err := st.AllServices()
if err != nil {
return nil, nil, err
}
for _, s := range services {
units, err := s.AllUnits()
if err != nil {
return nil, nil, err
}
svcUnitMap := make(map[string]*state.Unit)
for _, u := range units {
if !unitMatcher.matchUnit(u) {
continue
}
svcUnitMap[u.Name()] = u
}
if unitMatcher.matchesAny() || len(svcUnitMap) > 0 {
unitMap[s.Name()] = svcUnitMap
svcMap[s.Name()] = s
}
}
return svcMap, unitMap, nil
}
// fetchAllServices returns a map representing a mapping of service
// names to services.
func fetchAllServices(st *state.State) (map[string]*state.Service, error) {
v := make(map[string]*state.Service)
services, err := st.AllServices()
if err != nil {
return nil, err
}
for _, s := range services {
v[s.Name()] = s
}
return v, nil
}
// fetchAllServicesAndUnits returns a map from service name to service
// and a map from service name to unit name to unit.
func fetchAllServicesAndUnits(st *state.State) (map[string]*state.Service, map[string]map[string]*state.Unit, error) {
svcMap := make(map[string]*state.Service)
unitMap := make(map[string]map[string]*state.Unit)
services, err := st.AllServices()
if err != nil {
return nil, nil, err
}
for _, s := range services {
svcMap[s.Name()] = s
units, err := s.AllUnits()
if err != nil {
return nil, nil, err
}
svcUnitMap := make(map[string]*state.Unit)
for _, u := range units {
svcUnitMap[u.Name()] = u
}
unitMap[s.Name()] = svcUnitMap
}
return svcMap, unitMap, nil
}
