博客

引言

最近在写一些代码，需要判断自身是否处于container，如果处于container中，则有些数据获取方式需要改变。

k8s源码如何获取容器id

查了一些资料，因为容器种类较多，其实不太好判断。后来查到k8s源码，发现k8s获取容器id的代码可以借鉴使用。

var (
	containerID         containerIDProvider = getContainerIDFromCGroup
	cgroupContainerIDRe                     = regexp.MustCompile(`^.*/(?:.*-)?([0-9a-f]+)(?:\.|\s*$)`)
)
const cgroupPath = "/proc/self/cgroup"

// getContainerIDFromCGroup returns the id of the container from the cgroup file.
// If no container id found, an empty string will be returned.
func getContainerIDFromCGroup() (string, error) {
	if _, err := osStat(cgroupPath); errors.Is(err, os.ErrNotExist) {
		// File does not exist, skip
		return "", nil
	}

	file, err := osOpen(cgroupPath)
	if err != nil {
		return "", err
	}
	defer file.Close()

	return getContainerIDFromReader(file), nil
}

// getContainerIDFromReader returns the id of the container from reader.
func getContainerIDFromReader(reader io.Reader) string {
	scanner := bufio.NewScanner(reader)
	for scanner.Scan() {
		line := scanner.Text()

		if id := getContainerIDFromLine(line); id != "" {
			return id
		}
	}
	return ""
}

// getContainerIDFromLine returns the id of the container from one string line.
func getContainerIDFromLine(line string) string {
	matches := cgroupContainerIDRe.FindStringSubmatch(line)
	if len(matches) <= 1 {
		return ""
	}
	return matches[1]
}

代码其实很好理解，打开/proc/self/cgroup，然后读取所有行数，逐行正则匹配，匹配上了，就返回ContainerID。

如何判断

其实思路很简单了。打开/proc/self/cgroup，获取所有cgroup信息，然后做正则匹配。

匹配上了，就表示自身处于container中。

如果所有行数都没有匹配上，则表示自身不处于container。其实最核心的还是正则表达式：

^.*/(?:.*-)?([0-9a-f]+)(?:\.|\s*$)