Clean up cache code

This commit is contained in:
Zed 2020-06-02 22:36:02 +02:00
parent 9ad458ae25
commit d01f62c5ae
2 changed files with 11 additions and 20 deletions

View File

@ -25,10 +25,7 @@ proc initRedisPool*(cfg: Config) =
quit(1) quit(1)
template toKey(p: Profile): string = "p:" & toLower(p.username) template toKey(p: Profile): string = "p:" & toLower(p.username)
template toKey(v: Video): string = "v:" & v.videoId
template toKey(c: Card): string = "c:" & c.id
template toKey(l: List): string = toLower("l:" & l.username & '/' & l.name) template toKey(l: List): string = toLower("l:" & l.username & '/' & l.name)
template toKey(t: Token): string = "t:" & t.tok
template to(s: string; typ: typedesc): untyped = template to(s: string; typ: typedesc): untyped =
var res: typ var res: typ
@ -40,22 +37,20 @@ proc get(query: string): Future[string] {.async.} =
pool.withAcquire(r): pool.withAcquire(r):
result = await r.get(query) result = await r.get(query)
proc uncache*(id: int64) {.async.} = proc setex(key: string; time: int; data: string) {.async.} =
pool.withAcquire(r): pool.withAcquire(r):
discard await r.del("v:" & $id) discard await r.setex(key, time, data)
proc cache*[T](data: T; time=baseCacheTime) {.async.} = proc cache*(data: List) {.async.} =
pool.withAcquire(r): await setex(data.toKey, listCacheTime, data.pack)
discard await r.setex(data.toKey, time, pack(data))
proc cache*(data: PhotoRail; id: string) {.async.} = proc cache*(data: PhotoRail; id: string) {.async.} =
pool.withAcquire(r): await setex("pr:" & id, baseCacheTime, data.pack)
discard await r.setex("pr:" & id, baseCacheTime, pack(data))
proc cache*(data: Profile; time=baseCacheTime) {.async.} = proc cache*(data: Profile) {.async.} =
pool.withAcquire(r): pool.withAcquire(r):
r.startPipelining() r.startPipelining()
discard await r.setex(data.toKey, time, pack(data)) discard await r.setex(data.toKey, baseCacheTime, pack(data))
discard await r.hset("p:", toLower(data.username), data.id) discard await r.hset("p:", toLower(data.username), data.id)
discard await r.flushPipeline() discard await r.flushPipeline()
@ -63,7 +58,8 @@ proc cacheRss*(query, rss, cursor: string) {.async.} =
let key = "rss:" & query let key = "rss:" & query
pool.withAcquire(r): pool.withAcquire(r):
r.startPipelining() r.startPipelining()
await r.hmset(key, @[("rss", rss), ("min", cursor)]) discard await r.hset(key, "rss", rss)
discard await r.hset(key, "min", cursor)
discard await r.expire(key, rssCacheTime) discard await r.expire(key, rssCacheTime)
discard await r.flushPipeline() discard await r.flushPipeline()
@ -73,11 +69,6 @@ proc getProfileId*(username: string): Future[string] {.async.} =
if result == redisNil: if result == redisNil:
result.setLen(0) result.setLen(0)
proc hasCachedProfile*(username: string): Future[Option[Profile]] {.async.} =
let prof = await get("p:" & toLower(username))
if prof != redisNil:
result = some prof.to(Profile)
proc getCachedProfile*(username: string; fetch=true): Future[Profile] {.async.} = proc getCachedProfile*(username: string; fetch=true): Future[Profile] {.async.} =
let prof = await get("p:" & toLower(username)) let prof = await get("p:" & toLower(username))
if prof != redisNil: if prof != redisNil:
@ -107,7 +98,7 @@ proc getCachedList*(username=""; name=""; id=""): Future[List] {.async.} =
result = await getGraphListById(id) result = await getGraphListById(id)
else: else:
result = await getGraphList(username, name) result = await getGraphList(username, name)
await cache(result, time=listCacheTime) await cache(result)
proc getCachedRss*(key: string): Future[(string, string)] {.async.} = proc getCachedRss*(key: string): Future[(string, string)] {.async.} =
var res: Table[string, string] var res: Table[string, string]

View File

@ -41,5 +41,5 @@ proc createListRouter*(cfg: Config) =
let list = await getCachedList(id=(@"id")) let list = await getCachedList(id=(@"id"))
if list.id.len == 0: if list.id.len == 0:
resp Http404 resp Http404
await cache(list, time=listCacheTime) await cache(list)
redirect("/" & list.username & "/lists/" & list.name) redirect("/" & list.username & "/lists/" & list.name)