kubernetes/vendor/github.com/emicklei/go-restful/v3/custom_verb.go

package restful

// Copyright 2025 Ernest Micklei. All rights reserved.
// Use of this source code is governed by a license
// that can be found in the LICENSE file.

import (
	"fmt"
	"regexp"
	"sync"
)

var (
	customVerbReg          = regexp.MustCompile(":([A-Za-z]+)$")
	customVerbCache        sync.Map // Cache for compiled custom verb regexes
	customVerbCacheEnabled = true   // Enable/disable custom verb regex caching
)

// SetCustomVerbCacheEnabled enables or disables custom verb regex caching.
// When disabled, custom verb regex patterns will be compiled on every request.
// When enabled (default), compiled custom verb regex patterns are cached for better performance.
func SetCustomVerbCacheEnabled(enabled bool) {
	customVerbCacheEnabled = enabled
}

func hasCustomVerb(routeToken string) bool {
	return customVerbReg.MatchString(routeToken)
}

func isMatchCustomVerb(routeToken string, pathToken string) bool {
	rs := customVerbReg.FindStringSubmatch(routeToken)
	if len(rs) < 2 {
		return false
	}

	customVerb := rs[1]
	regexPattern := fmt.Sprintf(":%s$", customVerb)

	// Check cache first (if enabled)
	if customVerbCacheEnabled {
		if specificVerbReg, found := getCachedRegexp(&customVerbCache, regexPattern); found {
			return specificVerbReg.MatchString(pathToken)
		}
	}

	// Compile the regex
	specificVerbReg := regexp.MustCompile(regexPattern)

	// Cache the regex (if enabled)
	if customVerbCacheEnabled {
		customVerbCache.Store(regexPattern, specificVerbReg)
	}

	return specificVerbReg.MatchString(pathToken)
}

func removeCustomVerb(str string) string {
	return customVerbReg.ReplaceAllString(str, "")
}