summaryrefslogtreecommitdiffstats
path: root/modules/regexplru
diff options
context:
space:
mode:
authorDaniel Baumann <daniel@debian.org>2024-10-18 20:33:49 +0200
committerDaniel Baumann <daniel@debian.org>2024-12-12 23:57:56 +0100
commite68b9d00a6e05b3a941f63ffb696f91e554ac5ec (patch)
tree97775d6c13b0f416af55314eb6a89ef792474615 /modules/regexplru
parentInitial commit. (diff)
downloadforgejo-e68b9d00a6e05b3a941f63ffb696f91e554ac5ec.tar.xz
forgejo-e68b9d00a6e05b3a941f63ffb696f91e554ac5ec.zip
Adding upstream version 9.0.3.
Signed-off-by: Daniel Baumann <daniel@debian.org>
Diffstat (limited to '')
-rw-r--r--modules/regexplru/regexplru.go44
-rw-r--r--modules/regexplru/regexplru_test.go27
2 files changed, 71 insertions, 0 deletions
diff --git a/modules/regexplru/regexplru.go b/modules/regexplru/regexplru.go
new file mode 100644
index 0000000..8f66dcf
--- /dev/null
+++ b/modules/regexplru/regexplru.go
@@ -0,0 +1,44 @@
+// Copyright 2022 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package regexplru
+
+import (
+ "regexp"
+
+ "code.gitea.io/gitea/modules/log"
+
+ lru "github.com/hashicorp/golang-lru/v2"
+)
+
+var lruCache *lru.Cache[string, any]
+
+func init() {
+ var err error
+ lruCache, err = lru.New[string, any](1000)
+ if err != nil {
+ log.Fatal("failed to new LRU cache, err: %v", err)
+ }
+}
+
+// GetCompiled works like regexp.Compile, the compiled expr or error is stored in LRU cache
+func GetCompiled(expr string) (r *regexp.Regexp, err error) {
+ v, ok := lruCache.Get(expr)
+ if !ok {
+ r, err = regexp.Compile(expr)
+ if err != nil {
+ lruCache.Add(expr, err)
+ return nil, err
+ }
+ lruCache.Add(expr, r)
+ } else {
+ r, ok = v.(*regexp.Regexp)
+ if !ok {
+ if err, ok = v.(error); ok {
+ return nil, err
+ }
+ panic("impossible")
+ }
+ }
+ return r, nil
+}
diff --git a/modules/regexplru/regexplru_test.go b/modules/regexplru/regexplru_test.go
new file mode 100644
index 0000000..8c0c722
--- /dev/null
+++ b/modules/regexplru/regexplru_test.go
@@ -0,0 +1,27 @@
+// Copyright 2022 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package regexplru
+
+import (
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+ "github.com/stretchr/testify/require"
+)
+
+func TestRegexpLru(t *testing.T) {
+ r, err := GetCompiled("a")
+ require.NoError(t, err)
+ assert.True(t, r.MatchString("a"))
+
+ r, err = GetCompiled("a")
+ require.NoError(t, err)
+ assert.True(t, r.MatchString("a"))
+
+ assert.EqualValues(t, 1, lruCache.Len())
+
+ _, err = GetCompiled("(")
+ require.Error(t, err)
+ assert.EqualValues(t, 2, lruCache.Len())
+}