Merge pull request #594 from oif/chore/optimize-netcollector-implementation

optimize netcollector implementation and custom `/proc` mount path

Author: k8s-ci-robot

pkg/systemstatsmonitor/cpu_collector.go

@@ -46,13 +46,17 @@ type cpuCollector struct {
 	mSystemInterruptsTotal *metrics.Int64Metric
 	mSystemCPUStat         *metrics.Float64Metric // per-cpu time from /proc/stats
 
-	config *ssmtypes.CPUStatsConfig
+	config   *ssmtypes.CPUStatsConfig
+	procPath string
 
 	lastUsageTime map[string]float64
 }
 
-func NewCPUCollectorOrDie(cpuConfig *ssmtypes.CPUStatsConfig) *cpuCollector {
-	cc := cpuCollector{config: cpuConfig}
+func NewCPUCollectorOrDie(cpuConfig *ssmtypes.CPUStatsConfig, procPath string) *cpuCollector {
+	cc := cpuCollector{
+		config:   cpuConfig,
+		procPath: procPath,
+	}
 
 	var err error
 	cc.mRunnableTaskCount, err = metrics.NewFloat64Metric(

pkg/systemstatsmonitor/cpu_collector_linux.go

@@ -52,15 +52,14 @@ func (cc *cpuCollector) recordLoad() {
 }
 
 func (cc *cpuCollector) recordSystemStats() {
-
 	// don't collect the load metrics if the configs are not present.
 	if cc.mSystemCPUStat == nil && cc.mSystemInterruptsTotal == nil &&
 		cc.mSystemProcessesTotal == nil && cc.mSystemProcsBlocked == nil &&
 		cc.mSystemProcsRunning == nil {
 		return
 	}
 
-	fs, err := procfs.NewFS("/proc")
+	fs, err := procfs.NewFS(cc.procPath)
 	stats, err := fs.Stat()
 	if err != nil {
 		glog.Errorf("Failed to retrieve cpu/process stats: %v", err)

pkg/systemstatsmonitor/cpu_collector_test.go

@@ -67,6 +67,6 @@ func TestCpuCollector(t *testing.T) {
 	if err := json.Unmarshal([]byte(fakeCPUConfig), cfg); err != nil {
 		t.Fatalf("cannot load cpu config: %s", err)
 	}
-	mc := NewCPUCollectorOrDie(cfg)
+	mc := NewCPUCollectorOrDie(cfg, "/proc")
 	mc.collect()
 }