private void putDeviceCountIntoResponseHeaders(PushApplication app, ResponseBuilder response) {
   long appCount = 0;
   for (Variant variant : app.getVariants()) {
     long variantCount = installationDao.getNumberOfDevicesForVariantID(variant.getVariantID());
     appCount += variantCount;
     response.header("deviceCount_variant_" + variant.getVariantID(), variantCount);
   }
   response.header("deviceCount_app_" + app.getPushApplicationID(), appCount);
 }
 private void putActivityIntoResponseHeaders(PushApplication app, ResponseBuilder response) {
   response.header(
       "activity_app_" + app.getPushApplicationID(),
       metricsService.countMessagesForPushApplication(app.getPushApplicationID()));
   for (Variant variant : app.getVariants()) {
     response.header(
         "activity_variant_" + variant.getVariantID(),
         metricsService.countMessagesForVariant(variant.getVariantID()));
   }
 }
  @DELETE
  @Path("{token}")
  public Response unregisterInstallations(
      @PathParam("token") String token, @Context HttpServletRequest request) {

    // find the matching variation:
    final Variant variant = loadVariantWhenAuthorized(request);
    if (variant == null) {
      return appendAllowOriginHeader(
          Response.status(Status.UNAUTHORIZED)
              .header("WWW-Authenticate", "Basic realm=\"AeroGear UnifiedPush Server\"")
              .entity("Unauthorized Request"),
          request);
    }

    // look up all installations (with same token) for the given variant:
    Installation installation =
        clientInstallationService.findInstallationForVariantByDeviceToken(
            variant.getVariantID(), token);

    if (installation == null) {
      return appendAllowOriginHeader(Response.status(Status.NOT_FOUND), request);
    } else {
      logger.log(Level.INFO, "Deleting metadata Installation");
      // remove
      clientInstallationService.removeInstallation(installation);
    }

    return appendAllowOriginHeader(Response.noContent(), request);
  }
  @POST
  @Consumes(MediaType.APPLICATION_JSON)
  public Response registerInstallation(Installation entity, @Context HttpServletRequest request) {

    // find the matching variation:
    final Variant variant = loadVariantWhenAuthorized(request);
    if (variant == null) {
      return appendAllowOriginHeader(
          Response.status(Status.UNAUTHORIZED)
              .header("WWW-Authenticate", "Basic realm=\"AeroGear UnifiedPush Server\"")
              .entity("Unauthorized Request"),
          request);
    }

    // Poor validation: We require the Token! And the 'simplePushEndpoint' for SimplePush clients!
    if (entity.getDeviceToken() == null
        || (variant.getType() == VariantType.SIMPLE_PUSH
            && entity.getSimplePushEndpoint() == null)) {
      return appendAllowOriginHeader(Response.status(Status.BAD_REQUEST), request);
    }

    // look up all installations (with same token) for the given variant:
    Installation installation =
        clientInstallationService.findInstallationForVariantByDeviceToken(
            variant.getVariantID(), entity.getDeviceToken());

    // Needed for the Admin UI Only. Help for setting up Routes
    entity.setPlatform(variant.getType().getTypeName());

    // The 'mobile application' on the device/client was launched.
    // If the installation is already in the DB, let's update the metadata,
    // otherwise we register a new installation:
    logger.log(Level.FINEST, "Mobile Application on device was launched");

    // new device/client ?
    if (installation == null) {
      logger.log(Level.FINEST, "Performing new device/client registration");
      // store the installation:
      clientInstallationService.addInstallation(variant.getType(), entity);
      // add installation to the matching variant
      genericVariantService.addInstallation(variant, entity);
    } else {
      // We only update the metadata, if the device is enabled:
      if (installation.isEnabled()) {
        logger.log(Level.FINEST, "Updating received metadata for an 'enabled' installation");
        // update the entity:
        clientInstallationService.updateInstallation(installation, entity);
      }
    }

    return appendAllowOriginHeader(Response.ok(entity), request);
  }
  /**
   * Receives request for processing a {@link UnifiedPushMessage} and loads tokens for devices that
   * match requested parameters from database.
   *
   * <p>Device tokens are loaded in a stream and split to batches of configured size (see {@link
   * SenderConfiguration#batchSize()}). Once the pre-configured number of batches (see {@link
   * SenderConfiguration#batchesToLoad()}) is reached, this method resends message to the same queue
   * it took the request from, so that the transaction it worked in is split and further processing
   * may continue in next transaction.
   *
   * <p>Additionally it fires {@link BatchLoadedEvent} as CDI event (that is translated to JMS
   * event) that helps {@link MetricsCollector} to track how many batches were loaded. When all
   * batches were loaded for the given variant, it fires {@link AllBatchesLoadedEvent}.
   *
   * @param msg holder object containing the payload and info about the effected variants
   */
  public void loadAndQueueTokenBatch(@Observes @Dequeue MessageHolderWithVariants msg)
      throws IllegalStateException {
    final UnifiedPushMessage message = msg.getUnifiedPushMessage();
    final VariantType variantType = msg.getVariantType();
    final Collection<Variant> variants = msg.getVariants();
    final String lastTokenFromPreviousBatch = msg.getLastTokenFromPreviousBatch();
    final SenderConfiguration configuration =
        senderConfiguration.select(new SenderTypeLiteral(variantType)).get();
    final PushMessageInformation pushMessageInformation = msg.getPushMessageInformation();
    int serialId = msg.getLastSerialId();

    logger.debug("Received message from queue: " + message.getMessage().getAlert());

    final Criteria criteria = message.getCriteria();
    final List<String> categories = criteria.getCategories();
    final List<String> aliases = criteria.getAliases();
    final List<String> deviceTypes = criteria.getDeviceTypes();

    logger.info(
        String.format(
            "Preparing message delivery and loading tokens for the %s 3rd-party Push Network (for %d variants)",
            variantType, variants.size()));

    for (Variant variant : variants) {

      try {

        ResultsStream<String> tokenStream;
        final Set<String> topics = new TreeSet<>();
        final boolean isAndroid = variantType.equals(VariantType.ANDROID);

        // the entire batch size
        int batchesToLoad = configuration.batchesToLoad();

        // Some checks for GCM, because of GCM-3 topics
        boolean gcmTopicRequest = (isAndroid && TokenLoaderUtils.isGCMTopicRequest(criteria));
        if (gcmTopicRequest) {

          // If we are able to do push for GCM topics...

          // 1)
          // find all topics, BUT only on the very first round of batches
          // otherwise after 10 (or what ever the max. is) another request would be sent to that
          // topic
          if (serialId == 0) {
            topics.addAll(TokenLoaderUtils.extractGCMTopics(criteria, variant.getVariantID()));

            // topics are handled as a first extra batch,
            // therefore we have to adjust the number by adding this extra batch
            batchesToLoad = batchesToLoad + 1;
          }

          // 2) always load the legacy tokens, for all number of batch iterations
          tokenStream =
              clientInstallationService
                  .findAllOldGoogleCloudMessagingDeviceTokenForVariantIDByCriteria(
                      variant.getVariantID(),
                      categories,
                      aliases,
                      deviceTypes,
                      configuration.tokensToLoad(),
                      lastTokenFromPreviousBatch)
                  .fetchSize(configuration.batchSize())
                  .executeQuery();
        } else {
          tokenStream =
              clientInstallationService
                  .findAllDeviceTokenForVariantIDByCriteria(
                      variant.getVariantID(),
                      categories,
                      aliases,
                      deviceTypes,
                      configuration.tokensToLoad(),
                      lastTokenFromPreviousBatch)
                  .fetchSize(configuration.batchSize())
                  .executeQuery();
        }

        String lastTokenInBatch = null;
        int tokensLoaded = 0;
        for (int batchNumber = 0; batchNumber < batchesToLoad; batchNumber++) {

          // increasing the serial ID,
          // to make sure it's properly read from all block
          ++serialId;

          final Set<String> tokens = new TreeSet<>();

          // On Android, the first batch is for GCM3 topics
          // legacy tokens are submitted in the batch #2 and later
          if (isAndroid && batchNumber == 0 && !topics.isEmpty()) {
            tokens.addAll(topics);
          } else {
            for (int i = 0; i < configuration.batchSize() && tokenStream.next(); i++) {
              lastTokenInBatch = tokenStream.get();
              tokens.add(lastTokenInBatch);
              tokensLoaded += 1;
            }
          }

          if (tokens.size() > 0) {
            if (tryToDispatchTokens(
                new MessageHolderWithTokens(
                    msg.getPushMessageInformation(), message, variant, tokens, serialId))) {
              logger.info(
                  String.format(
                      "Loaded batch #%s, containing %d tokens, for %s variant (%s)",
                      serialId,
                      tokens.size(),
                      variant.getType().getTypeName(),
                      variant.getVariantID()));
            } else {
              logger.debug(
                  String.format(
                      "Failing token loading transaction for batch token #%s for %s variant (%s), since queue is full, will retry...",
                      serialId, variant.getType().getTypeName(), variant.getVariantID()));
              context.setRollbackOnly();
              return;
            }

            // using combined key of variant and PMI (AGPUSH-1585):
            batchLoaded.fire(
                new BatchLoadedEvent(
                    variant.getVariantID() + ":" + msg.getPushMessageInformation().getId()));
            if (serialId == MessageHolderWithVariants.INITIAL_SERIAL_ID) {
              triggerVariantMetricCollection.fire(
                  new TriggerVariantMetricCollectionEvent(
                      msg.getPushMessageInformation(), variant));
            }
          } else {
            logger.debug(
                String.format(
                    "Ending batch processing: No more tokens for batch #%s available", serialId));
            break;
          }
        }

        // should we trigger next transaction?
        if (tokensLoaded >= configuration.tokensToLoad()) {
          logger.debug(
              String.format(
                  "Ending token loading transaction for %s variant (%s)",
                  variant.getType().getTypeName(), variant.getVariantID()));
          nextBatchEvent.fire(
              new MessageHolderWithVariants(
                  msg.getPushMessageInformation(),
                  message,
                  msg.getVariantType(),
                  variants,
                  serialId,
                  lastTokenInBatch));
        } else {
          logger.debug(
              String.format(
                  "All batches for %s variant were loaded (%s)",
                  variant.getType().getTypeName(), pushMessageInformation.getId()));

          // using combined key of variant and PMI (AGPUSH-1585):
          allBatchesLoaded.fire(
              new AllBatchesLoadedEvent(
                  variant.getVariantID() + ":" + msg.getPushMessageInformation().getId()));
          triggerVariantMetricCollection.fire(
              new TriggerVariantMetricCollectionEvent(pushMessageInformation, variant));

          if (tokensLoaded == 0 && lastTokenFromPreviousBatch == null) {
            // no tokens were loaded at all!
            if (gcmTopicRequest) {
              logger.debug("No legacy(non-InstanceID) tokens found. Just pure GCM topic requests");
            } else {
              logger.warn("Check your push query: Not a single token was loaded from the DB!");
            }

            VariantMetricInformation variantMetricInformation = new VariantMetricInformation();
            variantMetricInformation.setPushMessageInformation(msg.getPushMessageInformation());
            variantMetricInformation.setVariantID(variant.getVariantID());
            variantMetricInformation.setDeliveryStatus(Boolean.TRUE);
            dispatchVariantMetricEvent.fire(variantMetricInformation);
          }
        }
      } catch (ResultStreamException e) {
        logger.error("Failed to load batch of tokens", e);
      }
    }
  }