func NewFlowProbeBundleFromConfig(tb *probes.TopologyProbeBundle, g *graph.Graph) *FlowProbeBundle { list := config.GetConfig().GetStringSlice("agent.flow.probes") logging.GetLogger().Infof("Flow probes: %v", list) gfe := mappings.NewGraphFlowEnhancer(g) var aclient *analyzer.Client addr, port, err := config.GetAnalyzerClientAddr() if err != nil { logging.GetLogger().Errorf("Unable to parse analyzer client: %s", err.Error()) return nil } if addr != "" { aclient, err = analyzer.NewClient(addr, port) if err != nil { logging.GetLogger().Errorf("Analyzer client error %s:%d : %s", addr, port, err.Error()) return nil } } probes := make(map[string]probe.Probe) for _, t := range list { if _, ok := probes[t]; ok { continue } switch t { case "ovssflow": ofe := mappings.NewOvsFlowEnhancer(g) pipeline := mappings.NewFlowMappingPipeline(gfe, ofe) o := NewOvsSFlowProbesHandler(tb, g, pipeline, aclient) if o != nil { probes[t] = o } case "pcap": pipeline := mappings.NewFlowMappingPipeline(gfe) o := NewPcapProbesHandler(tb, g, pipeline, aclient) if o != nil { probes[t] = o } default: logging.GetLogger().Errorf("unknown probe type %s", t) } } p := probe.NewProbeBundle(probes) return &FlowProbeBundle{ ProbeBundle: *p, Graph: g, } }
func NewTopologyProbeBundleFromConfig(g *graph.Graph, n *graph.Node) *TopologyProbeBundle { list := config.GetConfig().GetStringSlice("agent.topology.probes") // FIX(safchain) once viper setdefault on nested key will be fixed move this // to config init if len(list) == 0 { list = []string{"netlink", "netns"} } logging.GetLogger().Infof("Topology probes: %v", list) probes := make(map[string]probe.Probe) for _, t := range list { if _, ok := probes[t]; ok { continue } switch t { case "netlink": probes[t] = NewNetLinkProbe(g, n) case "netns": probes[t] = NewNetNSProbe(g, n) case "ovsdb": probes[t] = NewOvsdbProbeFromConfig(g, n) case "docker": probes[t] = NewDockerProbeFromConfig(g, n) case "neutron": neutron, err := NewNeutronMapperFromConfig(g) if err != nil { logging.GetLogger().Errorf("Failed to initialize Neutron probe: %s", err.Error()) continue } probes[t] = neutron default: logging.GetLogger().Errorf("unknown probe type %s", t) } } p := probe.NewProbeBundle(probes) return &TopologyProbeBundle{*p} }