com.sitewhere.microservice.api.schedule.BatchCommandInvocationJobParser Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) SiteWhere, LLC. All rights reserved. http://www.sitewhere.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package com.sitewhere.microservice.api.schedule;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.sitewhere.rest.model.batch.request.InvocationByAssignmentCriteriaRequest;
import com.sitewhere.rest.model.batch.request.InvocationByDeviceCriteriaRequest;
import com.sitewhere.spi.batch.request.IInvocationByAssignmentCriteriaRequest;
import com.sitewhere.spi.batch.request.IInvocationByDeviceCriteriaRequest;
import com.sitewhere.spi.scheduling.JobConstants;

public class BatchCommandInvocationJobParser {

    /**
     * Parse configuration data for a batch invocation based on device criteria.
     * 
     * @param data
     */
    public static IInvocationByDeviceCriteriaRequest parseInvocationByDeviceCriteria(Map data) {

	String deviceTypeToken = null;
	String commandToken = null;
	Map parameters = new HashMap();

	for (String key : data.keySet()) {
	    String value = data.get(key);
	    if (JobConstants.InvocationByDeviceCriteria.DEVICE_TYPE_TOKEN.equals(key)) {
		deviceTypeToken = value;
	    } else if (JobConstants.CommandInvocation.COMMAND_TOKEN.equals(key)) {
		commandToken = value;
	    } else if (key.startsWith(JobConstants.CommandInvocation.PARAMETER_PREFIX)) {
		String paramKey = key.substring(JobConstants.CommandInvocation.PARAMETER_PREFIX.length());
		parameters.put(paramKey, value);
	    }
	}

	InvocationByDeviceCriteriaRequest request = new InvocationByDeviceCriteriaRequest();
	request.setDeviceTypeToken(deviceTypeToken);
	request.setCommandToken(commandToken);
	request.setParameterValues(parameters);
	return request;
    }

    /**
     * Parse configuration data for a batch invocation based on device assignment
     * criteria.
     * 
     * @param data
     */
    public static IInvocationByAssignmentCriteriaRequest parseInvocationByAssignmentCriteria(Map data) {

	String deviceTypeToken = null;
	String commandToken = null;
	Map parameters = new HashMap();
	List customerTokens = new ArrayList<>();
	List areaTokens = new ArrayList<>();
	List assetTokens = new ArrayList<>();

	for (String key : data.keySet()) {
	    String value = data.get(key);
	    if (JobConstants.InvocationByAssignmentCriteria.DEVICE_TYPE_TOKEN.equals(key)) {
		deviceTypeToken = value;
	    } else if (JobConstants.CommandInvocation.COMMAND_TOKEN.equals(key)) {
		commandToken = value;
	    } else if (key.startsWith(JobConstants.CommandInvocation.PARAMETER_PREFIX)) {
		String paramKey = key.substring(JobConstants.CommandInvocation.PARAMETER_PREFIX.length());
		parameters.put(paramKey, value);
	    } else if (key.startsWith(JobConstants.InvocationByAssignmentCriteria.CUSTOMER_TOKEN_PREFIX)) {
		customerTokens.add(value);
	    } else if (key.startsWith(JobConstants.InvocationByAssignmentCriteria.AREA_TOKEN_PREFIX)) {
		areaTokens.add(value);
	    } else if (key.startsWith(JobConstants.InvocationByAssignmentCriteria.ASSET_TOKEN_PREFIX)) {
		assetTokens.add(value);
	    }
	}

	InvocationByAssignmentCriteriaRequest request = new InvocationByAssignmentCriteriaRequest();
	request.setDeviceTypeToken(deviceTypeToken);
	request.setCommandToken(commandToken);
	request.setParameterValues(parameters);
	request.setCustomerTokens(customerTokens);
	request.setAreaTokens(areaTokens);
	request.setAssetTokens(assetTokens);
	return request;
    }
}