dae/component/sniffing/sniffing_bench_test.go

49 lines
1.0 KiB
Go

/*
* SPDX-License-Identifier: AGPL-3.0-only
* Copyright (c) 2022-2024, daeuniverse Organization <dae@v2raya.org>
*/
package sniffing
import (
"fmt"
"testing"
"github.com/daeuniverse/dae/common"
"github.com/daeuniverse/outbound/pkg/fastrand"
)
var (
httpMethodSet map[string]struct{}
)
func init() {
httpMethods := []string{"GET", "POST", "PUT", "PATCH", "DELETE", "COPY", "HEAD", "OPTIONS", "LINK", "UNLINK", "PURGE", "LOCK", "UNLOCK", "PROPFIND", "CONNECT", "TRACE"}
httpMethodSet = make(map[string]struct{})
for _, method := range httpMethods {
httpMethodSet[method] = struct{}{}
}
}
func BenchmarkStringSet(b *testing.B) {
for i := 0; i < b.N; i++ {
var test [5]byte
fastrand.Read(test[:])
_, ok := httpMethodSet[string(test[:])]
if !ok {
fmt.Sprintf("%v", string(test[:]))
}
}
}
func BenchmarkStringSwitch(b *testing.B) {
for i := 0; i < b.N; i++ {
var test [5]byte
fastrand.Read(test[:])
if !common.IsValidHttpMethod(string(test[:])) {
fmt.Sprintf("%v", string(test[:]))
}
}
}