forked from openyurtio/openyurt
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
implement the markautonomous subcommand (openyurtio#39)
- Loading branch information
1 parent
785cd39
commit d8d1cd0
Showing
6 changed files
with
204 additions
and
83 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,136 @@ | ||
| /* | ||
| Copyright 2020 The OpenYurt Authors. | ||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
| */ | ||
|
|
||
| package markautonomous | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| "github.com/spf13/cobra" | ||
| "github.com/spf13/pflag" | ||
| "k8s.io/api/core/v1" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/client-go/kubernetes" | ||
| "k8s.io/klog" | ||
|
|
||
| "github.com/alibaba/openyurt/pkg/yurtctl/constants" | ||
| kubeutil "github.com/alibaba/openyurt/pkg/yurtctl/util/kubernetes" | ||
| ) | ||
|
|
||
| // MarkAutonomousOptions has the information that required by convert operation | ||
| type MarkAutonomousOptions struct { | ||
| *kubernetes.Clientset | ||
| AutonomousNodes []string | ||
| MarkAllEdgeNodes bool | ||
| } | ||
|
|
||
| // NewMarkAutonomousOptions creates a new MarkAutonomousOptions | ||
| func NewMarkAutonomousOptions() *MarkAutonomousOptions { | ||
| return &MarkAutonomousOptions{} | ||
| } | ||
|
|
||
| // NewMarkAutonomousCmd generates a new markautonomous command | ||
| func NewMarkAutonomousCmd() *cobra.Command { | ||
| co := NewMarkAutonomousOptions() | ||
| cmd := &cobra.Command{ | ||
| Use: "markautonomous -a AUTONOMOUSNODES", | ||
| Short: "mark the nodes as autonomous", | ||
| Run: func(cmd *cobra.Command, _ []string) { | ||
| if err := co.Complete(cmd.Flags()); err != nil { | ||
| klog.Fatalf("fail to complete the markautonomous option: %s", err) | ||
| } | ||
| if err := co.RunMarkAutonomous(); err != nil { | ||
| klog.Fatalf("fail to make nodes autonomous: %s", err) | ||
| } | ||
| }, | ||
| } | ||
|
|
||
| cmd.Flags().StringP("autonomous-nodes", "a", "", | ||
| "The list of nodes that will be marked as autonomous."+ | ||
| "(e.g. -a autonomousnode1,autonomousnode2)") | ||
|
|
||
| return cmd | ||
| } | ||
|
|
||
| // Complete completes all the required options | ||
| func (mao *MarkAutonomousOptions) Complete(flags *pflag.FlagSet) error { | ||
| anStr, err := flags.GetString("autonomous-nodes") | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if anStr == "" { | ||
| mao.AutonomousNodes = []string{} | ||
| } else { | ||
| mao.AutonomousNodes = strings.Split(anStr, ",") | ||
| } | ||
|
|
||
| // set mark-all-edge-node to false, as user has specifed autonomous nodes | ||
| if len(mao.AutonomousNodes) == 0 { | ||
| mao.MarkAllEdgeNodes = true | ||
| } | ||
|
|
||
| mao.Clientset, err = kubeutil.GenClientSet(flags) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // RunMarkAutonomous annotates specified edge nodes as autonomous | ||
| func (mao *MarkAutonomousOptions) RunMarkAutonomous() error { | ||
| var autonomousNodes []*v1.Node | ||
| if mao.MarkAllEdgeNodes { | ||
| // make all edge nodes autonomous | ||
| labelSelector := fmt.Sprintf("%s=true", constants.LabelEdgeWorker) | ||
| edgeNodeList, err := mao.CoreV1().Nodes(). | ||
| List(metav1.ListOptions{LabelSelector: labelSelector}) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if len(edgeNodeList.Items) == 0 { | ||
| klog.Warning("there is no edge nodes, please label the edge node first") | ||
| return nil | ||
| } | ||
| for _, node := range edgeNodeList.Items { | ||
| autonomousNodes = append(autonomousNodes, &node) | ||
| } | ||
| } else { | ||
| // make only the specified edge nodes autonomous | ||
| for _, nodeName := range mao.AutonomousNodes { | ||
| node, err := mao.CoreV1().Nodes().Get(nodeName, metav1.GetOptions{}) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if node.Labels[constants.LabelEdgeWorker] == "false" { | ||
| return fmt.Errorf("can't make cloud node(%s) autonomous", | ||
| node.GetName()) | ||
| } | ||
| autonomousNodes = append(autonomousNodes, node) | ||
| } | ||
| } | ||
|
|
||
| for _, anode := range autonomousNodes { | ||
| klog.Infof("mark %s as autonomous", anode.GetName()) | ||
| if _, err := kubeutil.AnnotateNode(mao.Clientset, | ||
| anode, constants.AnnotationAutonomy, "true"); err != nil { | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } |
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
Oops, something went wrong.