-
Notifications
You must be signed in to change notification settings - Fork 25
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #13 from deggja/network_map
feat: add network maps to dashboard
- Loading branch information
Showing
9 changed files
with
1,962 additions
and
265 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| package k8s | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "log" | ||
| "net/http" | ||
|
|
||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| ) | ||
|
|
||
| // VisualizationData represents the structure of network policy and pod data for visualization. | ||
| type VisualizationData struct { | ||
| Policies []PolicyVisualization `json:"policies"` | ||
| } | ||
|
|
||
| // PolicyVisualization represents a network policy and the pods it affects for visualization purposes. | ||
| type PolicyVisualization struct { | ||
| Name string `json:"name"` | ||
| Namespace string `json:"namespace"` | ||
| TargetPods []string `json:"targetPods"` | ||
| } | ||
|
|
||
| // gatherVisualizationData retrieves network policies and associated pods for visualization. | ||
| func gatherVisualizationData(namespace string) (*VisualizationData, error) { | ||
| clientset, err := GetClientset() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Retrieve all network policies in the specified namespace | ||
| policies, err := clientset.NetworkingV1().NetworkPolicies(namespace).List(context.TODO(), metav1.ListOptions{}) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| vizData := &VisualizationData{ | ||
| Policies: make([]PolicyVisualization, 0), // Initialize as empty slice | ||
| } | ||
|
|
||
| // Iterate over the retrieved policies to build the visualization data | ||
| for _, policy := range policies.Items { | ||
| // For each policy, find the pods that match its pod selector | ||
| selector, err := metav1.LabelSelectorAsSelector(&policy.Spec.PodSelector) | ||
| if err != nil { | ||
| log.Printf("Error parsing selector for policy %s: %v\n", policy.Name, err) | ||
| continue | ||
| } | ||
|
|
||
| pods, err := clientset.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{ | ||
| LabelSelector: selector.String(), | ||
| }) | ||
| if err != nil { | ||
| log.Printf("Error listing pods for policy %s: %v\n", policy.Name, err) | ||
| continue | ||
| } | ||
|
|
||
| podNames := make([]string, 0, len(pods.Items)) | ||
| for _, pod := range pods.Items { | ||
| podNames = append(podNames, pod.Name) | ||
| } | ||
|
|
||
| vizData.Policies = append(vizData.Policies, PolicyVisualization{ | ||
| Name: policy.Name, | ||
| Namespace: policy.Namespace, | ||
| TargetPods: podNames, | ||
| }) | ||
| } | ||
|
|
||
| return vizData, nil | ||
| } | ||
|
|
||
| // HandleVisualizationRequest handles the HTTP request for serving visualization data. | ||
| func HandleVisualizationRequest(w http.ResponseWriter, r *http.Request) { | ||
| if r.Method != http.MethodGet { | ||
| http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed) | ||
| return | ||
| } | ||
|
|
||
| namespace := r.URL.Query().Get("namespace") | ||
|
|
||
| vizData, err := gatherVisualizationData(namespace) | ||
| if err != nil { | ||
| http.Error(w, err.Error(), http.StatusInternalServerError) | ||
| return | ||
| } | ||
|
|
||
| w.Header().Set("Content-Type", "application/json") | ||
| if err := json.NewEncoder(w).Encode(vizData); err != nil { | ||
| http.Error(w, "Failed to encode visualization data", http.StatusInternalServerError) | ||
| } | ||
| } | ||
|
|
||
| // gatherNamespacesWithPolicies returns a list of all namespaces that contain network policies. | ||
| func GatherNamespacesWithPolicies() ([]string, error) { | ||
| clientset, err := GetClientset() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Retrieve all namespaces | ||
| namespaces, err := clientset.CoreV1().Namespaces().List(context.TODO(), metav1.ListOptions{}) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| var namespacesWithPolicies []string | ||
|
|
||
| // Check each namespace for network policies | ||
| for _, ns := range namespaces.Items { | ||
| policies, err := clientset.NetworkingV1().NetworkPolicies(ns.Name).List(context.TODO(), metav1.ListOptions{}) | ||
| if err != nil { | ||
| log.Printf("Error listing policies in namespace %s: %v\n", ns.Name, err) | ||
| continue | ||
| } | ||
|
|
||
| if len(policies.Items) > 0 { | ||
| namespacesWithPolicies = append(namespacesWithPolicies, ns.Name) | ||
| } | ||
| } | ||
|
|
||
| return namespacesWithPolicies, nil | ||
| } | ||
|
|
||
| // gatherClusterVisualizationData retrieves visualization data for all namespaces with network policies. | ||
| func GatherClusterVisualizationData() ([]VisualizationData, error) { | ||
| namespacesWithPolicies, err := GatherNamespacesWithPolicies() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Slice to hold the visualization data for the entire cluster | ||
| var clusterVizData []VisualizationData | ||
|
|
||
| for _, namespace := range namespacesWithPolicies { | ||
| vizData, err := gatherVisualizationData(namespace) | ||
| if err != nil { | ||
| log.Printf("Error gathering visualization data for namespace %s: %v\n", namespace, err) | ||
| continue | ||
| } | ||
| clusterVizData = append(clusterVizData, *vizData) | ||
| } | ||
|
|
||
| return clusterVizData, nil | ||
| } |
Large diffs are not rendered by default.
Oops, something went wrong.
Oops, something went wrong.