io.elastic.sailor.impl.ObjectStorageImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sailor-jvm Show documentation
Sailor for JVM
There is a newer version: 4.0.3
package io.elastic.sailor.impl;

import com.google.inject.Inject;
import com.google.inject.name.Named;
import io.elastic.sailor.Constants;
import io.elastic.sailor.ObjectStorage;
import org.apache.http.entity.ByteArrayEntity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.json.JsonObject;

public class ObjectStorageImpl implements ObjectStorage {

    private static final Logger logger = LoggerFactory.getLogger(ObjectStorageImpl.class);

    private CryptoServiceImpl cryptoService;
    private String objectStorageUri;
    private String objectStorageToken;

    @Override
    public JsonObject getJsonObject(final String id) {

        if (this.objectStorageUri == null) {
            logger.info("Object storage service URI is not set");
            return null;
        }

        if (this.objectStorageToken == null) {
            logger.info("Object storage auth token is not set");
            return null;
        }

        this.logger.info("About to retrieve object by id={}", id);

        final String endpoint = String.format("%s/objects/%s", this.objectStorageUri, id);

        final byte[] bytes = HttpUtils.get(endpoint,
                new HttpUtils.BearerAuthorizationHandler(this.objectStorageToken),
                5);
        return cryptoService.decryptMessageContent(bytes, MessageEncoding.UTF8);
    }

    @Override
    public JsonObject postJsonObject(final JsonObject object) {
        return post(object.toString());
    }

    @Override
    public JsonObject post(String object) {
        if (this.objectStorageUri == null) {
            logger.info("Object storage service URI is not set");
            return null;
        }

        if (this.objectStorageToken == null) {
            logger.info("Object storage auth token is not set");
            return null;
        }

        this.logger.info("About to post an object into the storage");

        final String endpoint = String.format("%s/objects/", this.objectStorageUri);

        final byte[] content = cryptoService.encrypt(object, MessageEncoding.UTF8);


        final JsonObject result = HttpUtils.post(endpoint,
                new ByteArrayEntity(content),
                new HttpUtils.BearerAuthorizationHandler(this.objectStorageToken),
                5);

        return result;
    }

    @Inject
    public void setCryptoService(final CryptoServiceImpl cryptoService) {
        this.cryptoService = cryptoService;
    }

    @Inject(optional = true)
    public void setObjectStorageUri(final @Named(Constants.ENV_VAR_OBJECT_STORAGE_URI) String objectStorageUri) {
        this.objectStorageUri = objectStorageUri;

        if (this.objectStorageUri != null && this.objectStorageUri.endsWith("/")) {
            this.objectStorageUri.substring(0, this.objectStorageUri.length() - 1);
        }
    }

    @Inject(optional = true)
    public void setObjectStorageToken(final @Named(Constants.ENV_VAR_OBJECT_STORAGE_TOKEN) String objectStorageToken) {
        this.objectStorageToken = objectStorageToken;
    }
}