io.p8e.grpc.client.ObjectClient.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of p8e-common Show documentation

A collection of services and libraries that iteract and run Provenance Java based contracts.

The newest version!

package io.p8e.grpc.client

import io.grpc.ManagedChannel
import io.p8e.proto.Common.Location
import io.p8e.proto.Common.WithAudience
import io.p8e.proto.ObjectGrpc
import io.p8e.proto.Objects
import io.p8e.proto.Objects.ObjectLoadRequest
import java.util.concurrent.TimeUnit

class ObjectClient(
    channel: ManagedChannel,
    challengeResponseInterceptor: ChallengeResponseInterceptor,
    private val deadlineMs: Long
) {
    private val client = ObjectGrpc.newBlockingStub(channel)
        .withInterceptors(challengeResponseInterceptor)

    fun store(
        withAudience: WithAudience
    ): Location {
        return client.withDeadlineAfter(deadlineMs, TimeUnit.MILLISECONDS)
            .store(withAudience)
    }

    fun load(
        uri: String
    ): ByteArray {
        return client.withDeadlineAfter(deadlineMs, TimeUnit.MILLISECONDS)
            .load(
                ObjectLoadRequest.newBuilder()
                    .setUri(uri)
                    .build()
            ).bytes.toByteArray()
    }

    fun loadJson(
        hash: String,
        className: String,
        contractSpecHash: String
    ): String {
        return client.withDeadlineAfter(deadlineMs, TimeUnit.MILLISECONDS)
            .loadJson(
                Objects.ObjectLoadJsonRequest.newBuilder()
                    .setHash(hash)
                    .setClassname(className)
                    .setContractSpecHash(contractSpecHash)
                    .build()
            ).json
    }
}