com.ibm.fhir.persistence.blob.app.ReadBlobValue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fhir-persistence-blob-app Show documentation
There is a newer version: 4.11.1
Show newest version
/*
 * (C) Copyright IBM Corp. 2022
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
package com.ibm.fhir.persistence.blob.app;

import java.nio.charset.StandardCharsets;
import java.util.logging.Level;
import java.util.logging.Logger;

import com.azure.core.util.BinaryData;
import com.azure.storage.blob.BlobAsyncClient;
import com.azure.storage.blob.BlobContainerAsyncClient;
import com.ibm.fhir.config.FHIRRequestContext;
import com.ibm.fhir.exception.FHIRException;
import com.ibm.fhir.persistence.blob.BlobContainerManager;
import com.ibm.fhir.persistence.blob.BlobManagedContainer;
import com.ibm.fhir.persistence.blob.BlobName;
import com.ibm.fhir.persistence.jdbc.dao.api.IResourceTypeMaps;

/**
 * Do a plain read of the resource blob value and print it to stdout for debugging
 */
public class ReadBlobValue {
    private static final Logger logger = Logger.getLogger(ReadBlobValue.class.getName());
    private final String tenantId;
    private final String dsId;
    private final IResourceTypeMaps resourceTypeMaps;
    private final BlobName blobName;

    /**
     * Public constructor
     * @param tenantId
     * @param dsId
     * @param resourceTypeMaps
     * @param blobName
     */
    public ReadBlobValue(String tenantId, String dsId, IResourceTypeMaps resourceTypeMaps, BlobName blobName) {
        this.tenantId = tenantId;
        this.dsId = dsId;
        this.resourceTypeMaps = resourceTypeMaps;
        this.blobName = blobName;
    }

    /**
     * Create the container for the configured tenant and datasource pair
     */
    public void run() throws FHIRException {
        // Set up the request context for the configured tenant and datastore
        FHIRRequestContext.set(new FHIRRequestContext(tenantId, dsId));

        // Check to see if the container already exists, and if not, then 
        // issue the create container command and wait for the response
        BlobManagedContainer bmc = BlobContainerManager.getSessionForTenantDatasource();
        try {
            if (blobName.isPartial()) {
                // resourceTypeId/logicalId/version
                listBlobs(bmc.getClient());                
            } else {
                // resourceTypeId/logicalId/version/resourcePayloadKey
                displaySingleValue(bmc.getClient(), this.blobName);
            }
        } catch (RuntimeException x) {
            logger.log(Level.SEVERE, "blob fetch failed: " + bmc.getContainerName() + ":" + blobName, x);
            throw x;
        }
    }
    /**
     * When the blobName does not include the resourcePayloadKey, list all matches
     * @param bcc
     * @throws FHIRException
     */
    private void listBlobs(BlobContainerAsyncClient bcc) {
        final String prefix = blobName.toBlobPath();
        logger.info("Scanning all entries with prefix: '" + prefix + "'");

        // List blobs using the key prefix
        bcc.listBlobsByHierarchy(prefix)
        .doOnNext(blobItem -> displaySingleValue(bcc, blobItem.getName()))
        .blockLast(); // wait for scan to complete
    }

    /**
     * Display the blob content for the given full blob path
     * @param bcc
     * @param fullBlobPath
     */
    private void displaySingleValue(BlobContainerAsyncClient bcc, String fullBlobPath) {
        // Need to parse the blob path provided by the database
        BlobName blobName = BlobName.create(resourceTypeMaps, fullBlobPath);
        BlobAsyncClient bac = bcc.getBlobAsyncClient(fullBlobPath);
        BinaryData result = bac.downloadContent().block();
        String blobValue = new String(result.toBytes(), StandardCharsets.UTF_8);
        // Use the human-readable form of the blob-name, which translates the resource type id
        // to the resource type name
        System.out.println(blobName.toString() + ": " + blobValue);
    }

    /**
     * Display the blob content for the given blob name (which is the complete
     * key, including the resourcePayloadKey).
     * @param bcc
     * @param fullBlobName
     */
    private void displaySingleValue(BlobContainerAsyncClient bcc, BlobName fullBlobName) {
        BlobAsyncClient bac = bcc.getBlobAsyncClient(fullBlobName.toBlobPath());
        BinaryData result = bac.downloadContent().block();
        String blobValue = new String(result.toBytes(), StandardCharsets.UTF_8);
        // Use the human-readable form of the blob-name, which translates the resource type id
        // to the resource type name
        System.out.println(fullBlobName.toString() + ": " + blobValue);
    }
}