com.bikfalvi.java.web.WebState.java Source code

Introduction

Here is the source code for com.bikfalvi.java.web.WebState.java

Source

/** 
 * Copyright (C) 2013 Alex Bikfalvi
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

package com.bikfalvi.java.web;

import java.io.DataOutputStream;
import java.io.IOException;
import java.io.StringReader;
import java.net.HttpURLConnection;
import java.nio.charset.Charset;

import org.apache.commons.io.IOUtils;

/**
 * A class representing the state of a web operation.
 * @author Alex Bikfalvi
 *
 */
public class WebState extends WebResult {
    private final HttpURLConnection connection;
    private byte[] data = null;

    /**
     * Creates a new web state instance using the specified URL.
     * @param url The web address.
     * @throws IOException 
     */
    public WebState(String url) throws IOException {
        super(url);
        // Open the web connection.
        this.connection = (HttpURLConnection) this.getUrl().openConnection();
    }

    /**
     * Gets the underlying HTTP URL connection.
     * @return The connection.
     */
    public HttpURLConnection getConnection() {
        return this.connection;
    }

    /**
     * Sets the data.
     * @param data The data as a string.
     * @param charset The charset.
     * @throws IOException 
     */
    public void setData(String data, Charset charset) throws IOException {
        StringReader reader = new StringReader(data);
        this.data = IOUtils.toByteArray(reader, charset);
    }

    /**
     * Creates a GET web request for the current URL.
     * @throws IOException 
     */
    public void execute() throws IOException {
        // If the data is not null.
        if (null != this.data) {
            // Write the data.
            this.connection.setDoOutput(true);
            DataOutputStream stream = new DataOutputStream(this.connection.getOutputStream());
            stream.write(this.data);
            stream.flush();
            stream.close();
        }

        // Execute the request and set the response code.
        int code = this.connection.getResponseCode();

        // Set the response data.
        this.setResponse(code, this.connection.getInputStream(), this.connection.getContentEncoding());

        // Complete the request.
        this.complete();
    }
}