Rework profile cache behavior, fix suspended cache

Fixes #480
This commit is contained in:
Zed 2022-01-16 20:32:45 +01:00
parent 23f87c115a
commit f3d6f53f6d
4 changed files with 20 additions and 21 deletions

View file

@ -34,7 +34,7 @@ proc getProfile*(username: string): Future[Profile] {.async.} =
let
ps = genParams({"screen_name": username})
json = await fetchRaw(userShow ? ps, Api.userShow)
result = parseUser(json)
result = parseUser(json, username)
proc getProfileById*(userId: string): Future[Profile] {.async.} =
let

View file

@ -38,10 +38,10 @@ proc getBanner(user: User): string =
return '#' & user.profileLinkColor
return "#161616"
proc parseUser*(json: string): Profile =
proc parseUser*(json: string; username=""): Profile =
handleErrors:
case error.code
of suspended: return Profile(suspended: true)
of suspended: return Profile(username: username, suspended: true)
of userNotFound: return
else: echo "[error - parseUser]: ", error

View file

@ -80,16 +80,16 @@ proc cache*(data: PhotoRail; name: string) {.async.} =
await setEx("pr:" & toLower(name), baseCacheTime, compress(toFlatty(data)))
proc cache*(data: Profile) {.async.} =
if data.username.len == 0 or data.id.len == 0: return
if data.username.len == 0: return
let name = toLower(data.username)
pool.withAcquire(r):
r.startPipelining()
dawait r.setEx(name.profileKey, baseCacheTime, compress(toFlatty(data)))
dawait r.setEx("i:" & data.id , baseCacheTime, data.username)
dawait r.hSet(name.pidKey, name, data.id)
if data.id.len > 0:
dawait r.hSet(name.pidKey, name, data.id)
dawait r.flushPipeline()
proc cacheProfileId*(username, id: string) {.async.} =
proc cacheProfileId(username, id: string) {.async.} =
if username.len == 0 or id.len == 0: return
let name = toLower(username)
pool.withAcquire(r):
@ -117,15 +117,21 @@ proc getCachedProfile*(username: string; fetch=true): Future[Profile] {.async.}
result = fromFlatty(uncompress(prof), Profile)
elif fetch:
result = await getProfile(username)
await cacheProfileId(result.username, result.id)
if result.suspended:
await cache(result)
proc getCachedProfileUsername*(userId: string): Future[string] {.async.} =
let username = await get("i:" & userId)
let
key = "i:" & userId
username = await get(key)
if username != redisNil:
result = username
else:
let profile = await getProfileById(userId)
result = profile.username
await cache(profile)
await setEx(key, baseCacheTime, result)
proc getCachedPhotoRail*(name: string): Future[PhotoRail] {.async.} =
if name.len == 0: return

View file

@ -30,20 +30,13 @@ proc fetchTimeline*(after: string; query: Query; skipRail=false):
if profileId.len == 0:
profile = await getCachedProfile(name)
profileId = if profile.suspended: "s"
else: profile.id
if profileId.len > 0:
await cacheProfileId(profile.username, profileId)
profileId = profile.id
fetched = true
if profileId.len == 0 or profile.protected:
result[0] = profile
return
elif profileId == "s":
result[0] = Profile(username: name, suspended: true)
return
if profile.protected or profile.suspended:
return (profile, Timeline(), @[])
elif profileId.len == 0:
return (Profile(username: name), Timeline(), @[])
var rail: Future[PhotoRail]
if skipRail or profile.protected or query.kind == media: