github.com/opencontainers/runc@v1.2.0-rc.1.0.20240520010911-492dc558cdd6/libcontainer/intelrdt/monitoring.go (about)

     1  package intelrdt
     2  
     3  import (
     4  	"bufio"
     5  	"io"
     6  	"os"
     7  	"path/filepath"
     8  
     9  	"github.com/sirupsen/logrus"
    10  )
    11  
    12  var enabledMonFeatures monFeatures
    13  
    14  type monFeatures struct {
    15  	mbmTotalBytes bool
    16  	mbmLocalBytes bool
    17  	llcOccupancy  bool
    18  }
    19  
    20  func getMonFeatures(intelRdtRoot string) (monFeatures, error) {
    21  	file, err := os.Open(filepath.Join(intelRdtRoot, "info", "L3_MON", "mon_features"))
    22  	if err != nil {
    23  		return monFeatures{}, err
    24  	}
    25  	defer file.Close()
    26  	return parseMonFeatures(file)
    27  }
    28  
    29  func parseMonFeatures(reader io.Reader) (monFeatures, error) {
    30  	scanner := bufio.NewScanner(reader)
    31  
    32  	monFeatures := monFeatures{}
    33  
    34  	for scanner.Scan() {
    35  		switch feature := scanner.Text(); feature {
    36  		case "mbm_total_bytes":
    37  			monFeatures.mbmTotalBytes = true
    38  		case "mbm_local_bytes":
    39  			monFeatures.mbmLocalBytes = true
    40  		case "llc_occupancy":
    41  			monFeatures.llcOccupancy = true
    42  		default:
    43  			logrus.Warnf("Unsupported Intel RDT monitoring feature: %s", feature)
    44  		}
    45  	}
    46  
    47  	return monFeatures, scanner.Err()
    48  }
    49  
    50  func getMonitoringStats(containerPath string, stats *Stats) error {
    51  	numaFiles, err := os.ReadDir(filepath.Join(containerPath, "mon_data"))
    52  	if err != nil {
    53  		return err
    54  	}
    55  
    56  	var mbmStats []MBMNumaNodeStats
    57  	var cmtStats []CMTNumaNodeStats
    58  
    59  	for _, file := range numaFiles {
    60  		if file.IsDir() {
    61  			numaPath := filepath.Join(containerPath, "mon_data", file.Name())
    62  			if IsMBMEnabled() {
    63  				numaMBMStats, err := getMBMNumaNodeStats(numaPath)
    64  				if err != nil {
    65  					return err
    66  				}
    67  				mbmStats = append(mbmStats, *numaMBMStats)
    68  			}
    69  			if IsCMTEnabled() {
    70  				numaCMTStats, err := getCMTNumaNodeStats(numaPath)
    71  				if err != nil {
    72  					return err
    73  				}
    74  				cmtStats = append(cmtStats, *numaCMTStats)
    75  			}
    76  		}
    77  	}
    78  
    79  	stats.MBMStats = &mbmStats
    80  	stats.CMTStats = &cmtStats
    81  
    82  	return err
    83  }