mirror of
https://github.com/arangodb/kube-arangodb.git
synced 2024-12-14 11:57:37 +00:00
115 lines
3.5 KiB
Go
115 lines
3.5 KiB
Go
//
|
|
// DISCLAIMER
|
|
//
|
|
// Copyright 2016-2022 ArangoDB GmbH, Cologne, Germany
|
|
//
|
|
// 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.
|
|
//
|
|
// Copyright holder is ArangoDB GmbH, Cologne, Germany
|
|
//
|
|
|
|
package k8sutil
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
core "k8s.io/api/core/v1"
|
|
|
|
"github.com/arangodb/kube-arangodb/pkg/util/constants"
|
|
"github.com/arangodb/kube-arangodb/pkg/util/errors"
|
|
)
|
|
|
|
const (
|
|
initLifecycleContainerName = "init-lifecycle"
|
|
LifecycleVolumeMountDir = "/lifecycle/tools"
|
|
lifecycleVolumeName = "lifecycle"
|
|
)
|
|
|
|
// InitLifecycleContainer creates an init-container to copy the lifecycle binary to a shared volume.
|
|
func InitLifecycleContainer(image string, resources *core.ResourceRequirements, securityContext *core.SecurityContext) (core.Container, error) {
|
|
binaryPath, err := os.Executable()
|
|
if err != nil {
|
|
return core.Container{}, errors.WithStack(err)
|
|
}
|
|
c := core.Container{
|
|
Name: initLifecycleContainerName,
|
|
Image: image,
|
|
Command: append([]string{binaryPath}, "lifecycle", "copy", "--target", LifecycleVolumeMountDir),
|
|
VolumeMounts: []core.VolumeMount{
|
|
LifecycleVolumeMount(),
|
|
},
|
|
ImagePullPolicy: core.PullIfNotPresent,
|
|
SecurityContext: securityContext,
|
|
}
|
|
|
|
if resources != nil {
|
|
c.Resources = ExtractPodResourceRequirement(*resources)
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
// NewLifecycleFinalizers creates a lifecycle structure with preStop handler which wait for finalizers to be removed.
|
|
func NewLifecycleFinalizers() (*core.Lifecycle, error) {
|
|
return NewLifecycle("finalizers")
|
|
}
|
|
|
|
// NewLifecyclePort creates a lifecycle structure with preStop handler which wait for port to be closed.
|
|
func NewLifecyclePort() (*core.Lifecycle, error) {
|
|
return NewLifecycle("port")
|
|
}
|
|
|
|
// NewLifecycle creates a lifecycle structure with preStop handler.
|
|
func NewLifecycle(t string) (*core.Lifecycle, error) {
|
|
binaryPath, err := os.Executable()
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
exePath := filepath.Join(LifecycleVolumeMountDir, filepath.Base(binaryPath))
|
|
lifecycle := &core.Lifecycle{
|
|
PreStop: &core.Handler{
|
|
Exec: &core.ExecAction{
|
|
Command: append([]string{exePath}, "lifecycle", "preStop", t),
|
|
},
|
|
},
|
|
}
|
|
|
|
return lifecycle, nil
|
|
}
|
|
|
|
func GetLifecycleEnv() []core.EnvVar {
|
|
return []core.EnvVar{
|
|
CreateEnvFieldPath(constants.EnvOperatorPodName, "metadata.name"),
|
|
CreateEnvFieldPath(constants.EnvOperatorPodNamespace, "metadata.namespace"),
|
|
CreateEnvFieldPath(constants.EnvOperatorNodeName, "spec.nodeName"),
|
|
CreateEnvFieldPath(constants.EnvOperatorNodeNameArango, "spec.nodeName"),
|
|
}
|
|
}
|
|
|
|
// LifecycleVolumeMount creates a volume mount structure for shared lifecycle emptyDir.
|
|
func LifecycleVolumeMount() core.VolumeMount {
|
|
return core.VolumeMount{
|
|
Name: lifecycleVolumeName,
|
|
MountPath: LifecycleVolumeMountDir,
|
|
}
|
|
}
|
|
|
|
// LifecycleVolume creates a volume mount structure for shared lifecycle emptyDir.
|
|
func LifecycleVolume() core.Volume {
|
|
return core.Volume{
|
|
Name: lifecycleVolumeName,
|
|
VolumeSource: core.VolumeSource{
|
|
EmptyDir: &core.EmptyDirVolumeSource{},
|
|
},
|
|
}
|
|
}
|