Skip to content

Commit

Permalink
add memory usage tests
Browse files Browse the repository at this point in the history
  • Loading branch information
phuslu committed Jan 5, 2024
1 parent c6d1914 commit e3ee3b6
Show file tree
Hide file tree
Showing 2 changed files with 134 additions and 0 deletions.
10 changes: 10 additions & 0 deletions .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,16 @@ jobs:
working-directory: ./bench
run: |
go test -v -cpu=8 -run=none -bench=. -benchtime=5s -benchmem bench_test.go
- name: Memory Usage
working-directory: ./bench
run: |
awk '{if($0 ~ "// memusage.go"){a=1;b=1};if($0 ~ "```" && b=1){b=0};if (a&&b) {print}}' ../README.md > memusage.go
go run memusage.go phuslu
go run memusage.go theine
go run memusage.go cloudflare
go run memusage.go ristretto
go run memusage.go otter
go run memusage.go ecache
- name: Upload coverage reports to Codecov
uses: codecov/codecov-action@v3
with:
Expand Down
124 changes: 124 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -292,6 +292,130 @@ PASS
ok command-line-arguments 47.529s
```

### Memory usage

The Memory usage result as below. Check github [actions][actions] for more results and details.
<details>
<summary>memory usage on keysize=16, cachesize=1000000</summary>

```go
// memusage.go
package main

import (
"fmt"
"os"
"runtime"
"time"

theine "github.com/Yiling-J/theine-go"
cloudflare "github.com/cloudflare/golibs/lrucache"
ristretto "github.com/dgraph-io/ristretto"
otter "github.com/maypok86/otter"
ecache "github.com/orca-zhang/ecache"
phuslu "github.com/phuslu/lru"
)

const (
keysize = 16
cachesize = 1000000
)

var keymap = func() (x []string) {
x = make([]string, cachesize)
for i := 0; i < cachesize; i++ {
x[i] = fmt.Sprintf(fmt.Sprintf("%%0%dd", keysize), i)
}
return
}()

func main() {
var o runtime.MemStats
runtime.ReadMemStats(&o)

name := os.Args[1]
switch name {
case "phuslu":
SetupPhuslu()
case "ristretto":
SetupRistretto()
case "otter":
SetupOtter()
case "ecache":
SetupEcache()
case "cloudflare":
SetupCloudflare()
case "theine":
SetupTheine()
default:
panic("no cache name")
}

mb := func(n uint64) uint64 { return n / 1024 / 1024 }

var m runtime.MemStats
runtime.ReadMemStats(&m)
fmt.Printf("%s: Alloc = %v MiB, TotalAlloc = %v MiB, Sys = %v MiB\n",
name, mb(m.Alloc-o.Alloc), mb(m.TotalAlloc-o.TotalAlloc), mb(m.Sys-o.Sys))
}

func SetupPhuslu() {
cache := phuslu.New[string, int](cachesize)
for i := 0; i < cachesize; i++ {
cache.SetWithTTL(keymap[i], i, time.Hour)
}
}

func SetupOtter() {
cache, _ := otter.MustBuilder[string, int](cachesize).Build()
for i := 0; i < cachesize; i++ {
cache.SetWithTTL(keymap[i], i, time.Hour)
}
}

func SetupEcache() {
cache := ecache.NewLRUCache(1024, cachesize/1024, time.Hour)
for i := 0; i < cachesize; i++ {
cache.Put(keymap[i], i)
}
}

func SetupRistretto() {
cache, _ := ristretto.NewCache(&ristretto.Config{
NumCounters: cachesize,
MaxCost: 2 << 30,
BufferItems: 64,
})
for i := 0; i < cachesize; i++ {
cache.SetWithTTL(keymap[i], i, 1, time.Hour)
}
}

func SetupTheine() {
cache, _ := theine.NewBuilder[string, int](cachesize).Build()
for i := 0; i < cachesize/2; i++ {
cache.SetWithTTL(keymap[i], i, 1, time.Hour)
}
}

func SetupCloudflare() {
cache := cloudflare.NewMultiLRUCache(1024, cachesize/1024)
for i := 0; i < cachesize; i++ {
cache.Set(keymap[i], i, time.Now().Add(time.Hour))
}
}
```
</details>

| | Alloc | TotalAlloc | Sys |
|---|---|---|---|
| phuslu | 48 MiB | 56 MiB | 61 MiB |
| theine | 84 MiB | 111 MiB | 99 MiB |
| ecache | 123 MiB | 131 MiB | 127 MiB |
| otter | 137 MiB | 211 MiB | 181 MiB |
| ristretto | 149 MiB | 235 MiB | 156 MiB |
| cloudflare | 183 MiB | 191 MiB | 188 MiB |

### License
LRU is licensed under the MIT License. See the LICENSE file for details.

Expand Down

0 comments on commit e3ee3b6

Please sign in to comment.