mirror of
https://github.com/andwati/telecom-platform.git
synced 2026-10-18 11:02:54 +00:00
feat: Add Kubernetes service manager with deployment operations and pod log streaming
- Add kubernetes package with ServiceManager struct containing clientset and namespace fields - Add Service struct with Name, Status, Version, Uptime, CPU, Memory, Replicas, and Available fields - Add NewServiceManager constructor with in-cluster config and kubeconfig fallback - Add KUBERNETES_NAMESPACE environment variable with telecom-platform-dev default - Add getInClusterNamespace helper to read namespace from service
This commit is contained in:
@@ -0,0 +1,218 @@
|
||||
package kubernetes
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/labels"
|
||||
)
|
||||
|
||||
// getInClusterNamespace reads the namespace from the in-cluster service account
|
||||
func getInClusterNamespace() (string, error) {
|
||||
data, err := os.ReadFile("/var/run/secrets/kubernetes.io/serviceaccount/namespace")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return string(data), nil
|
||||
}
|
||||
|
||||
// ListServices lists all services/deployments in the namespace
|
||||
func (sm *ServiceManager) ListServices() ([]Service, error) {
|
||||
deployments, err := sm.clientset.AppsV1().Deployments(sm.namespace).List(context.Background(), metav1.ListOptions{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to list deployments: %w", err)
|
||||
}
|
||||
|
||||
services := make([]Service, 0, len(deployments.Items))
|
||||
now := time.Now()
|
||||
|
||||
for _, d := range deployments.Items {
|
||||
uptime := "0m"
|
||||
if !d.CreationTimestamp.IsZero() {
|
||||
uptime = now.Sub(d.CreationTimestamp.Time).Round(time.Minute).String()
|
||||
}
|
||||
|
||||
status := "Running"
|
||||
if d.Status.UnavailableReplicas > 0 {
|
||||
status = "Degraded"
|
||||
}
|
||||
if d.Spec.Replicas != nil && *d.Spec.Replicas == 0 {
|
||||
status = "Stopped"
|
||||
}
|
||||
|
||||
version := "v1.0.0"
|
||||
if d.Annotations != nil {
|
||||
if v, ok := d.Annotations["version"]; ok {
|
||||
version = v
|
||||
}
|
||||
}
|
||||
|
||||
services = append(services, Service{
|
||||
Name: d.Name,
|
||||
Status: status,
|
||||
Version: version,
|
||||
Uptime: uptime,
|
||||
Replicas: *d.Spec.Replicas,
|
||||
Available: d.Status.AvailableReplicas,
|
||||
})
|
||||
}
|
||||
|
||||
return services, nil
|
||||
}
|
||||
|
||||
// GetServiceStatus gets detailed status for a specific service
|
||||
func (sm *ServiceManager) GetServiceStatus(serviceName string) (*Service, error) {
|
||||
deployment, err := sm.clientset.AppsV1().Deployments(sm.namespace).Get(context.Background(), serviceName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to get deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
uptime := "0m"
|
||||
if !deployment.CreationTimestamp.IsZero() {
|
||||
uptime = now.Sub(deployment.CreationTimestamp.Time).Round(time.Minute).String()
|
||||
}
|
||||
|
||||
status := "Running"
|
||||
if deployment.Status.UnavailableReplicas > 0 {
|
||||
status = "Degraded"
|
||||
}
|
||||
if deployment.Spec.Replicas != nil && *deployment.Spec.Replicas == 0 {
|
||||
status = "Stopped"
|
||||
}
|
||||
|
||||
version := "v1.0.0"
|
||||
if deployment.Annotations != nil {
|
||||
if v, ok := deployment.Annotations["version"]; ok {
|
||||
version = v
|
||||
}
|
||||
}
|
||||
|
||||
// Get pods for CPU/memory info
|
||||
pods, err := sm.clientset.CoreV1().Pods(sm.namespace).List(context.Background(), metav1.ListOptions{
|
||||
LabelSelector: labels.Set(deployment.Spec.Selector.MatchLabels).AsSelector().String(),
|
||||
})
|
||||
if err == nil && len(pods.Items) > 0 {
|
||||
// Simple estimation - in production use metrics API
|
||||
cpu := 45.0
|
||||
memory := "256MB"
|
||||
return &Service{
|
||||
Name: deployment.Name,
|
||||
Status: status,
|
||||
Version: version,
|
||||
Uptime: uptime,
|
||||
CPU: cpu,
|
||||
Memory: memory,
|
||||
Replicas: *deployment.Spec.Replicas,
|
||||
Available: deployment.Status.AvailableReplicas,
|
||||
}, nil
|
||||
}
|
||||
|
||||
return &Service{
|
||||
Name: deployment.Name,
|
||||
Status: status,
|
||||
Version: version,
|
||||
Uptime: uptime,
|
||||
CPU: 0.0,
|
||||
Memory: "0MB",
|
||||
Replicas: *deployment.Spec.Replicas,
|
||||
Available: deployment.Status.AvailableReplicas,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// RestartDeployment restarts a deployment by rolling its pods
|
||||
func (sm *ServiceManager) RestartDeployment(serviceName string) error {
|
||||
deployment, err := sm.clientset.AppsV1().Deployments(sm.namespace).Get(context.Background(), serviceName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
// Add restart annotation to trigger rolling restart
|
||||
if deployment.Spec.Template.Annotations == nil {
|
||||
deployment.Spec.Template.Annotations = make(map[string]string)
|
||||
}
|
||||
deployment.Spec.Template.Annotations["kubectl.kubernetes.io/restartedAt"] = time.Now().UTC().Format(time.RFC3339)
|
||||
|
||||
_, err = sm.clientset.AppsV1().Deployments(sm.namespace).Update(context.Background(), deployment, metav1.UpdateOptions{})
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to restart deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetPodLogs fetches recent logs from pods for a deployment
|
||||
func (sm *ServiceManager) GetPodLogs(serviceName string, tailLines int64) ([]string, error) {
|
||||
deployment, err := sm.clientset.AppsV1().Deployments(sm.namespace).Get(context.Background(), serviceName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to get deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
labelSelector := labels.Set(deployment.Spec.Selector.MatchLabels).AsSelector()
|
||||
pods, err := sm.clientset.CoreV1().Pods(sm.namespace).List(context.Background(), metav1.ListOptions{
|
||||
LabelSelector: labelSelector.String(),
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to list pods for deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
if len(pods.Items) == 0 {
|
||||
return []string{"No pods found for deployment " + serviceName}, nil
|
||||
}
|
||||
|
||||
// Get logs from the first running pod
|
||||
var pod *corev1.Pod
|
||||
for _, p := range pods.Items {
|
||||
if p.Status.Phase == corev1.PodRunning {
|
||||
pod = &p
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if pod == nil {
|
||||
pod = &pods.Items[0] // Use first pod even if not running
|
||||
}
|
||||
|
||||
req := sm.clientset.CoreV1().Pods(sm.namespace).GetLogs(pod.Name, &corev1.PodLogOptions{
|
||||
TailLines: &tailLines,
|
||||
})
|
||||
|
||||
logData, err := req.Stream(context.Background())
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to stream logs from pod %s: %w", pod.Name, err)
|
||||
}
|
||||
defer logData.Close()
|
||||
|
||||
buf := make([]byte, 1024*1024) // 1MB limit
|
||||
n, err := logData.Read(buf)
|
||||
if err != nil && err.Error() != "EOF" {
|
||||
return nil, fmt.Errorf("failed to read logs: %w", err)
|
||||
}
|
||||
|
||||
logs := string(buf[:n])
|
||||
// Split logs into lines
|
||||
logLines := []string{}
|
||||
currentLine := ""
|
||||
for _, c := range logs {
|
||||
if c == '\n' {
|
||||
logLines = append(logLines, currentLine)
|
||||
currentLine = ""
|
||||
} else {
|
||||
currentLine += string(c)
|
||||
}
|
||||
}
|
||||
if currentLine != "" {
|
||||
logLines = append(logLines, currentLine)
|
||||
}
|
||||
|
||||
// Return last 20 lines if there are more
|
||||
if len(logLines) > 20 {
|
||||
logLines = logLines[len(logLines)-20:]
|
||||
}
|
||||
|
||||
return logLines, nil
|
||||
}
|
||||
@@ -0,0 +1,95 @@
|
||||
package kubernetes
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/client-go/kubernetes"
|
||||
"k8s.io/client-go/rest"
|
||||
"k8s.io/client-go/tools/clientcmd"
|
||||
"k8s.io/client-go/util/homedir"
|
||||
)
|
||||
|
||||
// NewServiceManager creates a new Kubernetes service manager
|
||||
func NewServiceManager() (*ServiceManager, error) {
|
||||
var config *rest.Config
|
||||
var err error
|
||||
|
||||
// Try in-cluster config first
|
||||
if config, err = rest.InClusterConfig(); err != nil {
|
||||
// Fall back to kubeconfig
|
||||
kubeconfig := filepath.Join(homedir.HomeDir(), ".kube", "config")
|
||||
if config, err = clientcmd.BuildConfigFromFlags("", kubeconfig); err != nil {
|
||||
return nil, fmt.Errorf("failed to create Kubernetes config: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
clientset, err := kubernetes.NewForConfig(config)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to create Kubernetes clientset: %w", err)
|
||||
}
|
||||
|
||||
// Determine namespace
|
||||
namespace := os.Getenv("KUBERNETES_NAMESPACE")
|
||||
if namespace == "" {
|
||||
if ns, err := getInClusterNamespace(); err == nil {
|
||||
namespace = ns
|
||||
} else {
|
||||
namespace = "telecom-platform-dev"
|
||||
}
|
||||
}
|
||||
|
||||
return &ServiceManager{
|
||||
clientset: clientset,
|
||||
namespace: namespace,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// ScaleDeployment scales a deployment to the specified replica count
|
||||
func (sm *ServiceManager) ScaleDeployment(serviceName string, replicas int32) error {
|
||||
deployment, err := sm.clientset.AppsV1().Deployments(sm.namespace).Get(context.Background(), serviceName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
deployment.Spec.Replicas = &replicas
|
||||
_, err = sm.clientset.AppsV1().Deployments(sm.namespace).Update(context.Background(), deployment, metav1.UpdateOptions{})
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to scale deployment %s: %w", serviceName, err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// StartAllServices scales all platform services up
|
||||
func (sm *ServiceManager) StartAllServices() error {
|
||||
services := []string{"api-server", "charging-engine", "carrier-connector", "packet-gateway", "web-dashboard"}
|
||||
replicas := int32(1)
|
||||
|
||||
for _, svc := range services {
|
||||
err := sm.ScaleDeployment(svc, replicas)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to start service %s: %w", svc, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// StopAllServices scales all platform services down
|
||||
func (sm *ServiceManager) StopAllServices() error {
|
||||
services := []string{"web-dashboard", "packet-gateway", "carrier-connector", "charging-engine", "api-server"}
|
||||
replicas := int32(0)
|
||||
|
||||
for _, svc := range services {
|
||||
err := sm.ScaleDeployment(svc, replicas)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to stop service %s: %w", svc, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,21 @@
|
||||
package kubernetes
|
||||
|
||||
import "k8s.io/client-go/kubernetes"
|
||||
|
||||
// ServiceManager handles Kubernetes service operations
|
||||
type ServiceManager struct {
|
||||
clientset *kubernetes.Clientset
|
||||
namespace string
|
||||
}
|
||||
|
||||
// Service represents a Kubernetes service/deployment status
|
||||
type Service struct {
|
||||
Name string
|
||||
Status string
|
||||
Version string
|
||||
Uptime string
|
||||
CPU float64
|
||||
Memory string
|
||||
Replicas int32
|
||||
Available int32
|
||||
}
|
||||
Reference in New Issue
Block a user