Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: v.oleynikov <vasily.oleynikov@flant.com>
  • Loading branch information
duckhawk committed Mar 22, 2024
1 parent 196a640 commit 8d7782c
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 13 deletions.
15 changes: 8 additions & 7 deletions images/webhooks/src/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ import (
kwhhttp "github.com/slok/kubewebhook/v2/pkg/http"
kwhlogrus "github.com/slok/kubewebhook/v2/pkg/log/logrus"
kwhmutating "github.com/slok/kubewebhook/v2/pkg/webhook/mutating"
kwhvalidating "github.com/slok/kubewebhook/v2/pkg/webhook/validating"
corev1 "k8s.io/api/core/v1"
"net/http"
"os"
Expand Down Expand Up @@ -83,15 +84,15 @@ func main() {
os.Exit(1)
}

scuMF := kwhmutating.MutatorFunc(validators.StorageClassUpdate)
scuMF := kwhvalidating.ValidatorFunc(validators.StorageClassUpdate)

scuMCfg := kwhmutating.WebhookConfig{
ID: "StorageClassUpdate",
Obj: &v1alpha1.LocalStorageClass{},
Mutator: scuMF,
Logger: logger,
scuMCfg := kwhvalidating.WebhookConfig{
ID: "StorageClassUpdate",
Obj: &v1alpha1.LocalStorageClass{},
Validator: scuMF,
Logger: logger,
}
scuWh, err := kwhmutating.NewWebhook(scuMCfg)
scuWh, err := kwhvalidating.NewWebhook(scuMCfg)
if err != nil {
fmt.Fprintf(os.Stderr, "error creating webhook: %s", err)
os.Exit(1)
Expand Down
12 changes: 6 additions & 6 deletions images/webhooks/src/validators/storageClassUpdate.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,15 +22,15 @@ import (
"webhooks/v1alpha1"

"github.com/slok/kubewebhook/v2/pkg/model"
kwhmutating "github.com/slok/kubewebhook/v2/pkg/webhook/mutating"
kwhvalidating "github.com/slok/kubewebhook/v2/pkg/webhook/validating"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func StorageClassUpdate(ctx context.Context, _ *model.AdmissionReview, obj metav1.Object) (*kwhmutating.MutatorResult, error) {
func StorageClassUpdate(ctx context.Context, _ *model.AdmissionReview, obj metav1.Object) (*kwhvalidating.ValidatorResult, error) {
sc, ok := obj.(*v1alpha1.LocalStorageClass)
if !ok {
// If not a storage class just continue the mutation chain(if there is one) and do nothing.
return &kwhmutating.MutatorResult{}, nil
return &kwhvalidating.ValidatorResult{}, nil
}

thickExists := false
Expand All @@ -44,10 +44,10 @@ func StorageClassUpdate(ctx context.Context, _ *model.AdmissionReview, obj metav
}

if thickExists == true && thinExists == true {
return &kwhmutating.MutatorResult{}, errors.New("There must be only thin or thick pools simultaneously")
return &kwhvalidating.ValidatorResult{Valid: false}, errors.New("there must be only thin or thick pools simultaneously")
}

return &kwhmutating.MutatorResult{
MutatedObject: sc,
return &kwhvalidating.ValidatorResult{
Valid: true,
}, nil
}

0 comments on commit 8d7782c

Please sign in to comment.