使用Django怎么對(duì)gzip數(shù)據(jù)流進(jìn)行處理

這期內(nèi)容當(dāng)中小編將會(huì)給大家?guī)?lái)有關(guān)使用Django怎么對(duì)gzip數(shù)據(jù)流進(jìn)行處理，文章內(nèi)容豐富且以專(zhuān)業(yè)的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

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

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

因此，我要求對(duì)方給一個(gè)推送的簡(jiǎn)單示例，沒(méi)想到對(duì)方不講武德，仍過(guò)來(lái)一段沒(méi)法單獨(dú)運(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();
      }
    }
  });
}

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

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

• gzip_page()

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

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

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

# flask_inflate.py
import gzip
from flask import request

GZIP_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 wrapper


def _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, 問(wèn)題完美解決。還可以用如下方式測(cè)試請(qǐng)求：

import gzip
import requests
import json

data = {}

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怎么對(duì)gzip數(shù)據(jù)流進(jìn)行處理了，如果剛好有類(lèi)似的疑惑，不妨參照上述分析進(jìn)行理解。如果想知道更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道。