func main() {
n := flag.Int("n", 1000, "cache size")
alg := flag.String("alg", "", "algorithm")
file := flag.String("f", "", "input file")
cpuprofile := flag.Bool("cpuprofile", false, "cpuprofile")
memprofile := flag.Bool("memprofile", false, "memprofile")
flag.Parse()
if *alg == "" {
log.Fatalln("no algorithm provided (-alg)")
}
if *cpuprofile {
defer profile.Start(profile.CPUProfile).Stop()
}
count := 0
miss := 0
t0 := time.Now()
var f func(string) bool
switch *alg {
case "arc":
cache := arc.New(*n)
f = func(s string) bool {
var miss bool
cache.Get(s, func() interface{} {
miss = true
return s
})
return miss
}
case "random":
cache := random.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
}
return false
}
case "tworand":
cache := tworand.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
}
return false
}
case "lru":
cache := lru.New(*n)
f = func(s string) bool {
if _, ok := cache.Get(s); !ok {
cache.Add(s, s)
return true
}
return false
}
case "lfu":
cache := lfucache.New(*n)
f = func(s string) bool {
if _, ok := cache.Access(s); !ok {
cache.Insert(s, s)
return true
}
return false
}
case "clock":
cache := clock.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
}
return false
}
case "clockpro":
cache := clockpro.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
}
return false
}
case "slru":
cache := slru.New(int(float64(*n)*0.2), int(float64(*n)*0.8))
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
}
return false
}
case "s4lru":
cache := s4lru.New(*n)
f = func(s string) bool {
if _, ok := cache.Get(s); !ok {
cache.Set(s, s)
return true
}
return false
}
default:
log.Fatalln("unknown algorithm")
}
var inputFile = os.Stdin
if *file != "" {
var err error
inputFile, err = os.Open(*file)
if err != nil {
log.Fatalln(err)
}
defer inputFile.Close()
}
in := bufio.NewScanner(inputFile)
for in.Scan() {
if f(in.Text()) {
miss++
}
count++
}
if *memprofile {
var m runtime.MemStats
runtime.ReadMemStats(&m)
e := json.NewEncoder(os.Stdout)
e.Encode(m)
}
fmt.Printf("%s: %s %d total %d misses (hit rate %d %%)\n", *alg, time.Since(t0), count, miss, int(100*(float64(count-miss)/float64(count))))
}
func main() {
n := flag.Int("n", 1000, "cache size")
alg := flag.String("alg", "", "algorithm")
file := flag.String("f", "", "input file")
door := flag.Bool("door", false, "use doorkeeper")
cpuprofile := flag.Bool("cpuprofile", false, "cpuprofile")
memprofile := flag.Bool("memprofile", false, "memprofile")
flag.Parse()
if *alg == "" {
log.Fatalln("no algorithm provided (-alg)")
}
if *cpuprofile {
defer profile.Start(profile.CPUProfile).Stop()
}
if *memprofile {
defer profile.Start(profile.MemProfile).Stop()
}
count := 0
miss := 0
t0 := time.Now()
var f func(string) bool
var bouncer *doorkeeper
if *door {
bouncer = newDoorkeeper(*n)
}
switch *alg {
case "arc":
cache := arc.New(*n)
f = func(s string) bool {
var miss bool
cache.Get(s, func() interface{} {
miss = true
return s
})
return miss
}
case "random":
cache := random.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
if bouncer.allow(s) {
cache.Set(s, s)
}
return true
} else {
if i.(string) != s {
panic("key != value")
}
}
return false
}
case "tworand":
cache := tworand.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
cache.Set(s, s)
return true
} else {
if i.(string) != s {
panic("key != value")
}
}
return false
}
case "lru":
cache := lru.New(*n)
f = func(s string) bool {
if v, ok := cache.Get(s); !ok {
if bouncer.allow(s) {
cache.Add(s, s)
}
return true
} else {
if v.(string) != s {
panic("key != value")
}
}
return false
}
case "tinylfu":
cache := tinylfu.New(*n, *n*10)
f = func(s string) bool {
if v, ok := cache.Get(s); !ok {
if bouncer.allow(s) {
cache.Add(s, s)
}
return true
} else {
if v.(string) != s {
panic("key != value")
}
}
return false
}
case "clock":
cache := clock.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
if bouncer.allow(s) {
cache.Set(s, s)
}
return true
} else {
if i.(string) != s {
panic("key != value")
}
}
return false
}
case "clockpro":
cache := clockpro.New(*n)
f = func(s string) bool {
if i := cache.Get(s); i == nil {
if bouncer.allow(s) {
cache.Set(s, s)
}
return true
} else {
if i.(string) != s {
panic("key != value")
}
}
return false
}
case "slru":
cache := slru.New(int(float64(*n)*0.2), int(float64(*n)*0.8))
f = func(s string) bool {
if i := cache.Get(s); i == nil {
if bouncer.allow(s) {
cache.Set(s, s)
}
return true
} else {
if i.(string) != s {
panic("key != value")
}
}
return false
}
case "s4lru":
cache := s4lru.New(*n)
f = func(s string) bool {
if v, ok := cache.Get(s); !ok {
if bouncer.allow(s) {
cache.Set(s, s)
}
return true
} else {
if v.(string) != s {
panic("key != value")
}
}
return false
}
default:
log.Fatalln("unknown algorithm")
}
var inputFile = os.Stdin
if *file != "" {
var err error
inputFile, err = os.Open(*file)
if err != nil {
log.Fatalln(err)
}
defer inputFile.Close()
}
in := bufio.NewScanner(inputFile)
for in.Scan() {
if f(in.Text()) {
miss++
}
count++
}
fmt.Printf("%s: %s %d total %d misses (hit rate %d %%)\n", *alg, time.Since(t0), count, miss, int(100*(float64(count-miss)/float64(count))))
}
