// 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 }