fix: environment variable(Connection) may not exist (#14)

Author: 0x5457

internal/webhook/v1/pod_webhook.go

@@ -139,7 +139,18 @@ func ParseTFReq(pod *corev1.Pod) []TFReq {
 			ConnectionName:      connectionNameEnv.Name,
 			ConnectionNamespace: connectionNamespaceEnv.Name,
 		}
-
+		connectionNameEnv, ok := lo.Find(container.Env, func(e corev1.EnvVar) bool {
+			return e.Name == constants.ConnectionNameEnv
+		})
+		if ok {
+			req.ConnectionName = connectionNameEnv.Name
+		}
+		connectionNamespaceEnv, ok := lo.Find(container.Env, func(e corev1.EnvVar) bool {
+			return e.Name == constants.ConnectionNamespaceEnv
+		})
+		if ok {
+			req.ConnectionNamespace = connectionNamespaceEnv.Name
+		}
 		// Parse TFLOPS requirement
 		if hasTflops {
 			tflops, err := resource.ParseQuantity(tflopsStr)
@@ -235,7 +246,6 @@ func (m *TensorFusionPodMutator) patchTFClient(pod *corev1.Pod, tfReq []TFReq) (
 	}
 
 	// Generate JSON patch operations by comparing original and patched pod
-
 	strategicpatches, err := jsonpatch.CreatePatch(currentBytes, resultBytes)
 	if err != nil {
 		return nil, fmt.Errorf("create json patch: %v", err)