diff --git a/nitter.nimble b/nitter.nimble index 48f9b74..c88509c 100644 --- a/nitter.nimble +++ b/nitter.nimble @@ -21,8 +21,8 @@ requires "packedjson#d11d167" requires "supersnappy#2.1.1" requires "redpool#f880f49" requires "https://github.com/zedeus/redis#d0a0e6f" -requires "https://github.com/disruptek/frosty#0.3.1" requires "zippy#0.7.3" +requires "flatty#0.2.3" # Tasks diff --git a/src/redis_cache.nim b/src/redis_cache.nim index afa8244..f16420d 100644 --- a/src/redis_cache.nim +++ b/src/redis_cache.nim @@ -1,5 +1,5 @@ import asyncdispatch, times, strutils, tables, hashes -import redis, redpool, frosty, supersnappy +import redis, redpool, flatty, supersnappy import types, api @@ -11,6 +11,13 @@ var rssCacheTime: int listCacheTime*: int +# flatty can't serialize DateTime, so we need to define this +proc toFlatty*(s: var string, x: DateTime) = + s.toFlatty(x.toTime().toUnix()) + +proc fromFlatty*(s: string, i: var int, x: var DateTime) = + x = fromUnix(s.fromFlatty(int64)).utc() + proc setCacheTimes*(cfg: Config) = rssCacheTime = cfg.rssCacheTime * 60 listCacheTime = cfg.listCacheTime * 60 @@ -28,11 +35,12 @@ proc migrate*(key, match: string) {.async.} = proc initRedisPool*(cfg: Config) {.async.} = try: - pool = await newRedisPool(cfg.redisConns, maxConns=cfg.redisMaxConns, - host=cfg.redisHost, port=cfg.redisPort, password=cfg.redisPassword) + pool = await newRedisPool(cfg.redisConns, cfg.redisMaxConns, + host=cfg.redisHost, port=cfg.redisPort, + password=cfg.redisPassword) + await migrate("flatty", "*:*") await migrate("snappyRss", "rss:*") - await migrate("oldFrosty", "*") await migrate("userBuckets", "p:*") await migrate("profileDates", "p:*") @@ -58,17 +66,17 @@ proc setex(key: string; time: int; data: string) {.async.} = discard await r.setex(key, time, data) proc cache*(data: List) {.async.} = - await setex(data.listKey, listCacheTime, compress(freeze(data))) + await setex(data.listKey, listCacheTime, compress(toFlatty(data))) proc cache*(data: PhotoRail; name: string) {.async.} = - await setex("pr:" & name, baseCacheTime, compress(freeze(data))) + await setex("pr:" & name, baseCacheTime, compress(toFlatty(data))) proc cache*(data: Profile) {.async.} = if data.username.len == 0 or data.id.len == 0: return let name = toLower(data.username) pool.withAcquire(r): r.startPipelining() - discard await r.setex(name.profileKey, baseCacheTime, compress(freeze(data))) + discard await r.setex(name.profileKey, baseCacheTime, compress(toFlatty(data))) discard await r.hset(name.pidKey, name, data.id) discard await r.flushPipeline() @@ -97,7 +105,7 @@ proc getProfileId*(username: string): Future[string] {.async.} = proc getCachedProfile*(username: string; fetch=true): Future[Profile] {.async.} = let prof = await get("p:" & toLower(username)) if prof != redisNil: - uncompress(prof).thaw(result) + result = fromFlatty(uncompress(prof), Profile) elif fetch: result = await getProfile(username) @@ -105,7 +113,7 @@ proc getCachedPhotoRail*(name: string): Future[PhotoRail] {.async.} = if name.len == 0: return let rail = await get("pr:" & toLower(name)) if rail != redisNil: - uncompress(rail).thaw(result) + result = fromFlatty(uncompress(rail), PhotoRail) else: result = await getPhotoRail(name) await cache(result, name) @@ -115,7 +123,7 @@ proc getCachedList*(username=""; name=""; id=""): Future[List] {.async.} = else: await get(toLower("l:" & username & '/' & name)) if list != redisNil: - uncompress(list).thaw(result) + result = fromFlatty(uncompress(list), List) else: if id.len > 0: result = await getGraphListById(id)