[Feature] Add Internal Port (#713)

Author: ajanikow

CHANGELOG.md

@@ -1,6 +1,7 @@
 # Change Log
 
 ## [master](https://github.com/arangodb/kube-arangodb/tree/master) (N/A)
+- Add InternalPort to ServerGroupSpec to allow user to expose tcp connection over localhost for sidecars
 
 ## [1.1.7](https://github.com/arangodb/kube-arangodb/tree/1.1.7) (2021-04-14)
 - Bump Kubernetes Dependencies to 1.19.x

pkg/apis/deployment/v1/server_group_spec.go

@@ -139,6 +139,8 @@ type ServerGroupSpec struct {
 	ShutdownMethod *ServerGroupShutdownMethod `json:"shutdownMethod,omitempty"`
 	// ShutdownDelay define how long operator should delay finalizer removal after shutdown
 	ShutdownDelay *int `json:"shutdownDelay,omitempty"`
+	// InternalPort define port used in internal communication, can be accessed over localhost via sidecar
+	InternalPort *int `json:"internalPort,omitempty"`
 }
 
 // ServerGroupSpecSecurityContext contains specification for pod security context
@@ -499,6 +501,12 @@ func (s ServerGroupSpec) Validate(group ServerGroup, used bool, mode DeploymentM
 	} else if s.GetCount() != 0 {
 		return errors.WithStack(errors.Wrapf(ValidationError, "Invalid count value %d for un-used group. Expected 0", s.GetCount()))
 	}
+	if port := s.InternalPort; port != nil {
+		switch p := *port; p {
+		case 8529:
+			return errors.WithStack(errors.Wrapf(ValidationError, "Port %d already in use", p))
+		}
+	}
 	return nil
 }
 

pkg/apis/deployment/v1/zz_generated.deepcopy.go

@@ -23,6 +23,7 @@
 package v2alpha1
 
 import (
+	"github.com/arangodb/kube-arangodb/pkg/apis/deployment"
 	meta "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
@@ -47,3 +48,17 @@ type ArangoMember struct {
 	Spec            ArangoMemberSpec   `json:"spec,omitempty"`
 	Status          ArangoMemberStatus `json:"status,omitempty"`
 }
+
+// AsOwner creates an OwnerReference for the given member
+func (a *ArangoMember) AsOwner() meta.OwnerReference {
+	trueVar := true
+	return meta.OwnerReference{
+		APIVersion: SchemeGroupVersion.String(),
+		Kind:       deployment.ArangoMemberResourceKind,
+		Name:       a.Name,
+		UID:        a.UID,
+		Controller: &trueVar,
+		// For now BlockOwnerDeletion does not work on OpenShift, so we leave it out.
+		//BlockOwnerDeletion: &trueVar,
+	}
+}

pkg/apis/deployment/v2alpha1/arango_member.go

@@ -285,3 +285,19 @@ func (ds DeploymentStatusMembers) MembersOfGroup(group ServerGroup) MemberStatus
 		return MemberStatusList{}
 	}
 }
+
+// PodNames returns all members pod names
+func (ds DeploymentStatusMembers) PodNames() []string {
+	var n []string
+
+	ds.ForeachServerGroup(func(group ServerGroup, list MemberStatusList) error {
+		for _, m := range list {
+			if m.PodName != "" {
+				n = append(n, m.PodName)
+			}
+		}
+		return nil
+	})
+
+	return n
+}

pkg/apis/deployment/v2alpha1/deployment_status_members.go

@@ -57,3 +57,8 @@ func (p MemberPhase) IsFailed() bool {
 func (p MemberPhase) IsCreatedOrDrain() bool {
 	return p == MemberPhaseCreated || p == MemberPhaseDrain
 }
+
+// String returns string from MemberPhase
+func (p MemberPhase) String() string {
+	return string(p)
+}

pkg/apis/deployment/v2alpha1/member_phase.go

@@ -26,6 +26,8 @@ import (
 	"reflect"
 	"time"
 
+	"github.com/arangodb/kube-arangodb/pkg/util/k8sutil"
+
 	"k8s.io/apimachinery/pkg/types"
 
 	driver "github.com/arangodb/go-driver"
@@ -155,3 +157,8 @@ func (s MemberStatus) IsNotReadySince(timestamp time.Time) bool {
 	// A
 	return s.CreatedAt.Time.Before(timestamp)
 }
+
+// ArangoMemberName create member name from given member
+func (s MemberStatus) ArangoMemberName(deploymentName string, group ServerGroup) string {
+	return k8sutil.CreatePodHostName(deploymentName, group.AsRole(), s.ID)
+}

pkg/apis/deployment/v2alpha1/member_status.go

@@ -23,6 +23,7 @@
 package v2alpha1
 
 import (
+	"encoding/json"
 	"time"
 )
 
@@ -34,14 +35,19 @@ func (g *ServerGroup) UnmarshalJSON(bytes []byte) error {
 		return nil
 	}
 
-	*g = ServerGroupFromRole(string(bytes))
+	var s string
+
+	if err := json.Unmarshal(bytes, &s); err != nil {
+		return err
+	}
+
+	*g = ServerGroupFromRole(s)
 
 	return nil
 }
 
 func (g ServerGroup) MarshalJSON() ([]byte, error) {
-	s := g.AsRole()
-	return []byte(s), nil
+	return json.Marshal(g.AsRole())
 }
 
 const (

pkg/apis/deployment/v2alpha1/server_group.go

@@ -137,6 +137,10 @@ type ServerGroupSpec struct {
 	InitContainers *ServerGroupInitContainers `json:"initContainers,omitempty"`
 	// ShutdownMethod describe procedure of member shutdown taken by Operator
 	ShutdownMethod *ServerGroupShutdownMethod `json:"shutdownMethod,omitempty"`
+	// ShutdownDelay define how long operator should delay finalizer removal after shutdown
+	ShutdownDelay *int `json:"shutdownDelay,omitempty"`
+	// InternalPort define port used in internal communication, can be accessed over localhost via sidecar
+	InternalPort *int `json:"internalPort,omitempty"`
 }
 
 // ServerGroupSpecSecurityContext contains specification for pod security context
@@ -497,6 +501,12 @@ func (s ServerGroupSpec) Validate(group ServerGroup, used bool, mode DeploymentM
 	} else if s.GetCount() != 0 {
 		return errors.WithStack(errors.Wrapf(ValidationError, "Invalid count value %d for un-used group. Expected 0", s.GetCount()))
 	}
+	if port := s.InternalPort; port != nil {
+		switch p := *port; p {
+		case 8529:
+			return errors.WithStack(errors.Wrapf(ValidationError, "Port %d already in use", p))
+		}
+	}
 	return nil
 }