最近在工作中遇到一個需求，就是要開一個接口來接收供應(yīng)商推送的數(shù)據(jù)。項目采用的python的django框架，我是想也沒想，就直接一梭哈，寫出了如下代碼：

class XXDataPushView(APIView): ''' 接收xx數(shù)據(jù)推送 '''# ... @white_list_required def post(self, request, **kwargs): req_data = request.data or {}# ...

但隨后，發(fā)現(xiàn)每日數(shù)據(jù)并沒有任何變化，質(zhì)問供應(yīng)商是否沒有做推送，在忽悠我們。然后對方給的答復(fù)是，他們推送的是gzip壓縮的數(shù)據(jù)流，接收端需要主動進(jìn)行解壓。此前從沒有處理過這種壓縮的數(shù)據(jù)，對方具體如何做的推送對我來說也是一個黑盒。

因此，我要求對方給一個推送的簡單示例，沒想到對方不講武德，仍過來一段沒法單獨運(yùn)行的java代碼：

private byte[] compress(JSONObject body) { try { ByteArrayOutputStream out = new ByteArrayOutputStream(); GZIPOutputStream gzip = new GZIPOutputStream(out); gzip.write(body.toString().getBytes()); gzip.close(); return out.toByteArray(); } catch (Exception e) { logger.error('Compress data failed with error: ' + e.getMessage()).commit(); } return JSON.toJSONString(body).getBytes();}public void post(JSONObject body, String url, FutureCallback<HttpResponse> callback) { RequestBuilder requestBuilder = RequestBuilder.post(url); requestBuilder.addHeader('Content-Type', 'application/json; charset=UTF-8'); requestBuilder.addHeader('Content-Encoding', 'gzip'); byte[] compressData = compress(body); int timeout = (int) Math.max(((float)compressData.length) / 5000000, 5000); RequestConfig.Builder requestConfigBuilder = RequestConfig.custom(); requestConfigBuilder.setSocketTimeout(timeout).setConnectTimeout(timeout); requestBuilder.setEntity(new ByteArrayEntity(compressData)); requestBuilder.setConfig(requestConfigBuilder.build()); excuteRequest(requestBuilder, callback);}private void excuteRequest(RequestBuilder requestBuilder, FutureCallback<HttpResponse> callback) { HttpUriRequest request = requestBuilder.build(); httpClient.execute(request, new FutureCallback<HttpResponse>() { @Override public void completed(HttpResponse httpResponse) { try {int responseCode = httpResponse.getStatusLine().getStatusCode();if (callback != null) { if (responseCode == 200) { callback.completed(httpResponse); } else { callback.failed(new Exception('Status code is not 200')); }} } catch (Exception e) {logger.error('Get error on ' + requestBuilder.getMethod() + ' ' + requestBuilder.getUri() + ': ' + e.getMessage()).commit();if (callback != null) { callback.failed(e);} } EntityUtils.consumeQuietly(httpResponse.getEntity()); } @Override public void failed(Exception e) { logger.error('Get error on ' + requestBuilder.getMethod() + ' ' + requestBuilder.getUri() + ': ' + e.getMessage()).commit(); if (callback != null) {callback.failed(e); } } @Override public void cancelled() { logger.error('Request cancelled on ' + requestBuilder.getMethod() + ' ' + requestBuilder.getUri()).commit(); if (callback != null) {callback.cancelled(); } } });}

從上述代碼可以看出，對方將json數(shù)據(jù)壓縮為了gzip數(shù)據(jù)流stream。于是搜索django的文檔，只有這段關(guān)于gzip處理的裝飾器描述:

django.views.decorators.gzip 里的裝飾器控制基于每個視圖的內(nèi)容壓縮。

如果瀏覽器允許 gzip 壓縮，那么這個裝飾器將壓縮內(nèi)容。它相應(yīng)的設(shè)置了 Vary 頭部，這樣緩存將基于 Accept-Encoding 頭進(jìn)行存儲。

但是，這個裝飾器只是壓縮請求響應(yīng)至瀏覽器的內(nèi)容，我們目前的需求是解壓縮接收的數(shù)據(jù)。這不是我們想要的。

幸運(yùn)的是，在flask中有一個擴(kuò)展叫flask-inflate，安裝了此擴(kuò)展會自動對請求來的數(shù)據(jù)做解壓操作。查看該擴(kuò)展的具體代碼處理：

# flask_inflate.pyimport gzipfrom flask import requestGZIP_CONTENT_ENCODING = ’gzip’class Inflate(object): def __init__(self, app=None): if app is not None: self.init_app(app) @staticmethod def init_app(app): app.before_request(_inflate_gzipped_content)def inflate(func): ''' A decorator to inflate content of a single view function ''' def wrapper(*args, **kwargs): _inflate_gzipped_content() return func(*args, **kwargs) return wrapperdef _inflate_gzipped_content(): content_encoding = getattr(request, ’content_encoding’, None) if content_encoding != GZIP_CONTENT_ENCODING: return # We don’t want to read the whole stream at this point. # Setting request.environ[’wsgi.input’] to the gzipped stream is also not an option because # when the request is not chunked, flask’s get_data will return a limited stream containing the gzip stream # and will limit the gzip stream to the compressed length. This is not good, as we want to read the # uncompressed stream, which is obviously longer. request.stream = gzip.GzipFile(fileobj=request.stream)

上述代碼的核心是:

request.stream = gzip.GzipFile(fileobj=request.stream)

于是，在django中可以如下處理：

class XXDataPushView(APIView): ''' 接收xx數(shù)據(jù)推送 '''# ... @white_list_required def post(self, request, **kwargs): content_encoding = request.META.get('HTTP_CONTENT_ENCODING', '') if content_encoding != 'gzip': req_data = request.data or {} else: gzip_f = gzip.GzipFile(fileobj=request.stream) data = gzip_f.read().decode(encoding='utf-8') req_data = json.loads(data) # ... handle req_data

ok, 問題完美解決。還可以用如下方式測試請求：

import gzipimport requestsimport jsondata = {}data = json.dumps(data).encode('utf-8')data = gzip.compress(data)resp = requests.post('http://localhost:8760/push_data/',data=data,headers={'Content-Encoding': 'gzip', 'Content-Type':'application/json;charset=utf-8'})print(resp.json())

以上就是如何用Django處理gzip數(shù)據(jù)流的詳細(xì)內(nèi)容，更多關(guān)于Django處理gzip數(shù)據(jù)流的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章！