convert to using library version of cache

This commit is contained in:
Ted Unangst 2019-10-12 19:25:23 -04:00
parent dd8400b66c
commit d357480a1f
6 changed files with 15 additions and 103 deletions

View File

@ -31,6 +31,7 @@ import (
"strings" "strings"
"time" "time"
"humungus.tedunangst.com/r/webs/cache"
"humungus.tedunangst.com/r/webs/htfilter" "humungus.tedunangst.com/r/webs/htfilter"
"humungus.tedunangst.com/r/webs/httpsig" "humungus.tedunangst.com/r/webs/httpsig"
"humungus.tedunangst.com/r/webs/image" "humungus.tedunangst.com/r/webs/image"
@ -256,7 +257,7 @@ type Box struct {
Shared string Shared string
} }
var boxofboxes = cacheNew(cacheOptions{Filler: func(ident string) (*Box, bool) { var boxofboxes = cache.New(cache.Options{Filler: func(ident string) (*Box, bool) {
var info string var info string
row := stmtGetXonker.QueryRow(ident, "boxes") row := stmtGetXonker.QueryRow(ident, "boxes")
err := row.Scan(&info) err := row.Scan(&info)
@ -1118,7 +1119,7 @@ func jonkjonk(user *WhatAbout, h *Honk) (junk.Junk, junk.Junk) {
return j, jo return j, jo
} }
var oldjonks = cacheNew(cacheOptions{Filler: func(xid string) ([]byte, bool) { var oldjonks = cache.New(cache.Options{Filler: func(xid string) ([]byte, bool) {
row := stmtAnyXonk.QueryRow(xid) row := stmtAnyXonk.QueryRow(xid)
honk := scanhonk(row) honk := scanhonk(row)
if honk == nil || !honk.Public { if honk == nil || !honk.Public {
@ -1185,7 +1186,7 @@ func honkworldwide(user *WhatAbout, honk *Honk) {
} }
} }
var oldjonkers = cacheNew(cacheOptions{Filler: func(name string) ([]byte, bool) { var oldjonkers = cache.New(cache.Options{Filler: func(name string) ([]byte, bool) {
user, err := butwhatabout(name) user, err := butwhatabout(name)
if err != nil { if err != nil {
return nil, false return nil, false
@ -1226,7 +1227,7 @@ func asjonker(name string) ([]byte, bool) {
return j, ok return j, ok
} }
var handfull = cacheNew(cacheOptions{Filler: func(name string) (string, bool) { var handfull = cache.New(cache.Options{Filler: func(name string) (string, bool) {
m := strings.Split(name, "@") m := strings.Split(name, "@")
if len(m) != 2 { if len(m) != 2 {
log.Printf("bad fish name: %s", name) log.Printf("bad fish name: %s", name)

View File

@ -1,94 +0,0 @@
//
// Copyright (c) 2019 Ted Unangst <tedu@tedunangst.com>
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
package main
import (
"reflect"
"sync"
"time"
)
type cacheFiller func(key interface{}) (interface{}, bool)
type cacheOptions struct {
Filler interface{}
Duration time.Duration
}
type Cache struct {
cache map[interface{}]interface{}
filler cacheFiller
lock sync.Mutex
stale time.Time
duration time.Duration
}
func cacheNew(options cacheOptions) *Cache {
c := new(Cache)
c.cache = make(map[interface{}]interface{})
fillfn := options.Filler
ftype := reflect.TypeOf(fillfn)
if ftype.Kind() != reflect.Func {
panic("cache filler is not function")
}
if ftype.NumIn() != 1 || ftype.NumOut() != 2 {
panic("cache filler has wrong argument count")
}
c.filler = func(key interface{}) (interface{}, bool) {
vfn := reflect.ValueOf(fillfn)
args := []reflect.Value{reflect.ValueOf(key)}
rv := vfn.Call(args)
return rv[0].Interface(), rv[1].Bool()
}
if options.Duration != 0 {
c.duration = options.Duration
c.stale = time.Now().Add(c.duration)
}
return c
}
func (cache *Cache) Get(key interface{}, value interface{}) bool {
cache.lock.Lock()
defer cache.lock.Unlock()
if !cache.stale.IsZero() && cache.stale.Before(time.Now()) {
cache.stale = time.Now().Add(cache.duration)
cache.cache = make(map[interface{}]interface{})
}
v, ok := cache.cache[key]
if !ok {
v, ok = cache.filler(key)
if ok {
cache.cache[key] = v
}
}
if ok {
ptr := reflect.ValueOf(v)
reflect.ValueOf(value).Elem().Set(ptr)
}
return ok
}
func (cache *Cache) Clear(key interface{}) {
cache.lock.Lock()
defer cache.lock.Unlock()
delete(cache.cache, key)
}
func (cache *Cache) Flush() {
cache.lock.Lock()
defer cache.lock.Unlock()
cache.cache = make(map[interface{}]interface{})
}

View File

@ -26,10 +26,11 @@ import (
"strings" "strings"
"time" "time"
"humungus.tedunangst.com/r/webs/cache"
"humungus.tedunangst.com/r/webs/login" "humungus.tedunangst.com/r/webs/login"
) )
var someusers = cacheNew(cacheOptions{Filler: func(name string) (*WhatAbout, bool) { var someusers = cache.New(cache.Options{Filler: func(name string) (*WhatAbout, bool) {
row := stmtWhatAbout.QueryRow(name) row := stmtWhatAbout.QueryRow(name)
user := new(WhatAbout) user := new(WhatAbout)
var options string var options string

3
fun.go
View File

@ -28,6 +28,7 @@ import (
"sync" "sync"
"golang.org/x/net/html" "golang.org/x/net/html"
"humungus.tedunangst.com/r/webs/cache"
"humungus.tedunangst.com/r/webs/htfilter" "humungus.tedunangst.com/r/webs/htfilter"
"humungus.tedunangst.com/r/webs/httpsig" "humungus.tedunangst.com/r/webs/httpsig"
) )
@ -365,7 +366,7 @@ func quickrename(s string, userid int64) string {
return s return s
} }
var shortnames = cacheNew(cacheOptions{Filler: func(userid int64) (map[string]string, bool) { var shortnames = cache.New(cache.Options{Filler: func(userid int64) (map[string]string, bool) {
honkers := gethonkers(userid) honkers := gethonkers(userid)
m := make(map[string]string) m := make(map[string]string)
for _, h := range honkers { for _, h := range honkers {

View File

@ -21,6 +21,8 @@ import (
"regexp" "regexp"
"sort" "sort"
"time" "time"
"humungus.tedunangst.com/r/webs/cache"
) )
type Filter struct { type Filter struct {
@ -63,7 +65,7 @@ func (ft filtType) String() string {
type afiltermap map[filtType][]*Filter type afiltermap map[filtType][]*Filter
var filtcache = cacheNew(cacheOptions{Filler: func(userid int64) (afiltermap, bool) { var filtcache = cache.New(cache.Options{Filler: func(userid int64) (afiltermap, bool) {
rows, err := stmtGetFilters.Query(userid) rows, err := stmtGetFilters.Query(userid)
if err != nil { if err != nil {
log.Printf("error querying filters: %s", err) log.Printf("error querying filters: %s", err)

5
web.go
View File

@ -33,6 +33,7 @@ import (
"time" "time"
"github.com/gorilla/mux" "github.com/gorilla/mux"
"humungus.tedunangst.com/r/webs/cache"
"humungus.tedunangst.com/r/webs/css" "humungus.tedunangst.com/r/webs/css"
"humungus.tedunangst.com/r/webs/htfilter" "humungus.tedunangst.com/r/webs/htfilter"
"humungus.tedunangst.com/r/webs/httpsig" "humungus.tedunangst.com/r/webs/httpsig"
@ -471,7 +472,7 @@ func xzone(w http.ResponseWriter, r *http.Request) {
} }
} }
var oldoutbox = cacheNew(cacheOptions{Filler: func(name string) ([]byte, bool) { var oldoutbox = cache.New(cache.Options{Filler: func(name string) ([]byte, bool) {
user, err := butwhatabout(name) user, err := butwhatabout(name)
if err != nil { if err != nil {
return nil, false return nil, false
@ -1235,7 +1236,7 @@ func showhonkers(w http.ResponseWriter, r *http.Request) {
} }
} }
var combocache = cacheNew(cacheOptions{Filler: func(userid int64) ([]string, bool) { var combocache = cache.New(cache.Options{Filler: func(userid int64) ([]string, bool) {
honkers := gethonkers(userid) honkers := gethonkers(userid)
var combos []string var combos []string
for _, h := range honkers { for _, h := range honkers {