com.windowsazure.messaging.NotificationHubsService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Notification-Hubs-java-sdk Show documentation
Azure Notification Hubs Java SDK for interacting with the data and management plane operations.
The newest version!
//----------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//----------------------------------------------------------------

package com.windowsazure.messaging;

import org.apache.hc.client5.http.async.methods.*;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.Method;

import java.net.URI;
import java.util.Arrays;
import java.util.function.Consumer;

public abstract class NotificationHubsService {
    protected static final String AUTHORIZATION_HEADER_NAME = "Authorization";
    protected static final String USER_AGENT_HEADER_NAME = "User-Agent";
    protected static final String USER_AGENT = "NHub/2020-06 (api-origin=JavaSDK;os=%s;os-version=%s)";
    protected static final String TRACKING_ID_HEADER = "TrackingId";

    protected SasTokenProvider tokenProvider;

    protected SimpleRequestBuilder createRequest(URI uri, Method method) {
        return addBaseHeaders(SimpleRequestBuilder.create(method), uri);
    }

    private SimpleRequestBuilder addBaseHeaders(SimpleRequestBuilder builder, URI uri) {
        final String trackingId = java.util.UUID.randomUUID().toString();

        return builder
            .setUri(uri)
            .setHeader(AUTHORIZATION_HEADER_NAME, tokenProvider.generateSasToken(uri))
            .setHeader(TRACKING_ID_HEADER, trackingId)
            .setHeader(USER_AGENT_HEADER_NAME, getUserAgent());
    }

    private String getUserAgent() {
        String os = System.getProperty("os.name");
        String osVersion = System.getProperty("os.version");
        return String.format(USER_AGENT, os, osVersion);
    }

    protected String getTrackingId(SimpleHttpRequest request) {
        return request.getFirstHeader(TRACKING_ID_HEADER).getValue();
    }

    protected  void executeRequest(
        final SimpleHttpRequest request,
        final FutureCallback callback,
        final int statusCode,
        Consumer consumer) {
        executeRequest(request, callback, new int[] { statusCode }, consumer);
    }

    protected  void executeRequest(
        final SimpleHttpRequest request,
        final FutureCallback callback,
        final int[] statusCodes,
        Consumer consumer) {
        HttpClientManager.getHttpAsyncClient().execute(
            SimpleRequestProducer.create(request),
            SimpleResponseConsumer.create(),
            new FutureCallback() {

                @Override
                public void completed(SimpleHttpResponse simpleHttpResponse) {
                    final int statusCode = simpleHttpResponse.getCode();
                    if (Arrays.stream(statusCodes).noneMatch(x -> x == statusCode)) {
                        callback.failed(NotificationHubsException.create(simpleHttpResponse, statusCode, getTrackingId(request)));
                        return;
                    }

                    consumer.accept(simpleHttpResponse);
                }

                @Override
                public void failed(Exception e) {
                    callback.failed(e);
                }

                @Override
                public void cancelled() {
                    callback.cancelled();
                }
            });
    }
}