@@ -64,29 +64,38 @@ func (o *Options) run() error {
6464 return o .runWithClient (kubeClient , clusterClient )
6565}
6666
67- func (o * Options ) runWithClient (kubeClient * kubernetes.Clientset , clusterClient * clusterclientset.Clientset ) error {
67+ func (o * Options ) runWithClient (kubeClient * kubernetes.Clientset , clusterClient * clusterclientset.Clientset ) ( err error ) {
6868 for _ , clusterName := range o .values .clusters {
69- err := wait .PollImmediate (1 * time .Second , time .Duration (o .timeout )* time .Second , func () (bool , error ) {
70- csrApproved , err := o .approveCSR (kubeClient , clusterName )
71- if err != nil {
72- return false , err
73- }
74- mcUpdated , err := o .updateManagedCluster (clusterClient , clusterName )
75- if err != nil {
76- return false , err
77- }
78- if csrApproved && mcUpdated {
79- return true , nil
80- }
81- return false , nil
82- })
69+ if o .wait == 0 {
70+ _ , err = o .accept (kubeClient , clusterClient , clusterName )
71+ } else {
72+ err = wait .PollImmediate (1 * time .Second , time .Duration (o .wait )* time .Second , func () (bool , error ) {
73+ return o .accept (kubeClient , clusterClient , clusterName )
74+ })
75+ }
8376 if err != nil {
8477 return err
8578 }
8679 }
8780 return nil
8881}
8982
83+ func (o * Options ) accept (kubeClient * kubernetes.Clientset , clusterClient * clusterclientset.Clientset , clusterName string ) (bool , error ) {
84+ csrApproved , err := o .approveCSR (kubeClient , clusterName )
85+ if err != nil {
86+ return false , err
87+ }
88+ mcUpdated , err := o .updateManagedCluster (clusterClient , clusterName )
89+ if err != nil {
90+ return false , err
91+ }
92+ if csrApproved && mcUpdated {
93+ return true , nil
94+ }
95+ return false , nil
96+
97+ }
98+
9099func (o * Options ) approveCSR (kubeClient * kubernetes.Clientset , clusterName string ) (bool , error ) {
91100 csrs , err := kubeClient .CertificatesV1 ().CertificateSigningRequests ().List (context .TODO (),
92101 metav1.ListOptions {
0 commit comments