com.applitools.eyes.ResourceFuture Maven / Gradle / Ivy

Go to download
package com.applitools.eyes;

import com.applitools.eyes.visualgridclient.services.IResourceFuture;
import com.applitools.eyes.visualgridclient.model.RGridResource;
import com.applitools.utils.GeneralUtils;

import javax.ws.rs.core.Response;
import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

public class ResourceFuture implements IResourceFuture {

    private Future future;
    private String url;
    private Logger logger;
    private IServerConnector serverConnector;
    private RGridResource rgResource;
    private int retryCount = 3;

    public ResourceFuture(Future future, String url, Logger logger, IServerConnector serverConnector) {
        this.future = future;
        this.url = url;
        this.logger = logger;
        this.serverConnector = serverConnector;
    }

    public ResourceFuture(RGridResource rgResource, Logger logger, IServerConnector serverConnector) {
        this.url = rgResource.getUrl();
        this.rgResource = rgResource;
        this.logger = logger;
        this.serverConnector = serverConnector;
    }

    @Override
    public boolean cancel(boolean mayInterruptIfRunning) {
        return future.cancel(mayInterruptIfRunning);
    }

    @Override
    public boolean isCancelled() {
        return future.isCancelled();
    }

    @Override
    public boolean isDone() {
        return future.isDone();
    }

    @Override
    public RGridResource get() throws InterruptedException {
        logger.verbose("enter - this.rgResource: " + this.rgResource);
        if (this.future == null) {
            try {
                IResourceFuture newFuture = serverConnector.downloadResource(new URL(this.url), true, null);
                this.future = ((ResourceFuture) newFuture).future;
            } catch (MalformedURLException malformedUrlException) {
                GeneralUtils.logExceptionStackTrace(logger, malformedUrlException);
            }
        }
        while (this.rgResource == null && retryCount > 0) {
            try {
                Response response = this.future.get(15, TimeUnit.SECONDS);
                logger.verbose("response: " + response);
                ByteArrayOutputStream outputStream = downloadFile(response);
                String contentType = Utils.getResponseContentType(response);
                String contentEncoding = Utils.getResponseContentEncoding(response);
                byte[] content = outputStream.toByteArray();
                if (contentEncoding.contains("gzip")) {
                    content = GeneralUtils.getUnGzipByteArrayOutputStream(content);
                }
                rgResource = new RGridResource(url, contentType, content, logger, "ResourceFuture");
                break;
            } catch (Exception e) {
                retryCount--;
                logger.verbose("Entering retry");
                try {
                    Thread.sleep(300);
                    IResourceFuture newFuture = serverConnector.downloadResource(new URL(this.url), true, null);
                    this.future = ((ResourceFuture) newFuture).future;
                } catch (MalformedURLException malformedUrlException) {
                    GeneralUtils.logExceptionStackTrace(logger, malformedUrlException);
                }
            }
        }
        return rgResource;
    }

    @Override
    public RGridResource get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
        if (this.rgResource == null) {
            Response response = future.get(timeout, unit);
            ByteArrayOutputStream outputStream = downloadFile(response);
            byte[] content = outputStream.toByteArray();
            String contentType = Utils.getResponseContentType(response);
            if (content.length > 0) {
                String contentEncoding = Utils.getResponseContentEncoding(response);
                if (contentEncoding != null && contentEncoding.contains("gzip")) {
                    try {
                        content = GeneralUtils.getUnGzipByteArrayOutputStream(content);
                    } catch (IOException e) {
                        GeneralUtils.logExceptionStackTrace(logger, e);
                    }
                }
            } else {
                logger.log("content is empty");
            }
            rgResource = new RGridResource(url, contentType, content, logger, "ResourceFuture");
        }
        return rgResource;
    }

    private ByteArrayOutputStream downloadFile(Response response) {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream(1024);
        InputStream inputStream = response.readEntity(InputStream.class);
        byte[] bytes = new byte[1024];
        try {
            int readBytes = inputStream.read(bytes);
            while (readBytes > 0) {
                outputStream.write(bytes, 0, readBytes);
                readBytes = inputStream.read(bytes);
            }
        } catch (IOException e) {
            GeneralUtils.logExceptionStackTrace(logger, e);
        }
        return outputStream;
    }

    @Override
    public String getUrl() {
        return this.url;
    }
}