82 lines
3.2 KiB
Go
82 lines
3.2 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes Authors All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package dockertools
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/golang/glog"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
)
|
|
|
|
// This file contains all docker label related constants and functions, including:
|
|
// * label setters and getters
|
|
// * label filters (maybe in the future)
|
|
|
|
const (
|
|
kubernetesPodNameLabel = "io.kubernetes.pod.name"
|
|
kubernetesPodNamespaceLabel = "io.kubernetes.pod.namespace"
|
|
kubernetesPodUID = "io.kubernetes.pod.uid"
|
|
|
|
kubernetesContainerRestartCountLabel = "io.kubernetes.container.restartCount"
|
|
kubernetesContainerTerminationMessagePath = "io.kubernetes.container.terminationMessagePath"
|
|
|
|
kubernetesPodLabel = "io.kubernetes.pod.data"
|
|
kubernetesTerminationGracePeriodLabel = "io.kubernetes.pod.terminationGracePeriod"
|
|
kubernetesContainerLabel = "io.kubernetes.container.name"
|
|
)
|
|
|
|
func newLabels(container *api.Container, pod *api.Pod, restartCount int) map[string]string {
|
|
// TODO (random-liu) Move more label initialization here
|
|
labels := map[string]string{}
|
|
labels[kubernetesPodNameLabel] = pod.Name
|
|
labels[kubernetesPodNamespaceLabel] = pod.Namespace
|
|
labels[kubernetesPodUID] = string(pod.UID)
|
|
|
|
labels[kubernetesContainerRestartCountLabel] = strconv.Itoa(restartCount)
|
|
labels[kubernetesContainerTerminationMessagePath] = container.TerminationMessagePath
|
|
|
|
return labels
|
|
}
|
|
|
|
func getRestartCountFromLabel(labels map[string]string) (restartCount int, err error) {
|
|
if restartCountString, found := labels[kubernetesContainerRestartCountLabel]; found {
|
|
restartCount, err = strconv.Atoi(restartCountString)
|
|
if err != nil {
|
|
// This really should not happen. Just set restartCount to 0 to handle this abnormal case
|
|
restartCount = 0
|
|
}
|
|
} else {
|
|
// Get restartCount from docker label. If there is no restart count label in a container,
|
|
// it should be an old container or an invalid container, we just set restart count to 0.
|
|
// Do not report error, because there should be many old containers without this label now
|
|
glog.V(3).Infof("Container doesn't have label %s, it may be an old or invalid container", kubernetesContainerRestartCountLabel)
|
|
}
|
|
return restartCount, err
|
|
}
|
|
|
|
func getTerminationMessagePathFromLabel(labels map[string]string) string {
|
|
if terminationMessagePath, found := labels[kubernetesContainerTerminationMessagePath]; found {
|
|
return terminationMessagePath
|
|
} else {
|
|
// Do not report error, because there should be many old containers without this label now.
|
|
// Return empty string "" for these containers, the caller will get terminationMessagePath by other ways.
|
|
glog.V(3).Infof("Container doesn't have label %s, it may be an old or invalid container", kubernetesContainerTerminationMessagePath)
|
|
return ""
|
|
}
|
|
}
|