KeyValueRequest.java

/**
 * 
 */
package com.intuit.tank.http.keyvalue;

/*
 * #%L
 * Intuit Tank Agent (apiharness)
 * %%
 * Copyright (C) 2011 - 2015 Intuit Inc.
 * %%
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * #L%
 */

import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import com.intuit.tank.http.BaseRequest;
import com.intuit.tank.http.TankHttpClient;
import com.intuit.tank.http.TankHttpLogger;

/**
 * @author ahernandez1
 * 
 */
public class KeyValueRequest extends BaseRequest {

    protected HashMap<String, String> postData = null;
    static protected Logger logger = LogManager.getLogger(KeyValueRequest.class);

    public KeyValueRequest(TankHttpClient client, TankHttpLogger logUtil) {
        super(client, logUtil);
        this.postData = new HashMap<String, String>();
        this.headerInformation = new HashMap<String, String>();
    }

    /**
     * 
     */
    @Override
    public void setKey(String key, String value) {
        this.postData.put(key, value);
        this.body = this.getBody();
    }

    /**
     * 
     */
    @SuppressWarnings("rawtypes")
    public String getBody() {
        Set set = this.postData.entrySet();
        Iterator iter = set.iterator();
        String body = "";

        while (iter.hasNext()) {
            try {
                Map.Entry mapEntry = (Map.Entry) iter.next();
                String key = mapEntry.getKey() != null ? URLEncoder.encode((String) mapEntry.getKey(), "UTF-8") : "";
                String value = mapEntry.getValue() != null ? URLEncoder.encode((String) mapEntry.getValue(), "UTF-8")
                        : "";

                if (StringUtils.isEmpty(body)) {
                    body = key + "=" + value;
                } else {
                    body = body + "&" + key + "=" + value;
                }
            } catch (Exception ex) {
                logger.error(ex.getMessage(), ex);
            }
        }
        return body;
    }

    @Override
    public String getKey(String key) {
        return this.postData.get(key);
    }

    public String removeKey(String key) {
        return this.postData.remove(key);
    }

    public void setNamespace(String name, String value) {
    }

}