io.muserver.SelectiveHttpContentCompressor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mu-server Show documentation

A simple but powerful web server framework

The newest version!

package io.muserver;

import io.netty.handler.codec.http.HttpContentCompressor;
import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpResponse;

import static io.muserver.NettyResponseAdaptor.getVaryWithAE;

class SelectiveHttpContentCompressor extends HttpContentCompressor {

    private final ServerSettings settings;

    SelectiveHttpContentCompressor(ServerSettings settings) {
        this.settings = settings;
    }

    @Override
    protected Result beginEncode(HttpResponse response, String acceptEncoding) throws Exception {
        String declaredLength = response.headers().get(HttpHeaderNames.CONTENT_LENGTH);
        String declaredType = response.headers().get(HttpHeaderNames.CONTENT_TYPE);
        if (settings.shouldCompress(declaredLength, declaredType)) {
            response.headers().set(HeaderNames.VARY, getVaryWithAE(response.headers().get(HeaderNames.VARY)));
            return super.beginEncode(response, acceptEncoding);
        } else {
            return null;
        }
    }

}