com.bettercloud.vault.response.VaultResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vault-java-driver Show documentation

Zero-dependency Java client for HashiCorp's Vault

There is a newer version: 5.1.0

package com.bettercloud.vault.response;

import com.bettercloud.vault.json.Json;
import com.bettercloud.vault.rest.RestResponse;

/**
 * VaultResponse is a common base class for the response objects returned by
 * all API methods.  It contains the bare minimum of information common to all Vault
 * responses (e.g. the raw HTTP response, the number of retry attempts if any).  API methods
 * which return additional information will use more specialized subclasses inheriting
 * from VaultResponse.
 */
public class VaultResponse {

    private RestResponse restResponse;
    private int retries;

    private String leaseId;
    private Boolean renewable;
    private Long leaseDuration;

    /**
     * @param restResponse The raw HTTP response from Vault.
     * @param retries The number of retry attempts that occurred during the API call (can be zero).
     */
    public VaultResponse(final RestResponse restResponse, final int retries) {
        this.restResponse = restResponse;
        this.retries = retries;
        parseMetadataFields();
    }

    public RestResponse getRestResponse() {
        return restResponse;
    }

    public void setRestResponse(final RestResponse restResponse) {
        this.restResponse = restResponse;
        parseMetadataFields();
    }

    public int getRetries() {
        return retries;
    }

    public void setRetries(final int retries) {
        this.retries = retries;
    }

    public String getLeaseId() {
        return leaseId;
    }

    public void setLeaseId(final String leaseId) {
        this.leaseId = leaseId;
    }

    public Boolean getRenewable() {
        return renewable;
    }

    public void setRenewable(final Boolean renewable) {
        this.renewable = renewable;
    }

    @Deprecated
    public void baseSetRenewable(final Boolean renewable) {
        this.renewable = renewable;
    }

    public Long getLeaseDuration() {
        return leaseDuration;
    }

    public void setLeaseDuration(final Long leaseDuration) {
        this.leaseDuration = leaseDuration;
    }

    private void parseMetadataFields() {
        String jsonString;
        try {
            jsonString = new String(this.restResponse.getBody(), "UTF-8");
        } catch (Exception e) {
            return;
        }
        try {
            this.leaseId = Json.parse(jsonString).asObject().get("lease_id").asString();
        } catch (Exception e) {
        }
        try {
            this.renewable = Json.parse(jsonString).asObject().get("renewable").asBoolean();
        } catch (Exception e) {
        }
        try {
            this.leaseDuration = Json.parse(jsonString).asObject().get("lease_duration").asLong();
        } catch (Exception e) {
        }
    }
}