summaryrefslogtreecommitdiffstats
path: root/vendor/gitea.com/go-chi/cache/memory.go
diff options
context:
space:
mode:
authorLunny Xiao <xiaolunwen@gmail.com>2021-01-26 23:36:53 +0800
committerGitHub <noreply@github.com>2021-01-26 16:36:53 +0100
commit6433ba0ec3dfde67f45267aa12bd713c4a44c740 (patch)
tree8813388f7e58ff23ad24af9ccbdb5f0350cb3a09 /vendor/gitea.com/go-chi/cache/memory.go
parent3adbbb4255c42cde04d59b6ebf5ead7e3edda3e7 (diff)
downloadgitea-6433ba0ec3dfde67f45267aa12bd713c4a44c740.tar.gz
gitea-6433ba0ec3dfde67f45267aa12bd713c4a44c740.zip
Move macaron to chi (#14293)
Use [chi](https://github.com/go-chi/chi) instead of the forked [macaron](https://gitea.com/macaron/macaron). Since macaron and chi have conflicts with session share, this big PR becomes a have-to thing. According my previous idea, we can replace macaron step by step but I'm wrong. :( Below is a list of big changes on this PR. - [x] Define `context.ResponseWriter` interface with an implementation `context.Response`. - [x] Use chi instead of macaron, and also a customize `Route` to wrap chi so that the router usage is similar as before. - [x] Create different routers for `web`, `api`, `internal` and `install` so that the codes will be more clear and no magic . - [x] Use https://github.com/unrolled/render instead of macaron's internal render - [x] Use https://github.com/NYTimes/gziphandler instead of https://gitea.com/macaron/gzip - [x] Use https://gitea.com/go-chi/session which is a modified version of https://gitea.com/macaron/session and removed `nodb` support since it will not be maintained. **BREAK** - [x] Use https://gitea.com/go-chi/captcha which is a modified version of https://gitea.com/macaron/captcha - [x] Use https://gitea.com/go-chi/cache which is a modified version of https://gitea.com/macaron/cache - [x] Use https://gitea.com/go-chi/binding which is a modified version of https://gitea.com/macaron/binding - [x] Use https://github.com/go-chi/cors instead of https://gitea.com/macaron/cors - [x] Dropped https://gitea.com/macaron/i18n and make a new one in `code.gitea.io/gitea/modules/translation` - [x] Move validation form structs from `code.gitea.io/gitea/modules/auth` to `code.gitea.io/gitea/modules/forms` to avoid dependency cycle. - [x] Removed macaron log service because it's not need any more. **BREAK** - [x] All form structs have to be get by `web.GetForm(ctx)` in the route function but not as a function parameter on routes definition. - [x] Move Git HTTP protocol implementation to use routers directly. - [x] Fix the problem that chi routes don't support trailing slash but macaron did. - [x] `/api/v1/swagger` now will be redirect to `/api/swagger` but not render directly so that `APIContext` will not create a html render. Notices: - Chi router don't support request with trailing slash - Integration test `TestUserHeatmap` maybe mysql version related. It's failed on my macOS(mysql 5.7.29 installed via brew) but succeed on CI. Co-authored-by: 6543 <6543@obermui.de>
Diffstat (limited to 'vendor/gitea.com/go-chi/cache/memory.go')
-rw-r--r--vendor/gitea.com/go-chi/cache/memory.go179
1 files changed, 179 insertions, 0 deletions
diff --git a/vendor/gitea.com/go-chi/cache/memory.go b/vendor/gitea.com/go-chi/cache/memory.go
new file mode 100644
index 0000000000..2d5a8fac4d
--- /dev/null
+++ b/vendor/gitea.com/go-chi/cache/memory.go
@@ -0,0 +1,179 @@
+// Copyright 2013 Beego Authors
+// Copyright 2014 The Macaron Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License"): you may
+// not use this file except in compliance with the License. You may obtain
+// a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+// License for the specific language governing permissions and limitations
+// under the License.
+
+package cache
+
+import (
+ "errors"
+ "sync"
+ "time"
+)
+
+// MemoryItem represents a memory cache item.
+type MemoryItem struct {
+ val interface{}
+ created int64
+ expire int64
+}
+
+func (item *MemoryItem) hasExpired() bool {
+ return item.expire > 0 &&
+ (time.Now().Unix()-item.created) >= item.expire
+}
+
+// MemoryCacher represents a memory cache adapter implementation.
+type MemoryCacher struct {
+ lock sync.RWMutex
+ items map[string]*MemoryItem
+ interval int // GC interval.
+}
+
+// NewMemoryCacher creates and returns a new memory cacher.
+func NewMemoryCacher() *MemoryCacher {
+ return &MemoryCacher{items: make(map[string]*MemoryItem)}
+}
+
+// Put puts value into cache with key and expire time.
+// If expired is 0, it will be deleted by next GC operation.
+func (c *MemoryCacher) Put(key string, val interface{}, expire int64) error {
+ c.lock.Lock()
+ defer c.lock.Unlock()
+
+ c.items[key] = &MemoryItem{
+ val: val,
+ created: time.Now().Unix(),
+ expire: expire,
+ }
+ return nil
+}
+
+// Get gets cached value by given key.
+func (c *MemoryCacher) Get(key string) interface{} {
+ c.lock.RLock()
+ defer c.lock.RUnlock()
+
+ item, ok := c.items[key]
+ if !ok {
+ return nil
+ }
+ if item.hasExpired() {
+ go c.Delete(key)
+ return nil
+ }
+ return item.val
+}
+
+// Delete deletes cached value by given key.
+func (c *MemoryCacher) Delete(key string) error {
+ c.lock.Lock()
+ defer c.lock.Unlock()
+
+ delete(c.items, key)
+ return nil
+}
+
+// Incr increases cached int-type value by given key as a counter.
+func (c *MemoryCacher) Incr(key string) (err error) {
+ c.lock.RLock()
+ defer c.lock.RUnlock()
+
+ item, ok := c.items[key]
+ if !ok {
+ return errors.New("key not exist")
+ }
+ item.val, err = Incr(item.val)
+ return err
+}
+
+// Decr decreases cached int-type value by given key as a counter.
+func (c *MemoryCacher) Decr(key string) (err error) {
+ c.lock.RLock()
+ defer c.lock.RUnlock()
+
+ item, ok := c.items[key]
+ if !ok {
+ return errors.New("key not exist")
+ }
+
+ item.val, err = Decr(item.val)
+ return err
+}
+
+// IsExist returns true if cached value exists.
+func (c *MemoryCacher) IsExist(key string) bool {
+ c.lock.RLock()
+ defer c.lock.RUnlock()
+
+ _, ok := c.items[key]
+ return ok
+}
+
+// Flush deletes all cached data.
+func (c *MemoryCacher) Flush() error {
+ c.lock.Lock()
+ defer c.lock.Unlock()
+
+ c.items = make(map[string]*MemoryItem)
+ return nil
+}
+
+func (c *MemoryCacher) checkRawExpiration(key string) {
+ item, ok := c.items[key]
+ if !ok {
+ return
+ }
+
+ if item.hasExpired() {
+ delete(c.items, key)
+ }
+}
+
+func (c *MemoryCacher) checkExpiration(key string) {
+ c.lock.Lock()
+ defer c.lock.Unlock()
+
+ c.checkRawExpiration(key)
+}
+
+func (c *MemoryCacher) startGC() {
+ c.lock.Lock()
+ defer c.lock.Unlock()
+
+ if c.interval < 1 {
+ return
+ }
+
+ if c.items != nil {
+ for key, _ := range c.items {
+ c.checkRawExpiration(key)
+ }
+ }
+
+ time.AfterFunc(time.Duration(c.interval)*time.Second, func() { c.startGC() })
+}
+
+// StartAndGC starts GC routine based on config string settings.
+func (c *MemoryCacher) StartAndGC(opt Options) error {
+ c.lock.Lock()
+ c.interval = opt.Interval
+ c.lock.Unlock()
+
+ go c.startGC()
+ return nil
+}
+
+func init() {
+ Register("memory", NewMemoryCacher())
+}