Mastodon
/
forget-cancellare-vecchi-toot
mirror of https://github.com/codl/forget
2
0
Fork 0
Code Issues Releases Wiki Activity

brotli: use text mode when appropriate

This commit is contained in:
codl 2017-08-11 21:05:11 +02:00
parent 5656f86fcb
commit 98255c00a8
No known key found for this signature in database
GPG Key ID: 6CD7C8891ED1233A
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lib/brotli.py
View File

@ -1,4 +1,4 @@
from brotli import compress
import brotli as brotli_
from flask import request, make_response
from functools import wraps
from threading import Thread
@ -13,8 +13,8 @@ class BrotliCache(object):
self.max_wait = max_wait
self.expire = expire
def compress(self, cache_key, lock_key, body):
encbody = compress(body)
def compress(self, cache_key, lock_key, body, mode=brotli_.MODE_GENERIC):
encbody = brotli_.compress(body, mode=mode)
self.redis.set(cache_key, encbody, ex=self.expire)
self.redis.delete(lock_key)
@ -30,7 +30,8 @@ class BrotliCache(object):
if not encbody:
lock_key = 'brotlicache:lock:{}'.format(digest)
if self.redis.set(lock_key, 1, nx=True, ex=10):
t = Thread(target=self.compress, args=(cache_key, lock_key, body))
mode = brotli_.MODE_TEXT if response.content_type.startswith('text/') else brotli_.MODE_GENERIC
t = Thread(target=self.compress, args=(cache_key, lock_key, body, mode))
t.start()
if self.max_wait > 0:
t.join(self.max_wait)