no.nav.common.cxf.TimeoutFeature Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cxf Show documentation

There is a newer version: 2.2023.01.10_13.49-81ddc732df3a

package no.nav.common.cxf;

import org.apache.cxf.Bus;
import org.apache.cxf.endpoint.Client;
import org.apache.cxf.feature.AbstractFeature;
import org.apache.cxf.transport.Conduit;
import org.apache.cxf.transport.http.HTTPConduit;
import org.apache.cxf.transports.http.configuration.HTTPClientPolicy;

/**
 * Timeoutfeature for å konfigurer timeout på cxf-klient
 */
public class TimeoutFeature extends AbstractFeature {

    public static final int DEFAULT_RECEIVE_TIMEOUT = 10000;
    public static final int DEFAULT_CONNECTION_TIMEOUT = 10000;

    private int receiveTimeout = DEFAULT_RECEIVE_TIMEOUT;
    private int connectionTimeout = DEFAULT_CONNECTION_TIMEOUT;

    public TimeoutFeature() {
    }

    public TimeoutFeature(int receiveTimeout, int connectionTimeout) {
		this.receiveTimeout = receiveTimeout;
		this.connectionTimeout = connectionTimeout;
	}

	@Override
    public void initialize(Client client, Bus bus) {
        Conduit conduit = client.getConduit();
        if (conduit instanceof HTTPConduit) {
            HTTPClientPolicy policy = new HTTPClientPolicy();
            policy.setReceiveTimeout(receiveTimeout);
            policy.setConnectionTimeout(connectionTimeout);
            HTTPConduit httpConduit = (HTTPConduit) conduit;
            httpConduit.setClient(policy);
        }
        super.initialize(client, bus);
    }

    public TimeoutFeature withReceiveTimeout(int timeoutInMillis) {
        receiveTimeout = timeoutInMillis;
        return this;
    }

    public TimeoutFeature withConnectionTimeout(int timeoutInMillis) {
        connectionTimeout = timeoutInMillis;
        return this;
    }
}