feat: add session affinity for service (#27)

Co-authored-by: hardy <[email protected]>

Author: luohoufu

pkg/apis/common/types.go

@@ -104,6 +104,16 @@ type ServiceSpecPart struct {
 	// Annotations specific to the Service resource.
 	// +optional
 	Annotations map[string]string `json:"annotations,omitempty"`
+
+	// SessionAffinity specifies the session affinity for the Service.
+	// Supports "ClientIP" and "None".
+	// Defaults to "None" if not specified.
+	// +optional
+	SessionAffinity *corev1.ServiceAffinity `json:"sessionAffinity,omitempty"`
+
+	// SessionAffinityConfig contains the configurations of session affinity.
+	// +optional
+	SessionAffinityConfig *corev1.SessionAffinityConfig `json:"sessionAffinityConfig,omitempty"`
 }
 
 // PersistenceSpec defines configuration for a shared PersistentVolumeClaim (for Deployment).

pkg/apis/common/zz_generated.deepcopy.go

@@ -254,7 +254,7 @@ func (b *RuntimeBuilderStrategy) BuildObjects(ctx context.Context, k8sClient cli
 		regularServiceMetadata := builders.BuildObjectMeta(regularServiceName, namespace, commonLabels, runtimeConfig.Service.Annotations)
 		clientServicePorts := builders.BuildServicePorts(runtimeConfig.Service.Ports)
 
-		var serviceType corev1.ServiceType = corev1.ServiceTypeClusterIP // Default
+		var serviceType = corev1.ServiceTypeClusterIP // Default
 		if runtimeConfig.Service.Type != nil {
 			serviceType = *runtimeConfig.Service.Type
 		}
@@ -263,7 +263,12 @@ func (b *RuntimeBuilderStrategy) BuildObjects(ctx context.Context, k8sClient cli
 			Selector: selectorLabels,
 			Ports:    clientServicePorts,
 			Type:     serviceType,
-			// Other service spec fields...
+			// Session affinity configuration
+			SessionAffinityConfig: runtimeConfig.Service.SessionAffinityConfig,
+		}
+		// Set session affinity if specified
+		if runtimeConfig.Service.SessionAffinity != nil {
+			clientServiceSpec.SessionAffinity = *runtimeConfig.Service.SessionAffinity
 		}
 
 		logger.V(1).Info("Building Client Service object", "name", regularServiceName, "type", serviceType)