github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-05 02:09:56 +00:00
Code Issues Projects Releases Wiki Activity

Adding termination grace period to Deployment, RC, RCSet, and Job

Browse Source
This commit is contained in:
Vivek Bagade 2019-08-30 12:57:08 +02:00
parent 96439cc97f
commit 6392b69a1d
1 changed files with 34 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/utils/runners.go
View File

@ -130,6 +130,7 @@ type RCConfig struct {
ReadinessProbe *v1.Probe ReadinessProbe *v1.Probe
DNSPolicy *v1.DNSPolicy DNSPolicy *v1.DNSPolicy
PriorityClassName string PriorityClassName string
TerminationGracePeriodSeconds *int64
// Env vars, set the same for every pod. // Env vars, set the same for every pod.
Env map[string]string Env map[string]string
@ -322,6 +323,7 @@ func (config *DeploymentConfig) create() error {
}, },
Spec: v1.PodSpec{ Spec: v1.PodSpec{
Affinity: config.Affinity, Affinity: config.Affinity,
TerminationGracePeriodSeconds: config.getTerminationGracePeriodSeconds(nil),
Containers: []v1.Container{ Containers: []v1.Container{
{ {
Name: config.Name, Name: config.Name,
@ -402,6 +404,7 @@ func (config *ReplicaSetConfig) create() error {
}, },
Spec: v1.PodSpec{ Spec: v1.PodSpec{
Affinity: config.Affinity, Affinity: config.Affinity,
TerminationGracePeriodSeconds: config.getTerminationGracePeriodSeconds(nil),
Containers: []v1.Container{ Containers: []v1.Container{
{ {
Name: config.Name, Name: config.Name,
@ -474,6 +477,7 @@ func (config *JobConfig) create() error {
}, },
Spec: v1.PodSpec{ Spec: v1.PodSpec{
Affinity: config.Affinity, Affinity: config.Affinity,
TerminationGracePeriodSeconds: config.getTerminationGracePeriodSeconds(nil),
Containers: []v1.Container{ Containers: []v1.Container{
{ {
Name: config.Name, Name: config.Name,
@ -598,7 +602,7 @@ func (config *RCConfig) create() error {
DNSPolicy: *config.DNSPolicy, DNSPolicy: *config.DNSPolicy,
NodeSelector: config.NodeSelector, NodeSelector: config.NodeSelector,
Tolerations: config.Tolerations, Tolerations: config.Tolerations,
TerminationGracePeriodSeconds: &one, TerminationGracePeriodSeconds: config.getTerminationGracePeriodSeconds(&one),
PriorityClassName: config.PriorityClassName, PriorityClassName: config.PriorityClassName,
}, },
}, },
@ -1537,6 +1541,13 @@ func attachConfigMaps(template *v1.PodTemplateSpec, configMapNames []string) {
template.Spec.Containers[0].VolumeMounts = mounts template.Spec.Containers[0].VolumeMounts = mounts
} }
func (config *RCConfig) getTerminationGracePeriodSeconds(defaultGrace *int64) *int64 {
if config.TerminationGracePeriodSeconds == nil || *config.TerminationGracePeriodSeconds < 0 {
return defaultGrace
}
return config.TerminationGracePeriodSeconds
}
func attachServiceAccountTokenProjection(template *v1.PodTemplateSpec, name string) { func attachServiceAccountTokenProjection(template *v1.PodTemplateSpec, name string) {
template.Spec.Containers[0].VolumeMounts = append(template.Spec.Containers[0].VolumeMounts, template.Spec.Containers[0].VolumeMounts = append(template.Spec.Containers[0].VolumeMounts,
v1.VolumeMount{ v1.VolumeMount{