Skip to content

Commit

Permalink
filters/builtin: performance optimization for sanitize in StripQuery …
Browse files Browse the repository at this point in the history
…Filter (#2940)

pkg: github.com/zalando/skipper/filters/builtin
           │   HEAD~1    │                HEAD                │
           │   sec/op    │   sec/op     vs base               │
Sanitize-8   802.1n ± 0%   748.3n ± 0%  -6.71% (p=0.000 n=10)

           │   HEAD~1   │                HEAD                │
           │    B/op    │    B/op     vs base                │
Sanitize-8   472.0 ± 0%   192.0 ± 0%  -59.32% (p=0.000 n=10)

           │   HEAD~1   │               HEAD                │
           │ allocs/op  │ allocs/op   vs base               │
Sanitize-8   15.00 ± 0%   14.00 ± 0%  -6.67% (p=0.000 n=10)

Signed-off-by: Troy Kohler <troyalekson@gmail.com>

---------

Signed-off-by: Troy Kohler <troyalekson@gmail.com>
  • Loading branch information
trkohler authored Feb 23, 2024
1 parent 07b5566 commit e7dd553
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 5 deletions.
7 changes: 3 additions & 4 deletions filters/builtin/stripquery.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
package builtin

import (
"bytes"
"fmt"
"net/http"
"strconv"
Expand Down Expand Up @@ -52,14 +51,14 @@ func validHeaderFieldByte(b byte) bool {

// make sure we don't generate invalid headers
func sanitize(input string) string {
var s strings.Builder
toAscii := strconv.QuoteToASCII(input)
var b bytes.Buffer
for _, i := range toAscii {
if validHeaderFieldByte(byte(i)) {
b.WriteRune(i)
s.WriteRune(i)
}
}
return b.String()
return s.String()
}

// Strips the query parameters and optionally preserves them in the X-Query-Param-xyz headers.
Expand Down
21 changes: 20 additions & 1 deletion filters/builtin/stripquery_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,12 @@
package builtin

import (
"github.com/zalando/skipper/filters/filtertest"
"net/http"
"net/url"
"strings"
"testing"

"github.com/zalando/skipper/filters/filtertest"
)

func TestCreateStripQueryFilter(t *testing.T) {
Expand Down Expand Up @@ -91,3 +92,21 @@ func TestPreserveQuery(t *testing.T) {
}
}
}

func BenchmarkSanitize(b *testing.B) {
piece := "query=cXVlcnkgUGRwKCRjb25maWdTa3U6IElEIS&variables=%7B%0A%20%20%20%20%22beautyColorImageWidth%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20200,%0A%20%20%20%20%22portraitGalleryWidth%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20828,%0A%20%20%20%20%22segmentedBannerHeaderLogoWidth%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%2084,%0A%20%20%20%20%22shouldIncludeCtaTrackingKey%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20false,%0A%20%20%20%20%22shouldIncludeFlagInfo%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20false,%0A%20%20%20%20%22shouldIncludeHistogramValues%22:%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20false,%0A%20%20%20%20%22shouldIncludeOfferSelectionValues%22:%20%20%20%20%20%20%20%20%20%20%20true,%0A%20%20%20%20%22shouldIncludeOmnibusConfigModeChanges%22:%20%20%20%20%20%20%20false,%0A%20%20%20%20%22shouldIncludeOmnibusPriceLabelChanges%22:%20%20%20%20%20%20%20false,&apiEndpoint=https%253A%252F%252Fmodified%252Fsecret%252Fgraphql%252Fsecret&frontendType=secret&zalandoFeature=secret"
q := strings.Repeat(piece, 2)
v, e := url.ParseQuery(q)

if e != nil {
b.Error(e)
}

b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
for k := range v {
sanitize(k)
}
}
}

0 comments on commit e7dd553

Please sign in to comment.