Files
my-rook-config/pkg/operator/edgefs/s3x/s3x.go
T
Anton Skriptsov c0fe6af712 Initial v1
Signed-off-by: Anton Skriptsov <sabbotagge@gmail.com>
2019-08-28 15:18:38 -07:00

504 lines
14 KiB
Go

/*
Copyright 2019 The Rook 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 s3x for the Edgefs manager.
package s3x
import (
"fmt"
"strings"
edgefsv1 "github.com/rook/rook/pkg/apis/edgefs.rook.io/v1"
"github.com/rook/rook/pkg/clusterd"
"github.com/rook/rook/pkg/operator/k8sutil"
apps "k8s.io/api/apps/v1"
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
const (
appName = "rook-edgefs-s3x"
/* Volumes definitions */
serviceAccountName = "rook-edgefs-cluster"
sslCertVolumeName = "ssl-cert-volume"
defaultS3Image = "edgefs/edgefs-restapi"
sslMountPath = "/opt/nedge/etc/ssl/"
dataVolumeName = "edgefs-datadir"
stateVolumeFolder = ".state"
etcVolumeFolder = ".etc"
defaultPort = 4000
defaultSecurePort = 4443
defaultServiceType = "ClusterIP"
)
// Start the rgw manager
func (c *S3XController) CreateService(s edgefsv1.S3X, ownerRefs []metav1.OwnerReference) error {
return c.CreateOrUpdate(s, false, ownerRefs)
}
func (c *S3XController) UpdateService(s edgefsv1.S3X, ownerRefs []metav1.OwnerReference) error {
return c.CreateOrUpdate(s, true, ownerRefs)
}
// Start the s3x instance
func (c *S3XController) CreateOrUpdate(s edgefsv1.S3X, update bool, ownerRefs []metav1.OwnerReference) error {
logger.Infof("starting update=%v service=%s", update, s.Name)
logger.Infof("S3X Base image is %s", c.rookImage)
// validate S3X service settings
if err := validateService(c.context, s); err != nil {
return fmt.Errorf("invalid S3X service %s arguments. %+v", s.Name, err)
}
//check http settings
if s.Spec.Port == 0 {
s.Spec.Port = defaultPort
}
if s.Spec.SecurePort == 0 {
s.Spec.SecurePort = defaultSecurePort
}
if s.Spec.ServiceType == "" {
s.Spec.ServiceType = defaultServiceType
}
// check if S3X service already exists
exists, err := serviceExists(c.context, s)
if err == nil && exists {
if !update {
logger.Infof("S3X service %s exists in namespace %s", s.Name, s.Namespace)
return nil
}
logger.Infof("S3X service %s exists in namespace %s. checking for updates", s.Name, s.Namespace)
}
// start the deployment
deployment := c.makeDeployment(s.Name, s.Namespace, c.rookImage, s.Spec)
if _, err := c.context.Clientset.AppsV1().Deployments(s.Namespace).Create(deployment); err != nil {
if !errors.IsAlreadyExists(err) {
return fmt.Errorf("failed to create %s deployment. %+v", appName, err)
}
logger.Infof("%s deployment already exists", appName)
if _, err := c.context.Clientset.AppsV1().Deployments(s.Namespace).Update(deployment); err != nil {
return fmt.Errorf("failed to update %s deployment. %+v", appName, err)
}
logger.Infof("%s deployment updated", appName)
} else {
logger.Infof("%s deployment started", appName)
}
// create the s3x service
service := c.makeS3XService(instanceName(s.Name), s.Name, s.Namespace, s.Spec)
if _, err := c.context.Clientset.CoreV1().Services(s.Namespace).Create(service); err != nil {
if !errors.IsAlreadyExists(err) {
return fmt.Errorf("failed to create s3x service. %+v", err)
}
logger.Infof("s3x service %s already exists", service)
} else {
logger.Infof("s3x service %s started", service)
}
return nil
}
func (c *S3XController) makeS3XService(name, svcname, namespace string, s3xSpec edgefsv1.S3XSpec) *v1.Service {
labels := getLabels(name, svcname, namespace)
httpPort := v1.ServicePort{Name: "port", Port: int32(s3xSpec.Port), Protocol: v1.ProtocolTCP}
httpsPort := v1.ServicePort{Name: "secure-port", Port: int32(s3xSpec.SecurePort), Protocol: v1.ProtocolTCP}
if s3xSpec.ExternalPort != 0 {
httpPort.NodePort = int32(s3xSpec.ExternalPort)
}
if s3xSpec.SecureExternalPort != 0 {
httpsPort.NodePort = int32(s3xSpec.SecureExternalPort)
}
svc := &v1.Service{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Namespace: namespace,
Labels: labels,
},
Spec: v1.ServiceSpec{
Selector: labels,
Type: k8sutil.ParseServiceType(s3xSpec.ServiceType),
Ports: []v1.ServicePort{
{Name: "grpc", Port: 49000, Protocol: v1.ProtocolTCP},
httpPort,
httpsPort,
},
},
}
k8sutil.SetOwnerRef(&svc.ObjectMeta, &c.ownerRef)
return svc
}
func (c *S3XController) makeDeployment(svcname, namespace, rookImage string, s3xSpec edgefsv1.S3XSpec) *apps.Deployment {
name := instanceName(svcname)
volumes := []v1.Volume{}
if c.useHostLocalTime {
volumes = append(volumes, edgefsv1.GetHostLocalTimeVolume())
}
// add ssl certificate volume if defined
if len(s3xSpec.SSLCertificateRef) > 0 {
volumes = append(volumes, v1.Volume{
Name: sslCertVolumeName,
VolumeSource: v1.VolumeSource{
Secret: &v1.SecretVolumeSource{
SecretName: s3xSpec.SSLCertificateRef,
Items: []v1.KeyToPath{
{
Key: "sslkey",
Path: "ssl.key",
},
{
Key: "sslcert",
Path: "ssl.crt",
},
},
},
},
})
}
if c.dataVolumeSize.Value() > 0 {
// dataVolume case
volumes = append(volumes, v1.Volume{
Name: dataVolumeName,
VolumeSource: v1.VolumeSource{
PersistentVolumeClaim: &v1.PersistentVolumeClaimVolumeSource{
ClaimName: dataVolumeName,
},
},
})
} else {
// dataDir case
volumes = append(volumes, v1.Volume{
Name: dataVolumeName,
VolumeSource: v1.VolumeSource{
HostPath: &v1.HostPathVolumeSource{
Path: c.dataDirHostPath,
},
},
})
}
var rookImageVer string
rookImageComponents := strings.Split(c.rookImage, ":")
if len(rookImageComponents) == 2 {
rookImageVer = rookImageComponents[1]
} else {
rookImageVer = "latest"
}
podSpec := v1.PodTemplateSpec{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Labels: getLabels(name, svcname, namespace),
},
Spec: v1.PodSpec{
Containers: []v1.Container{c.s3xContainer(svcname, name, rookImage, s3xSpec), c.s3ProxyContainer(svcname, "s3-proxy", defaultS3Image+":"+rookImageVer, "s3", s3xSpec)},
RestartPolicy: v1.RestartPolicyAlways,
Volumes: volumes,
HostIPC: true,
HostNetwork: c.NetworkSpec.IsHost(),
NodeSelector: map[string]string{namespace: "cluster"},
ServiceAccountName: serviceAccountName,
},
}
if c.NetworkSpec.IsHost() {
podSpec.Spec.DNSPolicy = v1.DNSClusterFirstWithHostNet
} else if c.NetworkSpec.IsMultus() {
k8sutil.ApplyMultus(c.NetworkSpec, &podSpec.ObjectMeta)
}
// apply current S3X CRD options to pod's specification
s3xSpec.Annotations.ApplyToObjectMeta(&podSpec.ObjectMeta)
s3xSpec.Placement.ApplyToPodSpec(&podSpec.Spec)
d := &apps.Deployment{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Namespace: namespace,
},
Spec: apps.DeploymentSpec{
Selector: &metav1.LabelSelector{
MatchLabels: podSpec.Labels,
},
Template: podSpec,
Replicas: &s3xSpec.Instances,
},
}
k8sutil.SetOwnerRef(&d.ObjectMeta, &c.ownerRef)
s3xSpec.Annotations.ApplyToObjectMeta(&d.ObjectMeta)
return d
}
func (c *S3XController) s3xContainer(svcname, name, containerImage string, s3xSpec edgefsv1.S3XSpec) v1.Container {
runAsUser := int64(0)
readOnlyRootFilesystem := false
securityContext := &v1.SecurityContext{
RunAsUser: &runAsUser,
ReadOnlyRootFilesystem: &readOnlyRootFilesystem,
Capabilities: &v1.Capabilities{
Add: []v1.Capability{"SYS_NICE", "SYS_RESOURCE", "IPC_LOCK"},
},
}
volumeMounts := []v1.VolumeMount{
{
Name: dataVolumeName,
MountPath: "/opt/nedge/etc.target",
SubPath: etcVolumeFolder,
},
{
Name: dataVolumeName,
MountPath: "/opt/nedge/var/run",
SubPath: stateVolumeFolder,
},
}
if c.useHostLocalTime {
volumeMounts = append(volumeMounts, edgefsv1.GetHostLocalTimeVolumeMount())
}
if len(s3xSpec.SSLCertificateRef) > 0 {
volumeMounts = append(volumeMounts, v1.VolumeMount{Name: sslCertVolumeName, MountPath: sslMountPath})
}
cont := v1.Container{
Name: name,
Image: containerImage,
ImagePullPolicy: v1.PullAlways,
Args: []string{"s3x"},
Env: []v1.EnvVar{
{
Name: "CCOW_LOG_LEVEL",
Value: "5",
},
{
Name: "CCOW_SVCNAME",
Value: svcname,
},
{
Name: "HOST_HOSTNAME",
ValueFrom: &v1.EnvVarSource{
FieldRef: &v1.ObjectFieldSelector{
FieldPath: "spec.nodeName",
},
},
},
{
Name: "K8S_NAMESPACE",
ValueFrom: &v1.EnvVarSource{
FieldRef: &v1.ObjectFieldSelector{
FieldPath: "metadata.namespace",
},
},
},
{
Name: "EFSS3X_HTTP_PORT",
Value: fmt.Sprint(s3xSpec.Port),
},
{
Name: "EFSS3X_HTTPS_PORT",
Value: fmt.Sprint(s3xSpec.SecurePort),
},
},
SecurityContext: securityContext,
Resources: s3xSpec.Resources,
Ports: []v1.ContainerPort{
{Name: "grpc", ContainerPort: 49000, Protocol: v1.ProtocolTCP},
{Name: "port", ContainerPort: int32(s3xSpec.Port), Protocol: v1.ProtocolTCP},
{Name: "secure-port", ContainerPort: int32(s3xSpec.SecurePort), Protocol: v1.ProtocolTCP},
},
VolumeMounts: volumeMounts,
}
cont.Env = append(cont.Env, edgefsv1.GetInitiatorEnvArr("s3x",
c.resourceProfile == "embedded" || s3xSpec.ResourceProfile == "embedded",
s3xSpec.ChunkCacheSize, s3xSpec.Resources)...)
return cont
}
func (c *S3XController) s3ProxyContainer(svcname, name, containerImage, args string, s3xSpec edgefsv1.S3XSpec) v1.Container {
runAsUser := int64(0)
readOnlyRootFilesystem := false
securityContext := &v1.SecurityContext{
RunAsUser: &runAsUser,
ReadOnlyRootFilesystem: &readOnlyRootFilesystem,
Capabilities: &v1.Capabilities{
Add: []v1.Capability{"SYS_NICE", "SYS_RESOURCE", "IPC_LOCK"},
},
}
volumeMounts := []v1.VolumeMount{
{
Name: dataVolumeName,
MountPath: "/opt/nedge/etc.target",
SubPath: etcVolumeFolder,
},
{
Name: dataVolumeName,
MountPath: "/opt/nedge/var/run",
SubPath: stateVolumeFolder,
},
}
if len(s3xSpec.SSLCertificateRef) > 0 {
volumeMounts = append(volumeMounts, v1.VolumeMount{Name: sslCertVolumeName, MountPath: sslMountPath})
}
cont := v1.Container{
Name: name,
Image: containerImage,
ImagePullPolicy: v1.PullAlways,
Args: []string{args},
Env: []v1.EnvVar{
{
Name: "CCOW_LOG_LEVEL",
Value: "5",
},
{
Name: "CCOW_SVCNAME",
Value: svcname,
},
{
Name: "DEBUG",
Value: "alert,error,info",
},
{
Name: "HOST_HOSTNAME",
ValueFrom: &v1.EnvVarSource{
FieldRef: &v1.ObjectFieldSelector{
FieldPath: "spec.nodeName",
},
},
},
{
Name: "K8S_NAMESPACE",
ValueFrom: &v1.EnvVarSource{
FieldRef: &v1.ObjectFieldSelector{
FieldPath: "metadata.namespace",
},
},
},
{
Name: "GW_PORT",
Value: "9982",
},
{
Name: "GW_PORT_SSL",
Value: "9443",
},
},
SecurityContext: securityContext,
Resources: s3xSpec.Resources,
Ports: []v1.ContainerPort{
{Name: "port", ContainerPort: 9982, Protocol: v1.ProtocolTCP},
},
VolumeMounts: volumeMounts,
}
cont.Env = append(cont.Env, edgefsv1.GetInitiatorEnvArr("s3",
c.resourceProfile == "embedded" || s3xSpec.ResourceProfile == "embedded",
s3xSpec.ChunkCacheSize, s3xSpec.Resources)...)
return cont
}
// Delete S3X service and possibly some artifacts.
func (c *S3XController) DeleteService(s edgefsv1.S3X) error {
// check if service exists
exists, err := serviceExists(c.context, s)
if err != nil {
return fmt.Errorf("failed to detect if there is a S3X service to delete. %+v", err)
}
if !exists {
logger.Infof("S3X service %s does not exist in namespace %s", s.Name, s.Namespace)
return nil
}
logger.Infof("Deleting S3X service %s from namespace %s", s.Name, s.Namespace)
var gracePeriod int64
propagation := metav1.DeletePropagationForeground
options := &metav1.DeleteOptions{GracePeriodSeconds: &gracePeriod, PropagationPolicy: &propagation}
// Delete the s3x service
err = c.context.Clientset.CoreV1().Services(s.Namespace).Delete(instanceName(s.Name), options)
if err != nil && !errors.IsNotFound(err) {
logger.Warningf("failed to delete S3X service. %+v", err)
}
// Make a best effort to delete the S3X pods
err = k8sutil.DeleteDeployment(c.context.Clientset, s.Namespace, instanceName(s.Name))
if err != nil {
logger.Warningf(err.Error())
}
logger.Infof("Completed deleting S3X service %s", s.Name)
return nil
}
func getLabels(name, svcname, namespace string) map[string]string {
return map[string]string{
k8sutil.AppAttr: appName,
k8sutil.ClusterAttr: namespace,
"edgefs_svcname": svcname,
"edgefs_svctype": "s3x",
}
}
// Validate the S3X arguments
func validateService(context *clusterd.Context, s edgefsv1.S3X) error {
if s.Name == "" {
return fmt.Errorf("missing name")
}
if s.Namespace == "" {
return fmt.Errorf("missing namespace")
}
return nil
}
func instanceName(svcname string) string {
return fmt.Sprintf("%s-%s", appName, svcname)
}
// Check if the S3X service exists
func serviceExists(context *clusterd.Context, s edgefsv1.S3X) (bool, error) {
_, err := context.Clientset.AppsV1().Deployments(s.Namespace).Get(instanceName(s.Name), metav1.GetOptions{})
if err == nil {
// the deployment was found
return true, nil
}
if err != nil && !errors.IsNotFound(err) {
return false, err
}
// not found
return false, nil
}