make pd start timeout configurable

docs/api-references/docs.md

@@ -11528,6 +11528,17 @@ string
 <p>Start up script version</p>
 </td>
 </tr>
+<tr>
+<td>
+<code>startTimeout</code></br>
+<em>
+int
+</em>
+</td>
+<td>
+<p>Timeout threshold when pd get started</p>
+</td>
+</tr>
 </tbody>
 </table>
 <h3 id="pdstatus">PDStatus</h3>

manifests/crd.yaml

@@ -19683,6 +19683,9 @@ spec:
                     type: object
                   serviceAccount:
                     type: string
+                  startTimeout:
+                    default: 30
+                    type: integer
                   startUpScriptVersion:
                     enum:
                     - ""

manifests/crd/v1/pingcap.com_tidbclusters.yaml

@@ -5065,6 +5065,9 @@ spec:
                     type: object
                   serviceAccount:
                     type: string
+                  startTimeout:
+                    default: 30
+                    type: integer
                   startUpScriptVersion:
                     enum:
                     - ""

manifests/crd/v1beta1/pingcap.com_tidbclusters.yaml

@@ -5059,6 +5059,8 @@ spec:
                   type: object
                 serviceAccount:
                   type: string
+                startTimeout:
+                  type: integer
                 startUpScriptVersion:
                   enum:
                   - ""

manifests/crd_v1beta1.yaml

@@ -19661,6 +19661,8 @@ spec:
                   type: object
                 serviceAccount:
                   type: string
+                startTimeout:
+                  type: integer
                 startUpScriptVersion:
                   enum:
                   - ""

pkg/apis/pingcap/v1alpha1/types.go

@@ -514,6 +514,10 @@ type PDSpec struct {
 	// +optional
 	// +kubebuilder:validation:Enum:="";"v1"
 	StartUpScriptVersion string `json:"startUpScriptVersion,omitempty"`
+
+	// Timeout threshold when pd get started
+	// +kubebuilder:default=30
+	StartTimeout int `json:"startTimeout,omitempty"`
 }
 
 // TiKVSpec contains details of TiKV members

pkg/manager/member/startscript/v1/render_script.go

@@ -63,8 +63,9 @@ func RenderPDStartScript(tc *v1alpha1.TidbCluster) (string, error) {
 			AcrossK8s:     tc.AcrossK8s(),
 			ClusterDomain: tc.Spec.ClusterDomain,
 		},
-		Scheme:  tc.Scheme(),
-		DataDir: filepath.Join(constants.PDDataVolumeMountPath, tc.Spec.PD.DataSubDir),
+		Scheme:         tc.Scheme(),
+		DataDir:        filepath.Join(constants.PDDataVolumeMountPath, tc.Spec.PD.DataSubDir),
+		PDStartTimeout: tc.Spec.PD.StartTimeout,
 	}
 	if tc.Spec.PD.StartUpScriptVersion == "v1" {
 		model.CheckDomainScript = checkDNSV1

pkg/manager/member/startscript/v1/template.go

@@ -155,7 +155,7 @@ encoded_domain_url=` + "`" + `echo ${domain}:2380 | base64 | tr "\n" " " | sed "
 	`
 elapseTime=0
 period=1
-threshold=30
+threshold={{ .PDStartTimeout }}
 while true; do
 sleep ${period}
 elapseTime=$(( elapseTime+period ))
@@ -245,6 +245,7 @@ type PDStartScriptModel struct {
 	Scheme            string
 	DataDir           string
 	CheckDomainScript string
+	PDStartTimeout    int
 }
 
 var tikvStartScriptTplText = `#!/bin/sh

pkg/manager/member/startscript/v1/template_test.go

@@ -1227,7 +1227,9 @@ exec /pd-server ${ARGS}
 
 			tc := &v1alpha1.TidbCluster{
 				Spec: v1alpha1.TidbClusterSpec{
-					PD: &v1alpha1.PDSpec{},
+					PD: &v1alpha1.PDSpec{
+						StartTimeout: 30,
+					},
 				},
 			}
 			tc.Name = "test-pd"

pkg/manager/member/startscript/v2/pd_start_script.go

@@ -35,6 +35,7 @@ type PDStartScriptModel struct {
 	AdvertiseClientURL string
 	DiscoveryAddr      string
 	ExtraArgs          string
+	PDStartTimeout     int
 }
 
 // RenderPDStartScript renders PD start script from TidbCluster
@@ -66,6 +67,8 @@ func RenderPDStartScript(tc *v1alpha1.TidbCluster) (string, error) {
 
 	m.DiscoveryAddr = fmt.Sprintf("%s-discovery.%s:10261", tcName, tcNS)
 
+	m.PDStartTimeout = tc.Spec.PD.StartTimeout
+
 	return renderTemplateFunc(pdStartScriptTpl, m)
 }
 
@@ -80,7 +83,7 @@ PD_DOMAIN={{ .PDDomain }}
 
 elapseTime=0
 period=1
-threshold=30
+threshold={{ .PDStartTimeout }}
 while true; do
     sleep ${period}
     elapseTime=$(( elapseTime+period ))

pkg/manager/member/startscript/v2/pd_start_script_test.go

@@ -543,7 +543,9 @@ exec /pd-server ${ARGS}
 
 		tc := &v1alpha1.TidbCluster{
 			Spec: v1alpha1.TidbClusterSpec{
-				PD: &v1alpha1.PDSpec{},
+				PD: &v1alpha1.PDSpec{
+					StartTimeout: 30,
+				},
 			},
 		}
 		tc.Name = "start-script-test"