com.servioticy.api.commons.data.Group.java Source code

Introduction

Here is the source code for com.servioticy.api.commons.data.Group.java

Source

/*******************************************************************************
 * Copyright 2014 Barcelona Supercomputing Center (BSC)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package com.servioticy.api.commons.data;

import java.io.IOException;
import java.util.ArrayList;

import javax.ws.rs.core.Response;

import org.apache.log4j.Logger;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.servioticy.api.commons.datamodel.Data;
import com.servioticy.api.commons.elasticsearch.SearchEngine;
import com.servioticy.api.commons.exceptions.ServIoTWebApplicationException;

public class Group {
    private static ObjectMapper mapper = new ObjectMapper();
    private static Logger LOG = org.apache.log4j.Logger.getLogger(Group.class);

    private String streamId, groupId;
    private ArrayList<String> soIds;

    /** Create a Group from a JsonNode document.
     *
     * @param root The JsonNode Root
     */
    public Group(final JsonNode root) {
        try {
            // Check if exist stream
            if (root.path("stream").isMissingNode())
                throw new ServIoTWebApplicationException(Response.Status.NOT_FOUND,
                        "The stream field was not found");
            // Check if exist soIds
            if (root.path("soIds").isMissingNode())
                throw new ServIoTWebApplicationException(Response.Status.NOT_FOUND,
                        "The soIds field was not found");

            streamId = ((ObjectNode) root).get("stream").asText();
            try {
                soIds = mapper.readValue(mapper.writeValueAsString(root.get("soIds")),
                        new TypeReference<ArrayList<String>>() {
                        });
            } catch (JsonProcessingException e) {
                throw new ServIoTWebApplicationException(Response.Status.BAD_REQUEST,
                        "soIds has to be an array of Strings");
            }

        } catch (JsonProcessingException e) {
            LOG.error(e);
            throw new ServIoTWebApplicationException(Response.Status.BAD_REQUEST, e.getMessage());
        } catch (IOException e) {
            LOG.error(e);
            throw new ServIoTWebApplicationException(Response.Status.INTERNAL_SERVER_ERROR, e.getMessage());
        }

    }

    /** Create a Group form a String.
     *
     * @param body
     * @throws JsonProcessingException
     * @throws IOException
     */
    public Group(String body) throws JsonProcessingException, IOException {
        this(mapper.readTree(body));
    }

    /** Create a group with a groupId - to support CSOs
     * @param root
     * @param groupId
     */
    public Group(JsonNode root, String groupId) {
        this(root);
        this.groupId = groupId;
    }

    /**
     * @return The lastUpdate value of the streamId for all the soIds
     */
    public String lastUpdate() {
        Data data = null;

        String groupLastUpdateDataID = SearchEngine.getGroupLastUpdateDocId(streamId, soIds);
        if (groupLastUpdateDataID == null) {
            return "{}";
        }

        data = CouchBase.getData(groupLastUpdateDataID);

        return data.getString();
    }

    /** Create subscriptions to destination for all the soIds
     *
     * @param destination
     */
    public void createSubscriptions(String destination, String userId) {
        SO so;
        String body;

        if (this.groupId == null)
            throw new ServIoTWebApplicationException(Response.Status.INTERNAL_SERVER_ERROR,
                    "Error retrieving group info");

        for (String soId : soIds) {
            so = CouchBase.getSO(soId);

            body = "{ " + "\"callback\" : " + "\"internal\", \"destination\":  \"" + destination
                    + "\", \"customFields\": { \"groupId\": \"" + groupId + "\" }" + " }";

            // Create Subscription
            Subscription subs = new Subscription(so, userId, streamId, body);

            // Store in Couchbase
            CouchBase.setSubscription(subs);
        }

    }

    /** Check Group - All soIds has to exist and to have the stream
     *
     */
    public void checkSoIdsStream() {
        SO so;
        JsonNode stream;

        for (String soId : soIds) {
            so = CouchBase.getSO(soId);
            if (so == null)
                throw new ServIoTWebApplicationException(Response.Status.BAD_REQUEST,
                        "The Service Object: " + soId + " does not exist.");

            stream = so.getStream(streamId);
            if (stream == null)
                throw new ServIoTWebApplicationException(Response.Status.NOT_FOUND,
                        "The Service Object: " + soId + " does not have this stream");
        }
    }

    /**
     * @return groupId
     */
    public String getGroupId() {
        return groupId;
    }

}