check that ProbeThreshold is valid

Author: ushitora-anqou

api/pie/v1alpha1/pieprobe_types.go

@@ -23,7 +23,7 @@ type PieProbeSpec struct {
 	//+kubebuilder:validation:Minimum:=1
 	ProbePeriod int `json:"probePeriod"`
 
-	ProbeThreshold string `json:"probeThreshold"`
+	ProbeThreshold metav1.Duration `json:"probeThreshold"`
 }
 
 // PieProbeStatus defines the observed state of PieProbe

api/pie/v1alpha1/zz_generated.deepcopy.go

@@ -4,9 +4,11 @@ import (
 	"context"
 	"crypto/sha1"
 	"encoding/hex"
+	"errors"
 	"fmt"
 	"hash/crc32"
 	"io"
+	"time"
 
 	piev1alpha1 "github.com/topolvm/pie/api/pie/v1alpha1"
 	"github.com/topolvm/pie/constants"
@@ -68,6 +70,10 @@ func (r *PieProbeReconciler) Reconcile(ctx context.Context, req ctrl.Request) (c
 		return ctrl.Result{}, err
 	}
 
+	if time.Duration(pieProbe.Spec.ProbePeriod)*time.Minute <= pieProbe.Spec.ProbeThreshold.Duration {
+		return ctrl.Result{}, errors.New("probe period should be larger than probe threshold")
+	}
+
 	var storageClassForGet storagev1.StorageClass
 	err = r.client.Get(ctx, client.ObjectKey{Name: pieProbe.Spec.MonitoringStorageClass}, &storageClassForGet)
 	if err != nil {

internal/controller/pie/pieprobe_controller.go

@@ -77,10 +77,7 @@ func (r *ProbePodReconciler) Reconcile(ctx context.Context, req ctrl.Request) (c
 	}
 	r.po.setPodPieProbeName(pod.Namespace, pod.Name, pieProbeName)
 
-	probeThreshold, err := time.ParseDuration(pieProbe.Spec.ProbeThreshold)
-	if err != nil {
-		return ctrl.Result{}, err
-	}
+	probeThreshold := pieProbe.Spec.ProbeThreshold.Duration
 	r.po.setProbeThreshold(pod.Namespace, pod.Name, probeThreshold)
 
 	for _, status := range pod.Status.ContainerStatuses {