PostRequest.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package br.ufrgs.inf.prosoft.requestssimulator.requests;

import br.ufrgs.inf.prosoft.requestssimulator.Session;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.stream.Collectors;

/**
 *
 * @author romulo
 */
public class PostRequest extends Request {

    private final String data;

    protected PostRequest(RequestPlan requestPlan, Session session, String URL, String data) {
        super(requestPlan, session, URL);
        this.data = data;
    }

    protected PostRequest(RequestPlan requestPlan, Session session, String URL, String headers, String data) {
        super(requestPlan, session, URL, headers);
        this.data = data;
    }

    protected String getData() {
        return this.data;
    }

    @Override
    public void fire() {
        try {
            URL url = new URL(getURL());
            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
            connection.setRequestMethod("POST");
            connection.setDoOutput(true);
            try (OutputStream outputStream = connection.getOutputStream()) {
                outputStream.write(this.data.getBytes());
            }
            forEachHeader((key, value) -> {
                value = fillPropertyVariable(value);
                connection.setRequestProperty(key, value);
            });
            int responseCode = connection.getResponseCode();
            if (responseCode < 400) {
                Logger.getGlobal().log(Level.INFO, "{0} {1}", new Object[]{getMethod(), getURL()});
                try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(connection.getInputStream()))) {
                    String response = bufferedReader.lines().collect(Collectors.joining());
                    processResponse(response);
                }
            } else {
                Logger.getGlobal().log(Level.SEVERE, "error {0} on {1} {2}", new Object[]{responseCode, getMethod(), getURL()});
            }
            connection.disconnect();
        } catch (MalformedURLException ex) {
            Logger.getGlobal().log(Level.SEVERE, "Malormed URL");
        } catch (IOException ex) {
            Logger.getGlobal().log(Level.SEVERE, "IOException");
        }
    }
}